Update MPI test to work with python3