U
    Kvf                     @   s8   d dl Zd dlmZ d dlmZ d dlmZ dd ZdS )    N)assert_equal)	add_indep)linalgc                  C   s  t dddddddddddg} t dddddddddddg}t t|}t || d d d f t d |d d d f t d g}dgdd t dD  dd t dD  }t||\}}t|t || |f tt	|d	k
 d t|dd
dg d S )Nr            constc                 S   s   g | ]}d | qS )zvar1_%d .0ir	   r	   G/tmp/pip-unpacked-wheel-2v6byqio/statsmodels/tools/tests/test_catadd.py
<listcomp>   s     z"test_add_indep.<locals>.<listcomp>c                 S   s   g | ]}d | qS )zvar2_%dr	   r
   r	   r	   r   r      s     g-q=Zvar1_1Zvar2_1)nparrayZoneslenZcolumn_stackZaranger   r   r   Zsvdvalssum)x1Zx2Zx0xvarnamesZxoZvor	   r	   r   test_add_indep   s      <r   )	Znumpyr   Znumpy.testingr   Zstatsmodels.tools.cataddr   Zscipyr   r   r	   r	   r	   r   <module>   s   