
    M/Ph                        d Z ddlZddlmZmZ ddlmZ ddlm	Z	m
Z
  G d d          Z G d d	          Z G d
 dee          Z G d dee          Z G d dee          Zedk    rY e            Ze                                 e                                 ej          e            Ze                                 dS dS )zL
Created on Fri Nov 04 10:51:39 2011

Author: Josef Perktold
License: BSD-3
    N)assert_almost_equalassert_equal)	smoothers)OLSWLSc                        e Zd Zd Zd Zd ZdS )CheckSmootherc                 
   t          | j                            | j                  | j        j        d           t          | j                            | j        d d                   | j        j        d d         d           d S )N   decimal
   )r   res_pspredictxres2fittedvaluesselfs    v/var/www/html/test/jupyter/venv/lib/python3.11/site-packages/statsmodels/sandbox/nonparametric/tests/test_smoothers.pytest_predictzCheckSmoother.test_predict   s    DK//77 I2B	@ 	@ 	@ 	@DK//ss<< I23B37	E 	E 	E 	E 	E 	E    c                 v    t          | j        j                                        | j        j        d           d S )N   r   )r   r   coefravelr   paramsr   s    r   	test_coefzCheckSmoother.test_coef   s@    DK,2244 I,b	: 	: 	: 	: 	: 	:r   c                 8   t          | j                                        | j        j        dz              t          | j                                        | j        j        dz              t          | j                                        | j        j                   d S )N   )r   r   df_modelr   df_fitdf_residr   s    r   test_dfzCheckSmoother.test_df   s{    T[))++TY-?-ABBBT[''))49+=a+?@@@T[))++TY-?@@@@@r   N)__name__
__module____qualname__r   r   r$    r   r   r	   r	      sI        E E E: : :
A A A A Ar   r	   c                   $    e Zd Zed             ZdS )BasePolySmootherc                 l   d}d}d}d\  }}t          j        |||          x| _        }|d d d f         t          j        |dz             z  x| _        }|                    d          }t           j                            d           ||t           j                            |          z  z   x| _	        }	d S )N   g      ?d   )   r    i )
nplinspacer   arangeexogsumrandomseedrandny)
clsordersigma_noisenobslbubr   r3   y_truer8   s
             r   setup_classzBasePolySmoother.setup_class&   s     BKB---AAAdF)RYuQw%7%7774!
	v[29??4+@+@@@@r   N)r%   r&   r'   classmethodr@   r(   r   r   r*   r*   $   s2        
A 
A [
A 
A 
Ar   r*   c                   .     e Zd Ze fd            Z xZS )TestPolySmoother1c                 :   t                                                       | j        | j        | j        }}}t          j        d|          }|                    |           || _        t          ||d d d df                                                   | _
        d S )Nr/   r,   )superr@   r8   r   r3   r   PolySmootherfitr   r   r   r9   r8   r   r3   pmod	__class__s        r   r@   zTestPolySmoother1.setup_class6   s    UCE38d1 %a++
q$qqq#v,''++--r   r%   r&   r'   rA   r@   __classcell__rJ   s   @r   rC   rC   4   sB        
. 
. 
. 
. [
. 
. 
. 
. 
.r   rC   c                   .     e Zd Ze fd            Z xZS )TestPolySmoother2c                 :   t                                                       | j        | j        | j        }}}t          j        d|          }|                    |           || _        t          ||d d d df                   
                                | _        d S )Nr,      )rE   r@   r8   r   r3   r   rF   smoothr   r   rG   r   rH   s        r   r@   zTestPolySmoother2.setup_classE   s    UCE38d1 %a++A
q$qqq#v,''++--r   rK   rM   s   @r   rO   rO   C   sB        . . . . [. . . . .r   rO   c                   .     e Zd Ze fd            Z xZS )TestPolySmoother3c                    t                                                       | j        | j        | j        }}}|j        d         }t          j        |          }d|d |dz  <   d|| dz  d <   t          j	        d|          }|
                    ||           || _        t          ||d d d df         |          
                                | _        d S )Nr   g?r,   r/      )weights)rE   r@   r8   r   r3   shaper0   onesr   rF   rG   r   r   r   )r9   r8   r   r3   r<   rW   rI   rJ   s          r   r@   zTestPolySmoother3.setup_classU   s    UCE38d1wqz'$--qq		 %a++G$$$
q$qqq#v,888<<>>r   rK   rM   s   @r   rT   rT   S   sB        ? ? ? ? [? ? ? ? ?r   rT   __main__)__doc__numpyr0   numpy.testingr   r   !statsmodels.sandbox.nonparametricr   #statsmodels.regression.linear_modelr   r   r	   r*   rC   rO   rT   r%   t1r   r   r$   t3r(   r   r   <module>rb      s        ; ; ; ; ; ; ; ; 7 7 7 7 7 7 8 8 8 8 8 8 8 8A A A A A A A A*A A A A A A A A . . . . .(- . . .. . . . .(- . . . ? ? ? ? ?(- ? ? ?( z				BOOLLNNNJJ				BOO r   