Added test cases for serial/mpi python wrapper