ó
    ,ÊžjòT  ã                  óˆ  • S SK Jr  S SKJr  S SKJr  S SKrS SKrS SKrS SK	r	S SK
Jr  S SK
Jr  S SK
Jr  S SK
Jr  S S	K
Jr  S S
KJr  S SKJr  S SKJr  S SKJr  \(       a  S SKJr  S SKJr  S SKJr  S SKJr  S S jrS!S jr " S S\5      r " S S5      r  " S S\ 5      r! " S S\ 5      r" " S S\ 5      r# " S S\ 5      r$g)"é    )Úannotations)ÚAny)ÚTYPE_CHECKINGN)ÚBaseDistribution)ÚCategoricalChoiceType)ÚCategoricalDistribution)ÚFloatDistribution)ÚIntDistribution)ÚBaseSampler)ÚFrozenTrial)ÚTrial)Ú
TrialState)ÚCallable)ÚSequence)ÚMarkDecorator)ÚStudyc           
     ó†   ^ • [         R                  R                  ST  3U 4S jU 4S jU 4S jU 4S jU 4S jU 4S j/5      $ )NÚsuggest_method_c                ó*   >• U R                  TSS5      $ ©Nr   é
   ©Úsuggest_float©ÚtÚnames    €Ú¬C:\Users\johnn\AppData\Local\Packages\PythonSoftwareFoundation.Python.3.13_qbz5n2kfra8p0\LocalCache\local-packages\Python313\site-packages\optuna/testing/pytest_samplers.pyÚ<lambda>Ú,parametrize_suggest_method.<locals>.<lambda>#   s   ø€ �a—o‘o d¨A¨rÔ2ó    c                ó*   >• U R                  TSS5      $ r   ©Úsuggest_intr   s    €r   r   r   $   s   ø€ �a—m‘m D¨!¨RÔ0r    c                ó,   >• U R                  T/ SQ5      $ )N)r   é   é   )Úsuggest_categoricalr   s    €r   r   r   %   s   ø€ �a×+Ñ+¨D²)Ô<r    c                ó(   >• U R                  TSSSS9$ )Nr   r   ç      à?©Ústepr   r   s    €r   r   r   &   s   ø€ �a—o‘o d¨A¨r¸�oÑ<r    c                ó(   >• U R                  TSSSS9$ )NçH¯¼šò×z>r   T©Úlogr   r   s    €r   r   r   '   s   ø€ �a—o‘o d¨D°"¸$�oÑ?r    c                ó(   >• U R                  TSSSS9$ )Nr%   r   Tr.   r"   r   s    €r   r   r   (   s   ø€ �a—m‘m D¨!¨R°T�mÑ:r    )ÚpytestÚmarkÚparametrize)r   s   `r   Úparametrize_suggest_methodr4      s;   ø€ Ü�;‰;×"Ñ"Ø
