§
    §éUg  ã                   ó¤  — d dl mZ d dlZd dlZd dlmZ d dlmZ d dl	m
Z
mZmZmZ d dlmZ d dlmZ  e¦   «         Zej                             ej        ¦  «        Zej                             ej        ¦  «        Z ej        ej        ej        ¬¦  «        Zej                             ej        ¦  «        Zej                             ej        ¦  «        Z ej        ej        ej        ¬¦  «        Zej        ej        gZ d	„ Z!d
„ Z"d„ Z#d„ Z$d„ Z%d„ Z&ej'         (                    de¦  «        ej'         (                    d e$¦   «          e%¦   «         z   ¦  «        d„ ¦   «         ¦   «         Z)ej'         (                    dd„  ee e¦  «        D ¦   «         ¦  «        d„ ¦   «         Z*ej'         (                    d e&¦   «         ¦  «        d„ ¦   «         Z+d„ Z,dS )é    )ÚproductN)Úassert_array_equal)Ú	load_iris)ÚArrayDataset32ÚArrayDataset64ÚCSRDataset32ÚCSRDataset64)Úassert_allclose)ÚCSR_CONTAINERS)Údtypec                 óž  — |                       ¦   «          |                      ¦   «          |                     | j        ¦  «        }| j        d         |j        d         k    sJ ‚| j        d         |j        d         k    sJ ‚t	          | j        |j        ¦  «         t	          | j        |j        ¦  «         t	          | j        |j        ¦  «         d S )Nr   é   )Úeliminate_zerosÚastyper   Úshaper   ÚdataÚindicesÚindptr)ÚcurrentÚexpecteds     úc/var/www/surfInsights/venv3-11/lib/python3.11/site-packages/sklearn/utils/tests/test_seq_dataset.pyÚassert_csr_equal_valuesr   "   sº   € Ø×ÒÑÔÐØ×ÒÑÔÐØ�Š˜wœ}Ñ-Ô-€HØŒ=˜Ô˜xœ~¨aÔ0Ò0Ð0Ð0Ð0ØŒ=˜Ô˜xœ~¨aÔ0Ò0Ð0Ð0Ð0Ý�w”| X¤]Ñ3Ô3Ð3Ý�w”¨Ô(8Ñ9Ô9Ð9Ý�w”~ x¤Ñ7Ô7Ð7Ð7Ð7ó    c                 óª   — | t           j        k    r"t          t          t          t
          d¬¦  «        S t          t          t          t          d¬¦  «        S ©Né*   ©Úseed)
ÚnpÚfloat32r   ÚX32Úy32Úsample_weight32r   ÚX64Úy64Úsample_weight64)Úfloat_dtypes    r   Ú_make_dense_datasetr(   -   s=   € Ø•b”jÒ Ð Ý�c¥3­¸bÐAÑAÔAÐAÝ�#�s¥O¸"Ð=Ñ=Ô=Ð=r   c                 óö   — |t           j        k    r t          t          t          t
          f\  }}}}nt          t          t          t          f\  }}}} | |¦  «        } ||j
        |j        |j        ||d¬¦  «        S 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_datasetr/   3   sq   € Ø•b”jÒ Ð Ý+.µµ_ÅlÐ+RÑ(ˆˆ1ˆm˜[˜[å+.µµ_ÅlÐ+RÑ(ˆˆ1ˆm˜[Øˆ�aÑÔ€AØˆ;�q”v˜qœx¨¬°A°}È2ÐNÑNÔNÐNr   c                  ó$   — d„ t           D ¦   «         S )Nc                 ó,   — g | ]}t          |¦  «        ‘ŒS © )r(   )Ú.0r'   s     r   ú
