ó
áp7]c           @   sT   d  Z  d d l Z d d l m Z m Z d d l m Z m Z d „  Z	 d „  Z
 d S(   sH   
unit test for spline and other smoother classes

Author: Luca Puggini

iÿÿÿÿN(   t   assert_allcloset   assert_equal(   t   UnivariatePolynomialSmoothert   PolynomialSmootherc          C   sh   t  j d d d ƒ }  t |  d d ƒ} t | j j d ƒ t | j d  d  … d f |  j ƒ  d ƒ d  S(   Ni    i   i   t   degreei   i   (   i   i   (   t   npt   linspaceR   R   t   basist   shapeR    t   ravel(   t   xt   pol(    (    sF   lib/python2.7/site-packages/statsmodels/gam/tests/test_smooth_basis.pyt#   test_univariate_polynomial_smoother   s    c          C   s    t  j j d ƒ t  j j d d d ƒ }  d d g } t |  | ƒ } xV t | ƒ D]H \ } } t |  d  d  … | f d | ƒj } t | j	 | j | ƒ qP Wd  S(	   Ni   i    i
   i   i   i   R   (   i
   i   (
   R   t   randomt   seedt   normalR   t	   enumerateR   R   R    t	   smoothers(   R
   t   degreest   mpst   it   degt   uv_basis(    (    sF   lib/python2.7/site-packages/statsmodels/gam/tests/test_smooth_basis.pyt"   test_multivariate_polynomial_basis   s    %(   t   __doc__t   numpyR   t   numpy.testingR    R   t   statsmodels.gam.smooth_basisR   R   R   R   (    (    (    sF   lib/python2.7/site-packages/statsmodels/gam/tests/test_smooth_basis.pyt   <module>   s
   	