B
    (b                 @   sp   d dl mZmZmZ d dlmZmZmZ d dlm	Z	 d ddddgd ddddgddddd ggZ
d	d
 Zdd ZdS )    )assert_array_equalassert_equalassert_raises)
bsr_matrix
csc_matrix
csr_matrix)VarianceThreshold               c              C   s   xFt tt tt tt gD ]*} t | }tddddg|jdd qW tt	t jddddgg tt	t jddgddgg d S )Nr   r	   r   r   T)indicesr
   )
datar   r   r   r   Zfitr   Zget_supportr   
ValueError)Xsel r   f/home/dcms/DCMS/lib/python3.7/site-packages/sklearn/feature_selection/tests/test_variance_threshold.pytest_zero_variance   s
    r   c              C   s>   x8t tt gD ](} tdd| } ttt df| j qW d S )Ng?)	thresholdr	   )r   r   r   Zfit_transformr   lenshape)r   r   r   r   test_variance_threshold   s    r   N)Zsklearn.utils.testingr   r   r   Zscipy.sparser   r   r   Zsklearn.feature_selectionr   r   r   r   r   r   r   r   <module>   s   