<listcomp>z(_make_dense_datasets.<locals>.<listcomp>=   s!   € ÐIÐIÐI°Õ Ñ,Ô,ÐIÐIÐIr   )Úfloatingr2   r   r   Ú_make_dense_datasetsr6   <   s   € ØIÐIÅÐIÑIÔIÐIr   c                  óJ   — d„ t          t          t          ¦  «        D ¦   «         S )Nc                 ó4   — g | ]\  }}t          ||¦  «        ‘ŒS r2   )r/   )r3   r*   r'   s      r   r4   z)_make_sparse_datasets.<locals>.<listcomp>A   s6   € ð ð ð á&ˆM˜;õ 	˜]¨KÑ8Ô8ðð ð r   )r   r   r5   r2   r   r   Ú_make_sparse_datasetsr9   @   s-   € ðð å*1µ.Å(Ñ*KÔ*Kðñ ô ð r   c                  ó’   ‡ — t          ¦   «         t          ¦   «         z   Š ˆ fd„t          dt          ‰ ¦  «        d¦  «        D ¦   «         S )Nc              3   ó2   •K  — | ]}‰||d z   …         V — ŒdS )é   Nr2   )r3   ÚidxÚall_datasetss     €r   ú	<genexpr>z-_make_fused_types_datasets.<locals>.<genexpr>J   s0   øè è € ÐRÐR¨CˆL˜˜s Q™w˜Ô'ÐRÐRÐRÐRÐRÐRr   r   r<   )r6   r9   ÚrangeÚlen)r>   s   @r   Ú_make_fused_types_datasetsrB   G   sH   ø€ Ý'Ñ)Ô)Õ,AÑ,CÔ,CÑC€LàRÐRÐRÐRµ5¸½CÀÑ<MÔ<MÈqÑ3QÔ3QÐRÑRÔRÐRr   r*   Údatasetc                 ó"  — d} |t           ¦  «        }t          |¦  «        D ]ì}|                      ¦   «         \  }}}} ||dt           j        d         f¬¦  «        }	t	          |	||g         ¦  «         |t
          |         k    sJ ‚|t          |         k    sJ ‚|                      ¦   «         \  }}}} ||dt           j        d         f¬¦  «        }	t	          |	||g         ¦  «         |t
          |         k    sJ ‚|t          |         k    sJ ‚Œíd S )Né   r   )r   )r$   r@   Ú_next_pyr   r   r%   r&   Ú
_random_py)
rC   r*   ÚNUMBER_OF_RUNSÚX_csr64Ú_Úxi_ÚyiÚswir=   Úxis
             r   Ú test_seq_dataset_basic_iterationrO   M   s*  € ð €NØˆm�CÑ Ô €GÝ�>Ñ"Ô"ð +ð +ˆà#×,Ò,Ñ.Ô.ÑˆˆR��cØˆ]˜3 q­#¬)°A¬,Ð&7Ð8Ñ8Ô8ˆå  G¨S¨E¤NÑ3Ô3Ð3Ø•S˜”XŠ~ˆ~ˆ~ˆ~Ø•o cÔ*Ò*Ð*Ð*Ð*ð $×.Ò.Ñ0Ô0ÑˆˆR��cØˆ]˜3 q­#¬)°A¬,Ð&7Ð8Ñ8Ô8ˆå  G¨S¨E¤NÑ3Ô3Ð3Ø•S˜”XŠ~ˆ~ˆ~ˆ~Ø•o cÔ*Ò*Ð*Ð*Ð*Ð*ð+ð +r   zdense_dataset,sparse_datasetc                 óR   — g | ]$\  }}t          |¦  «        t          ||¦  «        f‘Œ%S r2   )r(   r/   )r3   r'   r*   s      r   r4   r4   f   sH   € ð ð ð ñ
 'ˆK˜õ   Ñ,Ô,Ý  °Ñ<Ô<ð	
