refactor: modify test_cube_partition_mpi.py - divide direct/PBS