# MPI tests are set up to run on 3 processes.
N_MPI=3
export OMPI_MCA_plm_rsh_agent=/bin/false
export OMPI_MCA_rmaps_base_oversubscribe=1
export OMPI_MCA_btl_base_warn_component_unused=0

echo "=== python unit test (serial) ==="
python3 -m pytest python/test/unit/

echo "=== python unit test (MPI) ==="
mpirun -n ${N_MPI} python3 -m pytest python/test/unit/ --color=no
