Adds unit tests for mpi plan construction