
    M/Ph              
       $   d Z ddlZddlZddlmZmZ ddlmZm	Z	m
Z
 d Zd Zej                            dej         ej        ddd	           ej        dd
d	          f         ddgddgf ej        ddd	          ddfg          d             ZdS )zH
unit test for spline and other smoother classes

Author: Luca Puggini

    N)assert_allcloseassert_equal)UnivariatePolynomialSmootherPolynomialSmootherBSplinesc                      t          j        ddd          } t          | d          }t          |j        j        d           t          |j        d d df         |                                 dz             d S )Nr            degree)r
   r      )nplinspacer   r   basisshaper   ravel)xpols     g/var/www/html/test/jupyter/venv/lib/python3.11/site-packages/statsmodels/gam/tests/test_smooth_basis.py#test_univariate_polynomial_smootherr      sk    
Aq!A
&q
3
3
3C&)))CIaaadOQWWYY!^44444    c                  X   t           j                            d           t           j                            ddd          } ddg}t	          | |          }t          |          D ]E\  }}t          | d d |f         |          j        }t          |j	        |         j        |           Fd S )Nr	   r   )
   r   r      r   )
r   randomseednormalr   	enumerater   r   r   	smoothers)r   degreesmpsideguv_basiss         r   "test_multivariate_polynomial_basisr&      s    INN1
	Aw''A!fG
Q
(
(CG$$ : :3/!!!Q$DDDJa(.9999: :r   zx, df, degreer	   d   r   r
      r   c                 R    t          | ||          }|                    |            d S )N)r   	transform)r   dfr   bsplines       r   test_bsplinesr-   !   s/     q"f%%Gar   )__doc__pytestnumpyr   numpy.testingr   r   statsmodels.gam.smooth_basisr   r   r   r   r&   markparametrizec_r   r-    r   r   <module>r7      s0         7 7 7 7 7 7 7 74 4 4 4 4 4 4 4 4 4
5 5 5: : :  E+"+aC((+"+aS*A*AABFF	

 
Q3		A&
 
 
 
  r   