˜$˜Ð ä2Ü0Ü<Ü<Ü?Ü:ð	
ó
ð 
r    c                ó‚   • U R                   R                  U R                  5      nU R                   R                  U5      $ ©N)Ú_storageÚcreate_new_trialÚ	_study_idÚ	get_trial)ÚstudyÚtrial_ids     r   Ú_create_new_trialr=   -   s/   € Ø�~‰~×.Ñ.¨u¯©Ó?€HØ�>‰>×#Ñ# HÓ-Ð-r    c                  ó|   • \ rS rSr        SS jr      S	S jr        S
S jr          SS jrSrg)ÚFixedSampleré2   c                ó(   • Xl         X l        X0l        g r6   )Úrelative_search_spaceÚrelative_paramsÚunknown_param_value)ÚselfrB   rC   rD   s       r   Ú__init__ÚFixedSampler.__init__3   s   € ð &;Ô"Ø.ÔØ#6Õ r    c                ó   • U R                   $ r6   )rB   )rE   r;   Útrials      r   Úinfer_relative_search_spaceÚ(FixedSampler.infer_relative_search_space=   s   € ð ×)Ñ)Ð)r    c                ó   • U R                   $ r6   )rC   )rE   r;   rI   Úsearch_spaces       r   Úsample_relativeÚFixedSampler.sample_relativeB   s   € ð ×#Ñ#Ð#r    c                ó   • U R                   $ r6   )rD   )rE   r;   rI   Ú
param_nameÚparam_distributions        r   Úsample_independentÚFixedSampler.sample_independentG   s   € ð ×'Ñ'Ð'r    )rC   rB   rD   N)rB   údict[str, BaseDistribution]rC   údict[str, Any]rD   r   ÚreturnÚNone)r;   r   rI   r   rW   rU   )r;   r   rI   r   rM   rU   rW   rV   )
r;   r   rI   r   rQ   ÚstrrR   r   rW   r   )	Ú__name__Ú
__module__Ú__qualname__Ú__firstlineno__rF   rJ   rN   rS   Ú__static_attributes__© r    r   r?   r?   2   s¤   † ð7à:ð7ð (ð7ð !ð	7ð
 
ô7ð*Øð*Ø#.ð*à	$ô*ð
$Øð$Ø#.ð$Ø>Yð$à	ô$ð
(àð(ð ð(ð ð	(ð
 -ð(ð 
÷(r    r?   c                  ó<   • \ rS rSr\R
                  SS j5       rSrg)Ú_BaseSamplerTestCaseéQ   c                ó   • [         er6   )ÚNotImplementedError)rE   s    r   ÚsamplerÚ_BaseSamplerTestCase.samplerR   s   € ä!Ð!r    r_   N)rW   úCallable[[], BaseSampler])rZ   r[   r\   r]   r1   Úfixturere   r^   r_   r    r   ra   ra   Q   s   † Ø‡^�^ó"ó ó"r    ra   c                  óÊ  • \ rS rSr\R
                  R                  S\" SS5      \" SS5      \" SS5      \" SSSS9\" S	S
SS9\" SSSS9/5            S&S j5       r\R
                  R                  S\	" S	S
5      \	" SS
5      \	" S	S5      \	" S	S
SS9\	" SS
SS9\	" S	SSS9\	" SSSS9/5      S'S j5       r
\R
                  R                  S/ SQ5            S(S j5       rS)S jrS)S jrS)S jrS)S jr\" S5            S*S j5       rS)S jr\" S5      \" S5              S+S  j5       5       r\R
                  R                  S!/ S"Q5              S,S# j5       rS$rg%)-ÚBasicSamplerTestCaseéW   Údistributionç      ð¿ç      ð?ç        r-   Tr.   éöÿÿÿr   çš™™™™™¹?r*   çffffff$Àçffffff$@c                óê  • [         R                  R                  U" 5       S9n[        R                  " [        S5       Vs/ s H*  nUR                  R                  U[        U5      SU5      PM,     sn5      n[        R                  " XRR                  :¬  5      (       d   e[        R                  " XRR                  :*  5      (       d   e[        UR                  R                  U[        U5      SU5      [        R                  5      (       a   eUR                  bR  XRR                  -  nXRR                  -  n[        R                  " U5      n[        R                   R#                  Xe5        g g s  snf ©N©re   éd   Úx)Úoptunar;   Úcreate_studyÚnpÚarrayÚrangere   rS   r=   ÚallÚlowÚhighÚ
isinstanceÚfloatingr+   ÚroundÚtestingÚassert_almost_equal)rE   re   rl   r;   Ú_ÚpointsÚround_pointss          r   Ú
test_floatÚBasicSamplerTestCase.test_floatX   s=  € ô  —‘×)Ñ)±'³)Ð)Ð<ˆÜ—’ô
 ˜sœó	ò $�Að —‘×0Ñ0ØÔ,¨UÓ3°S¸,öñ $ñ	ó
ˆô �vŠv�f× 0Ñ 0Ñ0×1Ñ1Ð1Ð1Ü�vŠv�f× 1Ñ 1Ñ1×2Ñ2Ð2Ð2ÜØ�M‰M×,Ñ,¨UÔ4EÀeÓ4LÈcÐS_Ó`Ü�K‰K÷
ñ 
ð 	
ð 
ð
 ×ÑÑ(à×&Ñ&Ñ&ˆFØ×'Ñ'Ñ'ˆFÜŸ8š8 FÓ+ˆLÜ�J‰J×*Ñ*¨<Õ@ð )ùòs   Á 1E0r   r&   r%   rw   c                ó,  • [         R                  R                  U" 5       S9n[        R                  " [        S5       Vs/ s H*  nUR                  R                  U[        U5      SU5      PM,     sn5      n[        R                  " XRR                  :¬  5      (       d   e[        R                  " XRR                  :*  5      (       d   e[        UR                  R                  U[        U5      SU5      [        R                  5      (       a   eg s  snf ru   )ry   r;   rz   r{   r|   r}   re   rS   r=   r~   r   r€   r�   Úinteger)rE   re   rl   r;   r†   r‡   s         r   Útest_intÚBasicSamplerTestCase.test_int   só   € ô —‘×)Ñ)±'³)Ð)Ð<ˆÜ—’ô
 ˜sœó	ò $�Að —‘×0Ñ0ØÔ,¨UÓ3°S¸,öñ $ñ	ó
ˆô �vŠv�f× 0Ñ 0Ñ0×1Ñ1Ð1Ð1Ü�vŠv�f× 1Ñ 1Ñ1×2Ñ2Ð2Ð2ÜØ�M‰M×,Ñ,¨UÔ4EÀeÓ4LÈcÐS_Ó`Ü�J‰J÷
ñ 
ð 	
ð 
ð 
ùòs   Á 1DÚchoices)©r%   r&   é   ©ÚaÚbÚc©r%   r“   c                óü  ^^• [        U5      m[        R                  R                  U" 5       S9mSUU4S jjn[        R
                  " [        S5       Vs/ s H	  oC" 5       PM     sn5      n[        R                  " US:¬  5      (       d   e[        R                  " U[        TR                  5      S-
  :*  5      (       d   e[        R                  " U5      n[        R                  R                  Xe5        g s  snf )Nrv   c                 óŠ   >• [        T5      n TR                  R                  TU ST5      n[        TR	                  U5      5      $ )Nrx   )r=   re   rS   ÚfloatÚto_internal_repr)rI   Úparam_valuerl   r;   s     €€r   ÚsampleÚ5BasicSamplerTestCase.test_categorical.<locals>.sample¤   s=   ø€ Ü% eÓ,ˆEØŸ-™-×:Ñ:¸5À%ÈÈlÓ[ˆKÜ˜×6Ñ6°{ÓCÓDÐDr    rw   r   r%   )rW   r™   )r   ry   r;   rz   r{   Úasarrayr}   r~   Úlenr�   rƒ   r„   r…   )	rE   re   r�   rœ   Úir‡   rˆ   rl   r;   s	          @@r   Útest_categoricalÚ%BasicSamplerTestCase.test_categoricalœ   sÃ   ù€ ô /¨wÓ7ˆä—‘×)Ñ)±'³)Ð)Ð<ˆ÷	Eð 	Eô
 —’¬u°S¬zÓ:ªz¨!˜VžX©zÑ:Ó;ˆô �vŠv�f ‘k×"Ñ"Ð"Ð"Ü�vŠv�f¤ L×$8Ñ$8Ó 9¸AÑ =Ñ=×>Ñ>Ð>Ð>Ü—x’x Ó'ˆÜ
�
‰
×&Ñ& |Õ<ùò ;s   ÁC9c                óÄ  • [         R                  R                  U" 5       S9n[        SS/S9n[        SS/S9nUR	                  [         R
                  " SSS.X4S.SS	95        UR	                  [         R
                  " SSS
.X4S
.SS	95        [        U5      nUR                  R                  X%SU5      nUS;   d   eUR                  R                  X%Xd5      nUS;   d   eg )Nrv   rx   Úy)r�   r“   r”   )Úcategoryrx   rq   )ÚparamsÚdistributionsÚvalue)r¥   r¤   r¥   ©rx   r¤   )r“   r”   )	ry   r;   rz   r   Ú	add_trialÚcreate_trialr=   re   rS   )rE   re   r;   Úcategorical_distributionÚdependent_distributionÚ_trialr¥   r¨   s           r   Ú#test_conditional_sample_independentÚ8BasicSamplerTestCase.test_conditional_sample_independent±   sï   € ô —‘×)Ñ)±'³)Ð)Ð<ˆÜ#:ÀCÈÀ:Ñ#NÐ Ü!8À#ÀsÀÑ!LÐà�‰Ü×ÒØ$'¨cÑ2Ø+CÑaØñô	
ð 	�‰Ü×ÒØ$'¨cÑ2Ø+CÑaØñô	
ô # 5Ó)ˆØ—=‘=×3Ñ3Ø˜:Ð'?ó
ˆð ˜:Ó%Ð%Ð%Ø—‘×0Ñ0°ÀÓaˆØ˜
Ó"Ð"Ñ"r    c                óŽ  ^^• [         R                  " U" 5       S9nSS jm[        SSS5       H  mUR                  UU4S jSSS9  M     [	        UR
                  5      S	:X  d   eUR                  U4S
 jSSS9  [	        UR
                  5      S	:X  d   eUR                  U4S jSSS9  [	        UR
                  5      S:X  d   eg )Nrv   c                ó.   • U R                  SSS5      U-   $ )Nrx   rq   gš™™™™™É?r   )rI   Ú
