Ë
    ÷Q(ht  ã                   óä   — d Z ddlZddlmZ ddlZddlZddlmZm	Z	 ddl
mZ ddlmZ g d¢Z ej                  d¬	«      d
„ «       Z ej                  d¬	«      d„ «       Zd„ Zd„ Zd„ Zd„ Zd„ Zd„ Zy)av  This test for the LFW require medium-size data downloading and processing

If the data has not been already downloaded by running the examples,
the tests won't run (skipped).

If the test are run, the first execution will be long (typically a bit
more than a couple of minutes) but as the dataset loader is leveraging
joblib, successive runs will be fast (less than 200ms).
é    N)Úpartial)Úfetch_lfw_pairsÚfetch_lfw_people)Úcheck_return_X_y)Úassert_array_equal)ÚAbdelatif_SmithÚAbhati_KeplerÚCamara_AlvaroÚChen_DupontÚJohn_LeeÚ
Lin_BaumanÚ
Onur_LopezÚmodule)Úscopec              #   ó6   K  — | j                  d«      }|–— y ­w)NÚscikit_learn_empty_test)Úmktemp)Útmp_path_factoryÚdata_dirs     ú]/var/www/skyplay_api_hub/venv/lib/python3.12/site-packages/sklearn/datasets/tests/test_lfw.pyÚmock_empty_data_homer       s   è ø€ à×&Ñ&Ð'@ÓA€Hà
ƒNùs   ‚c           	   #   ó   K  — t        j                  d«      }| j                  d«      }|dz  }|j                  dd¬«       t	        j
                  d«      }t        j                  j                  d«      }i }t        D ]¢  }|dz  |z  }|j                  dd¬«       |j                  dd	«      }	|	||<   t        |	«      D ]`  }
||d
|
z  z   z  }|j                  ddd¬«      }|j                  |j                  t        j                  «      «      }|j                  |«       Œb Œ¤ |dz  dz  j                  d«       t!        |dz  d«      5 }|j#                  d«       |j%                  «       D ��cg c]  \  }}|dk\  sŒ|‘Œ }}}t        d	«      D ]Y  }
|j'                  |«      }|j)                  t        ||   «      d«      \  }}|j#                  d|||fz  j+                  «       «       Œ[ t        d	«      D ]�  }
|j)                  t        d«      \  }}|j'                  t        j,                  ||   «      «      }|j'                  t        j,                  ||   «      «      }|j#                  d||||fz  j+                  «       «       Œ‘ 	 ddd«       |dz  j                  d«       |dz  j                  d«       |–— yc c}}w # 1 sw Y   Œ<xY w­w)z<Test fixture run once and common to all tests of this modulez	PIL.ImageÚscikit_learn_lfw_testÚlfw_homeT)ÚparentsÚexist_oké*   Úlfw_funneledé   é   z	_%04d.jpgr   éÿ   )éú   r"   é   )Úsizez	.test.swps.   Text file to be ignored by the dataset loader.zpairsDevTrain.txtÚwbs   10
é   z	%s	%d	%d
z%s	%d	%s	%d
NzpairsDevTest.txts&   Fake place holder that won't be testedz	pairs.txt)ÚpytestÚimportorskipr   ÚmkdirÚrandomÚRandomÚnpÚRandomStateÚ
FAKE_NAMESÚrandintÚrangeÚ	fromarrayÚastypeÚuint8ÚsaveÚwrite_bytesÚopenÚwriteÚitemsÚchoiceÚsampleÚencodeÚarange)r   ÚImager   r   Úrandom_stateÚnp_rngÚcountsÚnameÚfolder_nameÚn_facesÚiÚ	file_pathÚunifaceÚimgÚfÚcountÚmore_than_twoÚfirstÚsecondÚ
first_nameÚsecond_nameÚfirst_indexÚsecond_indexs                          r   Úmock_data_homerQ   '   s¶  è ø€ ô ×Ñ Ó,€Eà×&Ñ&Ð'>Ó?€HØ˜*Ñ$€HØ‡N�N˜4¨$€NÔ/ä—=‘= Ó$€LÜ�Y‰Y×"Ñ" 2Ó&€Fð €FÜò 
 ˆØ Ñ/°$Ñ6ˆØ×Ñ $°ÐÔ6à—.‘.  AÓ&ˆØˆˆt‰Ü�w“ò 	 ˆAØ# t¨k¸A©oÑ'=Ñ>ˆIØ—n‘n Q¨°-�nÓ@ˆGØ—/‘/ '§.¡.´·±Ó":Ó;ˆCØ�H‰H�YÕñ		 ð
 ð �Ñ Ñ,×9Ñ9Ø9ôô
 
ˆhÐ,Ñ,¨dÓ	3ð °qØ	�‰�ÔØ17·±³×N¡+ $¨À5ÈAÃ:šÐNˆÑNÜ�q“ò 	GˆAØ×&Ñ& }Ó5ˆDØ(×/Ñ/´°f¸T±lÓ0CÀQÓG‰MˆE�6Ø�G‰G�^ t¨U°FÐ&;Ñ;×CÑCÓEÕFð	Gô
 �q“ò 		ˆAØ&2×&9Ñ&9¼*ÀaÓ&HÑ#ˆJ˜Ø Ÿ-™-¬¯	©	°&¸Ñ2DÓ(EÓFˆKØ!Ÿ=™=¬¯©°6¸+Ñ3FÓ)GÓHˆLØ�G‰Gà&Ø! ;°¸\ÐJñKç‘&“(õ	ñ			÷ð& Ð"Ñ"×/Ñ/Ø1ôð �Ñ×(Ñ(Ð)RÔSà
ƒNùó- O÷ð üs7   ‚EKÅ%KÅ-J<Å;J<Å?DKÊ5KÊ<KËKËKc                 óz   — t        j                  t        «      5  t        | d¬«       d d d «       y # 1 sw Y   y xY w©NF©Ú	data_homeÚdownload_if_missing)r'   ÚraisesÚOSErrorr   ©r   s    r   Útest_load_empty_lfw_peoplerZ   b   s2   € Ü	�‰”wÓ	ñ TÜÐ#7ÈUÕS÷T÷ Tñ Túó   š1±:c                 ó"  — t        | dd¬«      }|j                  j                  dk(  sJ ‚|j                  j                  dk(  sJ ‚t	        |j
                  g d¢«       g d¢}t	        |j                  |«       t        | d d dd¬	«      }|j                  j                  d
k(  sJ ‚|j                  j                  d«      sJ ‚t	        |j
                  g d¢«       t	        |j                  g d¢«       t        t         | d d dd¬	«      }t        ||«       y )Nr#   F©rU   Úmin_faces_per_personrV   )é
   é>   é/   )r_   ib  )
r&   r   r   r   r&   r   r&   r   r   r&   )úAbdelatif SmithúAbhati Keplerú
Onur LopezT©rU   ÚresizeÚslice_ÚcolorrV   )é   r"   r"   r#   ú&.. _labeled_faces_in_the_wild_dataset:)r   r   r   é   r    rk   r#   rk   r   r#   rk   r   r&   é   r    r   r&   )rb   rc   zCamara AlvarozChen DupontzJohn Leez
Lin Baumanrd   )r   ÚimagesÚshapeÚdatar   ÚtargetÚtarget_namesÚDESCRÚ
startswithr   r   )rQ   Ú
lfw_peopleÚexpected_classesÚ
fetch_funcs       r   Útest_load_fake_lfw_peoplerw   g   s  € Ü!Ø °qÈeô€Jð ×Ñ×"Ñ" lÒ2Ð2Ð2Ø�?‰?× Ñ  JÒ.Ð.Ð.ô �z×(Ñ(Ò*HÔIò JÐÜ�z×.Ñ.Ð0@ÔAô "Ø ØØØØ!ô€Jð ×Ñ×"Ñ"Ð&7Ò7Ð7Ð7Ø×Ñ×&Ñ&Ð'OÔPÐPÐPô Ø×ÑÒNôô Ø×Ñò	
ôô ÜØ ØØØØ!ô€Jô �Z Õ,ó    c                 ó|   — t        j                  t        «      5  t        | dd¬«       d d d «       y # 1 sw Y   y xY w)Néd   Fr]   )r'   rW   Ú
ValueErrorr   )rQ   s    r   Ú)test_load_fake_lfw_people_too_restrictiver|   ¡   s4   € Ü	�‰”zÓ	"ñ 
ÜØ$Ø!$Ø %õ	
÷
÷ 
ñ 
ús   š2²;c                 óz   — t        j                  t        «      5  t        | d¬«       d d d «       y # 1 sw Y   y xY wrS   )r'   rW   rX   r   rY   s    r   Útest_load_empty_lfw_pairsr~   ª   s2   € Ü	�‰”wÓ	ñ SÜÐ"6ÈEÕR÷S÷ Sñ Súr[   c                 ó¤  — t        | d¬«      }|j                  j                  dk(  sJ ‚t        |j                  g d¢«       ddg}t        |j
                  |«       t        | d d dd¬«      }|j                  j                  d	k(  sJ ‚t        |j                  g d¢«       t        |j
                  |«       |j                  j                  d
«      sJ ‚y )NFrT   )r_   r&   r`   ra   )
r   r   r   r   r   r   r   r   r   r   zDifferent personszSame personTre   )r_   r&   r"   r"   r#   rj   )r   Úpairsrn   r   rp   rq   rr   rs   )rQ   Úlfw_pairs_trainru   s      r   Útest_load_fake_lfw_pairsr‚   ¯   sÕ   € Ü%Ø °eô€Oð × Ñ ×&Ñ&¨/Ò9Ð9Ð9ô �×-Ñ-Ò/MÔNð ,¨]Ð;ÐÜ�×3Ñ3Ð5EÔFô &Ø ØØØØ!ô€Oð × Ñ ×&Ñ&Ð*>Ò>Ð>Ð>ô �×-Ñ-Ò/MÔNÜ�×3Ñ3Ð5EÔFà× Ñ ×+Ñ+Ð,TÔUÐUÑUrx   c                 ó  — t        dd«      t        dd«      f}t        | ddd|¬«      }|j                  d	   j                  |d	   j                  |d	   j
                  z
  |d
   j                  |d
   j
                  z
  fk(  sJ ‚y)zˆCheck that we properly crop the images.

    Non-regression test for:
    https://github.com/scikit-learn/scikit-learn/issues/24942
    éF   éÃ   éN   é¬   r#   FN)rU   r^   rV   rf   rg   r   r   )Úslicer   rm   rn   ÚstopÚstart)rQ   rg   Úlfws      r   Ú'test_fetch_lfw_people_internal_croppingrŒ   Ñ   sŠ   € ô �B˜‹nœe B¨›nÐ-€FÜ
Ø ØØ!ØØô€Cð �:‰:�a‰=×ÑØˆq‰	�‰˜ ™Ÿ™Ñ(Øˆq‰	�‰˜ ™Ÿ™Ñ(ð#ò ð ñ rx   )Ú__doc__r*   Ú	functoolsr   Únumpyr,   r'   Úsklearn.datasetsr   r   Ú"sklearn.datasets.tests.test_commonr   Úsklearn.utils._testingr   r.   Úfixturer   rQ   rZ   rw   r|   r~   r‚   rŒ   © rx   r   ú<module>r•      sŠ   ðñó Ý ã Û ç >Ý ?Ý 5ò€
ð €‡��hÔñó  ðð €‡��hÔñ7ó  ð7òtTò
7-òt
òSò
VóDrx   