U md@sddlZddlZddlZddlZddlZzddlZeddZ Wne k rZdZ YnXe e Z dZz eejddZeejj_Wnek rYnXere dn e ddd Zd d Zd d ZeddZeddZdS)NaggTFZSM_TEST_COPY_ON_WRITEzCopy on Write Enabled!zCopy on Write disabledcCsd|jdddd|jdddd|jdddd|jd dd d|jd dd d|jd ddddS)N --skip-slow store_truezskip slow tests)actionhelp --only-slowzrun only slow tests--skip-exampleszskip tests of examples--skip-matplotlibz$skip tests that depend on matplotlib --skip-smokezskip smoke tests --only-smokezrun only smoke tests)Z addoption)parserr M/home/sam/Atlas/atlas_env/lib/python3.8/site-packages/statsmodels/conftest.pypytest_addoption s$rcCsd|jkr |jdr tdd|jkr@|jdr@tdd|jkr`|jdr`tdd |jkr|jd rtd d |jkrtstd d |jkr|jdrtdd |jkr|jdrtddS)NZslowrzskipping due to --skip-slowrzskipping due to --only-slowZexamplerzskipping due to --skip-examples matplotlibr z!skipping due to --skip-matplotlibz)skipping since matplotlib is not intalledZsmoker zskipping due to --skip-smoker zskipping due to --only-smoke)keywordsconfigZ getoptionpytestskipHAVE_MATPLOTLIB)itemr r rpytest_runtest_setup/s         rcCs\zBddl}|dzddlm}|Wntk r>YnXWntk rVYnXdS)Nrr)register_matplotlib_converters)ruseZpandas.plottingr ImportError)rrrr r rpytest_configureGs    rc#sFzddlfdd}Wntk r4dd}YnX|V|dS)a Fixture that closes all figures after a test function has completed Returns ------- closer : callable Function that will close all figures when called. Notes ----- Used by passing as an argument to the function that produces a plot, for example def test_some_plot(close_figures): If a function creates many figures, then these can be destroyed within a test function by calling close_figures to ensure that the number of figures does not become too large. def test_many_plots(close_figures): for i in range(100000): plt.plot(x,y) close_figures() rNcsjddS)Nall)Zpyplotcloser rr rrrszclose_figures..closecSsdS)Nr r r r rrvs)Zmatplotlib.pyplotr)rr rr close_figuresTsrccs,tj}tjddVtj|dS)a' Fixture that set the global RandomState to the fixed seed 1 Notes ----- Used by passing as an argument to the function that uses the global RandomState def test_some_plot(reset_randomstate): Returns the state after the test function exits N)nprandomZ get_stateseedZ set_state)stater r rreset_randomstate}s  r%)loggingosnumpyr!Zpandaspdrrrrr getLogger__name__loggercowboolenvirongetoptionsmodeZ copy_on_writeAttributeErrorcriticalrrrZfixturerr%r r r rs6       (