§
    rŠtj£  ã                   óœ  — d dl mZ 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 ee"e¦  «        ¦  «        d„ ¦   «         Z,ej)         *                    d e(¦   «         ¦  «        d„ ¦   «         Z-d„ Z.dS )é    )Úpartial)Ú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     úb/var/www/html/CA-Chatbot/venv/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_datasetr0   2   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                 ó8   — g | ]}t          t          |¦  «        ‘ŒS © )r   r)   )Ú.0r(   s     r   ú
<listcomp>z,_dense_dataset_factories.<locals>.<listcomp><   s#   € ÐRÐRÐR¸+�GÕ'¨Ñ5Ô5ÐRÐRÐRr   )Úfloatingr3   r   r   Ú_dense_dataset_factoriesr7   ;   s   € ØRÐRÍÐRÑRÔRÐRr   c                  óJ   — d„ t          t          t          ¦  «        D ¦   «         S )Nc                 ó@   — g | ]\  }}t          t          ||¦  «        ‘ŒS r3   )r   r0   )r4   r+   r(   s      r   r5   z-_sparse_dataset_factories.<locals>.<listcomp>@   s9   € ð ð ð á&ˆM˜;õ 	Õ$ m°[ÑAÔAðð ð r   )r   r   r6   r3   r   r   Ú_sparse_dataset_factoriesr:   ?   s-   € ðð å*1µ.Å(Ñ*KÔ*Kðñ ô ð r   c                  ó’   ‡ — t          ¦   «         t          ¦   «         z   Š ˆ fd„t          dt          ‰ ¦  «        d¦  «        D ¦   «         S )Nc                 ó*   •— g | ]}‰||d z   …         ‘ŒS )é   r3   )r4   ÚidxÚall_factoriess     €r   r5   z2_fused_types_dataset_factories.<locals>.<listcomp>I   s&   ø€ ÐTÐTÐT¨SˆM˜#  a¡˜-Ô(ÐTÐTÐTr   r   r=   )r7   r:   ÚrangeÚlen)r?   s   @r   Ú_fused_types_dataset_factoriesrB   F   sI   ø€ Ý,Ñ.Ô.Õ1JÑ1LÔ1LÑL€MàTÐTÐTÐTµE¸!½SÀÑ=OÔ=OÐQRÑ4SÔ4SÐTÑTÔTÐTr   r+   Údataset_factoryc                 ó6  — 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ÚdatasetÚ_Úxi_ÚyiÚswir>   Úxis              r   Ú test_seq_dataset_basic_iterationrP   L   s7  € ð
 €NØˆm�CÑ Ô €GØˆoÑÔ€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float_dtype, csr_containerc                 ó  — t          | ¦  «        }t          || ¦  «        }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)   r0   r@   rF   rG   Ú_shuffle_pyÚzip)r(   r+   Údense_datasetÚsparse_datasetÚirK   Úidx1Úidx2r   Úidx_nextÚidx_shuffleÚjs               r   Útest_seq_dataset_shufflerl   f   sä  € õ (¨Ñ4Ô4€MÝ)¨-¸ÑEÔE€Nå�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_factory, dataset_64_factoryc                 ón  —  | ¦   «          |¦   «         }}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_factoryÚdataset_64_factoryÚ
dataset_32Ú
dataset_64rH   rK   Ú	xi_data32Úyi32Ú	xi_data64Úyi64s
             r   Útest_fused_types_consistencyrx   ‹   sß   € ð 0Ð/Ñ1Ô1Ð3EÐ3EÑ3GÔ3G�
€JØ€NÝ�>Ñ"Ô"ð 	/ð 	/ˆà(2×(;Ò(;Ñ(=Ô(=Ñ%Ñˆ�A�q˜4  AØ(2×(;Ò(;Ñ(=Ô(=Ñ%Ñˆ�A�q˜4  AàŒ¥"¤*Ò,Ð,Ð,Ð,ØŒ¥"¤*Ò,Ð,Ð,Ð,å˜	 9°4Ð8Ñ8Ô8Ð8Ý˜˜d¨Ð.Ñ.Ô.Ð.Ð.ð	/ð 	/r   c            
      ó2  — t          j        t          d¬¦  «        5  t          t          t
          t          d¬¦  «         d d d ¦  «         n# 1 swxY w Y   t          j        t          d¬¦  «        5  t          t          t          t          d¬¦  «         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¬¦  «         d d d ¦  «         n# 1 swxY w Y   t          j        t          d¬¦  «        5  t!          |j        |j        |j        t          t          d¬¦  «         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   �   sx  € Ý	Œ•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Á1#B Â B$Â'B$Ã//D*Ä*D.	Ä1D.	Å/FÆF	ÆF	)/Ú	functoolsr   Ú	itertoolsr   Únumpyr    r{   Ú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$   r6   r   r)   r0   r7   r:   rB   ÚmarkÚparametrizerP   rl   rx   r   r3   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ðSð Sð Sðð ð ðUð Uð Uð „×Ò˜¨.Ñ9Ô9Ø„×ÒØÐ/Ð/Ñ1Ô1Ð4MÐ4MÑ4OÔ4OÑOñô ð+ð +ñô ñ :Ô9ð+ð, „×ÒØ  ' '¨(°NÑ"CÔ"Cñô ðð ñô ððD „×ÒØ,Ð.LÐ.LÑ.NÔ.Nñô ð/ð /ñô ð/ðð ð ð ð r   