
    Vfd                     ~    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 d Zd Zd Zd	 Zd
 Zd ZdS )    )datetimeN)ValueWarning)assert_equal)TimeSeriesModelc                  6   g d} g d}t          j        | |          }g d} t          j        g d          }t          j        | |          }|d                             d          t	          |d         j                  z   }t          |d           t          j        t          d	          5  t          |          }d d d            n# 1 swxY w Y   |                    dd
          \  }}}}	t          t          |j        j                  d           d S )Ni  i3  i  )abcindex)d   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   pytestwarnsr   r   _get_prediction_indexlendatapredict_dates)
r   datessr   
actual_strmodstartendout_of_sample_s
             Dlib/python3.11/site-packages/statsmodels/tsa/base/tests/test_base.pytest_pandas_nodates_indexr)      sT   ??DOOE
	$e$$$A ??DN???++E
	$e$$$A(##$:;;eAhn%%&J<===	l*D	E	E	E ! !a  ! ! ! ! ! ! ! ! ! ! ! ! ! ! ! $'#<#<Q#B#B E3qSX+,,a00000s   0CCCc                  x   t          j        dd          } t          j        ddd          }t          j        | |          }t          |          }t          |j        j        d           |	                    dd          \  }}}}|j
        j        }t          j        ddd          }	t          ||	           d S )	N         B@	1972-4-30	2006-4-30A-APRfreqr   z	2016-4-30)nparanger   
date_ranger   r   r   _indexfreqstrr   r   r   )
xr    seriesmodelr$   r%   r&   r'   r   expected_datess
             r(   test_predict_freqr;   (   s    
	!CA M+{AAAEYq&&&FF##E%w/// 	##K== !E3q J,M ];'JJJN/////    c                      t          j        dd          } t          j        ddd          }t          j        | |          }t          |          }t          j        t          |j	        dd            d S )	Nr+   r,   r-   r.   r/   r0   r   z	1970-4-30)
r2   r3   r   r4   r   r   nptassert_raisesKeyErrorr   )r7   r    r8   r9   s       r(   test_keyerror_start_daterA   @   sl    
	!CA M+{AAAEYq&&&FF##Eh ;[$OOOOOr<   c                  b   t          j        ddd          } t          j        dd          }t	          t          j        ||                     }t          |j        j        d           t	          t          j        ||                     }t          j
        |j        j        dk               d S )Nz1/1/1990   M)r$   periodsr1   r+   g      5@r   )r   period_ranger2   r3   r   r   r   r5   r6   r>   assert_r   r1   )r    r7   r9   s      r(   test_period_indexrH   K   s     O*bsCCCE
	!SABIau55566E%s+++BIau55566EK
3&'''''r<   c                  ,   g d} g d}t          j        |          }t          j        | |d          }t          j        d| it          j        |d                    }t          |d                   }t          |j        j        |j	                   d S )Nr   )z2016-01-01 12:00:00z2016-02-01 12:00:00z2016-03-01 12:00:00price)r   r   nameMSr0   )r   r   )
r   r   r   	DataFrameDatetimeIndexr   r   r   r    r   )r   r    datetime_datesresultdfr9   s         r(   test_pandas_datesrR   W   s    ??DQQQE^E**NYDWEEEF	GT?"2B5t2T2T2T	U	U	UBBwK((E!6<00000r<   c                     t          j        ddd          } t          j        t          j        d          | d d                   }t          |          }ddt          ddd          | d         g}d	dt          d
dd          | d         g}d}|D ]3}|D ].}t          |                    ||          d d         |           /4d S )Nz
1970-01-01z
1990-01-01AS)r$   r%   r1   
   r+   z
1971-01-01i  rC   i  )r+   	         )	r   r4   r   r2   zerosr   r   r   r   )r   endogr9   predict_startspredict_endsdesiredr$   r%   s           r(   test_get_predict_start_endr_   f   s    M,TJJJEIbhrllE#2#J//EE""Exa';';U1XFNhtQ&:&:E"IFLG O O 	O 	OC44UC@@!DgNNNN	OO Or<   )r   numpyr2   numpy.testingtestingr>   pandasr   r   statsmodels.tools.sm_exceptionsr   statsmodels.tools.testingr   statsmodels.tsa.base.tsa_modelr   r)   r;   rA   rH   rR   r_    r<   r(   <module>rh      s                         8 8 8 8 8 8 2 2 2 2 2 2 : : : : : :1 1 160 0 00P P P	( 	( 	(1 1 1O O O O Or<   