U md @sFdZddlZddlmZddlmZGdddZddZd d ZdS) azglobal_random_seed fixture The goal of this fixture is to prevent tests that use it to be sensitive to a specific seed value while still being deterministic by default. See the documentation for the SKLEARN_TESTS_GLOBAL_RANDOM_SEED variable for insrtuctions on how to use this fixture. https://scikit-learn.org/dev/computing/parallelism.html#sklearn-tests-global-random-seed N)environ)Randomc@seZdZddddZdS) XDistHooksN)returncCs|jd}||jd<dS)N random_seeds)configZ getoption workerinput)selfnoderr R/home/sam/Atlas/atlas_env/lib/python3.8/site-packages/sklearn/tests/random_seed.pypytest_configure_nodes z XDistHooks.pytest_configure_node)__name__ __module__ __qualname__r r r r r rsrcs|jdr|jtttd}td}t|drPd|j krP|j dn|dkr`dgn|dkrxt |gnp|dkr|nbd |kr| d \}}ttt |t |d n t |gtd kstd krtd ||j_Gfddd}|j|dS)NZxdistd SKLEARN_TESTS_GLOBAL_RANDOM_SEEDrr*anyall-rczThe value(s) of the environment variable SKLEARN_TESTS_GLOBAL_RANDOM_SEED must be in the range [0, 99] (or 'any' or 'all'), got: cs eZdZejdddZdS)z0pytest_configure..GlobalRandomSeedPlugin)paramscss |jVdS)aFixture to ask for a random yet controllable random seed. All tests that use this fixture accept the contract that they should deterministically pass for any seed value from 0 to 99 included. See the documentation for the SKLEARN_TESTS_GLOBAL_RANDOM_SEED variable for insrtuctions on how to use this fixture. https://scikit-learn.org/dev/computing/parallelism.html#sklearn-tests-global-random-seed N)param)r requestr r r global_random_seed8s zCpytest_configure..GlobalRandomSeedPlugin.global_random_seedN)rrrpytestZfixturerr rr r GlobalRandomSeedPlugin7s r)Z pluginmanagerZ haspluginregisterrlistrangergethasattrrrchoicesplitintminmax ValueErroroptionr)rZRANDOM_SEED_RANGErandom_seed_varstartstoprr rr pytest_configures.     r/cCs0td}|dkr,dd|jjdddgSdS)NrrzCTo reproduce this test run, set the following environment variable:z& SKLEARN_TESTS_GLOBAL_RANDOM_SEED="r"z]See: https://scikit-learn.org/dev/computing/parallelism.html#sklearn-tests-global-random-seed)rr#r+r)rr,r r r pytest_report_headerIs  r1) __doc__rosrrandomrrr/r1r r r r s    2