
    "gW                     ~    d dl Zd dlZd dlmZmZmZmZ d dl	m
Z
mZ d Zd Zd Zd Zd Zd	 Zd
 Zd Zd Zd Zd Zy)    N)assert_assert_equalassert_allcloseassert_raises)specificationparamsc            	         t        j                  dggdg      } t        j                  | dd      }t	        j
                  |      }t        |j                  |       t        |j                  dg       t        |j                  dg       t        |j                  d	g       t        |j                  d
g       t        |j                  dg       t        |j                  g d       t        |j                  d       t        |j                  d       t        |j                   d       t        |j"                  d       t        |j$                  d       t        |j&                  d       t        |j                  t(        j*                         t        |j,                  t(        j*                  g       t        |j.                  t(        j*                  g       t        |j0                  t(        j*                  g       t        |j2                  t(        j*                  g       t        |j4                  t(        j*                         t        |j6                  j8                  t(        j:                  dt(        j*                  f          t        |j<                  j8                  t(        j:                  dt(        j*                  f          t        |j>                  j8                  t(        j:                  ddddt(        j*                  f          t        |j@                  j8                  t(        j:                  ddddt(        j*                  f          t        |jB                  j8                  t(        j:                  dt(        j*                  gdz  f          t        |jD                  j8                  t(        j:                  dt(        j*                  gdz  f          tG        |jH                          tG        |jJ                          tM        tN        |jP                  d       tM        tN        |jP                  d       t(        j*                  gt(        j*                  gt(        j*                  gt(        j*                  gt(        j*                  gt(        j*                  d}t        |jS                         |       t        jT                  t(        j*                  g|j&                  z  |j                        }tW        |jY                         |       t        j                  ddggddg      } t        j                  | dd      }t	        j
                  |      }t        |j                  d       t        |j                  d       t        |j                   d       t        |j"                  d       t        |j$                  d       t        |j&                  d       y )Nr   acolumns   r   r   r   r   r      exogorderseasonal_orderspecar.L1ma.L1ar.S.L4ma.S.L4)r
   r   r   r   r   sigma2r         is_stationaryis_invertible)exog_params	ar_params	ma_paramsseasonal_ar_paramsseasonal_ma_paramsr   )indexb)   r      )r   r   r   r   r(   r'      )-pd	DataFramer   SARIMAXSpecificationr   SARIMAXParamsr   r   
exog_namesar_namesma_namesseasonal_ar_namesseasonal_ma_namesparam_namesk_exog_paramsk_ar_paramsk_ma_paramsk_seasonal_ar_paramsk_seasonal_ma_paramsk_paramsnpnanr!   r"   r#   r$   r   ar_polycoefr_ma_polyseasonal_ar_polyseasonal_ma_polyreduced_ar_polyreduced_ma_polyr   is_completeis_validr   
ValueError__getattribute__to_dictSeriesr   	to_pandas)r   r   pdesireds       d/var/www/dash_apps/app1/venv/lib/python3.12/site-packages/statsmodels/tsa/arima/tests/test_params.py	test_initrN   	   s    <<!u-D--<AD$'A u%gY'gY'$$yk2$$yk2 !+ , !$""''+''+Q 266"rvvh'rvvh'%%x0%%x0266"q"&&y!12q"&&y!12##(("%%1aBFF0B*CD##(("%%1aBFF0B*CD""''q266(Q,)?@""''q266(Q,)?@ 

N*a00/B*a00/BxffXffX!vvh!vvh&&G g&ii4==08H8HIGAKKM7+ <<!Q3*5D--<AD$'A !$""''+''+23    c                     t        j                  dggdg      } t        j                  | dd      }t	        j
                  |      d2fd	}d	_         |        d
