Awesome job on the `partition.py` tests - very neat!