ó
áp7]c           @   s  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
 d e f d „  ƒ  YZ d e f d	 „  ƒ  YZ d
 e e f d „  ƒ  YZ d e e f d „  ƒ  YZ d e e f d „  ƒ  YZ e d k re ƒ  Z e j ƒ  e j ƒ  e j e ƒ  Z e j ƒ  n  d S(   sL   
Created on Fri Nov 04 10:51:39 2011

Author: Josef Perktold
License: BSD-3
iÿÿÿÿN(   t   assert_almost_equalt   assert_equal(   t	   smoothers(   t   OLSt   WLSt   CheckSmootherc           B   s#   e  Z d  „  Z d „  Z d „  Z RS(   c         C   s\   t  |  j j |  j ƒ |  j j d d ƒt  |  j j |  j d  ƒ |  j j d  d d ƒd  S(   Nt   decimali   i
   (   R    t   res_pst   predictt   xt   res2t   fittedvalues(   t   self(    (    sU   lib/python2.7/site-packages/statsmodels/sandbox/nonparametric/tests/test_smoothers.pyt   test_predict   s    c         C   s)   t  |  j j j ƒ  |  j j d d ƒd  S(   NR   i   (   R    R   t   coeft   ravelR
   t   params(   R   (    (    sU   lib/python2.7/site-packages/statsmodels/sandbox/nonparametric/tests/test_smoothers.pyt	   test_coef   s    c         C   s`   t  |  j j ƒ  |  j j d ƒ t  |  j j ƒ  |  j j d ƒ t  |  j j ƒ  |  j j ƒ d  S(   Ni   (   R   R   t   df_modelR
   t   df_fitt   df_resid(   R   (    (    sU   lib/python2.7/site-packages/statsmodels/sandbox/nonparametric/tests/test_smoothers.pyt   test_df   s      (   t   __name__t
   __module__R   R   R   (    (    (    sU   lib/python2.7/site-packages/statsmodels/sandbox/nonparametric/tests/test_smoothers.pyR      s   		t   BasePolySmootherc           B   s   e  Z e d  „  ƒ Z RS(   c   
      C   s¬   d } d } d } d \ } } t  j | | | ƒ |  _ } | d  d  … d  f t  j | d ƒ |  _ } | j d ƒ } t  j j d ƒ | | t  j j	 | ƒ |  _
 }	 d  S(	   Ni   g      à?id   iÿÿÿÿi   i   i¯ (   iÿÿÿÿi   (   t   npt   linspaceR	   t   Nonet   aranget   exogt   sumt   randomt   seedt   randnt   y(
   t   clst   ordert   sigma_noiset   nobst   lbt   ubR	   R   t   y_trueR"   (    (    sU   lib/python2.7/site-packages/statsmodels/sandbox/nonparametric/tests/test_smoothers.pyt   setup_class'   s    .(   R   R   t   classmethodR*   (    (    (    sU   lib/python2.7/site-packages/statsmodels/sandbox/nonparametric/tests/test_smoothers.pyR   %   s   t   TestPolySmoother1c           B   s   e  Z e d  „  ƒ Z RS(   c         C   sŠ   t  t |  ƒ j ƒ  |  j |  j |  j } } } t j d | ƒ } | j | ƒ | |  _	 t
 | | d  d  … d  d … f ƒ j ƒ  |  _ d  S(   Ni   i   i   (   t   superR,   R*   R"   R	   R   R   t   PolySmoothert   fitR   R   R
   (   R#   R"   R	   R   t   pmod(    (    sU   lib/python2.7/site-packages/statsmodels/sandbox/nonparametric/tests/test_smoothers.pyR*   7   s    	(   R   R   R+   R*   (    (    (    sU   lib/python2.7/site-packages/statsmodels/sandbox/nonparametric/tests/test_smoothers.pyR,   5   s   t   TestPolySmoother2c           B   s   e  Z e d  „  ƒ Z RS(   c         C   sŠ   t  t |  ƒ j ƒ  |  j |  j |  j } } } t j d | ƒ } | j | ƒ | |  _	 t
 | | d  d  … d  d … f ƒ j ƒ  |  _ d  S(   Ni   i   i   (   R-   R1   R*   R"   R	   R   R   R.   t   smoothR   R   R/   R
   (   R#   R"   R	   R   R0   (    (    sU   lib/python2.7/site-packages/statsmodels/sandbox/nonparametric/tests/test_smoothers.pyR*   F   s    	(   R   R   R+   R*   (    (    (    sU   lib/python2.7/site-packages/statsmodels/sandbox/nonparametric/tests/test_smoothers.pyR1   D   s   t   TestPolySmoother3c           B   s   e  Z e d  „  ƒ Z RS(   c         C   sÏ   t  t |  ƒ j ƒ  |  j |  j |  j } } } | j d } t j | ƒ } d | | d *d | | d )t	 j
 d | ƒ } | j | d | ƒ| |  _ t | | d  d  … d  d … f d | ƒj ƒ  |  _ d  S(	   Ni    gš™™™™™¹?i   i   i   t   weightsi   i   (   R-   R3   R*   R"   R	   R   t   shapeR   t   onesR   R.   R/   R   R   R
   (   R#   R"   R	   R   R&   R4   R0   (    (    sU   lib/python2.7/site-packages/statsmodels/sandbox/nonparametric/tests/test_smoothers.pyR*   V   s    	(   R   R   R+   R*   (    (    (    sU   lib/python2.7/site-packages/statsmodels/sandbox/nonparametric/tests/test_smoothers.pyR3   T   s   t   __main__(   t   __doc__t   numpyR   t   numpy.testingR    R   t!   statsmodels.sandbox.nonparametricR   t#   statsmodels.regression.linear_modelR   R   t   objectR   R   R,   R1   R3   R   t   t1R   R   R   t   t3(    (    (    sU   lib/python2.7/site-packages/statsmodels/sandbox/nonparametric/tests/test_smoothers.pyt   <module>   s    	

	