U
    md                     @   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se ZedZeeejejfse 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<   



 
