
    /Цi                        S SK Jr  S SKJr  S SKrS SKrS SKJr  S SK	J
r
  S SKJrJrJrJr  S SKJr  S SKJr  \
" 5       r\R*                  R-                  \R.                  5      r\R2                  R-                  \R.                  5      r\R6                  " \R8                  \R.                  S	9r\R*                  R-                  \R<                  5      r\R2                  R-                  \R<                  5      r \R6                  " \ R8                  \R<                  S	9r!\R<                  \R.                  /r"S
 r#S r$S r%S r&S r'S r(\RR                  RU                  S\5      \RR                  RU                  S\&" 5       \'" 5       -   5      S 5       5       r+\RR                  RU                  S\" \"\5      5      S 5       r,\RR                  RU                  S\(" 5       5      S 5       r-S r.g)    )partial)productN)assert_array_equal)	load_iris)ArrayDataset32ArrayDataset64CSRDataset32CSRDataset64)assert_allclose)CSR_CONTAINERS)dtypec                    U R                  5         UR                  5         UR                  U R                  5      nU R                  S   UR                  S   :X  d   eU R                  S   UR                  S   :X  d   e[	        U R
                  UR
                  5        [	        U R                  UR                  5        [	        U R                  UR                  5        g )Nr      )eliminate_zerosastyper   shaper   dataindicesindptr)currentexpecteds     c/var/www/html/ai-image-ml/venv/lib/python3.13/site-packages/sklearn/utils/tests/test_seq_dataset.pyassert_csr_equal_valuesr   !   s    w}}-H==x~~a0000==x~~a0000w||X]]3w(8(89w~~x7    c                     U [         R                  :X  a  [        [        [        [
        SS9$ [        [        [        [        SS9$ N*   seed)
npfloat32r   X32y32sample_weight32r   X64y64sample_weight64float_dtypes    r   _make_dense_datasetr*   ,   s0    bjj c3bAA#sO"==r   c           	          U[         R                  :X  a  [        [        [        [
        4u  p#pEO[        [        [        [        4u  p#pEU " U5      nU" UR                  UR                  UR                  X4SS9$ r   )r    r!   r"   r#   r$   r	   r%   r&   r'   r
   r   r   r   )csr_containerr)   Xysample_weightcsr_datasets         r   _make_sparse_datasetr1   2   s_    bjj +._l+R(m[+._l+R(maAqvvqxxA2NNr   c                  V    [          V s/ s H  n [        [        U 5      PM     sn $ s  sn f N)floatingr   r*   r(   s    r   _dense_dataset_factoriesr5   ;   s"    IQR+G'5RRRs   &c            	      |    [        [        [        5       V Vs/ s H  u  p[        [        X5      PM     snn $ s  snn f r3   )r   r   r4   r   r1   )r,   r)   s     r   _sparse_dataset_factoriesr7   ?   s;     +2.(*K*K&M 	$mA*K  s   8c                      [        5       [        5       -   n [        S[        U 5      S5       Vs/ s H	  oXS-    PM     sn$ s  snf )Nr      )r5   r7   rangelen)all_factoriesidxs     r   _fused_types_dataset_factoriesr>   F   sC    ,.1J1LLM49!S=OQR4ST4SS#a(4STTTs   Ar,   dataset_factoryc                    SnU" [         5      nU " 5       n[        U5       H  nUR                  5       u  pgpU" US[         R                  S   4S9n
[	        XU	/   5        U[
        U	   :X  d   eU[        U	   :X  d   eUR                  5       u  pgpU" US[         R                  S   4S9n
[	        XU	/   5        U[
        U	   :X  d   eU[        U	   :X  a  M   e   g )N   r   )r   )r%   r:   _next_pyr   r   r&   r'   
_random_py)r?   r,   NUMBER_OF_RUNSX_csr64dataset_xi_yiswir=   xis              r    test_seq_dataset_basic_iterationrL   L   s    
 NC GG>"#,,.3q#))A,&78SEN3SX~~oc**** $..03q#))A,&78SEN3SX~~oc**** #r   zfloat_dtype, csr_containerc                    [        U 5      n[        X5      n[        S5       H9  nUR                  5       u      pVUR                  5       u      pWXd:X  d   eXt:X  a  M9   e   S H9  nUR	                  5       u      pVUR	                  5       u      pWXd:X  d   eXt:X  a  M9   e   SnUR                  U5        UR                  U5        / SQn	/ SQn
[        X5       Hq  u  pKUR                  5       u      pVUR                  5       u      pWXd:X  d   eXt:X  d   eUR	                  5       u      pVUR	                  5       u      pWXk:X  d   eX{:X  a  Mq   e   g )NrA   )   2   	      :   M   )?   [      W      )   }   8   y      )r*   r1   r:   rB   rC   _shuffle_pyzip)r)   r,   dense_datasetsparse_datasetirG   idx1idx2r   idx_nextidx_shufflejs               r   test_seq_dataset_shufflerh   f   s`    (4M)-EN1X%..01a&//11ayyyy	  "%0021a&1131ayyyy	 " Dd#t$$H*KH*%..01a&//11ayyyy%0021a&1131ayyyy +r   z&dataset_32_factory, dataset_64_factoryc                 N   U " 5       U" 5       p2Sn[        U5       H  nUR                  5       u  u  n  pW  nUR                  5       u  u  n  pY  nUR                  [        R                  :X  d   eUR                  [        R
                  :X  d   e[        XSS9  [        XSS9  M     g )NrA   gh㈵>)rtol)r:   rB   r   r    r!   float64r   )
dataset_32_factorydataset_64_factory
dataset_32
dataset_64rD   rG   	xi_data32yi32	xi_data64yi64s
             r   test_fused_types_consistencyrt      s     013E3G
N>"(2(;(;(=%AqA(2(;(;(=%AqA"**,,,"**,,,	48. #r   c                     [         R                  " [        SS9   [        [        [
        [        SS9  S S S 5        [         R                  " [        SS9   [        [        [        [        SS9  S S S 5        [         H  n U " [        5      nU " [        5      n[         R                  " [        SS9   [        UR                  UR                  UR                  [
        [        SS9  S S S 5        [         R                  " [        SS9   [!        UR                  UR                  UR                  [        [        SS9  S S S 5        M     g ! , (       d  f       GN= f! , (       d  f       N= f! , (       d  f       N= f! , (       d  f       GM  = f)NzBuffer dtype mismatch)matchr   r   )pytestraises
ValueErrorr   r"   r#   r$   r   r%   r&   r'   r   r
   r   r   r   r	   )r,   X_csr32rE   s      r    test_buffer_dtype_mismatch_errorr{      s   	z)@	AsCr: 
B 
z)@	AsCr: 
B ($$]]:-DE F ]]:-DE FE ( 
B	A 
B	A FE FEs/   EE/5E*5E;
E
E'*
E8	;
F	)/	functoolsr   	itertoolsr   numpyr    rw   numpy.testingr   sklearn.datasetsr   sklearn.utils._seq_datasetr   r   r	   r
   sklearn.utils._testingr   sklearn.utils.fixesr   irisr   r   rk   r%   targetr&   arangesizer'   r!   r"   r#   r$   r4   r   r*   r1   r5   r7   r>   markparametrizerL   rh   rt   r{    r   r   <module>r      s       , &  3 .{
iirzz"
kk$))CHHBJJ7
iirzz"
kk$))CHHBJJ7JJ

#8>OSU .9/14M4OO+ :+,  '(N"CD ,.L.N//r   