_         |        d_         |        d_	         |d       d_
         |dd       d_        t        j                         t        j                          t        j                  g d       t        j                  dg       t        j                  dg       t        j                  dg       t        j                  dg       t        j                  dg       t        j                  d       t        j                   j"                  t$        j&                  d          t        j(                  j"                  t$        j&                  d          t        j*                  j"                  t$        j&                  d          t        j,                  j"                  t$        j&                  d          t        j.                  j"                  t$        j&                  d          t        j0                  j"                  t$        j&                  d          dg_        dg_        d g_        d!g_	        d"g_
        d#g_        g d$_        t        j                  g d$       t        j                  d%g       t        j                  d&g       t        j                  d'g       t        j                  d(g       t        j                  d)g       t        j                  d#       t%        j2                  d#      _        t%        j2                  d"      _        t%        j2                  d!      _        t%        j2                  d       _	        t%        j2                  d      _
        t%        j2                  d      _        t        j                  g d*       t        j                  d+g       t        j                  d)g       t        j                  d(g       t        j                  d'g       t        j                  d&g       t        j                  d       g d$_        t        j                  g d$       t        j                  d%g       t        j                  d&g       t        j                  d'g       t        j                  d(g       t        j                  d)g       t        j                  d#       t        j                   j"                  t$        j&                  d,          t        j(                  j"                  t$        j&                  d-          t        j*                  j"                  t$        j&                  d.          t        j,                  j"                  t$        j&                  d/          t        j.                  j"                  t$        j&                  d0          t        j0                  j"                  t$        j&                  d1          y )3Nr   r
   r   r   r   r   r   c                 :   t        j                          t        j                          | dk(  rt        t        j
                  d       nt        j                  |        |dk(  rt        t        j
                  d       y t        j                  |       y Nraiser   r   	r   rD   rE   r   rF   rG   r   r   r   r   r   rK   s     rM   checkz%test_set_params_single.<locals>.checkZ   p    AMM!"AJJG#*a&8&8/J-8G#*a&8&8/J-8rO                           Fr          r   r         ra   rb   rc   rd   re   )r   r   )r   rc   )r   r   r   r   r'   )r   r   r   r   re   )r   r   r   r   r'      )r   rc   r   r   re            ?g       @g      @g      @g      @      @r   r(   r'   r   r   r   r   r(   r'   r   r   )r   r   r   r'   r(   r   r   )r   re   )r   r'   )r   r   r   r   rc   )r   r   r   r   r   )r   re   r   r   rc   rh   )r   r'   r   r   r   rg   rS   rS   )r*   r+   r   r,   r   r-   r    r!   r"   r#   r$   r   r   rD   rE   r   r<   r=   r:   r>   r?   r@   rA   rB   rC   arrayr   r   rV   rK   s      @rM   test_set_params_singlero   Q   s    <<!u-D--<AD$'A
9 AM	GAK	GAK	GA	A	U3AHAMM

N34%rd#rd#%%t,%%t,3 t-u.##(("%%*>?##(("%%*?@""''/@)AB""''/A)BC DAM$AK$AK4A4AtAH!AH-.$qc"qc"%%s+%%s+2 HHRLAM((2,AK((2,AK88B<A88B<Axx|AH-.$qc"qc"%%s+%%s+2 "AH-.$qc"qc"%%s+%%s+2 u.t-##(("%%*?@##(("%%*>?""''/A)BC""''/@)ABrO   c                     t        j                  dggdg      } t        j                  | ddgdddgfddgdddgdf      }t	        j
                  |      d)fd	}d	_         |        d
_         |        d_         |        d_	         |d       d_
         |dd       d_        t        j                         t        j                          t        j                  g d       t        j                  dg       t        j                  dg       t        j                  dg       t        j                  dg       t        j                  dg       t        j                  d       t        j                   j"                  g d       t        j$                  j"                  g d       t        j&                  j"                  g d       t        j(                  j"                  g d       t        j*                  j"                  g d       t        j,                  j"                  g d       g d_        t        j                  g d       t        j                  dg       t        j                  dg       t        j                  d g       t        j                  dg       t        j                  d!g       t        j                  d"       t        j                   j"                  t.        j0                  d#          t        j$                  j"                  t.        j0                  d$          t        j&                  j"                  g d%       t        j(                  j"                  g d&       t        j*                  j"                  g d'       t        j,                  j"                  g d(       y )*Nr   r
   r   r   r   r   r   c                 :   t        j                          t        j                          | dk(  rt        t        j
                  d       nt        j                  |        |dk(  rt        t        j
                  d       y t        j                  |       y rR   rT   rU   s     rM   rV   z4test_set_params_single_nonconsecutive.<locals>.check   rW   rO   rX   rY   rZ   r[   Fr\   r]   r^   r_   r`   ra   rb   rc   rd   re   )r   r   r   )r   r   rc   )	r   r   r   r   r   r   r   r   r'   )	r   r   r   r   r   r   r   r   re   )r   r   r   r   r   r   r   r   r'   r   rg   )r   r   rc   r   r   r   r   r   re   r   rh   rk   r(   r'   r   rj   )r   r   re   )r   r   r'   )	r   r   r   r   r   r   r   r   rc   )	r   r   r   r   r   r   r   r   r   )r   r   re   r   r   r   r   r   rc   r   rh   )r   r   r'   r   r   r   r   r   r   r   rg   rl   )r*   r+   r   r,   r   r-   r    r!   r"   r#   r$   r   r   rD   rE   r   r<   r=   r?   r@   rA   rB   rC   r:   r>   rn   s      @rM   %test_set_params_single_nonconsecutiverr      s   
 <<!u-D--1a&!aV,AAq61-/D 	$'A
9 AM	GAK	GAK	GA	A	U3AHAMM

N34%rd#rd#%%t,%%t,3 +,##((*EF##((*FG""'')KL""'')LM "AH-.$qc"qc"%%s+%%s+2 x1w0##((*FG##((*EF""'')LM""'')KLrO   c                     t        j                  ddggddg      } t        j                  | dd      }t	        j
                  |      }g d	|_        t        |j                  g d	       t        |j                  d
dg       t        |j                  ddg       t        |j                  ddg       t        |j                  ddg       t        |j                  ddg       t        |j                  d       t        |j                  j                  t        j                   d          t        |j"                  j                  t        j                   d          t        |j$                  j                  t        j                   d          t        |j&                  j                  t        j                   d          t        |j(                  j                  g d       t        |j*                  j                  g d       y )Nr   r
   r&   r   r(   r   r(   r(   r   r(   r   r   r   rf   r(   rd   r   rb   r   rh   
   rf   r(   rd   r   rb   r   rw   rh   rx   ry   rz   r   r'   rc   r   rb   r   	r   r   r   r      r   r   r   	r   r   r   r   rx   r   r   r   ry   )r   r'   rc   r   r~      ir   r   i    )r   rb   r   r   rx   -   ir   ry   i<   )r*   r+   r   r,   r   r-   r   r    r!   r"   r#   r$   r   r<   r=   r:   r>   r?   r@   rA   rB   rC   r   r   rK   s      rM   test_set_params_multipler     sy    <<!Q3*5D--<AD$'A8AH:<Q(r1g&r1g&%%Aw/%%Bx03 x1x1##(("%%0J*KL##(("%%0K*LM ""''NP ""''./rO   c                     t        j                  ddggddg      } t        j                  | dd      }t	        j
                  |      }d	d
g|_        t        |j                  dg       t        j                  j                  d	dg      |_        t        |j                  dg       d	dg|_        t        |j                  dg       t        j                  j                  d	dg      |_        t        |j                  dg       g d|_        t        |j                  dg       t        j                  j                  g d      |_        t        |j                  dg       g d|_        t        |j"                  dg       t        j                  j                  g d      |_        t        |j"                  dg       t%        t&        |j(                  ddd
g       t%        t&        |j(                  dddg       t%        t&        |j(                  dg d       t%        t&        |j(                  dg d       t%        t&        |j(                  dd	       t%        t&        |j(                  dd	g       t%        t&        |j(                  dd	       t%        t&        |j(                  dd	g       t%        t&        |j(                  dd	       t%        t&        |j(                  dd	g       t%        t&        |j(                  dg d       t%        t&        |j(                  dd	       t%        t&        |j(                  dd	g       t%        t&        |j(                  dg d       t%        t&        |j(                  dg d        t%        t&        |j(                  dg d!       t%        t&        |j(                  dg d"       t%        t&        |j(                  dg d"       t%        t&        |j(                  dg d#       t%        t&        |j(                  dg d$       y )%Nr   r
   r&   r   r   r   r   r   r               ?皙皙?333333?ffffff?)r   r   r   r   皙ɿ皙?)r   r   r   r         п      ?)r   r   r   r   皙?r   )r   r   r   r   333333?r   r<   r(   r?   r@   )r(   r   r   r   r   rA   )r(   r   r   r   r   )r   r   r   r   )r   r   r   )r   r   r   )r   r   r   r   r   r   )r   r   r   r   r   r   )r   r   r   r   r   r   r*   r+   r   r,   r   r-   r<   r   r!   r:   
polynomial
Polynomialr?   r"   r@   r#   rA   r$   r   rF   __setattr__r   s      rM   test_set_poly_short_lagsr   9  s   <<!Q3*5D--<AD$'A D	AIse$((!U4AItf%CAIse$((!T3AItf%+A%%u-112EFA%%v.*A%%u-112DEA%%v. *ammYD	B*ammYCA*amm-?$&*amm-?#% *ammY:*ammY<*ammY:*ammY<*amm-?C*amm-?!E*amm-?N*amm-?C*amm-?!E*amm-?N*ammYG*ammYF*amm-?&(*amm-?&( *amm-?')*amm-?&(rO   c                     t        j                  ddggddg      } t        j                  | ddgdddgfddgdddgdf      }t	        j
                  |      }g d	|_        t        |j                  d
g       t        j                  j                  g d      |_        t        |j                  dg       g d|_        t        |j                  dg       t        j                  j                  g d      |_        t        |j                  dg       g d|_        t        |j                  dg       t        j                  j                  g d      |_        t        |j                  dg       g d|_        t        |j"                  dg       t        j                  j                  g d      |_        t        |j"                  dg       t%        t&        |j(                  dg d       t%        t&        |j(                  dg d       t%        t&        |j(                  dg d       t%        t&        |j(                  dg d        y )!Nr   r
   r&   r   r   r   r   r   )r   r   r   r   )r   r   r   r   )r   r   r   r   )r   r   r   r   )	r   r   r   r   r   r   r   r   r   r   )	r   r   r   r   r   r   r   r   r   r   )	r   r   r   r   r   r   r   r   r   r   )	r   r   r   r   r   r   r   r   r   r   r<   )r   r   r   r?   )r   r   r   r@   )	r   r   r   r   ri   r   r   r   r   rA   )	r   r   r   r   ri   r   r   r   r   r   r   s      rM   'test_set_poly_short_lags_nonconsecutiver   t  s   <<!Q3*5D--1a&!aV,AAq61-/D 	$'A AIse$((7AItf%AIse$((6AItf%7A%%u-
  !@A %%v.6A%%u-
  !?@ %%v.
 *ammYE*ammYD*amm-?13*amm-?02rO   c                     t        j                  ddggddg      } t        j                  | dd      }t	        j
                  |      }d	d
g|_        d|_        t        j                  d   |_
        t        j                  d   |_        t        j                  d   |_        t        j                  d   |_        t        |j                  g d       t        |j                  d	d
g       t        |j                  ddg       t        |j                   ddg       t        |j"                  ddg       t        |j$                  ddg       t        |j                  d       y )Nr   r
   r&   r   rt   ru   r   r   rf   r(   rz   r{   r|   r}   r   rv   rd   r   rb   r   rw   rh   rx   ry   )r*   r+   r   r,   r   r-   r    r   r:   r>   r<   r?   r@   rA   r   r!   r"   r#   r$   r   s      rM   test_set_poly_longer_lagsr     s   <<!Q3*5D--<AD$'A GAMAH hAIhAI9:A:;A :<Q(r1g&r1g&%%Aw/%%Bx03rO   c                     t        j                  dd      } t        j                  |       }t	        t
        |j                  d       dg|_        dg|_        t        |j                         dg|_        t        |j                          dg|_        dg|_        t        |j                         dg|_        t        |j                          d	g|_        d	g|_        t        |j                         d
g|_        d
g|_        t        |j                         dg|_        dg|_        t        |j                          y )Nr   r   r   r   r   r   r   r   ri   r   Gz?)r   r,   r   r-   r   rF   rG   r!   r#   r   r   r   rK   s     rM   test_is_stationaryr     
   --6D$'A *a00/B%AK3AAOO%AK #AK5AAOO5A %AK5AAOO&AK 6AAOO$AK4A rO   c                     t        j                  dd      } t        j                  |       }t	        t
        |j                  d       dg|_        dg|_        t        |j                         dg|_        t        |j                          dg|_        dg|_        t        |j                         dg|_        t        |j                          d	g|_        d	g|_        t        |j                         d
g|_        d
g|_        t        |j                         dg|_        dg|_        t        |j                          y )Nr   r   r   r   r   r   r   ri   r   r   )r   r,   r   r-   r   rF   rG   r"   r$   r   r   r   s     rM   test_is_invertibler     r   rO   c                     t        j                  dddd      } t        j                  |       }t	        |j
                          g d|_        t	        |j
                         g d|_        t	        |j
                          g d|_        t	        |j
                          g d	|_        t	        |j
                          g d
|_        t	        |j
                          y )Nr   r   T)r   r   enforce_stationarityenforce_invertibilityr   )r   r   r   r   ri   )ri   r   r   r   ri   )r   ri   r   r   ri   )r   r   ri   r   ri   )r   r   r   ri   ri   )r   r,   r   r-   r   rE   r   s     rM   test_is_validr     s     --!?D 	$'A 

N'AHAJJ&AH

N&AH

N&AH

N&AH

NrO   c                     t        j                  ddggddg      } t        j                  | dd      }t	        j
                  |      }t        t        |      d	       d
dg|_        t        t        |      d       dg|_	        t        t        |      d       dg|_
        t        t        |      d       dg|_        t        t        |      d       dg|_        t        t        |      d       d|_        t        t        |      d       y )Nr   r
   r&   r   r   r   r   r   zcSARIMAXParams(exog=[nan nan], ar=[nan], ma=[nan], seasonal_ar=[nan], seasonal_ma=[nan], sigma2=nan)r   r(   zaSARIMAXParams(exog=[1. 2.], ar=[nan], ma=[nan], seasonal_ar=[nan], seasonal_ma=[nan], sigma2=nan)r   zaSARIMAXParams(exog=[1. 2.], ar=[0.5], ma=[nan], seasonal_ar=[nan], seasonal_ma=[nan], sigma2=nan)r   zaSARIMAXParams(exog=[1. 2.], ar=[0.5], ma=[0.2], seasonal_ar=[nan], seasonal_ma=[nan], sigma2=nan)gMbP?zcSARIMAXParams(exog=[1. 2.], ar=[0.5], ma=[0.2], seasonal_ar=[0.001], seasonal_ma=[nan], sigma2=nan)gMbPzfSARIMAXParams(exog=[1. 2.], ar=[0.5], ma=[0.2], seasonal_ar=[0.001], seasonal_ma=[-0.001], sigma2=nan)g">$@ziSARIMAXParams(exog=[1. 2.], ar=[0.5], ma=[0.2], seasonal_ar=[0.001], seasonal_ma=[-0.001], sigma2=10.123))r*   r+   r   r,   r   r-   r   reprr    r!   r"   r#   r$   r   r   s      rM   test_repr_strr     s   <<!Q3*5D--<AD$'A a O P FAMa O P %AKa O P %AKa O P "7Aa ) * #8Aa ) * AHa , -rO   )numpyr:   pandasr*   numpy.testingr   r   r   r   statsmodels.tsa.arimar   r   rN   ro   rr   r   r   r   r   r   r   r   r    rO   rM   <module>r      sV      O O 7E4PkC\LM^(/V8(v%2P 8!>!>0+-rO   