Improving how partitions are created/removed during the tests.