
    !gQ                        d Z ddlZddlZddlmZmZ ddlZddl	Z	ddl
mZ dZej                  j                  ej                  j                  e            ZdZ ej&                  ej                  j)                  ee            Zed   j-                  d      Zed	   j-                  d      Zedk\  Z ej4                  ee         Zee   Z ej6                  ej9                         ej;                         d
      Z G d d      Z G d de      Z  G d de      Z! G d de      Z" G d de      Z# G d de      Z$ G d de      Z%d Z&y)z>

Created on Sat Dec 14 17:23:25 2013

Author: Josef Perktold
    N)assert_allcloseassert_array_less)kernelsz%results/results_kernel_regression.csvaccidentT)copyservice(   c                       e Zd ZdZdZdZdZd Zej                  j                  ej                  j                  d               Zy)CheckKernelMixingffffff?g?皙?333333?c           
         | j                   }| j                  }t        j                  t        D cg c]  }|j                  t        t        |       c}      }|| _        t        d|z      }t        d|z      }d}|d d df   |d d df   z
  |z  }|d d df   }	t        |	|dd       t        |	|d	d
       || _        || _        t        j                  |      }
t        ||
   ||
   | j                  d       t        j                  ||z
        dd|z  z   kD  }t!        | d      sd}n| j"                  }t%        |j'                         |dz          |||z  z   }|||z  z
  }t        j(                  |||f      | _        t        ||
df   ||
   | j,                  d       t        ||
d	f   ||
   | j.                  | j0                         y c c}w )Ns_se_1\?      gƠ>g#B;)rtolatolr   gư>r   	se_n_diffg      @)	kern_namekernnparrayxg
smoothconfxyfittedgresultsr   seres_seisfinitese_rtolabshasattrr   r   sumcolumn_stackres_fittedgupp_rtollow_rtollow_atol)selfr   r   xir   
res_fittedr"   critr!   fittedse_validmaskr   res_uppres_lows                  i/var/www/dash_apps/app1/venv/lib/python3.12/site-packages/statsmodels/nonparametric/tests/test_kernels.pytest_smoothconfz CheckKernelMixin.test_smoothconf4   s   NN	yy((C"DOOAq"5CDTI-.
*+!admgadm+t3A 	
EB
>
 ;;v& 	8fX&6T\\PSTvvb6k"cC&L&89t[)"II$((*i!m4 tf},tf},??GZ+IJ!,gh.?!]]	6!,gh.?!]]	@M Ds   "Gc           
          | j                   }d}t        j                  t        D cg c]  }|j	                  t        t
        |       c}      }y c c}w )Nr   )r   r   r   r   r   r   )r-   r   r0   r.   fitted_xs        r6   test_smoothconf_dataz%CheckKernelMixin.test_smoothconf_datac   s;     yy!88C2T__Q26CDCs   "AN)__name__
__module____qualname__r$   r*   r+   r,   r7   pytestmarkslowsmoker:        r6   r   r   -   sO    GHHH+@^ [[[[E  ErC   r   c                   2    e Zd ZdZ ej
                         Zy)TestEpanepan2N)r;   r<   r=   r   r   Epanechnikovr   rB   rC   r6   rE   rE   l   s    I7!DrC   rE   c                   2    e Zd ZdZ ej
                         Zy)TestGaugauN)r;   r<   r=   r   r   Gaussianr   rB   rC   r6   rI   rI   q   s    I7DrC   rI   c                   F    e Zd ZdZ ej
                         ZdZdZdZ	dZ
dZy)TestUniformrecg?   g?r   N)r;   r<   r=   r   r   Uniformr   r$   r   r*   r+   r,   rB   rC   r6   rM   rM   v   s.    I7??DGIHHHrC   rM   c                   >    e Zd ZdZ ej
                         ZdZdZdZ	y)TestTriangulartri
   g333333?r   N)
r;   r<   r=   r   r   
Triangularr   r   r*   r+   rB   rC   r6   rR   rR      s&    I7DIHHrC   rR   c                        e Zd ZdZ ej
                         Zej                  j                  de
d       fd       Z xZS )
TestCosinecoszNaN mismatchT)reasonraisesstrictc                 "    t         |           y )N)superr7   )r-   	__class__s    r6   r7   zTestCosine.test_smoothconf   s     	!rC   )r;   r<   r=   r   r   Cosine2r   r>   r?   xfailAssertionErrorr7   __classcell__)r^   s   @r6   rW   rW      sC    I7??D[[n,T  ;";"rC   rW   c                   :    e Zd ZdZ ej
                         ZdZdZy)TestBiweightbi	   r   N)	r;   r<   r=   r   r   Biweightr   r   r+   rB   rC   r6   rd   rd      s!    I7DIHrC   rd   c                      g d} t        j                  ddd      } t        j                         |      }t	        || d       y )N)	        6?,R_?)O?    `?g?rl   rk   rj   ri   r   rf   g|=)r   )r   linspacer   Tricuber   )res_kxxxkxs      r6   test_tricubers      s<    
F
 
RA	B		2	BBU+rC   )'__doc__osnumpyr   numpy.testingr   r   pandaspdr>   !statsmodels.sandbox.nonparametricr   DEBUGpathdirnameabspath__file__curdirfnameread_csvjoinr    to_numpyr   r   positivelogrn   minmaxr   r   rE   rI   rM   rR   rW   rd   rs   rB   rC   r6   <module>r      s<   
  <   5		2	3/
"++bggll651
2J  d +IT*6
BFF1X;hKR[[!%%'2&$<E <E~" "
 
" % "! "# ,rC   