a
    h'	                     @   sL   d Z ddlmZ ddlZddlmZ ddlm	Z	 ddl
mZmZ dd ZdS )zyTest the rcv1 loader, if the data is available,
or if specifically requested via environment variable
(e.g. for CI jobs).    )partialN)check_return_X_y)assert_almost_equalassert_array_equalc                 C   s  | dd}|j |j }}|j |j }}t|s8J t|sFJ d|j jksVJ d|j jksfJ d|jkstJ d|jksJ d|jksJ dt	|ksJ |j
d	sJ g d
}t||d d  d}d}	t|	|D ].\}
}||}|
|d d |f j jksJ q| dd|d}|j |j }}|j}t| ddd}t|| tt|d d t| d}|D ]}| |}| |}||d d f  }||d d f  }t|| ||d d f  }||d d f  }t|| qld S )NF)shufflei}i' )>F i  )r   g   )r   r   z.. _rcv1_dataset:)ZC11ZC12ZC13ZC14ZC15ZC151   )ZGMILZE143ZCCAT)   i  i Ttrain)r   subsetZrandom_state)r   r   imZ  )i  i  i6  )datatargetZtarget_namestolist	sample_idspissparsesizeshapelenZDESCR
startswithr   zipindexr   r   npsortZtoarrayr   )Zfetch_rcv1_fxtZglobal_random_seedZdata1ZX1ZY1Zcat_lists1Zfirst_categoriesZsome_categoriesZnumber_non_zero_in_catnumcatjZdata2ZX2ZY2s2Z
fetch_funcZsome_sample_idsr   Zidx1Zidx2Zfeature_values_1Zfeature_values_2Ztarget_values_1Ztarget_values_2 r    \/var/www/html/assistant/venv/lib/python3.9/site-packages/sklearn/datasets/tests/test_rcv1.pytest_fetch_rcv1   sJ    



r"   )__doc__	functoolsr   numpyr   Zscipy.sparsesparser   Z"sklearn.datasets.tests.test_commonr   Zsklearn.utils._testingr   r   r"   r    r    r    r!   <module>   s   