
    'YHh                         d dl mZ d dlZd dlZd dlZd dlmZ d dl	m
Z
  ej                  ddg      d        Zd	 Zej                  j                  d
ddd ej                          g      d        Zy)    )	MONTH_ENDN)stats)PredictionResultsTF)paramsc                    t        j                  d      }t        j                  dd      }| j                  s||fS t        j                  ddt
              }t        j                  ||      t        j                  ||      fS )Ng      $@   g      &@z2000-1-1
   )periodsfreq)index)nparangeparampd
date_ranger   Series)requestmeanvarianceidxs       c/var/www/html/planif/env/lib/python3.12/site-packages/statsmodels/tsa/base/tests/test_prediction.pydatar      sc    99T?DyyD!H==X~
--
BY
?C99T%ryy'EEE    c                    t        | d   t        j                        }t        | d   | d         }t        j
                  j                  | d   |j                         t        j
                  j                  | d   |j                         |rrt        |j                  t        j                        sJ t        |j                  t        j                        sJ t        |j                  t        j                        sJ |j                         }t        |t        j                        sJ t        |j                  g dk(        sJ y )Nr   r   )r   mean_semean_ci_lowermean_ci_upper)
isinstancer   r   r   r   testingassert_allclosepredicted_meanvar_pred_meanse_meansummary_frame	DataFramelistcolumns)r   	is_pandaspredframes       r   
test_basicr+      s    47BII.IT!Wd1g.DJJtAw(;(;<JJtAw(:(:;$--ryy999$,,bii888$,,		222 EeR\\***NN  r   distnormtc                 d   |dk(  rdnd }t        | d   | d   ||      }t        | d   | d         }|j                         }|j                         }|dk(  r.t        j                  t        j                  ||k7              sJ y t        j                  t        j                  ||k(              sJ y )Nr.   r	   r   r   )r,   df)r   conf_intr   allasarray)r   r,   r0   r)   basiccibasic_cis          r   	test_distr7   %   s    s{BT!Wd1gDR@Dd1gtAw/E	B~~Hs{vvbjjx0111vvbjjx0111r   )statsmodels.compat.pandasr   numpyr   pandasr   pytestscipyr   statsmodels.tsa.base.predictionr   fixturer   r+   markparametrizer-   r7    r   r   <module>rB      sx    /     = e}%F &F  $ZUZZ\!BC	2 D	2r   