base_values     r   Ú	objectiveÚ@BasicSamplerTestCase.test_nan_objective_value.<locals>.objectiveÔ   s   € Ø×&Ñ& s¨C°Ó5¸
ÑBÐBr    r   r%   éÿÿÿÿc                ó   >• T" U T5      $ r6   r_   )r   r    r´   s    €€r   r   Ú?BasicSamplerTestCase.test_nan_objective_value.<locals>.<lambda>Ù   s   ø€ ¡Y¨q°!¤_r    r_   )Ún_trialsÚcatchr&   c                ó(   >• T" U [        S5      5      $ )NÚnan)r™   ©r   r´   s    €r   r   r¸   Ý   s   ø€ ¡¨1¬e°E«lÔ!;r    c                ó   >• T" U S5      $ )Nr%   r_   r½   s    €r   r   r¸   á   s   ø€ ¡¨1¨a¤r    )rI   r   r³   r™   rW   r™   )ry   rz   r}   ÚoptimizeÚintÚ
best_value)rE   re   r;   r    r´   s      @@r   Útest_nan_objective_valueÚ-BasicSamplerTestCase.test_nan_objective_valueÑ   s¸   ù€ Ü×#Ò#©G«IÑ6ˆô	Cô �r˜1˜bÖ!ˆAØ�N‰NÕ4¸qÈˆNÓKñ "ä�5×#Ñ#Ó$¨Ó)Ð)Ð)ð 	�‰Ô;ÀaÈrˆÑRÜ�5×#Ñ#Ó$¨Ó)Ð)Ð)ð 	�‰Ô0¸1ÀBˆÑGÜ�5×#Ñ#Ó$¨Ó)Ð)Ñ)r    c                óî  • [         R                  " U" 5       S9nS	S jnUR                  USS9  SS0n[        R                  " 5          [        R
                  " S[         R                  R                  5        [         R                  R                  XBR                  5      Ul
        S S S 5        UR                  USS9  UR                  S   R                  nUS   US   :X  d   eg ! , (       d  f       NF= f)
