Skip to main content

Fixtures for seeding tests and making randomness reproducible

Project description

pytest-rng

pytest-rng provides fixtures for ensuring “randomness” in your tests is reproducible from one run to the next. It also allows the seed for all tests to be changed if requested, to help ensure that test successes are not dependent on particular random number seeds.

  • Use the rng fixture to get a pre-seeded random number generator (RNG) that exposes NumPy’s ~numpy.random.mtrand.RandomState interface.

  • Use the seed fixture to get an integer seed that can be used to initialize your own RNG.

The following example prints the same four random numbers every time the test is run.

import numpy as np

def test_rectification(rng, seed):
    print(rng.uniform(-1, 1, size=3))
    print(seed)

To use these fixtures, install with

pip install pytest-rng

Once installed, you can use these fixtures like any other fixture: add rng or seed to the arguments of a test function or class.

Seed generation

For the seed fixture, we generate a seed by doing the following:

  1. Concatenate the test’s nodeid and a salt value, if provided.

  2. Hash that string to yield an integer seed.

For the rng fixture, we also add the string "rng" to the salt value before generating the seed as above. The seed is used to instantiate a ~numpy.random.mtrand.RandomState, which is returned.

salt

salt is a string that is added to the test’s nodeid in order to change the seed for all tests. It is advantageous to change seeds regularly to ensure that your test suite is robust to different seeds.

The salt value can be specified in a configuration file like setup.cfg or pytest.ini.

[tool:pytest]

rng_salt = v0.3.0

The salt value can also be specified through the command line.

pytest --rng-salt "v0.4.0"

The salt value passed through the command line takes precedence over the value set in the configuration file so that you can change seeds on-the-fly.


See the full documentation for more details.

Release History

1.0.0 (August 8, 2019)

Initial release of pytest-rng! Thanks to all of the contributors for making this possible!

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-rng-1.0.0.tar.gz (14.8 kB view hashes)

Uploaded Source

Built Distribution

pytest_rng-1.0.0-py3-none-any.whl (8.0 kB view hashes)

Uploaded Python 3

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