Skip to main content

pytest plugin for running individual tests with mpiexec

Project description

pytest-mpixec

pytest plugin for running individual tests with mpiexec

A test marked with mark.mpiexec will be run in a separate instance with mpiexec, instead of in the current process.

The subprocess test will be run with pytest, so fixtures and everything should still work!

The number of processes can be parametrized if you parametrize an argument called mpiexec_n.

Try it out

pip install pytest-mpiexec

And write tests that use mpiexec:

from mpi4py import MPI

@pytest.mark.mpiexec(n=2)
def test_my_mpi_code(fixtures):

@pytest.mark.mpiexec
@pytest.mark.parametrize("mpiexec_n", [1, 2, 3])
def test_my_mpi_code(mpiexec_n):
    assert MPI.COMM_WORLD.size == mpiexec_n

Caveats

If you use module or session-scoped fixtures, another instance will be running, so these can't conflict with other pytest runs (e.g. conflicting on ports, files, etc.)

Prior art

  • pytest-mpi (helpers for tests run inside mpi - compatible with this package!)
  • pytest-easyMPI (similar goal to this one, but takes a different approach)

Project details


Download files

Download the file for your platform. If you're not sure which to choose, learn more about installing packages.

Source Distribution

pytest_mpiexec-0.0.1a1.tar.gz (7.0 kB view hashes)

Uploaded source

Built Distribution

pytest_mpiexec-0.0.1a1-py3-none-any.whl (5.3 kB view hashes)

Uploaded py3

Supported by

AWS AWS Cloud computing and Security Sponsor Datadog Datadog Monitoring Fastly Fastly CDN Google Google Download Analytics Microsoft Microsoft PSF Sponsor Pingdom Pingdom Monitoring Sentry Sentry Error logging StatusPage StatusPage Status page