
    %gR                        d dl mZ d dlZd dlZ ej
                  d      dej                  fd       Z ej
                  d      dej                  fd       Z ej
                  d      dej                  fd       Z	d	 Z
d
 Zd Zd Zy)    )PathNpackage)scopereturnc                  f    t        j                  t        t              j                  dz  dg      S )z1Daily univariate time series with 2 years of datazdata.csvdsparse_datespdread_csvr   __file__parent     S/var/www/dash_apps/app1/venv/lib/python3.12/site-packages/prophet/tests/conftest.pydaily_univariate_tsr      s&     ;;tH~,,z9vNNr   c                  f    t        j                  t        t              j                  dz  dg      S )z Sub-daily univariate time seriesz	data2.csvr   r	   r   r   r   r   subdaily_univariate_tsr      &     ;;tH~,,{:OOr   c                  f    t        j                  t        t              j                  dz  dg      S )z8Univariate time series with large values to test scalingz	data3.csvr   r	   r   r   r   r   large_numbers_tsr      r   r   c                 (    | j                  dd       y )Nmarkersz:slow: mark tests as slow (include in run with --test-slow))addinivalue_line)configs    r   pytest_configurer      s    
I'cdr   c                 Z    | j                  dddd       | j                  dddgd	
       y )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                     | j                  d      ry t        j                  j                  d      }|D ]"  }d|j                  v s|j                  |       $ y )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   '   sR    &  ([ \I 'T]]"OOI&'r   c                 x    d| j                   v r,| j                  d| j                  j                  d             yy)a  
    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   