ðð ð r   c                 óØ  — t          d¦  «        D ]D}|                      ¦   «         \  }}}}|                     ¦   «         \  }}}}||k    sJ ‚||k    sJ ‚ŒEdD ]D}|                      ¦   «         \  }}}}|                     ¦   «         \  }}}}||k    sJ ‚||k    sJ ‚ŒEd}|                      |¦  «         |                     |¦  «         g d¢}g d¢}t	          ||¦  «        D ]‰\  }}	|                      ¦   «         \  }}}}|                     ¦   «         \  }}}}||k    sJ ‚||k    sJ ‚|                      ¦   «         \  }}}}|                     ¦   «         \  }}}}||	k    sJ ‚||	k    sJ ‚ŒŠd S )NrE   )é„   é2   é	   é   é:   éM   )é?   é[   é”   éW   é   )é‰   é}   é8   éy   é   )r@   rF   rG   Ú_shuffle_pyÚzip)
Údense_datasetÚsparse_datasetÚirJ   Úidx1Úidx2r   Úidx_nextÚidx_shuffleÚjs
             r   Útest_seq_dataset_shufflerl   d   sÈ  € õ �1‰XŒXð ð ˆØ%×.Ò.Ñ0Ô0‰ˆˆ1ˆa�Ø&×/Ò/Ñ1Ô1‰ˆˆ1ˆa�Ø�qŠyˆyˆyˆyØ�qŠyˆyˆyˆyˆyà!ð ð ˆØ%×0Ò0Ñ2Ô2‰ˆˆ1ˆa�Ø&×1Ò1Ñ3Ô3‰ˆˆ1ˆa�Ø�qŠyˆyˆyˆyØ�qŠyˆyˆyˆyˆyà€DØ×Ò˜dÑ#Ô#Ð#Ø×Ò˜tÑ$Ô$Ð$à$Ð$Ð$€HØ*Ð*Ð*€KÝ�H˜kÑ*Ô*ð 	ð 	‰ˆˆ1Ø%×.Ò.Ñ0Ô0‰ˆˆ1ˆa�Ø&×/Ò/Ñ1Ô1‰ˆˆ1ˆa�Ø�qŠyˆyˆyˆyØ�qŠyˆyˆyˆyà%×0Ò0Ñ2Ô2‰ˆˆ1ˆa�Ø&×1Ò1Ñ3Ô3‰ˆˆ1ˆa�Ø�qŠyˆyˆyˆyØ�qŠyˆyˆyˆyˆyð	ð 	r   zdataset_32,dataset_64c                 óF  — d}t          |¦  «        D ]Ž}|                      ¦   «         \  \  }}}}}}|                     ¦   «         \  \  }}}}}}|j        t          j        k    sJ ‚|j        t          j        k    sJ ‚t          ||d¬¦  «         t          ||d¬¦  «         Œ�d S )NrE   gñhãˆµøä>)Úrtol)r@   rF   r   r   r    Úfloat64r
   )Ú
