
    -i'	                     J    S r SSKJr  SSKrSSKJr  SSKJ	r	  SSK
JrJr  S rg)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                 d   U " SS9nUR                   UR                  pCUR                  R                  5       UR                  pe[
        R                  " U5      (       d   e[
        R                  " U5      (       d   eSUR                   R                  :X  d   eSUR                   R                  :X  d   eSUR                  :X  d   eSUR                  :X  d   eSUR                  :X  d   eS[        U5      :X  d   eUR                  R                  S	5      (       d   e/ S
Qn[        XuS S 5        SnSn	[        X5       H8  u  pUR                  U5      nXS S 2U4   R                   R                  :X  a  M8   e   U " SSUS9nUR                   UR                  pUR                  n[        U SSS9n[!        UU5        [        ["        R$                  " US S 5      ["        R$                  " U5      5        SnU H  nUR                  5       R                  U5      nUR                  5       R                  U5      nUUS S 24   R'                  5       nUUS S 24   R'                  5       n[)        UU5        UUS S 24   R'                  5       nUUS S 24   R'                  5       n[)        UU5        M     g )NF)shufflei}i' )>F i  )r	   g   )r	   r
   z.. _rcv1_dataset:)C11C12C13C14C15C151   )GMILE143CCAT)   i  i Ttrain)r   subsetrandom_state)r   r   imZ  )i  i  i6  )datatargettarget_namestolist	sample_idspissparsesizeshapelenDESCR
startswithr   zipindexr   r   npsorttoarrayr   )fetch_rcv1_fxtglobal_random_seeddata1X1Y1cat_lists1first_categoriessome_categoriesnumber_non_zero_in_catnumcatjdata2X2Y2s2
fetch_funcsome_sample_idsr   idx1idx2feature_values_1feature_values_2target_values_1target_values_2s                             S/var/www/html/venv/lib/python3.13/site-packages/sklearn/datasets/tests/test_rcv1.pytest_fetch_rcv1rD      sk   5)EZZ%%,,.b ;;r???;;r???rww||###bggll""" bhh&&&BHH$$$   #h- ;;!!"56666 C'"16 /O..@NN3Ahmm((((( A
 W3EE ZZ	B wGJUJ' rwwr&5z*BGGBK8 *O$	yy{  +yy{  +dAg;..0dAg;..0,.>?T1W+--/T1W+--/O_= %    )__doc__	functoolsr   numpyr'   scipy.sparsesparser   "sklearn.datasets.tests.test_commonr   sklearn.utils._testingr   r   rD    rE   rC   <module>rN      s        ? J9>rE   