U
    Ãmœd¡  ã                   @   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	   ú\/home/sam/Atlas/atlas_env/lib/python3.8/site-packages/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   )	Únumpyr   Znumpy.testingr   Zstatsmodels.tools.cataddr   Zscipyr   r   r	   r	   r	   r   Ú<module>   s   