
    ,YHhn	                     j   d dl mZmZ d dlmZ d dlmZ d dlmZ d dl	Z
d dlZ edddd	      Z ed
ddd      ZdZd Zd Zd Zej$                  j'                  d eg dddgd       eeedd       eg g g d       eg g g d       edgdgg dd       eg g g d      g      d         Zd! Zd" Zy)#    )approx_regularize)c)ADFTest)assert_array_almost_equalNgS?gI+?g㥛 ?gX9v?{Gz?g?g?g?)\(?c                      t        dddd      t        dddd      }} t        | |d      \  } }t        | t        j                  g d	             t        |t        j                  g d
             y )N      ?      ?      ?            mean)r   r   r   )r   g      @g      @)r   r   r   nparrayxys     Y/var/www/html/planif/env/lib/python3.12/site-packages/pmdarima/arima/tests/test_approx.pytest_regularizer      sX    S#sC !Aq!Q-qAq!V$DAqa/!:;a/!:;    c                      t        t        t        t        d      \  } }t	        | t        d             t	        |t        t        j                               y )Nr   ruler	   )r   tabletablepstatr   r   r   nanr   s     r   test_approx_rule1r"      s6    %A.DAqa4)a266+r   c                      t        t        t        t        d      \  } }t	        | t        d             t	        |t        d             y )Nr   r   r	   r   )r   r   r   r    r   r   r   s     r   test_approx_rule2r$       s2    %A.DAqa4)a4)r   kwargs)r   r   r   r   r   r   )r   r   xoutz
bad-string)r   r   r&   methodr   )r   r   r&   tiesconstantlinearorderedr   r   r&   r'   r(   c                 v    t        j                  t              5  t        di |  d d d        y # 1 sw Y   y xY w)N )pytestraises
ValueErrorr   )r%   s    r   test_corner_errorsr2   '   s0    0 
z	"   s   /8c                  (    t        dgdgg dd       y )Nr   r)   r+   r,   )r   r.   r   r   test_valid_cornerr4   C   s    
aSQCb)Dr   c            
          t        j                  dgdgdgdgdgdgdgdgg      } t        | t        j                  d	d
      \  }}t        j
                  |d      sJ y )NgNё\Cg8EGrgV-gsA	gk	gg6;NёgOegX5;Nѿg"nNer   )r&   r   g
f)?)r   r   r   r   r   allclose)tableipl_interpols      r   test_approx_precisionr:   H   sl     xx'! !!!!!$ %H 7>>	JKAx;;x,,,r   )pmdarima.arima.approxr   r   pmdarima.utils.arrayr   pmdarima.arima.stationarityr   numpy.testingr   numpyr   r/   r   r   r    r   r"   r$   markparametrizedictr2   r4   r:   r.   r   r   <module>rC      s    6 " / 3  	%u%	
4d	#<,*  	yQF- 	uC= 	rRbv. 	rRb4 	sqc8)D 	rRb4%./.
E
-r   