working with demo_mock.py in single and mpi