Nrv   c                ó‚   • U R                  SSS5      nU R                  SSS5      nU R                  SSS5      nX-   U-   $ )Nrx   r¶   r%   r¤   Úz©r   r#   )rI   rx   r¤   rÆ   s       r   r´   ÚCBasicSamplerTestCase.test_partial_fixed_sampling.<locals>.objectiveç   sJ   € Ø×#Ñ# C¨¨QÓ/ˆAØ×!Ñ! # r¨1Ó-ˆAØ×#Ñ# C¨¨QÓ/ˆAØ‘5˜1‘9Ðr    r%   ©r¹   r¤   r   Úignorer¶   ©rI   r   rW   r™   )ry   rz   r¿   ÚwarningsÚcatch_warningsÚsimplefilterÚ
exceptionsÚExperimentalWarningÚsamplersÚPartialFixedSamplerre   Útrialsr¦   )rE   re   r;   r´   Úfixed_paramsÚtrial_paramss         r   Útest_partial_fixed_samplingÚ0BasicSamplerTestCase.test_partial_fixed_samplingä   sÄ   € Ü×#Ò#©G«IÑ6ˆô	ð 	�‰�y¨1ˆÑ-ð ˜Q�xˆÜ×$Ò$Õ&Ü×!Ò! (¬F×,=Ñ,=×,QÑ,QÔRÜ"ŸO™O×?Ñ?ÀÏmÉmÓ\ˆEŒM÷ 'ð 	�‰�y¨1ˆÑ-Ø—|‘| BÑ'×.Ñ.ˆØ˜CÑ  L°Ñ$5Ó5Ð5Ñ5÷ 'Õ&ús   ÁAC&Ã&
C4c                óB  • [        S/5      [        SSS9[        SSSS9[        SSS9[        SSSS9[        SSSS9S.n[        R                  " 5          [        R
                  " S[        R                  R                  5        U" 5       nS S S 5        [        R                  R                  WS	9n[        S
5       HP  nUR                  US9nUR                  US5        UR                  5        H  nUR                  U   S:X  a  M   e   MR     g ! , (       d  f       N‹= f)Nr%   )r   r€   T)r   r€   r/   rn   )r   r€   r+   )r“   r”   r•   ÚdÚeÚfrÊ   rv   r&   )Úfixed_distributions)r   r
   r	   rÌ   rÍ   rÎ   ry   rÏ   rÐ   r;   rz   r}   ÚaskÚtellÚkeysr¦   )rE   re   rB   Úsampler_r;   r†   rI   rQ   s           r   Útest_sample_single_distributionÚ4BasicSamplerTestCase.test_sample_single_distributionù   sý   € ä(¨!¨Ó-Ü  Q¨QÑ/Ü  Q¨Q°DÑ9Ü" s°Ñ5Ü" s°¸$Ñ?Ü" s°¸3Ñ?ñ!
Ðô ×$Ò$Õ&Ü×!Ò! (¬F×,=Ñ,=×,QÑ,QÔRÙ“yˆH÷ 'ô —‘×)Ñ)°(Ð)Ð;ˆô �q–ˆAØ—I‘IÐ2G�IÐHˆEØ�J‰J�u˜cÔ"Ø3×8Ñ8Ö:�
Ø—|‘| JÑ/°1Õ4Ð4Ð4ó ;ò ÷ 'Õ&ús   Á7DÄ
Drx   c                ó¶  ^• SU4S jjn[         R                  " 5          [         R                  " S[        R                  R
                  5        U" 5       nS S S 5        [        R                  R                  WS9nUR                  USS9  [        UR                  5      S:X  d   e[        S UR                   5       5      (       d   eg ! , (       d  f       Nz= f)Nc                ó   >• T" U 5      $ r6   r_   )rI   Úsuggest_method_xs    €r   r´   ÚGBasicSamplerTestCase.test_single_parameter_objective.<locals>.objective  s   ø€ Ù# EÓ*Ð*r    rÊ   rv   r   rÉ   c              3  óZ   #   • U  H!  oR                   [        R                  :H  v •  M#     g 7fr6   ©Ústater   ÚCOMPLETE©Ú.0r   s     r   Ú	<genexpr>ÚGBasicSamplerTestCase.test_single_parameter_objective.<locals>.<genexpr>  ó   é € ÐHº<°a—7‘7œj×1Ñ1Ö1º<ùó   ‚)+rË   ©rÌ   rÍ   rÎ   ry   rÏ   rÐ   r;   rz   r¿   rŸ   rÓ   r~   )rE   re   rå   r´   rà   r;   s     `   r   Útest_single_parameter_objectiveÚ4BasicSamplerTestCase.test_single_parameter_objective  s¢   ø€ ÷	+ô ×$Ò$Õ&Ü×!Ò! (¬F×,=Ñ,=×,QÑ,QÔRÙ“yˆH÷ 'ô —‘×)Ñ)°(Ð)Ð;ˆØ�‰�y¨2ˆÑ.ä�5—<‘<Ó  BÓ&Ð&Ð&ÜÑH¸5¿<º<ÓH×HÑHÐHÑH÷ 'Õ&ús   Ÿ7C
Ã

Cc                ó®  • SS jn[         R                  " 5          [         R                  " S[        R                  R
                  5        U" 5       nS S S 5        [        R                  R                  WS9nUR                  USS9  [        UR                  5      S:X  d   e[        S UR                   5       5      (       d   eg ! , (       d  f       Nz= f)Nc                ó„   • U R                  SSS/5      nU(       a  U R                  SSS5      $ U R                  SSS5      $ )Nrx   TFr¤   r   r%   rÆ   )r'   r   )rI   rx   s     r   r´   ÚLBasicSamplerTestCase.test_conditional_parameter_objective.<locals>.objective!  sE   € Ø×)Ñ)¨#°°e¨}Ó=ˆAÞØ×*Ñ*¨3°°1Ó5Ð5Ø×&Ñ& s¨A¨qÓ1Ð1r    rÊ   rv   r   rÉ   c              3  óZ   #   • U  H!  oR                   [        R                  :H  v •  M#     g 7fr6   rè   rë   s     r   rí   ÚLBasicSamplerTestCase.test_conditional_parameter_objective.<locals>.<genexpr>/  rï   rð   rË   rñ   )rE   re   r´   rà   r;   s        r   Ú$test_conditional_parameter_objectiveÚ9BasicSamplerTestCase.test_conditional_parameter_objective   s¡   € ô	2ô ×$Ò$Õ&Ü×!Ò! (¬F×,=Ñ,=×,QÑ,QÔRÙ“yˆH÷ 'ô —‘×)Ñ)°(Ð)Ð;ˆØ�‰�y¨2ˆÑ.ä�5—<‘<Ó  BÓ&Ð&Ð&ÜÑH¸5¿<º<ÓH×HÑHÐHÑH÷ 'Õ&ús   ›7CÃ
Cr¤   c                óº  ^^• SUU4S jjn[         R                  " 5          [         R                  " S[        R                  R
                  5        U" 5       nS S S 5        [        R                  R                  WS9nUR                  USS9  [        UR                  5      S:X  d   e[        S UR                   5       5      (       d   eg ! , (       d  f       Nz= f)Nc                ó&   >• T" U 5      T" U 5      -   $ r6   r_   )rI   rå   Úsuggest_method_ys    €€r   r´   Ú]BasicSamplerTestCase.test_combination_of_different_distributions_objective.<locals>.objective9  s   ø€ Ù# EÓ*Ñ-=¸eÓ-DÑDÐDr    rÊ   rv   r‘   rÉ   c              3  óZ   #   • U  H!  oR                   [        R                  :H  v •  M#     g 7fr6   rè   rë   s     r   rí   Ú]BasicSamplerTestCase.test_combination_of_different_distributions_objective.<locals>.<genexpr>D  rï   rð   rË   rñ   )rE   re   rå   rý   r´   rà   r;   s     ``   r   Ú5test_combination_of_different_distributions_objectiveÚJBasicSamplerTestCase.test_combination_of_different_distributions_objective1  s©   ù€ ÷	Eð 	Eô ×$Ò$Õ&Ü×!Ò! (¬F×,=Ñ,=×,QÑ,QÔRÙ“yˆH÷ 'ô —‘×)Ñ)°(Ð)Ð;ˆØ�‰�y¨1ˆÑ-ä�5—<‘<Ó  AÓ%Ð%Ð%ÜÑH¸5¿<º<ÓH×HÑHÐHÑH÷ 'Õ&ús   ¡7CÃ
Czsecond_low,second_high))r   é   )r   é   )r  é   c                óè  ^^^• S
S jm[         R                  " 5          [         R                  " S[        R                  R
                  5        U" 5       nS S S 5        [        R                  R                  WS9nUR                  U4S jSS9  UR                  UUU4S jSS9  [        UR                  5      S:X  d   e[        S	 UR                   5       5      (       d   eg ! , (       d  f       N”= f)Nc                óR   • U R                  SX5      nX0R                  SX5      -  nU$ )Nrx   r¤   rÇ   )rI   r   r€   Úvs       r   r´   ÚDBasicSamplerTestCase.test_dynamic_range_objective.<locals>.objectiveQ  s.   € Ø×#Ñ# C¨Ó3ˆAØ×"Ñ" 3¨Ó2Ñ2ˆAØˆHr    rÊ   rv   c                ó   >• T" U SS5      $ r   r_   r½   s    €r   r   ÚCBasicSamplerTestCase.test_dynamic_range_objective.<locals>.<lambda>[  s   ø€ ¡¨1¨a°Ô!4r    r   rÉ   c                ó   >• T" U TT5      $ r6   r_   )r   r´   Úsecond_highÚ
second_lows    €€€r   r   r  \  s   ø€ ¡¨1¨j¸+Ô!Fr    r  c              3  óZ   #   • U  H!  oR                   [        R                  :H  v •  M#     g 7fr6   rè   rë   s     r   rí   ÚDBasicSamplerTestCase.test_dynamic_range_objective.<locals>.<genexpr>_  rï   rð   )rI   r   r   rÀ   r€   rÀ   rW   r™   rñ   )rE   re   r  r  rà   r;   r´   s     ``  @r   Útest_dynamic_range_objectiveÚ1BasicSamplerTestCase.test_dynamic_range_objectiveF  sµ   ú€ ô	ô
 ×$Ò$Õ&Ü×!Ò! (¬F×,=Ñ,=×,QÑ,QÔRÙ“yˆH÷ 'ô —‘×)Ñ)°(Ð)Ð;ˆØ�‰Ô4¸rˆÑBØ�‰ÖFÐQSˆÑTä�5—<‘<Ó  BÓ&Ð&Ð&ÜÑH¸5¿<º<ÓH×HÑHÐHÑH÷ 'Õ&ús   ž7C#Ã#
C1r_   N)re   rg   rl   r	   rW   rX   )re   rg   rl   r
   rW   rX   )re   rg   r�   zSequence[CategoricalChoiceType]rW   rX   ©re   rg   rW   rX   )re   rg   rå   úCallable[[Trial], float]rW   rX   )re   rg   rå   r  rý   r  rW   rX   )re   rg   r  rÀ   r  rÀ   rW   rX   )rZ   r[   r\   r]   r1   r2   r3   r	   r‰   r
   r�   r¡   r¯   rÂ   rÖ   rá   r4   rò   rù   r  r  r^   r_   r    r   rj   rj   W   s:  † Ø‡[�[×ÑØá˜d CÓ(Ù˜c 3Ó'Ù˜d CÓ(Ù˜d C¨TÑ2Ù˜c 2¨CÑ0Ù˜e T°Ñ4ð	
ó
ðAà*ðAð (ðAð 
ó	Aó
ðAð8 ‡[�[×ÑØá˜C Ó$Ù˜A˜rÓ"Ù˜C Ó#Ù˜C ¨!Ñ,Ù˜A˜r¨Ñ*Ù˜C ¨Ñ+Ù˜A˜s¨Ñ-ð	
óó
óð
ð" ‡[�[×Ñ˜YÒ(NÓOð=Ø0ð=Ø;Zð=à	ó=ó Pð=ô(#ô@*ô&6ô*5ñ,   Ó$ðIØ0ðIØD\ðIà	óIó %ðIô Iñ"   Ó$Ù Ó$ðIà*ðIð 3ðIð 3ð	Ið
 
óIó %ó %ðIð& ‡[�[×ÑØ ò	
óðIØ0ðIØ>AðIØPSðIà	óIóóIr    rj   c                  óN  • \ rS rSr\R
                  R                  S\" SS5      \" SSSS9\" SS	S
S9\" SS	5      \" SSSS9\" SSSS9/5      \R
                  R                  S\" SS5      \" SSSS9\" SS	S
S9\" SS	5      \" SSSS9\" SSSS9/5              SS j5       5       r	SS jr
\R
                  R                  S\" SS5      \" SSSS9\" SS	S
S9\" SS	5      \" SSSS9/5            SS j5       r\R
                  R                  SSS/5      SS j5       rSrg)ÚRelativeSamplerTestCaseib  Úx_distributionrm   rn   r-   Tr.   rp   r   r)   r*   r‘   r%   rw   é	   r&   Úy_distributionc                ó
  ^^• [        X#S9m[        R                  R                  U" 5       S9mTR	                  T5      nTR                  U[        UR                  R                  5       5      5        SUU4S jjn[        R                  " [        S5       Vs/ s H	  oe" 5       PM     sn5      n[        TR                  5       5       H„  u  p‰[        U	[        [        45      (       d   e[        R                   " US S 2U4   U	R"                  :¬  5      (       d   e[        R                   " US S 2U4   U	R$                  :*  5      (       a  M„   e   ['        U" 5       TR                  5       5       HŒ  u  p©[        U
[        R(                  5      (       a   e[        U
[        R*                  5      (       a   e[        U	[        5      (       a  [        U
[,        5      (       d   eMu  [        U
[.        5      (       a  MŒ   e   g s  snf )Nr©   rv   c                 ó„   >• TR                   R                  T[        T5      T5      n T Vs/ s H  oU   PM	     sn$ s  snf r6   ©re   rN   r=   ©r¦   r   rM   r;   s     €€r   rœ   ÚFRelativeSamplerTestCase.test_sample_relative_numerical.<locals>.sample„  ó<   ø€ Ø—]‘]×2Ñ2°5Ô:KÈEÓ:RÐT`ÓaˆFÙ-9Ó:ª\ T˜4”L©\Ñ:Ð:ùÒ:ó   ¬=r   )rW   zlist[int | float])Údictry   r;   rz   rÝ   rÞ   Úsumr¦   Úvaluesr{   r|   r}   Ú	enumerater�   r	   r
   r~   r   r€   Úzipr‚   rŒ   rÀ   r™   )rE   re   r  r  rI   rœ   r†   r‡   r    rl   r›   rM   r;   s              @@r   Útest_sample_relative_numericalÚ6RelativeSamplerTestCase.test_sample_relative_numericalc  sŸ  ù€ ô8 59¸>Ñ4\ˆÜ—‘×)Ñ)±'³)Ð)Ð<ˆØ—	‘	˜,Ó'ˆØ�
‰
�5œ#˜eŸl™l×1Ñ1Ó3Ó4Ô5÷	;ð 	;ô —’¬U°2¬YÓ7ªY¨˜6ž8©YÑ7Ó8ˆÜ(¨×)<Ñ)<Ó)>Ö?‰OˆAÜØä%Ü#ð÷ñ ð ð ô —6’6˜&¢ A ™,¨,×*:Ñ*:Ñ:×;Ñ;Ð;Ð;Ü—6’6˜&¢ A ™,¨,×*;Ñ*;Ñ;×<Ó<Ð<Ð<ñ  @ô *-©V«X°|×7JÑ7JÓ7LÖ)MÑ%ˆKÜ! +¬r¯{©{×;Ñ;Ð;Ð;Ü! +¬r¯z©z×:Ñ:Ð:Ð:Ü˜,¬×8Ñ8Ü! +¬s×3Ñ3Ð3Ò3ä! +¬u×5Ó5Ð5Ð5ò *Nùò 8s   ÂH c                ód  ^
^• [        [        / SQ5      [        / SQ5      S9m
[        R                  R	                  U" 5       S9mTR                  T
5      nTR                  U[        UR                  R                  5       5      5        SU
U4S jjn[        R                  " [        S5       Vs/ s H	  oC" 5       PM     sn5      n[        T
R                  5       5       H]  u  pg[        U[        5      (       d   e[        R                  " US S 2U4    Vs/ s H  oˆUR                   ;   PM     sn5      (       a  M]   e   U" 5        H\  n	[        U	[        R"                  5      (       a   e[        U	[        R$                  5      (       a   e[        U	[&        5      (       a  M\   e   g s  snf s  snf )N)r%   r   rw   ©r¶   rp   iœÿÿÿr©   rv   c                 ó„   >• TR                   R                  T[        T5      T5      n T Vs/ s H  oU   PM	     sn$ s  snf r6   r  r  s     €€r   rœ   ÚHRelativeSamplerTestCase.test_sample_relative_categorical.<locals>.sample£  r  r   r   ©rW   zlist[float])r!  r   ry   r;   rz   rÝ   rÞ   r"  r¦   r#  r{   r|   r}   r$  r�   r~   r�   r‚   rŒ   rÀ   )rE   re   rI   rœ   r†   r‡   r    rl   r  r›   rM   r;   s             @@r   Ú test_sample_relative_categoricalÚ8RelativeSamplerTestCase.test_sample_relative_categorical›  sS  ù€ Ü48Ü%¢lÓ3Ô7NÊÓ7_ñ5
ˆô —‘×)Ñ)±'³)Ð)Ð<ˆØ—	‘	˜,Ó'ˆØ�
‰
�5œ#˜eŸl™l×1Ñ1Ó3Ó4Ô5÷	;ð 	;ô —’¬U°2¬YÓ7ªY¨˜6ž8©YÑ7Ó8ˆÜ(¨×)<Ñ)<Ó)>Ö?‰OˆAÜ˜lÔ,C×DÑDÐDÐDÜ—6’6¸fÂQÈÀTºlÓKºl¸ × 4Ñ 4Ô4¹lÑK×LÓLÐLÐLñ  @ñ "ž8ˆKÜ! +¬r¯{©{×;Ñ;Ð;Ð;Ü! +¬r¯z©z×:Ñ:Ð:Ð:Ü˜k¬3×/Ó/Ð/Ð/ò $ùò	 8ùò Ls   Â/F(ÄF-
c                óÂ  ^
^• [        U[        / SQ5      S9m
[        R                  R	                  U" 5       S9mTR                  T
5      nTR                  U[        UR                  R                  5       5      5        S
U
U4S jjn[        R                  " [        S5       Vs/ s H	  oT" 5       PM     sn5      n[        T
S   [        [        45      (       d   e[        R                   " US S 2S4   T
S   R"                  :¬  5      (       d   e[        R                   " US S 2S4   T
S   R$                  :*  5      (       d   e[        T
S   [        5      (       d   e[        R                   " US S 2S	4    Vs/ s H  owT
S   R&                  ;   PM     sn5      (       d   e[)        U" 5       T
R                  5       5       H’  u  p‰[        U[        R*                  5      (       a   e[        U[        R,                  5      (       a   e[        U	[        [        45      (       a  [        U[.        5      (       d   eM{  [        U[0        5      (       a  M’   e   g s  snf s  snf )Nr)  r©   rv   c                 ó„   >• TR                   R                  T[        T5      T5      n T Vs/ s H  oU   PM	     sn$ s  snf r6   r  r  s     €€r   rœ   ÚBRelativeSamplerTestCase.test_sample_relative_mixed.<locals>.sampleÄ  r  r   r   rx   r   r¤   r%   r,  )r!  r   ry   r;   rz   rÝ   rÞ   r"  r¦   r#  r{   r|   r}   r�   r	   r
   r~   r   r€   r�   r%  r‚   rŒ   rÀ   r™   )rE   re   r  rI   rœ   r†   r‡   r  r›   rl   rM   r;   s             @@r   Útest_sample_relative_mixedÚ2RelativeSamplerTestCase.test_sample_relative_mixed°  s  ù€ ô 59ØÔ 7ºÓ Hñ5
ˆô —‘×)Ñ)±'³)Ð)Ð<ˆØ—	‘	˜,Ó'ˆØ�
‰
�5œ#˜eŸl™l×1Ñ1Ó3Ó4Ô5÷	;ð 	;ô —’¬U°2¬YÓ7ªY¨˜6ž8©YÑ7Ó8ˆÜØ˜Ñä!Üð÷
ñ 
ð 	
ð 
ô �vŠv�fšQ ˜T‘l l°3Ñ&7×&;Ñ&;Ñ;×<Ñ<Ð<Ð<Ü�vŠv�fšQ ˜T‘l l°3Ñ&7×&<Ñ&<Ñ<×=Ñ=Ð=Ð=Ü˜, sÑ+Ô-D×EÑEÐEÐEÜ�vŠv¸vÂaÈÀdº|ÓLº|¸!˜L¨Ñ-×5Ñ5Ô5¹|ÑL×MÑMÐMÐMÜ),©V«X°|×7JÑ7JÓ7LÖ)MÑ%ˆKÜ! +¬r¯{©{×;Ñ;Ð;Ð;Ü! +¬r¯z©z×:Ñ:Ð:Ð:ÜØä#Ü+ð÷ñ ô " +¬s×3Ñ3Ð3Ò3ä! +¬u×5Ó5Ð5Ð5ò *Nùò 8ùò Ms   Â$IÅ8IÚn_jobsc                óx   • U" 5       n[         R                  R                  US9nSS jnUR                  USUS9  g )Nrv   c                óŒ   • U R                   b   eU R                  SSS5        U R                  SSS5        U R                   c   eg)Nrx   rp   r   r¤   r¶   )Ú_relative_paramsr   )rI   s    r   r´   ÚERelativeSamplerTestCase.test_trial_relative_params.<locals>.objectiveè  sM   € Ø×)Ñ)Ñ1Ð1Ð1à×Ñ  S¨"Ô-Ø×Ñ  S¨"Ô-Ø×)Ñ)Ñ5Ð5Ð5Ør    r   )r¹   r4  rË   )ry   r;   rz   r¿   )rE   r4  re   rà   r;   r´   s         r   Útest_trial_relative_paramsÚ2RelativeSamplerTestCase.test_trial_relative_paramsâ  s;   € ñ “9ˆÜ—‘×)Ñ)°(Ð)Ð;ˆô	ð 	�‰�y¨2°fˆÒ=r    r_   N)re   rg   r  r   r  r   rW   rX   r  )re   rg   r  r   rW   rX   )r4  rÀ   re   rg   rW   rX   )rZ   r[   r\   r]   r1   r2   r3   r	   r
   r&  r-  r2  r9  r^   r_   r    r   r  r  b  s¥  † Ø‡[�[×ÑØá˜d CÓ(Ù˜d C¨TÑ2Ù˜c 2¨CÑ0Ù˜A˜rÓ"Ù˜A˜s¨Ñ-Ù˜A˜q qÑ)ð	
ó
ð ‡[�[×ÑØá˜d CÓ(Ù˜d C¨TÑ2Ù˜c 2¨CÑ0Ù˜A˜rÓ"Ù˜A˜s¨Ñ-Ù˜A˜q qÑ)ð	
ó
ð 6à*ð 6ð )ð 6ð )ð	 6ð
 
ó 6ó
ó
ð, 6ôD0ð* ‡[�[×ÑØá˜d CÓ(Ù˜d C¨TÑ2Ù˜c 2¨CÑ0Ù˜A˜rÓ"Ù˜A˜s¨Ñ-ð	
ó	ð&6Ø0ð&6ØBRð&6à	ó&6ó	ð&6ðP ‡[�[×Ñ˜X¨¨1 vÓ.ó>ó /ó>r    r  c                  óT  • \ rS rSr\R
                  R                  S\" SS5      \" SS5      \" SS5      \" SSSS9\" S	S
SS9\" SSSS9\" S	S
5      \" SS
5      \" S	S5      \" S	S
SS9\" SS
SS9\" S	SSS9\" SSSS9\	" S5      \	" S5      \	" S5      /5            SS j5       r
Srg)ÚMultiObjectiveSamplerTestCaseió  rl   rm   rn   ro   r-   Tr.   rp   r   rq   r*   rr   rs   r   r&   r%   rw   r�   r’   r–   c                ój  • [         R                  R                  SS/U" 5       S9n[        S5       Hÿ  nUR                  R                  U[        U5      SU5      nUR                  UR                  U5      5      (       d   e[        U[        5      (       d!  [        U[        R                  5      (       a   e[        U[        5      (       d  Mž  UR                  c  M­  XRR                  -  nXRR                  -  n[        R                   " U5      n[        R"                  R%                  Xe5        GM     g )NÚminimizeÚmaximize©Ú
directionsre   rw   rx   )ry   r;   rz   r}   re   rS   r=   Ú	_containsrš   r�   r   r{   r‚   r	   r+   r   rƒ   r„   r…   )rE   re   rl   r;   r    r¨   Úround_values          r   Ú'test_multi_objective_sample_independentÚEMultiObjectiveSamplerTestCase.test_multi_objective_sample_independentô  sù   € ô4 —‘×)Ñ)°jÀ*Ð5MÑW^ÓW`Ð)ÐaˆÜ�s–ˆAØ—M‘M×4Ñ4ØÔ(¨Ó/°°lóˆEð  ×)Ñ)¨,×*GÑ*GÈÓ*N×OÑOÐOÐOä˜lÔ,C×DÑDô & e¬R¯[©[×9Ñ9Ð9Ð9ä˜,Ô(9×:Ó:Ø×$Ñ$Ó0ð ×-Ñ-Ñ-�EØ×.Ñ.Ñ.�EÜ"$§(¢(¨5£/�KÜ—J‘J×2Ñ2°;×Fò% r    r_   N)re   rg   rl   r   rW   rX   )rZ   r[   r\   r]   r1   r2   r3   r	   r
   r   rD  r^   r_   r    r   r<  r<  ó  sô   † Ø‡[�[×ÑØá˜d CÓ(Ù˜c 3Ó'Ù˜d CÓ(Ù˜d C¨TÑ2Ù˜c 2¨CÑ0Ù˜e T°Ñ4Ù˜C Ó$Ù˜A˜rÓ"Ù˜C Ó#Ù˜C ¨!Ñ,Ù˜A˜r¨Ñ*Ù˜C ¨Ñ+Ù˜A˜s¨Ñ-Ù# IÓ.Ù# OÓ4Ù# HÓ-ð!	
óð*Gà*ðGð 'ðGð 
ó	Gó+ó*Gr    r<  c                  ó&   • \ rS rSr    SS jrSrg)ÚSingleOnlySamplerTestCasei$  c           	     ó  • [         R                  R                  SS/U" 5       S9n[        SS5      n[        R
                  " [        5         UR                  R                  U[        U5      SU5        S S S 5        [        R
                  " [        5         [        U5      nUR                  R                  X$UR                  R                  X$5      5        S S S 5        g ! , (       d  f       Nr= f! , (       d  f       g = f)Nr?  r@  ro   rn   rx   )ry   r;   rz   r	   r1   ÚraisesÚ
ValueErrorre   rS   r=   rN   rJ   )rE   re   r;   rl   rI   s        r   Ú5test_raise_error_for_samplers_during_multi_objectivesÚOSingleOnlySamplerTestCase.test_raise_error_for_samplers_during_multi_objectives%  sº   € ô —‘×)Ñ)°jÀ*Ð5MÑW^ÓW`Ð)Ðaˆä(¨¨cÓ2ˆÜ�]Š]œ:Õ&Ø�M‰M×,Ñ,¨UÔ4EÀeÓ4LÈcÐS_Ô`÷ 'ô �]Š]œ:Õ&Ü% eÓ,ˆEØ�M‰M×)Ñ)Ø˜eŸm™m×GÑGÈÓUô÷ 'Ð&÷ 'Õ&ú÷ 'Õ&ús   Á(C ÂAC1Ã 
C.Ã1
C?r_   Nr  )rZ   r[   r\   r]   rK  r^   r_   r    r   rG  rG  $  s   † ðà*ðð 
÷r    rG  )r   rY   rW   r   )r;   r   rW   r   )%Ú
__future__r   Útypingr   r   rÌ   Únumpyr{   r1   ry   Úoptuna.distributionsr   r   r   r	   r
   Úoptuna.samplersr   Úoptuna.trialr   r   r   Úcollections.abcr   r   Ú_pytest.mark.structuresr   Úoptuna.studyr   r4   r=   r?   ra   rj   r  r<  rG  r_   r    r   Ú<module>rV     s¥   ðÝ "å Ý  Û ã Û ã Ý 1Ý 6Ý 8Ý 2Ý 0Ý 'Ý $Ý Ý #ö Ý(Ý(å5å"ôô.ô
(�;ô (÷>"ñ "ôHIÐ/ô HIôVN>Ð2ô N>ôb.GÐ$8ô .GôbÐ 4õ r    