# 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 "== running python tests =="

mkdir -p ${AUTOPKGTEST_TMP}/python/dist-packages
cp -ra python/dolfinx_utils ${AUTOPKGTEST_TMP}/python/dist-packages
export PYTHONPATH=${PYTHONPATH}:${AUTOPKGTEST_TMP}/python/dist-packages

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

python3 python/demo/generate-demo-files.py
echo "=== python demo test (serial) ==="
python3 -m pytest -v -k 'not demo_gmsh' python/demo/test.py
echo "=== python demo test (MPI) ==="
python3 -m pytest -v -m mpi -k 'not demo_gmsh' python/demo/test.py --mpiexec=mpiexec --num-proc=${N_MPI}
