U
    Kvfþ  ã                   @   s‚   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dd	„ ƒZd
d„ Zej dddde ¡ g¡dd„ ƒZdS )é    )Ú	MONTH_ENDN)Ústats)ÚPredictionResultsTF)Úparamsc                 C   sP   t  d¡}t  dd¡}| js$||fS tjddtd}tj||dtj||dfS )Ng      $@é   g      &@z2000-1-1é
   )ZperiodsÚfreq)Úindex)ÚnpZarangeÚparamÚpdZ
date_ranger   ÚSeries)ÚrequestÚmeanZvarianceÚidx© r   úN/tmp/pip-unpacked-wheel-2v6byqio/statsmodels/tsa/base/tests/test_prediction.pyÚdata   s    
r   c                 C   sº   t | d tjƒ}t| d | d ƒ}tj | d |j¡ tj | d |j¡ |r„t |jtjƒs`t	‚t |jtjƒsrt	‚t |j
tjƒs„t	‚| ¡ }t |tjƒsœt	‚t|jddddgkƒs¶t	‚d S )Nr   r   r   Zmean_seZmean_ci_lowerZmean_ci_upper)Ú
isinstancer   r   r   r
   ZtestingZassert_allcloseZpredicted_meanZvar_pred_meanÚAssertionErrorZse_meanZsummary_frameZ	DataFrameÚlistÚcolumns)r   Z	is_pandasÚpredÚframer   r   r   Ú
test_basic   s    ÿr   ÚdistÚnormÚtc                 C   sˆ   |dkrdnd }t | d | d ||d}t | d | d ƒ}| ¡ }| ¡ }|dkrlt t ||k¡¡s„t‚nt t ||k¡¡s„t‚d S )Nr   r   r   r   )r   Údf)r   Zconf_intr
   ÚallZasarrayr   )r   r   r   r   ÚbasicÚciZbasic_cir   r   r   Ú	test_dist%   s    r"   )Zstatsmodels.compat.pandasr   Znumpyr
   Zpandasr   ZpytestZscipyr   Zstatsmodels.tsa.base.predictionr   Zfixturer   r   ÚmarkZparametrizer   r"   r   r   r   r   Ú<module>   s   
	