o 0Gf@sddlZddlZddlZddlZddlZz ddlZeddZ Wn e y,dZ Ynwe e Z dZzeejddZeejj_Wn eyMYnwerVe dne 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 4lib/python3.10/site-packages/statsmodels/conftest.pypytest_addoption s$ rcCsd|jvr|jdrtdd|jvr |jdr tdd|jvr0|jdr0tdd |jvr@|jd r@td d |jvrLtsLtd d |jvr\|jdr\tdd |jvrn|jdrptddSdSdS)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"        rcCsZz"ddl}|dz ddlm}|WWdSty"YWdSwty,YdSw)Nrr)register_matplotlib_converters)ruseZpandas.plottingr ImportError)rrrr r rpytest_configureGs    rc#sFz ddlfdd}Wn tydd}Ynw|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_figuresTs   rccs.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)npZrandomZ get_stateZseedZ set_state)stater r rreset_randomstate}s   r#)ZloggingosZnumpyr!ZpandaspdrrrrrZ getLogger__name__loggerZcowboolenvirongetoptionsmodeZ copy_on_writeAttributeErrorZcriticalrrrZfixturerr#r r r rs<       (