
    M/Ph1                         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rdndZd	 Zd
 Zd Zd Zd Zd ZdS )    )PD_LT_2_2_0)datetimeN)ValueWarning)assert_equal)TimeSeriesModelA-APRYE-APRc                  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
             d/var/www/html/test/jupyter/venv/lib/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t                    }t          j        | |          }t          |          }|j        j        dv sJ |	                    dd          \  }}}}|j
        j        }t          j        ddt                    }	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freqstrr   r!   r"   r   )
xr#   seriesmodelr'   r(   r)   r*   r"   expected_datess
             r+   test_predict_freqr>   +   s    
	!CA
 M+{@@@EYq&&&FF##E <#????? 	##K== !E3q J,M ];&IIIN/////    c                      t          j        dd          } t          j        ddt                    }t          j        | |          }t          |          }t          j        t          |j
        dd            d S )Nr.   r/   r0   r1   r2   r   z	1970-4-30)r4   r5   r   r6   r7   r   r   nptassert_raisesKeyErrorr   )r:   r#   r;   r<   s       r+   test_keyerror_start_daterD   E   sl    
	!CA M+{@@@E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'   periodsr3   r.   g      5@r   )r   period_ranger4   r5   r   r   r   r8   r9   rA   assert_r!   r3   )r#   r:   r<   s      r+   test_period_indexrK   P   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MSr2   )r!   r   )
r   r   r   	DataFrameDatetimeIndexr   r   r!   r#   r   )r!   r#   datetime_datesresultdfr<   s         r+   test_pandas_datesrU   \   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YS)r'   r(   r3   
   r.   z
1971-01-01i  rF   i  )r.   	         )	r   r6   r   r4   zerosr   r   r   r   )r   endogr<   predict_startspredict_endsdesiredr'   r(   s           r+   test_get_predict_start_endrb   k   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?   )statsmodels.compat.pandasr   r   numpyr4   numpy.testingtestingrA   pandasr   r   statsmodels.tools.sm_exceptionsr   statsmodels.tools.testingr   statsmodels.tsa.base.tsa_modelr   r7   r,   r>   rD   rK   rU   rb    r?   r+   <module>rl      s   1 1 1 1 1 1                      8 8 8 8 8 8 2 2 2 2 2 2 : : : : : :	-X1 1 160 0 04P P P	( 	( 	(1 1 1O O O O Or?   