
    ^h                        S SK Jr  S SKrS SKr\R
                  " SS9S\R                  4S j5       r\R
                  " SS9S\R                  4S j5       r\R
                  " SS9S\R                  4S j5       r	S	 r
S
 rS rS rg)    )PathNpackage)scopereturnc                  b    [         R                  " [        [        5      R                  S-  S/S9$ )z1Daily univariate time series with 2 years of datazdata.csvdsparse_datespdread_csvr   __file__parent     iC:\Users\julio\OneDrive\Documentos\Trabajo\Ideas Frescas\venv\Lib\site-packages\prophet/tests/conftest.pydaily_univariate_tsr      s&     ;;tH~,,z9vNNr   c                  b    [         R                  " [        [        5      R                  S-  S/S9$ )z Sub-daily univariate time seriesz	data2.csvr   r	   r   r   r   r   subdaily_univariate_tsr      &     ;;tH~,,{:OOr   c                  b    [         R                  " [        [        5      R                  S-  S/S9$ )z8Univariate time series with large values to test scalingz	data3.csvr   r	   r   r   r   r   large_numbers_tsr      r   r   c                 (    U R                  SS5        g )Nmarkersz:slow: mark tests as slow (include in run with --test-slow))addinivalue_line)configs    r   pytest_configurer      s    
I'cdr   c                 N    U R                  SSSSS9  U R                  SSS/S	S
9  g )N--test-slow
store_trueFzRun slow tests)actiondefaulthelpz	--backend+	CMDSTANPYzAProbabilistic Programming Language backend to perform tests with.)nargsr"   r#   )	addoption)parsers    r   pytest_addoptionr)      s;    
]<M]^
P	  r   c                     U R                  S5      (       a  g [        R                  R                  SS9nU H&  nSUR                  ;   d  M  UR                  U5        M(     g )Nr   z1Skipped due to the lack of '--test-slow' argument)reasonslow)	getoptionpytestmarkskipkeywords
add_marker)r   items	skip_slowitems       r   pytest_collection_modifyitemsr6   '   sP    &&  ([ \IT]]"OOI& r   c                 |    SU R                   ;   a,  U R                  SU R                  R                  S5      5        gg)z
For each test, if `backend` is used as a fixture, add a parametrization equal to the value of the
--backend option.

This is used to re-run the test suite for different probabilistic programming language backends
(e.g. cmdstanpy, numpyro).
backendN)fixturenamesparametrizer   r-   )metafuncs    r   pytest_generate_testsr<   0   s6     H)))Y(A(A)(LM *r   )pathlibr   pandasr   r.   fixture	DataFramer   r   r   r   r)   r6   r<   r   r   r   <module>rA      s       i OR\\ O !O
 i P P !P
 i P",, P !P
e'	Nr   