
    !g1                         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y)    )PD_LT_2_2_0)datetimeN)ValueWarning)assert_equal)TimeSeriesModelA-APRYE-APRc                     g d} g d}t        j                  | |      }g d} t        j                  g d      }t        j                  | |      }|d   j                  d      t	        |d   j
                        z   }t        |d       t        j                  t        d	      5  t        |      }d d d        j                  dd
      \  }}}}	t        t        |j                  j                        d       y # 1 sw Y   JxY w)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
             a/var/www/dash_apps/app1/venv/lib/python3.12/site-packages/statsmodels/tsa/base/tests/test_base.pytest_pandas_nodates_indexr,      s    DE
		$e$A DNN?+E
		$e$A(##$:;eAhnn%&J<=	l*D	E !a ! $'#<#<Q#B E3qSXX++,a0	! !s   &C;;Dc                     t        j                  dd      } t        j                  ddt              }t        j
                  | |      }t        |      }|j                  j                  dv sJ |j                  dd      \  }}}}|j                  j                  }t        j                  ddt              }	t        ||	       y )	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
 MM+{@EYYq&FF#E <<#???? 	##K= !E3q JJ,,M ]];&IN/    c                      t        j                  dd      } t        j                  ddt              }t        j
                  | |      }t        |      }t        j                  t        |j                  dd        y )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   sZ    
		!CA MM+{@EYYq&FF#Eh ; ;[$Or?   c                  x   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(         y )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}     OO*bsCE
		!SABIIau56E%%s+BIIau56EKK

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                         y )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   \   ss    DQE^^E*NYYDWEF	GT?"2B2B5t2T	UBBwK(E!!6<<0r?   c                  P   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 ](  }|D ]!  }t        |j                  ||      d d |       # * y )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    MM,TJEIIbhhrlE#2J/EE"Exa';U1XFNhtQ&:E"IFLG O 	OC44UC@!DgN	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      sI    1      8 2 :X1604P	(1Or?   