U
    Kvf1                     @   s   d dl mZ d dlmZ d dlZd dlmZ d dlZ	d dl
Z
d dlmZ d dlmZ d dlmZ erhdndZd	d
 Zdd Zdd Zdd Zdd Zdd ZdS )    )PD_LT_2_2_0)datetimeN)ValueWarning)assert_equal)TimeSeriesModelA-APRYE-APRc            
   	   C   s   dddg} dddg}t j| |d}dddg} t dd	d
g}t j| |d}|d dt|d j }t|d tjt	dd t
|}W 5 Q R X |dd\}}}}	tt|jjd d S )N  3    abcindexd   e   f   r   z%Y-%m-%d %H:%M:%S.%fz1970-01-01 00:00:00.000000100zNo frequency information)match      )pdSeriesto_datetimestrftimestrvaluer   pytestZwarnsr   r   _get_prediction_indexlendatapredict_dates)
r    datessr   Z
actual_strmodstartendout_of_sample_ r)   H/tmp/pip-unpacked-wheel-2v6byqio/statsmodels/tsa/base/tests/test_base.pytest_pandas_nodates_index   s    



r+   c            
      C   s|   t dd} tjddtd}tj| |d}t|}|jjdksBt	|
dd\}}}}|jj}tjddtd}	t||	 d S )	N         B@	1972-4-30	2006-4-30freqr   )zY-APRr   r   z	2016-4-30)nparanger   
date_rangeYE_APRr   r   _indexfreqstrAssertionErrorr   r    r!   r   )
xr"   seriesmodelr%   r&   r'   r(   r!   Zexpected_datesr)   r)   r*   test_predict_freq+   s    

r<   c                  C   sH   t dd} tjddtd}tj| |d}t|}tt	|j
dd  d S )Nr,   r-   r.   r/   r0   r   z	1970-4-30)r2   r3   r   r4   r5   r   r   nptZassert_raisesKeyErrorr   )r9   r"   r:   r;   r)   r)   r*   test_keyerror_start_dateE   s
    r?   c                  C   sd   t jdddd} tdd}tt j|| d}t|jjd tt j|| d}t	
|jjdk d S )Nz1/1/1990   M)r%   Zperiodsr1   r,   g      5@r   )r   Zperiod_ranger2   r3   r   r   r   r6   r7   r=   Zassert_r    r1   )r"   r9   r;   r)   r)   r*   test_period_indexP   s    rB   c                  C   sj   dddg} dddg}t |}t j| |dd}t jd| it j|d	d
d}t|d }t|jj|j	 d S )Nr	   r
   r   z2016-01-01 12:00:00z2016-02-01 12:00:00z2016-03-01 12:00:00Zprice)r    r   nameZMSr0   )r    r   )
r   r   r   Z	DataFrameZDatetimeIndexr   r   r    r"   r   )r    r"   Zdatetime_datesresultZdfr;   r)   r)   r*   test_pandas_dates\   s    


rE   c                  C   s   t jdddd} t td| d d }t|}ddtddd| d g}d	dtd
dd| d g}d}|D ](}|D ]}t|||d d | qrqjd S )Nz
1970-01-01z
1990-01-01ZYS)r%   r&   r1   
   r,   z
1971-01-01i  r@   i  )r,   	         )	r   r4   r   r2   zerosr   r   r   r   )r   Zendogr;   Zpredict_startsZpredict_endsZdesiredr%   r&   r)   r)   r*   test_get_predict_start_endk   s    rL   )Zstatsmodels.compat.pandasr   r   Znumpyr2   Znumpy.testingZtestingr=   Zpandasr   r   Zstatsmodels.tools.sm_exceptionsr   Zstatsmodels.tools.testingr   Zstatsmodels.tsa.base.tsa_modelr   r5   r+   r<   r?   rB   rE   rL   r)   r)   r)   r*   <module>   s   