dataset_32Ú
dataset_64rH   rJ   Ú	xi_data32Úyi32Ú	xi_data64Úyi64s           r   Útest_fused_types_consistencyrv   Ž   sÁ   € à€NÝ�>Ñ"Ô"ð 	/ð 	/ˆà(2×(;Ò(;Ñ(=Ô(=Ñ%Ñˆ�A�q˜4  AØ(2×(;Ò(;Ñ(=Ô(=Ñ%Ñˆ�A�q˜4  AàŒ¥"¤*Ò,Ð,Ð,Ð,ØŒ¥"¤*Ò,Ð,Ð,Ð,å˜	 9°4Ð8Ñ8Ô8Ð8Ý˜˜d¨Ð.Ñ.Ô.Ð.Ð.ð	/ð 	/r   c            
      ó:  — t          j        t          d¬¦  «        5  t          t          t
          t          d¬¦  «        f d d d ¦  «         n# 1 swxY w Y   t          j        t          d¬¦  «        5  t          t          t          t          d¬¦  «        f d d d ¦  «         n# 1 swxY w Y   t          D ]æ}  | t          ¦  «        } | t          ¦  «        }t          j        t          d¬¦  «        5  t          |j        |j        |j        t
          t          d¬¦  «        f d d d ¦  «         n# 1 swxY w Y   t          j        t          d¬¦  «        5  t!          |j        |j        |j        t          t          d¬¦  «        f d d d ¦  «         n# 1 swxY w Y   Œçd S )NzBuffer dtype mismatch)Úmatchr   r   )ÚpytestÚraisesÚ
ValueErrorr   r!   r"   r#   r   r$   r%   r&   r   r	   r   r   r   r   )r*   ÚX_csr32rI   s      r   Ú test_buffer_dtype_mismatch_errorr}   �   sˆ  € Ý	Œ•zÐ)@Ð	AÑ	AÔ	Að <ð <Ý•s�C¥°rÐ:Ñ:Ô:Ð;Ð;ð<ð <ð <ñ <ô <ð <ð <ð <ð <ð <ð <øøøð <ð <ð <ð <õ 
Œ•zÐ)@Ð	AÑ	AÔ	Að <ð <Ý•s�C¥°rÐ:Ñ:Ô:Ð;Ð;ð<ð <ð <ñ <ô <ð <ð <ð <ð <ð <ð <øøøð <ð <ð <ð <õ (ð ð ˆØ�-¥Ñ$Ô$ˆØ�-¥Ñ$Ô$ˆÝŒ]�:Ð-DÐEÑEÔEð 	ð 	ÝØ”Ø”Ø”ÝÝØðñ ô ð ð ð	ð 	ð 	ñ 	ô 	ð 	ð 	ð 	ð 	ð 	ð 	øøøð 	ð 	ð 	ð 	õ Œ]�:Ð-DÐEÑEÔEð 	ð 	ÝØ”Ø”Ø”ÝÝØðñ ô ð ð ð	ð 	ð 	ñ 	ô 	ð 	ð 	ð 	ð 	ð 	ð 	øøøð 	ð 	ð 	ð 	øðð sG   œ$AÁAÁAÁ2$B"Â"B&Â)B&Ã10D-Ä-D1	Ä4D1	Å0FÆF	ÆF	)-Ú	itertoolsr   Únumpyr   ry   Únumpy.testingr   Úsklearn.datasetsr   Úsklearn.utils._seq_datasetr   r   r   r	   Úsklearn.utils._testingr
   Úsklearn.utils.fixesr   Úirisr   r   ro   r$   Útargetr%   ÚarangeÚsizer&   r    r!   r"   r#   r5   r   r(   r/   r6   r9   rB   ÚmarkÚparametrizerO   rl   rv   r}   r2   r   r   ú<module>r‹      sÖ  ðð Ð Ð Ð Ð Ð à Ð Ð Ð Ø €€€Ø ,Ð ,Ð ,Ð ,Ð ,Ð ,à &Ð &Ð &Ð &Ð &Ð &ðð ð ð ð ð ð ð ð ð ð ð ð 3Ð 2Ð 2Ð 2Ð 2Ð 2Ø .Ð .Ð .Ð .Ð .Ð .à€y�{„{€Ø
„i×Ò�r”zÑ"Ô"€Ø
„k×Ò˜œÑ$Ô$€Ø�"”)˜CœH¨B¬JÐ7Ñ7Ô7€à
„i×Ò�r”zÑ"Ô"€Ø
„k×Ò˜œÑ$Ô$€Ø�"”)˜CœH¨B¬JÐ7Ñ7Ô7€àŒJ˜œ
Ð#€ð8ð 8ð 8ð>ð >ð >ðOð Oð OðJð Jð Jðð ð ðSð Sð Sð „×Ò˜¨.Ñ9Ô9Ø„×Ò˜Ð$8Ð$8Ñ$:Ô$:Ð=RÐ=RÑ=TÔ=TÑ$TÑUÔUð+ð +ñ VÔUñ :Ô9ð+ð* „×ÒØ"ðð ð
 +2¨'°(¸NÑ*KÔ*Kðñ ô ñ	ô 	ðð ñ	ô 	ðð@ „×ÒÐ0Ð2LÐ2LÑ2NÔ2NÑOÔOð/ð /ñ PÔOð/ðð ð ð ð r   