Add plan unit tests for MPI