
    Vfd?                        d Z ddlZddlZddlmZmZ ddlZddlm	Z	 dZ
ej                            ej                            e                    ZdZ ej        ej                            ee                    Zed         Zed         Zedk    Z ej        ee                   Zee         Z ej        e                                e                                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#dS )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accidentservice(   c                   r    e Zd ZdZdZdZdZd Zej	        j
        ej	        j        d                         ZdS )CheckKernelMixingffffff?g?皙?333333?c                    | j         }| j        t          j        fdt          D                       }|| _        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   k    }	t          | d          sd}
n| j        }
t          |	                                |
dz              |||z  z   }|||z  z
  }t          j        |||f          | _        t          ||df         ||         | j        d	           t          ||d
f         ||         | j        | j        	           d S )Nc                 R    g | ]#}                     t          t          |          $S  
smoothconfxy.0xikerns     Llib/python3.11/site-packages/statsmodels/nonparametric/tests/test_kernels.py
<listcomp>z4CheckKernelMixin.test_smoothconf.<locals>.<listcomp>8   s)    CCC"DOOAq"55CCC    s_se_1\?      gƠ>g#B;)rtolatolr   gư>r   	se_n_diffg      @)	kern_namer   nparrayxgfittedgresultsr   seres_seisfinitese_rtolabshasattrr"   r   sumcolumn_stackres_fittedgupp_rtollow_rtollow_atol)selfr#   r'   
res_fittedr*   critr)   fittedse_validmaskr"   res_uppres_lowr   s                @r   test_smoothconfz CheckKernelMixin.test_smoothconf4   s   N	y(CCCCCCCDDTI-.
*+!aaadmgaaadm+t3A 	
EBBBB
>>>>
 ;v&& 	8fX&6T\PSTTTTvb6k""cC&L&89t[)) 	'"III$((**i!m444 tf},tf},?GZ+IJJ!,gh.?!]	6 	6 	6 	6!,gh.?!]	@ 	@ 	@ 	@ 	@ 	@r   c                 d    | j         d}t          j        fdt          D                       }d S )Nr   c                 R    g | ]#}                     t          t          |          $S r   r   r   s     r   r   z9CheckKernelMixin.test_smoothconf_data.<locals>.<listcomp>i   s)    CCC2T__Q266CCCr   )r   r$   r%   r   )r5   r7   fitted_xr   s      @r   test_smoothconf_dataz%CheckKernelMixin.test_smoothconf_datac   s9     y!8CCCCCCCDDr   N)__name__
__module____qualname__r,   r2   r3   r4   r=   pytestmarkslowsmokerA   r   r   r   r
   r
   -   sq        GHHH+@ +@ +@^ [[E E  E E Er   r
   c                   0    e Zd ZdZ ej                    ZdS )TestEpanepan2N)rB   rC   rD   r#   r   Epanechnikovr   r   r   r   rJ   rJ   l   s#        I7!!DDDr   rJ   c                   0    e Zd ZdZ ej                    ZdS )TestGaugauN)rB   rC   rD   r#   r   Gaussianr   r   r   r   rN   rN   q   s#        I7DDDr   rN   c                   D    e Zd ZdZ ej                    ZdZdZdZ	dZ
dZdS )TestUniformrecg?   g?r   N)rB   rC   rD   r#   r   Uniformr   r,   r"   r2   r3   r4   r   r   r   rR   rR   v   s;        I7?DGIHHHHHr   rR   c                   <    e Zd ZdZ ej                    ZdZdZdZ	dS )TestTriangulartri
   g333333?r   N)
rB   rC   rD   r#   r   
Triangularr   r"   r2   r3   r   r   r   rW   rW      s2        I7DIHHHHr   rW   c                        e Zd ZdZ ej                    Zej        	                    de
d           fd            Z xZS )
TestCosinecoszNaN mismatchT)reasonraisesstrictc                 V    t          t          |                                            d S )N)superr\   r=   )r5   	__class__s    r   r=   zTestCosine.test_smoothconf   s'     	j$//11111r   )rB   rC   rD   r#   r   Cosine2r   rE   rF   xfailAssertionErrorr=   __classcell__)rc   s   @r   r\   r\      sw        I7?D[n,T  ; ;2 2 2 2; ;2 2 2 2 2r   r\   c                   8    e Zd ZdZ ej                    ZdZdZdS )TestBiweightbi	   r   N)	rB   rC   rD   r#   r   Biweightr   r"   r3   r   r   r   ri   ri      s-        I7DIHHHr   ri   c                      g d} t          j        ddd          } t          j                    |          }t	          || d           d S )N)	        6?,R_?)O?    `?g?rq   rp   ro   rn   r   rk   g|=)r    )r$   linspacer   Tricuber   )res_kxxxkxs      r   test_tricuberx      sZ    
 
 
F
 
RA		B			2		BBU++++++r   )$__doc__osnumpyr$   numpy.testingr   r   rE   !statsmodels.sandbox.nonparametricr   DEBUGpathdirnameabspath__file__curdirfname
recfromcsvjoinr(   r   r   positivelogrs   minmaxr&   r
   rJ   rN   rR   rW   r\   ri   rx   r   r   r   <module>r      sS    
			     < < < < < < < <  5 5 5 5 5 5 	
	22	3	3/
"-VU33
4
4JI6
BF1X;hKR[!%%''2&&$<E <E <E <E <E <E <E <E~" " " " " " " "
       
    "       %   2 2 2 2 2! 2 2 2    #   , , , , ,r   