
    M/Ph                         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                            d
ddd ej                    g          d             ZdS )    )	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       j/var/www/html/test/jupyter/venv/lib/python3.11/site-packages/statsmodels/tsa/base/tests/test_prediction.pydatar      ss    9T??DyD!!H= X~
-
BY
?
?
?C9T%%%ry'E'E'EEE    c                    t          | d         t          j                  }t          | d         | d                   }t          j                            | d         |j                   t          j                            | d         |j                   |rct          |j        t          j                  sJ t          |j        t          j                  sJ t          |j	        t          j                  sJ |
                                }t          |t          j                  sJ t          |j        g dk              sJ d S )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7BI..IT!Wd1g..DJtAw(;<<<JtAw(:;;; 3$-ry99999$,bi88888$,	22222  EeR\*****NNNN      r   distnormtc                    |dk    rdnd }t          | d         | d         ||          }t          | d         | d                   }|                                }|                                }|dk    r.t          j        t          j        ||k                        sJ d S t          j        t          j        ||k                        sJ d S )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{{vbjx001111111vbjx001111111r   )statsmodels.compat.pandasr   numpyr   pandasr   pytestscipyr   statsmodels.tsa.base.predictionr   fixturer   r+   markparametrizer-   r7    r   r   <module>rB      s    / / / / / /                = = = = = = e}%%%F F &%F    $ZUZ\\!BCC	2 	2 DC	2 	2 	2r   