U
    ÃmœdÝ  ã                   @   s(  d dl Zd dlZd dlmZmZ d dlmZmZm	Z	 eƒ Z
ejje
dkdd�Ze ¡ jZejddd	d
�Zee_e ¡ Ze ¡ jZej ¡ ed< e d¡Zeeejejfƒs²e ¡ Ze d¡ZeeejejfƒsÖe ¡ Zeeejejejedg fZdZejeed�dd„ ƒZ dd„ Z!ejj"dd„ ƒZ#dS )é    N)Úco2Ú	macrodata)Ú	_find_x12Úx13_arima_analysisÚx13_arima_select_orderFzX13/X12 not available)ÚreasonZ1959Q1Z2009Q3ÚQ)ÚstartÚendÚfreqr   ÚMZMSÚrealgdp)ZmonthlyZmonthly_startZmonthly_co2Zmonthly_start_co2ZseriesZ	dataframe)ÚparamsÚidsc                 C   s   | j S ©N)Úparam)Úrequest© r   úW/home/sam/Atlas/atlas_env/lib/python3.8/site-packages/statsmodels/tsa/tests/test_x13.pyÚdataset'   s    r   c                 C   s,   t | ƒ}t|jtƒst‚t|jtƒs(t‚d S r   )r   Ú
isinstanceÚorderÚtupleÚAssertionErrorZsorder©r   Úresr   r   r   Útest_x13_arima_select_order,   s    r   c                 C   s   t | ƒ}| ¡  d S r   )r   Zplotr   r   r   r   Útest_x13_arima_plot2   s    r   )$ZpandasÚpdZpytestZstatsmodels.datasetsr   r   Zstatsmodels.tsa.x13r   r   r   Zx13pathÚmarkZskipifZ
pytestmarkZload_pandasÚdataZdtaZperiod_rangeÚindexZdropnaZquarterly_dataZinterpolateZresampleZmonthly_datar   Z	DataFrameZSeriesZmeanZmonthly_start_datar   r   Zfixturer   r   Z
matplotlibr   r   r   r   r   Ú<module>   s<   ÿ



 þ
