modify the test_cube_partition_mpi.py with the test generator