Ë
    þÍ:jéÃ  ã                  óÈ  — U d dl mZ d dlZd dlmZ d dlZd dlZd dlZd dlZd dlm	Z	 d dl
mZ d dl
mZ d dlZd dlZd dlZd dlmZ d dlmZ d d	lmZ d d
lmZ d dlmZ d dlmZ d dlmZ d dlmZ erd dlm Z  d dl!m"Z" ejF                  d)d„«       Z$ G d„ d«      Z% e&e«      Z'dddddgdœdœZ(de)d<   d ejT                  ejV                  jX                  ejV                  jX                   ejV                  jZ                  ejV                  jZ                    e.d«       e.d«        e.d«      dœ	Z/d*d„Z0d+d „Z1d+d!„Z2d+d"„Z3d+d#„Z4d+d$„Z5d+d%„Z6d+d&„Z7	 d,	 	 	 	 	 	 	 	 	 	 	 d-d'„Z8d.d(„Z9y)/é    )ÚannotationsN)Údatetime)Úsleep)ÚAny)ÚTYPE_CHECKING)ÚCategoricalDistribution)ÚFloatDistribution)ÚUpdateFinishedTrialError)ÚDEFAULT_STUDY_NAME_PREFIX)ÚFrozenStudy)ÚStudyDirection)ÚFrozenTrial)Ú
TrialState)ÚJSONSerializable)ÚBaseStoragec                 ó   — t         ‚©N)ÚNotImplementedError© ó    ús/home/mcse/projects/srt_converter/srt-converter-venv/lib/python3.12/site-packages/optuna/testing/pytest_storages.pyÚstorager       s   € ä
Ðr   c                  ó˜  — e Zd Zd<d„Zd<d„Zd<d„Zd<d„Zd<d„Zd<d„Zd<d„Z	d<d„Z
d<d	„Zd<d
„Zd<d„Zd<d„Zd<d„Zej"                  j%                  d ej(                  «        ej(                  «       f eddd«       eddd«      fg«      	 	 	 	 	 	 	 	 d=d„«       Zd<d„Zd<d„Zd<d„Zd<d„Zd<d„Zd<d„Zd<d„Zd<d„Zd<d„Zd<d„Zd<d„Z d<d„Z!d<d „Z"d<d!„Z#d<d"„Z$d<d#„Z%d<d$„Z&d<d%„Z'ej"                  j%                  d&d'd(gd(d'gg«      	 	 	 	 	 	 d>d)„«       Z(d<d*„Z)d<d+„Z*d<d,„Z+d<d-„Z,d<d.„Z-ej"                  j%                  d/e.j^                  e.j`                  g«      ej"                  j%                  d0g d1¢d2 e1d3«      d4gd2 e1d5«      d4g e1d3«      d2d4 e1d5«      g e1d3«      g e1d5«      gg«      	 	 	 	 	 	 	 	 d?d6„«       «       Z2d<d7„Z3d<d8„Z4d<d9„Z5d<d:„Z6y;)@ÚStorageTestCasec                óú  — |j                  t        j                  g¬«      }|j                  «       }t	        |«      dk(  sJ ‚|d   j
                  |k(  sJ ‚|d   j                  j                  t        «      sJ ‚|j                  t        j                  g¬«      }||k7  sJ ‚|j                  «       }t	        |«      dk(  sJ ‚|D �ch c]  }|j
                  ’Œ c}||hk(  sJ ‚t        d„ |D «       «      sJ ‚y c c}w )N©Ú
directionsé   r   é   c              3  óZ   K  — | ]#  }|j                   j                  t        «      –— Œ% y ­wr   )Ú
study_nameÚ
startswithr   )Ú.0Úss     r   ú	<genexpr>z8StorageTestCase.test_create_new_study.<locals>.<genexpr>4   s    è ø€ Ò^È!�1—<‘<×*Ñ*Ô+D×EÑ^ùs   ‚)+)
Úcreate_new_studyr   ÚMINIMIZEÚget_all_studiesÚlenÚ	_study_idr!   r"   r   Úall)Úselfr   Ústudy_idÚfrozen_studiesÚ	study_id2r$   s         r   Útest_create_new_studyz%StorageTestCase.test_create_new_study&   sþ   € Ø×+Ñ+¼×8OÑ8OÐ7PÐ+ÓQˆà ×0Ñ0Ó2ˆÜ�>Ó" aÒ'Ð'Ð'Ø˜aÑ ×*Ñ*¨hÒ6Ð6Ð6Ø˜aÑ ×+Ñ+×6Ñ6Ô7PÔQÐQÐQà×,Ñ,¼×9PÑ9PÐ8QÐ,ÓRˆ	à˜9Ò$Ð$Ð$Ø ×0Ñ0Ó2ˆÜ�>Ó" aÒ'Ð'Ð'Ø%3Ö4 �—“Ò4¸À9Ð8MÒMÐMÐMÜÑ^È~Ô^Ô^Ð^Ñ^ùò 5s   ÃC8c                óX  — |j                  t        j                  g¬«      }|j                  t        j                  g¬«      }|j                  |«       |j                  t        j                  g¬«      }|j	                  «       }|D �ch c]  }|j
                  ’Œ c}||hk(  sJ ‚y c c}w ©Nr   ©r&   r   r'   Údelete_studyr(   r*   )r,   r   r-   r/   Ú	study_id3r.   r$   s          r   Útest_create_new_study_unique_idz/StorageTestCase.test_create_new_study_unique_id6   s—   € Ø×+Ñ+¼×8OÑ8OÐ7PÐ+ÓQˆØ×,Ñ,¼×9PÑ9PÐ8QÐ,ÓRˆ	Ø×Ñ˜YÔ'Ø×,Ñ,¼×9PÑ9PÐ8QÐ,ÓRˆ	à ×0Ñ0Ó2ˆØ%3Ö4 �—“Ò4¸À9Ð8MÒMÐMÑMùÒ4s   Â	B'c                ól  — | j                   j                  }|}|j                  t        j                  g|¬«      }||j                  |«      k(  sJ ‚t        j                  t        j                  j                  «      5  |j                  t        j                  g|¬«       d d d «       y # 1 sw Y   y xY w)N©r   r!   )Útest_create_new_study_with_nameÚ__name__r&   r   r'   Úget_study_name_from_idÚpytestÚraisesÚoptunaÚ
exceptionsÚDuplicatedStudyError©r,   r   Úfunction_namer!   r-   s        r   r9   z/StorageTestCase.test_create_new_study_with_name?   s¤   € à×<Ñ<×EÑEˆØ"ˆ
Ø×+Ñ+Ü&×/Ñ/Ð0¸Zð ,ó 
ˆð ˜W×;Ñ;¸HÓEÒEÐEÐEä�]‰]œ6×,Ñ,×AÑAÓBñ 	bØ×$Ñ$´×1HÑ1HÐ0IÐV`Ð$Ôa÷	b÷ 	bñ 	bús   Á>#B*Â*B3c                ót  — |j                  t        j                  g¬«      }|j                  |«       |j	                  |«      }t        |«      dk(  sJ ‚t        j                  t        «      5  |j                  |dz   «       d d d «       |j                  |«       |j                  t        j                  g¬«      }|j	                  |«      }t        |«      dk(  sJ ‚|j                  |«       t        j                  t        «      5  |j                  |«       d d d «       y # 1 sw Y   Œ¡xY w# 1 sw Y   y xY w)Nr   r   r   )
r&   r   r'   Úcreate_new_trialÚget_all_trialsr)   r<   r=   ÚKeyErrorr4   )r,   r   r-   Útrialss       r   Útest_delete_studyz!StorageTestCase.test_delete_studyL   s  € Ø×+Ñ+¼×8OÑ8OÐ7PÐ+ÓQˆØ× Ñ  Ô*Ø×'Ñ'¨Ó1ˆÜ�6‹{˜aÒÐÐä�]‰]œ8Ó$ñ 	/à× Ñ  ¨A¡Ô.÷	/ð 	×Ñ˜XÔ&Ø×+Ñ+¼×8OÑ8OÐ7PÐ+ÓQˆØ×'Ñ'¨Ó1ˆÜ�6‹{˜aÒÐÐà×Ñ˜XÔ&Ü�]‰]œ8Ó$ñ 	+à× Ñ  Ô*÷	+ð 	+÷	/ð 	/ú÷	+ð 	+ús   Á-D"ÄD.Ä"D+Ä.D7c                ój  — |j                  t        j                  g¬«      }|j                  t        j                  g¬«      }|j                  t        j                  g¬«      }|j                  |«       |j	                  «       D �ci c]  }|j
                  |“Œ }}||v sJ ‚||vsJ ‚||v sJ ‚y c c}w r2   r3   )r,   r   Ú	study_id1r/   r5   r$   Ústudiess          r   Ú/test_delete_study_after_create_multiple_studiesz?StorageTestCase.test_delete_study_after_create_multiple_studies`   s¸   € Ø×,Ñ,¼×9PÑ9PÐ8QÐ,ÓRˆ	Ø×,Ñ,¼×9PÑ9PÐ8QÐ,ÓRˆ	Ø×,Ñ,¼×9PÑ9PÐ8QÐ,ÓRˆ	à×Ñ˜YÔ'à+2×+BÑ+BÓ+DÖE a�1—;‘; ‘>ÐEˆÐEØ˜GÑ#Ð#Ð#Ø Ñ'Ð'Ð'Ø˜GÑ#Ð#Ñ#ùò Fs   ÂB0c                óÒ  — | j                   j                  }|}|j                  t        j                  g|¬«      }|j                  |«      |k(  sJ ‚|j                  |«      |k(  sJ ‚t        j                  t        «      5  |j                  d«       d d d «       t        j                  t        «      5  |j                  |dz   «       d d d «       y # 1 sw Y   Œ@xY w# 1 sw Y   y xY w)Nr8   z
dummy-namer   )
Ú6test_get_study_id_from_name_and_get_study_name_from_idr:   r&   r   r'   r;   Úget_study_id_from_namer<   r=   rF   rA   s        r   rN   zFStorageTestCase.test_get_study_id_from_name_and_get_study_name_from_idl   sÚ   € à×SÑS×\Ñ\ˆØ"ˆ
Ø×+Ñ+Ü&×/Ñ/Ð0¸Zð ,ó 
ˆð
 ×-Ñ-¨hÓ7¸:ÒEÐEÐEØ×-Ñ-¨jÓ9¸XÒEÐEÐEô �]‰]œ8Ó$ñ 	9Ø×*Ñ*¨<Ô8÷	9ô �]‰]œ8Ó$ñ 	9Ø×*Ñ*¨8°a©<Ô8÷	9ð 	9÷	9ð 	9ú÷	9ð 	9ús   Â CÂ3CÃCÃC&c                ó  ‡‡‡— t         j                  ft         j                  ft         j                  t         j                  ft         j                  t         j                  ft         j                  t         j                  gt         j                  t         j                  gfD ][  Š‰j                  ‰¬«      Šdˆˆˆfd„} |«        ‰dz   }t	        j
                  t        «      5  ‰j                  |«       d d d «       Œ] y # 1 sw Y   ŒhxY w)Nr   c                 óR   •— ‰j                  ‰«      } | t        ‰«      k(  sJ d«       ‚y )NzCDirection of a study should be a tuple of `StudyDirection` objects.)Úget_study_directionsÚlist)Úgot_directionsr   r-   Útargets    €€€r   Ú	check_getzDStorageTestCase.test_set_and_get_study_directions.<locals>.check_getŠ   s/   ø€ Ø!(×!=Ñ!=¸hÓ!G�à%¬¨f«Ò5ð ØYóÑ5r   r   )ÚreturnÚNone)r   r'   ÚMAXIMIZEr&   r<   r=   rF   rR   )r,   r   rV   Únon_existent_study_idr-   rU   s    `  @@r   Ú!test_set_and_get_study_directionsz1StorageTestCase.test_set_and_get_study_directions   så   ú€ ä×$Ñ$Ð&Ü×$Ñ$Ð&Ü×$Ñ$¤n×&=Ñ&=Ð>Ü×$Ñ$¤n×&=Ñ&=Ð>Ü×$Ñ$¤n×&=Ñ&=Ð>Ü×$Ñ$¤n×&=Ñ&=Ð>ð
ò 	DˆFð ×/Ñ/¸6Ð/ÓBˆH÷ñ ŒKð %-¨q¡LÐ!ä—‘œxÓ(ñ DØ×,Ñ,Ð-BÔC÷Dð Dñ/	D÷.Dð Dús   Ã$DÄD
	c                óò  ‡‡— ‰j                  t        j                  g¬«      Šdˆˆfd„}t        j	                  «       D ]  \  }} |||«       Œ ‰j                  ‰«      t        k(  sJ ‚ |dd«       ‰dz   }t        j                  t        «      5  ‰j                  |«       d d d «       t        j                  t        «      5  ‰j                  |dd«       d d d «       y # 1 sw Y   Œ?xY w# 1 sw Y   y xY w)	Nr   ÚkeyÚvaluec                ó^   •— ‰j                  ‰| |«       ‰j                  ‰«      |    |k(  sJ ‚y r   )Úset_study_user_attrÚget_study_user_attrs©r]   r^   r   r-   s     €€r   Úcheck_set_and_getzLStorageTestCase.test_set_and_get_study_user_attrs.<locals>.check_set_and_get�   s3   ø€ Ø×'Ñ'¨°#°uÔ=Ø×/Ñ/°Ó9¸#Ñ>À%ÒGÐGÑGr   ÚdatasetÚImageNetr   ©r]   Ústrr^   r   rW   rX   )
r&   r   r'   ÚEXAMPLE_ATTRSÚitemsra   r<   r=   rF   r`   ©r,   r   rc   r]   r^   rZ   r-   s    `    @r   Ú!test_set_and_get_study_user_attrsz1StorageTestCase.test_set_and_get_study_user_attrsš   sò   ù€ Ø×+Ñ+¼×8OÑ8OÐ7PÐ+ÓQˆö	Hô
 (×-Ñ-Ó/ò 	*‰JˆC�Ù˜c 5Õ)ð	*à×+Ñ+¨HÓ5¼ÒFÐFÐFñ 	˜) ZÔ0ð !)¨1¡ÐÜ�]‰]œ8Ó$ñ 	@Ø×(Ñ(Ð)>Ô?÷	@ô �]‰]œ8Ó$ñ 	OØ×'Ñ'Ð(=¸uÀgÔN÷	Oð 	O÷		@ð 	@ú÷	Oð 	Oúó   ÂC!ÃC-Ã!C*Ã-C6c                ó   — t        |«       y r   )Ú-_test_set_and_get_study_user_attrs_for_floats©r,   r   s     r   Ú,test_set_and_get_study_user_attrs_for_floatsz<StorageTestCase.test_set_and_get_study_user_attrs_for_floats²   ó
   € Ü5°gÕ>r   c                óò  ‡‡— ‰j                  t        j                  g¬«      Šdˆˆfd„}t        j	                  «       D ]  \  }} |||«       Œ ‰j                  ‰«      t        k(  sJ ‚ |dd«       ‰dz   }t        j                  t        «      5  ‰j                  |«       d d d «       t        j                  t        «      5  ‰j                  |dd«       d d d «       y # 1 sw Y   Œ?xY w# 1 sw Y   y xY w)	Nr   r]   r^   c                ó^   •— ‰j                  ‰| |«       ‰j                  ‰«      |    |k(  sJ ‚y r   )Úset_study_system_attrÚget_study_system_attrsrb   s     €€r   rc   zNStorageTestCase.test_set_and_get_study_system_attrs.<locals>.check_set_and_get¸   ó3   ø€ Ø×)Ñ)¨(°C¸Ô?Ø×1Ñ1°(Ó;¸CÑ@ÀEÒIÐIÑIr   rd   re   r   rf   )
r&   r   r'   rh   ri   ru   r<   r=   rF   rt   rj   s    `    @r   Ú#test_set_and_get_study_system_attrsz3StorageTestCase.test_set_and_get_study_system_attrsµ   sò   ù€ Ø×+Ñ+¼×8OÑ8OÐ7PÐ+ÓQˆö	Jô
 (×-Ñ-Ó/ò 	*‰JˆC�Ù˜c 5Õ)ð	*à×-Ñ-¨hÓ7¼=ÒHÐHÐHñ 	˜) ZÔ0ð !)¨1¡ÐÜ�]‰]œ8Ó$ñ 	BØ×*Ñ*Ð+@ÔA÷	Bô �]‰]œ8Ó$ñ 	QØ×)Ñ)Ð*?ÀÈÔP÷	Qð 	Q÷		Bð 	Bú÷	Qð 	Qúrl   c                ó   — t        |«       y r   )Ú/_test_set_and_get_study_system_attrs_for_floatsro   s     r   Ú.test_set_and_get_study_system_attrs_for_floatsz>StorageTestCase.test_set_and_get_study_system_attrs_for_floatsÍ   s
   € Ü7¸Õ@r   c                ó  — |j                  t        j                  g¬«      }t        j	                  «       D ]  \  }}|j                  |||«       Œ |j                  |«      t        k(  sJ ‚|j                  |«      i k(  sJ ‚|j                  t        j                  g¬«      }t        j	                  «       D ]  \  }}|j                  |||«       Œ |j                  |«      t        k(  sJ ‚|j                  |«      i k(  sJ ‚y r2   )	r&   r   r'   rh   ri   rt   ru   ra   r`   )r,   r   r-   r]   r^   s        r   Ú*test_study_user_and_system_attrs_confusionz:StorageTestCase.test_study_user_and_system_attrs_confusionÐ   sü   € Ø×+Ñ+¼×8OÑ8OÐ7PÐ+ÓQˆÜ'×-Ñ-Ó/ò 	@‰JˆC�Ø×)Ñ)¨(°C¸Õ?ð	@à×-Ñ-¨hÓ7¼=ÒHÐHÐHØ×+Ñ+¨HÓ5¸Ò;Ð;Ð;à×+Ñ+¼×8OÑ8OÐ7PÐ+ÓQˆÜ'×-Ñ-Ó/ò 	>‰JˆC�Ø×'Ñ'¨°#°uÕ=ð	>à×+Ñ+¨HÓ5¼ÒFÐFÐFØ×-Ñ-¨hÓ7¸2Ò=Ð=Ñ=r   c                ó¤  — 	 	 	 	 	 	 	 	 	 	 	 	 dd„}|j                  t        j                  g¬«      }d}t        |«      D ]n  }t	        j
                  «       }t        d«       |j                  |«      }t        d«       t	        j
                  «       }|j                  |«      }	 ||	||||«       Œp t        j                  t        «      5  |j                  |dz   «       d d d «       |j                  t        j                  g¬«      }
t        |«      D ]X  }|j                  |
«       |j                  |
«      }	|	D �ch c]  }|j                  ’Œ c}t        t        |dz   «      «      k(  rŒXJ ‚ |j                  |«      |j                  |
«      z   }	t        |	D �ch c]  }|j                  ’Œ c}«      d|z  k(  sJ ‚y # 1 sw Y   ŒßxY wc c}w c c}w )Nc                ó¾  ‡‡‡— t        | «      |dz   k(  sJ ‚t        | D �ch c]  }|j                  ’Œ c}«      |dz   k(  sJ ‚‰| D �ch c]  }|j                  ’Œ c}v sJ ‚| D �ch c]  }|j                  ’Œ c}t        t	        |dz   «      «      k(  sJ ‚t        d„ | D «       «      sJ ‚t        d„ | D «       «      sJ ‚t        d„ | D «       «      sJ ‚t        d„ | D «       «      sJ ‚t        d„ | D «       «      sJ ‚t        ˆˆfd„| D «       «      sJ ‚t        ˆˆˆfd„| D «       «      sJ ‚t        d	„ | D «       «      sJ ‚t        d
„ | D «       «      sJ ‚y c c}w c c}w c c}w )Nr   c              3  óV   K  — | ]!  }|j                   t        j                  k(  –— Œ# y ­wr   )Ústater   ÚRUNNING©r#   Úts     r   r%   zOStorageTestCase.test_create_new_trial.<locals>._check_trials.<locals>.<genexpr>é   s   è ø€ ÒE¸�q—w‘w¤*×"4Ñ"4Õ4ÑEùó   ‚')c              3  ó:   K  — | ]  }|j                   i k(  –— Œ y ­wr   ©Úparamsr‚   s     r   r%   zOStorageTestCase.test_create_new_trial.<locals>._check_trials.<locals>.<genexpr>ê   s   è ø€ Ò6¨!�q—x‘x 2•~Ñ6ùó   ‚c              3  ó:   K  — | ]  }|j                   i k(  –— Œ y ­wr   ©Úintermediate_valuesr‚   s     r   r%   zOStorageTestCase.test_create_new_trial.<locals>._check_trials.<locals>.<genexpr>ë   s   è ø€ ÒC°q�q×,Ñ,°Õ2ÑCùrˆ   c              3  ó:   K  — | ]  }|j                   i k(  –— Œ y ­wr   ©Ú
user_attrsr‚   s     r   r%   zOStorageTestCase.test_create_new_trial.<locals>._check_trials.<locals>.<genexpr>ì   s   è ø€ Ò:¨a�q—|‘| rÕ)Ñ:ùrˆ   c              3  ó:   K  — | ]  }|j                   i k(  –— Œ y ­wr   ©Úsystem_attrsr‚   s     r   r%   zOStorageTestCase.test_create_new_trial.<locals>._check_trials.<locals>.<genexpr>í   s   è ø€ Ò<°�q—~‘~¨Õ+Ñ<ùrˆ   c              3  ór   •K  — | ].  }|j                   ‰k7  r|j                  �|j                  ‰k  –— Œ0 y ­wr   ©Ú	_trial_idÚdatetime_start)r#   rƒ   Útime_before_creationÚtrial_ids     €€r   r%   zOStorageTestCase.test_create_new_trial.<locals>._check_trials.<locals>.<genexpr>î   s<   øè ø€ ò àØ—;‘; (Ò*¨q×/?Ñ/?Ð/Kð × Ñ Ð#7Õ7ñùs   ƒ47c              3  óˆ   •K  — | ]9  }|j                   ‰k(  r(|j                  �‰|j                  cxk  xr ‰k  nc –— Œ; y ­wr   r“   )r#   rƒ   Útime_after_creationr–   r—   s     €€€r   r%   zOStorageTestCase.test_create_new_trial.<locals>._check_trials.<locals>.<genexpr>ó   sD   øè ø€ ò àØ—;‘; (Ò*¨q×/?Ñ/?Ð/Kð % q×'7Ñ'7ÖMÐ:M×MÐMñùs   ƒ?Ac              3  ó8   K  — | ]  }|j                   d u –— Œ y ­wr   ©Údatetime_completer‚   s     r   r%   zOStorageTestCase.test_create_new_trial.<locals>._check_trials.<locals>.<genexpr>ø   s   è ø€ ÒC°q�q×*Ñ*¨dÔ2ÑCùó   ‚c              3  ó8   K  — | ]  }|j                   d u –— Œ y ­wr   ©r^   r‚   s     r   r%   zOStorageTestCase.test_create_new_trial.<locals>._check_trials.<locals>.<genexpr>ù   s   è ø€ Ò7¨1�q—w‘w $”Ñ7ùr�   ©r)   r”   ÚnumberÚsetÚranger+   )rG   Úidxr—   r–   r™   rƒ   s     ``` r   Ú_check_trialsz<StorageTestCase.test_create_new_trial.<locals>._check_trialsÞ   sa  ú€ ô �v“; #¨¡'Ò)Ð)Ð)Ü¨VÖ4¨˜Ÿ›Ò4Ó5¸¸q¹Ò@Ð@Ð@Ø°VÖ<° §£Ò<Ñ<Ð<Ð<Ø&,Ö- �A—H“HÒ-´´U¸3À¹7³^Ó1DÒDÐDÐDÜÑE¸fÔEÔEÐEÐEÜÑ6¨vÔ6Ô6Ð6Ð6ÜÑC¸FÔCÔCÐCÐCÜÑ:°6Ô:Ô:Ð:Ð:ÜÑ<°VÔ<Ô<Ð<Ð<Üô àôô ð ð ô
 õ àôô ð ð ô
 ÑC¸FÔCÔCÐCÐCÜÑ7°Ô7Ô7Ð7Ñ7ùò' 5ùÚ<ùÚ-s    EÁEÁ$Er   é   çü©ñÒMbP?r   r   )rG   úlist[FrozenTrial]r¤   Úintr—   r©   r–   r   r™   r   rW   rX   )r&   r   r'   r£   r   Únowr   rD   rE   r<   r=   rF   r¡   r¢   r)   r”   )r,   r   r¥   r-   Ún_trial_in_studyÚir–   r—   r™   rG   r/   rƒ   s               r   Útest_create_new_trialz%StorageTestCase.test_create_new_trialÝ   s×  € ð	8Ø%ð	8àð	8ð ð	8ð #+ð		8ð
 "*ð	8ð ó	8ð: ×+Ñ+¼×8OÑ8OÐ7PÐ+ÓQˆØÐÜÐ'Ó(ò 	ZˆAÜ#+§<¡<£>Ð Ü�%ŒLØ×/Ñ/°Ó9ˆHÜ�%ŒLÜ"*§,¡,£.Ðà×+Ñ+¨HÓ5ˆFÙ˜& ! XÐ/CÐEXÕYð	Zô �]‰]œ8Ó$ñ 	3Ø×$Ñ$ X°¡\Ô2÷	3ð ×,Ñ,¼×9PÑ9PÐ8QÐ,ÓRˆ	ÜÐ'Ó(ò 	CˆAØ×$Ñ$ YÔ/à×+Ñ+¨IÓ6ˆFà&,Ö- �A—H“HÒ-´´U¸1¸q¹5³\Ó1BÓBÐBÐBð	Cð ×'Ñ'¨Ó1°G×4JÑ4JÈ9Ó4UÑUˆä¨Ö0 A�A—K“KÒ0Ó1°QÐ9IÑ5IÒIÐIÑI÷	3ð 	3üò .ùò 1s   Ã	F<Ä;GÆGÆ<Gzstart_time,complete_timeiæ  é	   r   r   c                óŽ  ‡— t        t        j                  d||ddidt        dd«      iddidd	id
dddœdd¬«      Šdˆfd„}|j	                  t
        j                  g¬«      }d}t        |«      D ]0  }|j                  |‰¬«      }|j                  |«      }	 ||	||«       Œ2 t        j                  t        «      5  |j                  |dz   «       d d d «       |j	                  t
        j                  g¬«      }
t        |«      D ]Z  }|j                  |
‰¬«       |j                  |
«      }	|	D �ch c]  }|j                  ’Œ c}t        t        |dz   «      «      k(  rŒZJ ‚ |j                  |«      |j                  |
«      z   }	t        |	D �ch c]  }|j                   ’Œ c}«      d|z  k(  sJ ‚y # 1 sw Y   ŒáxY wc c}w c c}w )Ni'  Úxç      à?r   r   ÚfooÚbarÚbazé{   é
   éd   iè  )r   r   r¦   é7   éÿÿÿÿ)r€   r^   r•   rœ   r‡   ÚdistributionsrŽ   r‘   r‹   r¡   r—   c                óÐ  •— t        | «      |dz   k(  sJ ‚t        | D �ch c]  }|j                  ’Œ c}«      |dz   k(  sJ ‚|| D �ch c]  }|j                  ’Œ c}v sJ ‚| D �ch c]  }|j                  ’Œ c}t        t	        |dz   «      «      k(  sJ ‚t        ˆfd„| D «       «      sJ ‚t        ˆfd„| D «       «      sJ ‚t        ˆfd„| D «       «      sJ ‚t        ˆfd„| D «       «      sJ ‚t        ˆfd„| D «       «      sJ ‚t        ˆfd„| D «       «      sJ ‚t        ˆfd„| D «       «      sJ ‚t        ˆfd	„| D «       «      sJ ‚t        ˆfd
„| D «       «      sJ ‚y c c}w c c}w c c}w )Nr   c              3  óP   •K  — | ]  }|j                   ‰j                   k(  –— Œ y ­wr   ©r€   ©r#   rƒ   Útemplate_trials     €r   r%   zcStorageTestCase.test_create_new_trial_with_template_trial.<locals>._check_trials.<locals>.<genexpr>1  ó    øè ø€ ÒG¸1�q—w‘w .×"6Ñ"6Õ6ÑGùó   ƒ#&c              3  óP   •K  — | ]  }|j                   ‰j                   k(  –— Œ y ­wr   r†   r¾   s     €r   r%   zcStorageTestCase.test_create_new_trial_with_template_trial.<locals>._check_trials.<locals>.<genexpr>2  s    øè ø€ ÒI¸Q�q—x‘x >×#8Ñ#8Õ8ÑIùrÁ   c              3  óP   •K  — | ]  }|j                   ‰j                   k(  –— Œ y ­wr   )rº   r¾   s     €r   r%   zcStorageTestCase.test_create_new_trial_with_template_trial.<locals>._check_trials.<locals>.<genexpr>3  s    øè ø€ ÒWÈ1�q—‘¨.×*FÑ*FÕFÑWùrÁ   c              3  óP   •K  — | ]  }|j                   ‰j                   k(  –— Œ y ­wr   rŠ   r¾   s     €r   r%   zcStorageTestCase.test_create_new_trial_with_template_trial.<locals>._check_trials.<locals>.<genexpr>4  s#   øè ø€ ÒcÐWX�q×,Ñ,°×0RÑ0RÕRÑcùrÁ   c              3  óP   •K  — | ]  }|j                   ‰j                   k(  –— Œ y ­wr   r�   r¾   s     €r   r%   zcStorageTestCase.test_create_new_trial_with_template_trial.<locals>._check_trials.<locals>.<genexpr>5  s    øè ø€ ÒQÀQ�q—|‘| ~×'@Ñ'@Õ@ÑQùrÁ   c              3  óP   •K  — | ]  }|j                   ‰j                   k(  –— Œ y ­wr   r�   r¾   s     €r   r%   zcStorageTestCase.test_create_new_trial_with_template_trial.<locals>._check_trials.<locals>.<genexpr>6  s    øè ø€ ÒUÈ�q—~‘~¨×)DÑ)DÕDÑUùrÁ   c              3  óP   •K  — | ]  }|j                   ‰j                   k(  –— Œ y ­wr   )r•   r¾   s     €r   r%   zcStorageTestCase.test_create_new_trial_with_template_trial.<locals>._check_trials.<locals>.<genexpr>7  s"   øè ø€ ÒYÈQ�q×'Ñ'¨>×+HÑ+HÕHÑYùrÁ   c              3  óP   •K  — | ]  }|j                   ‰j                   k(  –— Œ y ­wr   r›   r¾   s     €r   r%   zcStorageTestCase.test_create_new_trial_with_template_trial.<locals>._check_trials.<locals>.<genexpr>8  s#   øè ø€ Ò_ÐST�q×*Ñ*¨n×.NÑ.NÕNÑ_ùrÁ   c              3  óP   •K  — | ]  }|j                   ‰j                   k(  –— Œ y ­wr   rŸ   r¾   s     €r   r%   zcStorageTestCase.test_create_new_trial_with_template_trial.<locals>._check_trials.<locals>.<genexpr>9  rÀ   rÁ   r    )rG   r¤   r—   rƒ   r¿   s       €r   r¥   zPStorageTestCase.test_create_new_trial_with_template_trial.<locals>._check_trials,  sI  ø€ Ü�v“; #¨¡'Ò)Ð)Ð)Ü¨VÖ4¨˜Ÿ›Ò4Ó5¸¸q¹Ò@Ð@Ð@Ø°VÖ<° §£Ò<Ñ<Ð<Ð<Ø&,Ö- �A—H“HÒ-´´U¸3À¹7³^Ó1DÒDÐDÐDÜÓGÀÔGÔGÐGÐGÜÓIÀ&ÔIÔIÐIÐIÜÓWÐPVÔWÔWÐWÐWÜÓcÐ\bÔcÔcÐcÐcÜÓQÈ&ÔQÔQÐQÐQÜÓUÈfÔUÔUÐUÐUÜÓYÐRXÔYÔYÐYÐYÜÓ_ÐX^Ô_Ô_Ð_Ð_ÜÓGÀÔGÔGÐGÑGùò 5ùÚ<ùÚ-s   žEÁEÁ"E#r   r¦   ©r¿   r   )rG   r¨   r¤   r©   r—   r©   rW   rX   )r   r   ÚCOMPLETEr	   r&   r   r'   r£   rD   rE   r<   r=   rF   r¡   r¢   r)   r”   )r,   r   Ú
start_timeÚcomplete_timer¥   r-   r«   r¬   r—   rG   r/   rƒ   r¿   s               @r   Ú)test_create_new_trial_with_template_trialz9StorageTestCase.test_create_new_trial_with_template_trial  sÐ  ø€ ô %Ü×%Ñ%ØØ%Ø+Ø˜�:ØÔ 1°!°QÓ 7Ð8Ø˜u�~Ø ˜Ø$&¨3°4Ñ 8ØØô
ˆõ	Hð ×+Ñ+¼×8OÑ8OÐ7PÐ+ÓQˆàÐÜÐ'Ó(ò 	/ˆAØ×/Ñ/°ÈÐ/ÓXˆHØ×+Ñ+¨HÓ5ˆFÙ˜& ! XÕ.ð	/ô �]‰]œ8Ó$ñ 	3Ø×$Ñ$ X°¡\Ô2÷	3ð ×,Ñ,¼×9PÑ9PÐ8QÐ,ÓRˆ	ÜÐ'Ó(ò 	CˆAØ×$Ñ$ Y¸~Ð$ÔNØ×+Ñ+¨IÓ6ˆFØ&,Ö- �A—H“HÒ-´´U¸1¸q¹5³\Ó1BÓBÐBÐBð	Cð
 ×'Ñ'¨Ó1°G×4JÑ4JÈ9Ó4UÑUˆä¨Ö0 A�A—K“KÒ0Ó1°QÐ9IÑ5IÒIÐIÑI÷	3ð 	3üò .ùò 1s   Â<F1Ä0F=ÆGÆ1F:c                óf  — |j                  t        j                  g¬«      }|j                  |«      }|j	                  |«      dk(  sJ ‚|j                  |«      }|j	                  |«      dk(  sJ ‚t        j                  t        «      5  |j	                  |dz   «       d d d «       y # 1 sw Y   y xY w)Nr   r   r   )r&   r   r'   rD   Úget_trial_number_from_idr<   r=   rF   ©r,   r   r-   r—   s       r   Útest_get_trial_number_from_idz-StorageTestCase.test_get_trial_number_from_idQ  s£   € à×+Ñ+¼×8OÑ8OÐ7PÐ+ÓQˆà×+Ñ+¨HÓ5ˆØ×/Ñ/°Ó9¸QÒ>Ð>Ð>à×+Ñ+¨HÓ5ˆØ×/Ñ/°Ó9¸QÒ>Ð>Ð>ä�]‰]œ8Ó$ñ 	;Ø×,Ñ,¨X¸©\Ô:÷	;÷ 	;ñ 	;ús   Â	B'Â'B0c                ó  — |j                  t        j                  g¬«      }t        D �cg c]  }|j	                  |«      ‘Œ }}t        |t        «      D �](  \  }}|t        j                  k(  rŒ|j                  |«      j                  t        j                  k(  sJ ‚|j                  |«      j                  }|j                  |||j                  «       rdnd ¬«       |j                  |«      j                  |k(  sJ ‚|t        j                  k(  r |j                  |«      j                  |k(  sJ ‚|j                  «       r|j                  |«      j                  ��Œ
J ‚|j                  |«      j                  €�Œ)J ‚ t        j                   t"        «      5  t%        |«      dz   }|j                  |t        j&                  ¬«       d d d «       t        D ]‘  }|j                  «       sŒ|j	                  |«      }|j                  ||d¬«       t        D ]P  }	d }
|	t        j&                  k(  rdg}
t        j                   t(        «      5  |j                  ||	|
¬«       d d d «       ŒR Œ“ y c c}w # 1 sw Y   Œ©xY w# 1 sw Y   ŒpxY w)Nr   )ç        ©r€   Úvaluesr   r½   rÔ   )r&   r   r'   Ú
ALL_STATESrD   Úzipr   ÚWAITINGÚ	get_trialr€   r�   r•   Úset_trial_state_valuesÚis_finishedrœ   r<   r=   rF   ÚmaxrË   r
   )r,   r   r-   Ú_Ú	trial_idsr—   r€   Údatetime_start_prevÚnon_existent_trial_idÚstate2rÖ   s              r   Ú%test_set_trial_state_values_for_statez5StorageTestCase.test_set_trial_state_values_for_state^  sm  € Ø×+Ñ+¼×8OÑ8OÐ7PÐ+ÓQˆÜAKÖL¸A�W×-Ñ-¨hÕ7ÐLˆ	ÐLä" 9¬jÓ9ó 	M‰OˆH�eØœ
×*Ñ*Ò*ØØ×$Ñ$ XÓ.×4Ñ4¼
×8JÑ8JÒJÐJÐJØ")×"3Ñ"3°HÓ"=×"LÑ"LÐØ×*Ñ*Ø ¸×8IÑ8IÔ8K©fÐQUð +ô ð ×$Ñ$ XÓ.×4Ñ4¸Ò=Ð=Ð=àœ
×*Ñ*Ò*Ø×(Ñ(¨Ó2×AÑAÐEXÒXÐXÐXØ× Ñ Ô"Ø×(Ñ(¨Ó2×DÑDÒPÐPÐPà×(Ñ(¨Ó2×DÑDÒLÐLÐLð	Mô$ �]‰]œ8Ó$ñ 	Ü$'¨	£N°QÑ$6Ð!Ø×*Ñ*Ø%Ü ×)Ñ)ð +ô ÷	ô  ò 	ZˆEØ×$Ñ$Ô&ØØ×/Ñ/°Ó9ˆHØ×*Ñ*¨8¸5ÈÐ*ÔPÜ$ò Z�à-1�ØœZ×0Ñ0Ò0Ø!˜U�FÜ—]‘]Ô#;Ó<ñ ZØ×2Ñ2°8À6ÐRXÐ2ÔY÷Zð ZñZñ	Zùò7 M÷(	ð 	ú÷"Zð Zús   ªI,Æ0I1É
I=É1I:É=Jc                ó   — t        |«       y r   )Ú'_test_set_trial_state_values_for_floatsro   s     r   Ú&test_set_trial_state_values_for_floatsz6StorageTestCase.test_set_trial_state_values_for_floatsˆ  s
   € Ü/°Õ8r   c                óT  — t        |ddd¬«      \  }}|j                  «       D ]œ  \  }}|j                  «       D ]„  \  }}|j                  |«      |j                  k(  sJ ‚|j                  j	                  «       D ]B  }|j                  ||«      |j                  |   j                  |j                  |   «      k(  rŒBJ ‚ Œ† Œž t        d„ |j                  «       D «       «      dz   }t        j                  t        «      5  |j                  |«       d d d «       t        j                  t        «      5  |j                  |d«       d d d «       |dz
  }	t        j                  t        «      5  |j                  |	d«       d d d «       y # 1 sw Y   ŒwxY w# 1 sw Y   ŒOxY w# 1 sw Y   y xY w)Nr   é   r   ©Ún_studyÚn_trialÚseedc              3  óJ   K  — | ]  }|j                  «       D ]  }|–— Œ Œ y ­wr   ©Úkeys©r#   ÚtsÚtids      r   r%   zLStorageTestCase.test_get_trial_param_and_get_trial_params.<locals>.<genexpr>—  ó"   è ø€ ÒL˜À"Ç'Á'Ã)ÒL¸3”ÐL�ÑLùó   ‚!#ÚparamAz	dummy-key)Ú_setup_studiesri   Úget_trial_paramsr‡   rï   Úget_trial_paramrº   Úto_internal_reprrÝ   rÖ   r<   r=   rF   )
r,   r   rÞ   Ústudy_to_trialsÚtrial_id_to_trialr—   Úexpected_trialr]   rá   Úexistent_trial_ids
             r   Ú)test_get_trial_param_and_get_trial_paramsz9StorageTestCase.test_get_trial_param_and_get_trial_params‹  s±  € Ü+¨G¸QÈÐPQÔRÑˆˆ?à$3×$9Ñ$9Ó$;ò 	CÑ ˆAÐ Ø,=×,CÑ,CÓ,Eò CÑ(�˜.Ø×/Ñ/°Ó9¸^×=RÑ=RÒRÐRÐRØ)×0Ñ0×5Ñ5Ó7ò C�CØ"×2Ñ2°8¸SÓAÀ^×EaÑEaØñFç&Ñ& ~×'<Ñ'<¸SÑ'AÓBóCð Cð CñCñCð	Cô ÑL˜o×4Ñ4Ó6ÔLÓLÈqÑPð 	ô �]‰]œ8Ó$ñ 	<Ø×$Ñ$Ð%:Ô;÷	<ä�]‰]œ8Ó$ñ 	EØ×#Ñ#Ð$9¸8ÔD÷	Eà1°AÑ5ÐÜ�]‰]œ8Ó$ñ 	DØ×#Ñ#Ð$5°{ÔC÷	Dð 	D÷	<ð 	<ú÷	Eð 	Eú÷	Dð 	Dús$   Ã>FÄ1FÅ*FÆFÆFÆF'c           	     óú  — |j                  t        j                  g¬«      }|j                  |«      }|j                  |«      }|j                  |j                  t        j                  g¬«      «      }t	        dd¬«      }t        d¬«      }t        d¬«      }t	        ddd	¬
«      }	|j                  |dd|«       |j                  |dd|«       |j                  |d«      dk(  sJ ‚|j                  |d«      dk(  sJ ‚|j                  |«      j                  dddœk(  sJ ‚|j                  |«      dddœk(  sJ ‚|j                  |dd|«       |j                  |dd|	«       |j                  |d«      dk(  sJ ‚|j                  |d«      dk(  sJ ‚|j                  |«      j                  dddœk(  sJ ‚|j                  |«      dddœk(  sJ ‚t        j                  t        «      5  |j                  |dd|	«       d d d «       t        j                  t        «      5  |j                  |ddt        d¬«      «       d d d «       |j                  |t        j                   dg¬«       t        j                  t"        «      5  |j                  |dd|«       d d d «       t        j                  t$        «      5  |j                  |d«       d d d «       t        j                  t"        «      5  |j                  |dd|	«       d d d «       |j                  |dd|«       |j                  |d«      dk(  sJ ‚|j                  |«      j                  ddik(  sJ ‚|j                  |«      ddik(  sJ ‚t'        |||g«      dz   }
t        j                  t$        «      5  |j                  |
dd|«       d d d «       y # 1 sw Y   �ŒºxY w# 1 sw Y   �Œ‡xY w# 1 sw Y   �Œ;xY w# 1 sw Y   �ŒxY w# 1 sw Y   ŒêxY w# 1 sw Y   y xY w)Nr   ç      ð?ç       @©ÚlowÚhigh)ÚShibuyaÚEbisuÚMeguro©Úchoices)r  ÚShinseng      Y@T)r  r  Úlogr°   r±   Úyr   r  )r°   r  ç333333Ó?Úzçš™™™™™¹?)r°   r  )r  r  r  rÔ   rÕ   çš™™™™™Ù?r   r
  )r&   r   r'   rD   r	   r   Úset_trial_paramrø   rÚ   r‡   r÷   r<   r=   Ú
ValueErrorrÛ   r   rË   r
   rF   rÝ   )r,   r   r-   Ú
trial_id_1Ú
trial_id_2Ú
trial_id_3Údistribution_xÚdistribution_y_1Údistribution_y_2Údistribution_zrá   s              r   Útest_set_trial_paramz$StorageTestCase.test_set_trial_param¡  sõ  € à×+Ñ+¼×8OÑ8OÐ7PÐ+ÓQˆØ×-Ñ-¨hÓ7ˆ
Ø×-Ñ-¨hÓ7ˆ
Ø×-Ñ-Ø×$Ñ$´×1HÑ1HÐ0IÐ$ÓJó
ˆ
ô
 +¨s¸Ô=ˆÜ2Ð;YÔZÐÜ2Ð;QÔRÐÜ*¨s¸ÀDÔIˆð 	×Ñ 
¨C°°nÔEØ×Ñ 
¨C°Ð4DÔEØ×&Ñ& z°3Ó7¸3Ò>Ð>Ð>Ø×&Ñ& z°3Ó7¸1Ò<Ð<Ð<à× Ñ  Ó,×3Ñ3¸SÀxÑ7PÒPÐPÐPØ×'Ñ'¨
Ó3¸SÀxÑ7PÒPÐPÐPð 	×Ñ 
¨C°°nÔEØ×Ñ 
¨C°°nÔEØ×&Ñ& z°3Ó7¸3Ò>Ð>Ð>Ø×&Ñ& z°3Ó7¸3Ò>Ð>Ð>Ø× Ñ  Ó,×3Ñ3¸SÀsÑ7KÒKÐKÐKØ×'Ñ'¨
Ó3¸SÀsÑ7KÒKÐKÐKô �]‰]œ:Ó&ñ 	JØ×#Ñ# J°°S¸.ÔI÷	Jô �]‰]œ:Ó&ñ 	Ø×#Ñ#Ø˜C Ô$;ÐDbÔ$cô÷	ð
 	×&Ñ& z¼×9LÑ9LÐVYÐUZÐ&Ô[ä�]‰]Ô3Ó4ñ 	JØ×#Ñ# J°°QÐ8HÔI÷	Jô �]‰]œ8Ó$ñ 	5Ø×#Ñ# J°Ô4÷	5ô �]‰]Ô3Ó4ñ 	JØ×#Ñ# J°°S¸.ÔI÷	Jð 	×Ñ 
¨C°Ð4DÔEØ×&Ñ& z°3Ó7¸1Ò<Ð<Ð<Ø× Ñ  Ó,×3Ñ3¸¸YÐ7GÒGÐGÐGØ×'Ñ'¨
Ó3¸¸YÐ7GÒGÐGÐGô !$ Z°¸ZÐ$HÓ IÈAÑ MÐÜ�]‰]œ8Ó$ñ 	UØ×#Ñ#Ð$9¸3ÀÀ^ÔT÷	Uð 	U÷7	Jñ 	Jú÷	ñ 	ú÷	Jñ 	Jú÷	5ñ 	5ú÷	Jð 	Jú÷	Uð 	UúsH   Ç$N1ÈN>É=OÊ3OË'O%ÎO1Î1N;Î>OÏOÏO"Ï%O.Ï1O:c                ó   — t        |«       y r   )Ú(_test_set_and_get_trial_param_for_floatsro   s     r   Ú'test_set_and_get_trial_param_for_floatsz7StorageTestCase.test_set_and_get_trial_param_for_floatsà  s
   € Ü0°Õ9r   c                ó  — |j                  t        j                  g¬«      }|j                  |«      }|j                  |«      }|j                  |j                  t        j                  g¬«      «      }|j                  |«      }|j                  |«      }|j	                  |t
        j                  d¬«       |j	                  |t
        j                  t        d«      f¬«       |j	                  |t
        j                  d¬«       |j	                  |t
        j                  g d¢¬«       |j                  |«      j                  dk(  sJ ‚|j                  |«      j                  �J ‚|j                  |«      j                  t        d«      k(  sJ ‚|j                  |«      j                  g d¢k(  sJ ‚|j                  |«      j                  g d¢k(  sJ ‚t        |||||«      dz   }t        j                  t        «      5  |j	                  |t
        j                  d¬«       d d d «       t        j                  t         «      5  |j	                  |t
        j                  d¬«       d d d «       y # 1 sw Y   ŒNxY w# 1 sw Y   y xY w)	Nr   )r±   rÕ   Úinf)r  çš™™™™™É?r  r±   r   )r   )r&   r   r'   rD   rÛ   r   rË   ÚfloatrÙ   rÚ   r^   rÖ   rÝ   r<   r=   rF   r
   )	r,   r   r-   r  r  r  Ú
trial_id_4Ú
trial_id_5rá   s	            r   Ú&test_set_trial_state_values_for_valuesz6StorageTestCase.test_set_trial_state_values_for_valuesã  sV  € à×+Ñ+¼×8OÑ8OÐ7PÐ+ÓQˆØ×-Ñ-¨hÓ7ˆ
Ø×-Ñ-¨hÓ7ˆ
Ø×-Ñ-Ø×$Ñ$´×1HÑ1HÐ0IÐ$ÓJó
ˆ
ð ×-Ñ-¨hÓ7ˆ
Ø×-Ñ-¨hÓ7ˆ
ð 	×&Ñ& z¼×9LÑ9LÐU[Ð&Ô\Ø×&Ñ&Øœj×1Ñ1¼5À»<¸/ð 	'ô 	
ð 	×&Ñ&Øœj×0Ñ0¸ð 	'ô 	
ð 	×&Ñ&Øœj×0Ñ0ºð 	'ô 	
ð × Ñ  Ó,×2Ñ2°cÒ9Ð9Ð9Ø× Ñ  Ó,×2Ñ2Ð:Ð:Ð:Ø× Ñ  Ó,×2Ñ2´e¸E³lÒBÐBÐBØ× Ñ  Ó,×3Ñ3²ÒFÐFÐFØ× Ñ  Ó,×3Ñ3²ÒFÐFÐFä # J°
¸JÈ
ÐT^Ó _ÐbcÑ cÐÜ�]‰]œ8Ó$ñ 	Ø×*Ñ*Ø%¬Z×-@Ñ-@Èð +ô ÷	ô �]‰]Ô3Ó4ñ 	_Ø×*Ñ*¨:¼Z×=PÑ=PÐY]Ð*Ô^÷	_ð 	_÷	ð 	ú÷	_ð 	_ús   Ç?#I/É#I;É/I8É;Jc                óZ  — |j                  t        j                  g¬«      }|j                  |«      }|j                  |«      }|j                  |j                  t        j                  g¬«      «      }|j                  |«      }|j	                  |dd«       |j	                  |dd«       |j	                  |dd«       |j	                  |dd«       |j	                  |dd«       |j	                  |d	t        d
«      «       |j	                  |dt        d«      «       |j                  |«      j                  dddœk(  sJ ‚|j                  |«      j                  i k(  sJ ‚|j                  |«      j                  dddt        d
«      dœk(  sJ ‚t        j                  |j                  |«      j                  d   «      sJ ‚|j	                  |dd«       |j                  |«      j                  dddœk(  sJ ‚t        ||||«      dz   }t        j                  t        «      5  |j	                  |dd«       d d d «       |j                  |t        j                   dg¬«       t        j                  t"        «      5  |j	                  |dd«       d d d «       y # 1 sw Y   ŒbxY w# 1 sw Y   y xY w)Nr   r   r  r   r  r  r   r±   r¦   r  Únan)r   r   )r   r   r   r¦   r   rÔ   rÕ   )r&   r   r'   rD   Úset_trial_intermediate_valuer!  rÚ   r‹   ÚnpÚisnanrÝ   r<   r=   rF   rÛ   r   rË   r
   )r,   r   r-   r  r  r  r"  rá   s           r   Ú!test_set_trial_intermediate_valuez1StorageTestCase.test_set_trial_intermediate_value
  s�  € à×+Ñ+¼×8OÑ8OÐ7PÐ+ÓQˆØ×-Ñ-¨hÓ7ˆ
Ø×-Ñ-¨hÓ7ˆ
Ø×-Ñ-Ø×$Ñ$´×1HÑ1HÐ0IÐ$ÓJó
ˆ
ð ×-Ñ-¨hÓ7ˆ
ð 	×,Ñ,¨Z¸¸CÔ@Ø×,Ñ,¨Z¸¸CÔ@Ø×,Ñ,¨Z¸¸CÔ@Ø×,Ñ,¨Z¸¸CÔ@Ø×,Ñ,¨Z¸¸CÔ@Ø×,Ñ,¨Z¸¼EÀ%»LÔIØ×,Ñ,¨Z¸¼EÀ%»LÔIà× Ñ  Ó,×@Ñ@ÈÐPSÑDTÒTÐTÐTØ× Ñ  Ó,×@Ñ@ÀBÒFÐFÐFØ× Ñ  Ó,×@Ñ@ØØØÜ�U‹|ñ	E
ò 
ð 	
ð 
ô �x‰x˜×)Ñ)¨*Ó5×IÑIÈ!ÑLÔMÐMÐMð 	×,Ñ,¨Z¸¸CÔ@Ø× Ñ  Ó,×@Ñ@ÈÐPSÑDTÒTÐTÐTä # J°
¸JÈ
Ó SÐVWÑ WÐÜ�]‰]œ8Ó$ñ 	PØ×0Ñ0Ð1FÈÈ3ÔO÷	Pð 	×&Ñ& z¼×9LÑ9LÐVYÐUZÐ&Ô[ä�]‰]Ô3Ó4ñ 	EØ×0Ñ0°¸QÀÔD÷	Eð 	E÷	Pð 	Pú÷
	Eð 	Eús   È JÉ8J!ÊJÊ!J*c                ó   — t        |«       y r   )Ú-_test_set_trial_intermediate_value_for_floatsro   s     r   Ú,test_set_trial_intermediate_value_for_floatsz<StorageTestCase.test_set_trial_intermediate_value_for_floats4  rq   r   c                ó6  ‡— t        ‰ddd¬«      \  }}t        ˆfd„|j                  «       D «       «      sJ ‚t        d„ |j                  «       D «       «      dz   }t	        j
                  t        «      5  ‰j                  |«       d d d «       y # 1 sw Y   y xY w)Nr   rè   r¶   ré   c              3  óŠ   •K  — | ]:  }|j                  «       D ]%  \  }}‰j                  |«      |j                  k(  –— Œ' Œ< y ­wr   )ri   Úget_trial_user_attrsrŽ   ©r#   rG   r—   Útrialr   s       €r   r%   z<StorageTestCase.test_get_trial_user_attrs.<locals>.<genexpr>9  sN   øè ø€ ò 
àØ#)§<¡<£>ò
ñ  �˜%ð ×(Ñ(¨Ó2°e×6FÑ6FÕFð
ØFñ
ùó   ƒA Ac              3  óJ   K  — | ]  }|j                  «       D ]  }|–— Œ Œ y ­wr   rî   rð   s      r   r%   z<StorageTestCase.test_get_trial_user_attrs.<locals>.<genexpr>?  ó&   è ø€ Ò ]¨ÐSU×SZÑSZÓS\Ò ]ÈC¤Ð ] Ñ ]ùrô   r   )rö   r+   rÖ   rÝ   r<   r=   rF   r0  ©r,   r   rÞ   rú   Únon_existent_trials    `   r   Útest_get_trial_user_attrsz)StorageTestCase.test_get_trial_user_attrs7  s“   ø€ Ü+¨G¸QÈÐPRÔSÑˆˆ?Üó 
à)×0Ñ0Ó2ô
ô 
ð 	
ð 
ô !Ñ ]¨×/EÑ/EÓ/GÔ ]Ó]Ð`aÑaÐÜ�]‰]œ8Ó$ñ 	=Ø×(Ñ(Ð);Ô<÷	=÷ 	=ñ 	=úó   Á4BÂBc                ó�  ‡— ‰j                  ‰j                  t        j                  g¬«      «      }dˆfd„}t        j                  «       D ]  \  }} ||||«       Œ ‰j                  |«      j                  t        k(  sJ ‚ ||dd«       ‰j                  ‰j                  t        j                  g¬«      «      } ||dd«       t        ‰j                  |«      j                  «      d	k(  sJ ‚‰j                  |«      j                  d   dk(  sJ ‚t        ||h«      d	z   }t        j                  t        «      5  ‰j                  |dd«       d d d «       ‰j                  |t        j                   d
g¬«       t        j                  t"        «      5  ‰j                  |dd«       d d d «       y # 1 sw Y   ŒbxY w# 1 sw Y   y xY w)Nr   r]   r^   c                ór   •— ‰j                  | ||«       ‰j                  | «      j                  |   |k(  sJ ‚y r   )Úset_trial_user_attrrÚ   rŽ   ©r—   r]   r^   r   s      €r   rc   zCStorageTestCase.test_set_trial_user_attr.<locals>.check_set_and_getH  s9   ø€ Ø×'Ñ'¨°#°uÔ=Ø×$Ñ$ XÓ.×9Ñ9¸#Ñ>À%ÒGÐGÑGr   rd   re   Úbaseline_scorer§   r   rÔ   rÕ   ©r—   r©   r]   rg   r^   r   rW   rX   )rD   r&   r   r'   rh   ri   rÚ   rŽ   r)   rÝ   r<   r=   rF   r<  rÛ   r   rË   r
   )r,   r   r  rc   r]   r^   r  rá   s    `      r   Útest_set_trial_user_attrz(StorageTestCase.test_set_trial_user_attrC  s¼  ø€ Ø×-Ñ-Ø×$Ñ$´×1HÑ1HÐ0IÐ$ÓJó
ˆ
õ	Hô
 (×-Ñ-Ó/ò 	6‰JˆC�Ù˜j¨#¨uÕ5ð	6à× Ñ  Ó,×7Ñ7¼=ÒHÐHÐHñ 	˜* i°Ô<ð ×-Ñ-Ø×$Ñ$´×1HÑ1HÐ0IÐ$ÓJó
ˆ
ñ 	˜*Ð&6¸Ô>Ü�7×$Ñ$ ZÓ0×;Ñ;Ó<ÀÒAÐAÐAØ× Ñ  Ó,×7Ñ7Ð8HÑIÈUÒRÐRÐRô !$ Z°Ð$<Ó =ÀÑ AÐÜ�]‰]œ8Ó$ñ 	OØ×'Ñ'Ð(=¸uÀgÔN÷	Oð 	×&Ñ& z¼×9LÑ9LÐVYÐUZÐ&Ô[Ü�]‰]Ô3Ó4ñ 	DØ×'Ñ'¨
°E¸7ÔC÷	Dð 	D÷	Oð 	Oú÷
	Dð 	Dús   Ä;F0ÆF<Æ0F9Æ<Gc                ó   — t        |«       y r   )Ú,_test_set_and_get_trial_user_attr_for_floatsro   s     r   Ú+test_set_and_get_trial_user_attr_for_floatsz;StorageTestCase.test_set_and_get_trial_user_attr_for_floatsf  s
   € Ü4°WÕ=r   c                ó6  ‡— t        ‰ddd¬«      \  }}t        ˆfd„|j                  «       D «       «      sJ ‚t        d„ |j                  «       D «       «      dz   }t	        j
                  t        «      5  ‰j                  |«       d d d «       y # 1 sw Y   y xY w)Nr   rè   r¶   ré   c              3  óŠ   •K  — | ]:  }|j                  «       D ]%  \  }}‰j                  |«      |j                  k(  –— Œ' Œ< y ­wr   )ri   Úget_trial_system_attrsr‘   r1  s       €r   r%   z>StorageTestCase.test_get_trial_system_attrs.<locals>.<genexpr>k  sN   øè ø€ ò 
àØ#)§<¡<£>ò
ñ  �˜%ð ×*Ñ*¨8Ó4¸×8JÑ8JÕJð
ØJñ
ùr3  c              3  óJ   K  — | ]  }|j                  «       D ]  }|–— Œ Œ y ­wr   rî   rð   s      r   r%   z>StorageTestCase.test_get_trial_system_attrs.<locals>.<genexpr>q  r5  rô   r   )rö   r+   rÖ   rÝ   r<   r=   rF   rF  r6  s    `   r   Útest_get_trial_system_attrsz+StorageTestCase.test_get_trial_system_attrsi  s“   ø€ Ü+¨G¸QÈÐPRÔSÑˆˆ?Üó 
à)×0Ñ0Ó2ô
ô 
ð 	
ð 
ô !Ñ ]¨×/EÑ/EÓ/GÔ ]Ó]Ð`aÑaÐÜ�]‰]œ8Ó$ñ 	?Ø×*Ñ*Ð+=Ô>÷	?÷ 	?ñ 	?úr9  c                ó
  ‡— ‰j                  t        j                  g¬«      }‰j                  |«      }dˆfd„}t        j                  «       D ]  \  }} ||||«       Œ ‰j                  |«      j                  }|t        k(  sJ ‚ ||dd«       ‰j                  |«      } ||dd«       ‰j                  |«      j                  }|ddik(  sJ ‚t        ||h«      d	z   }	t        j                  t        «      5  ‰j                  |	dd«       d d d «       ‰j                  |t        j                  d
g¬«       t        j                  t         «      5  ‰j                  |dd«       d d d «       y # 1 sw Y   ŒbxY w# 1 sw Y   y xY w)Nr   r]   r^   c                ó^   •— ‰j                  | ||«       ‰j                  | «      |   |k(  sJ ‚y r   )Úset_trial_system_attrrF  r=  s      €r   rc   zEStorageTestCase.test_set_trial_system_attr.<locals>.check_set_and_gety  rv   r   rd   re   r>  r§   r   rÔ   rÕ   r?  )r&   r   r'   rD   rh   ri   rÚ   r‘   rÝ   r<   r=   rF   rK  rÛ   r   rË   r
   )
r,   r   r-   r  rc   r]   r^   r‘   r  rá   s
    `        r   Útest_set_trial_system_attrz*StorageTestCase.test_set_trial_system_attru  sˆ  ø€ Ø×+Ñ+¼×8OÑ8OÐ7PÐ+ÓQˆØ×-Ñ-¨hÓ7ˆ
õ	Jô
 (×-Ñ-Ó/ò 	6‰JˆC�Ù˜j¨#¨uÕ5ð	6à×(Ñ(¨Ó4×AÑAˆØœ}Ò,Ð,Ð,ñ 	˜* i°Ô<ð ×-Ñ-¨hÓ7ˆ
Ù˜*Ð&6¸Ô>Ø×(Ñ(¨Ó4×AÑAˆØÐ 0°%Ð8Ò8Ð8Ð8ô !$ Z°Ð$<Ó =ÀÑ AÐÜ�]‰]œ8Ó$ñ 	QØ×)Ñ)Ð*?ÀÈÔP÷	Qð 	×&Ñ& z¼×9LÑ9LÐVYÐUZÐ&Ô[Ü�]‰]Ô3Ó4ñ 	FØ×)Ñ)¨*°e¸WÔE÷	Fð 	F÷	Qð 	Qú÷
	Fð 	Fús   Ã8E-ÅE9Å-E6Å9Fc                ó   — t        |«       y r   )Ú._test_set_and_get_trial_system_attr_for_floatsro   s     r   Ú-test_set_and_get_trial_system_attr_for_floatsz=StorageTestCase.test_set_and_get_trial_system_attr_for_floats–  s
   € Ü6°wÕ?r   c                óÖ  — t        |ddd¬«      \  }}|j                  «       }t        |«      t        |«      k(  sJ ‚|j                  «       D ]›  \  }}d }|D ]  }|j                  |j                  k(  sŒ|} n |€J ‚|j
                  |j
                  k(  sJ ‚|j                  |j                  k(  sJ ‚|j                  |j                  k(  sJ ‚|j                  |j                  k(  rŒ›J ‚ y )Nr¶   é.   ré   )rö   r(   r)   ri   r!   Ú	directionrŽ   r‘   )r,   r   Úexpected_frozen_studiesrÞ   r.   Úexpected_frozen_studyÚfrozen_studyr$   s           r   Útest_get_all_studiesz$StorageTestCase.test_get_all_studies™  s
  € Ü%3°GÀRÐQSÐZ\Ô%]Ñ"Ð Ø ×0Ñ0Ó2ˆÜ�>Ó"¤cÐ*AÓ&BÒBÐBÐBØ(?×(EÑ(EÓ(Gò 
	SÑ$ˆAÐ$Ø/3ˆLØ#ò �Ø—<‘<Ð#8×#CÑ#CÓCØ#$�LÙðð  Ð+Ð+Ð+Ø×)Ñ)Ð-B×-LÑ-LÒLÐLÐLØ×*Ñ*Ð.C×.NÑ.NÒNÐNÐNØ×*Ñ*Ð.C×.NÑ.NÒNÐNÐNØ×,Ñ,Ð0E×0RÑ0RÓRÐRÐRñ
	Sr   c                óŠ  — t        |ddd¬«      \  }}|j                  «       D ]<  \  }}|j                  «       D ]$  }|j                  |j                  «      }||k(  rŒ$J ‚ Œ> t        d„ |j                  «       D «       «      dz   }t        j                  t        «      5  |j                  |«       d d d «       y # 1 sw Y   y xY w)Nr   é   é/   ré   c              3  óJ   K  — | ]  }|j                  «       D ]  }|–— Œ Œ y ­wr   rî   rð   s      r   r%   z1StorageTestCase.test_get_trial.<locals>.<genexpr>²  ró   rô   r   )	rö   ri   rÖ   rÚ   r”   rÝ   r<   r=   rF   )r,   r   rÞ   rú   Úexpected_trialsrü   r2  rá   s           r   Útest_get_trialzStorageTestCase.test_get_trial©  sÆ   € Ü+¨G¸QÈÐQSÔTÑˆˆ?à"1×"7Ñ"7Ó"9ò 	/ÑˆAˆØ"1×"8Ñ"8Ó":ò /�Ø×)Ñ)¨.×*BÑ*BÓC�Ø Ó.Ð.Ð.ñ/ð	/ô ÑL˜o×4Ñ4Ó6ÔLÓLÈqÑPð 	ô �]‰]œ8Ó$ñ 	5Ø×ÑÐ3Ô4÷	5÷ 	5ñ 	5ús   ÂB9Â9Cc                ój  — t        |ddd¬«      \  }}|j                  «       D ]3  \  }}|j                  |«      }|D ]  }||j                     }||k(  rŒJ ‚ Œ5 t	        |j                  «       «      dz   }	t        j                  t        «      5  |j                  |	«       d d d «       y # 1 sw Y   y xY w)Nr   rX  é0   ré   r   )	rö   ri   rE   r”   rÝ   rï   r<   r=   rF   )
r,   r   rÞ   rú   r-   r[  rG   r2  rü   rZ   s
             r   Útest_get_all_trialsz#StorageTestCase.test_get_all_trials·  s»   € Ü+¨G¸QÈÐQSÔTÑˆˆ?à)8×)>Ñ)>Ó)@ò 	/Ñ%ˆH�oØ×+Ñ+¨HÓ5ˆFØò /�Ø!0°·±Ñ!A�Ø Ó.Ð.Ð.ñ/ð	/ô !$ O×$8Ñ$8Ó$:Ó ;¸aÑ ?ÐÜ�]‰]œ8Ó$ñ 	:Ø×"Ñ"Ð#8Ô9÷	:÷ 	:ñ 	:ús   ÂB)Â)B2Úparam_namesÚaÚbc           
     ó†  — |j                  t        j                  g¬«      }|j                  |t        j
                  j                  t        j                  ¬«      «      }|D ]!  }|j                  ||dt        dd«      ¬«       Œ# |j                  |«      }t        |d   j                  j                  «       «      |k(  sJ ‚y )Nr   r½   r   rÔ   r  )Údistributionr   )r&   r   r'   rD   r>   r2  Úcreate_trialr   r�   r  r	   rE   rS   r‡   rï   )r,   r   r`  r-   r—   Ú
param_namerG   s          r   Ú test_get_all_trials_params_orderz0StorageTestCase.test_get_all_trials_params_orderÄ  sº   € ð ×+Ñ+¼×8OÑ8OÐ7PÐ+ÓQˆØ×+Ñ+Ø”f—l‘l×/Ñ/´j×6HÑ6HÐ/ÓIó
ˆð &ò 	ˆJØ×#Ñ#Ø˜* cÔ8IÈ#ÈsÓ8Sð $õ ð	ð
 ×'Ñ'¨Ó1ˆÜ�F˜1‘I×$Ñ$×)Ñ)Ó+Ó,°Ò;Ð;Ñ;r   c                ó  — t        |ddd¬«      \  }}|D ]r  }|j                  |d¬«      }t        |«      t        ||   «      k(  sJ ‚t        j                  |«      }d|d   j
                  d	<   |j                  |d
¬«      }||k(  rŒrJ ‚ y )Nr   rè   é1   ré   T©Údeepcopyr  r   r°   F)rö   rE   r)   Úcopyrk  r‡   )r,   r   r.   rú   r-   Útrials0Útrials0_originalÚtrials1s           r   Ú#test_get_all_trials_deepcopy_optionz3StorageTestCase.test_get_all_trials_deepcopy_optionÔ  s¡   € Ü*8¸È!ÐUVÐ]_Ô*`Ñ'ˆ˜à&ò 		/ˆHØ×,Ñ,¨XÀÐ,ÓEˆGÜ�w“<¤3 °xÑ'@Ó#AÒAÐAÐAô  $Ÿ}™}¨WÓ5ÐØ%(ˆG�A‰J×Ñ˜cÑ"à×,Ñ,¨XÀÐ,ÓFˆGØ# wÓ.Ð.Ð.ñ		/r   c                ó  — |j                  t        j                  g¬«      }t        j                  d«      }t
        j                  t
        j                  t
        j                  f}|D ]N  }t        |«      }||_	        |j                  t
        j                  k(  rdgnd |_
        |j                  ||¬«       ŒP |j                  |d ¬«      }t        |«      dk(  sJ ‚|j                  |t
        j                  f¬«      }t        |«      dk(  sJ ‚t        d„ |D «       «      sJ ‚|j                  |t
        j                  t
        j                  f¬«      }t        |«      dk(  sJ ‚t        d	„ |D «       «      sJ ‚|j                  |d
¬«      }t        |«      dk(  sJ ‚t        D �cg c],  }|t
        j                  k7  sŒ|t
        j                  k7  sŒ+|‘Œ. }	}|	D ]&  }|j                  ||f¬«      }t        |«      dk(  rŒ&J ‚ y c c}w )Nr   é3   rÔ   rÊ   )Ústatesr¦   r   c              3  óV   K  — | ]!  }|j                   t        j                  k(  –— Œ# y ­wr   )r€   r   rË   r‚   s     r   r%   zCStorageTestCase.test_get_all_trials_state_option.<locals>.<genexpr>÷  s   è ø€ ÒB°a�1—7‘7œj×1Ñ1Õ1ÑBùr„   c              3  ót   K  — | ]0  }|j                   t        j                  t        j                  fv –— Œ2 y ­wr   )r€   r   rË   ÚPRUNEDr‚   s     r   r%   zCStorageTestCase.test_get_all_trials_state_option.<locals>.<genexpr>û  s*   è ø€ ÒWÈ1�1—7‘7œz×2Ñ2´J×4EÑ4EÐFÔFÑWùs   ‚68r   r   )r&   r   rY   ÚrandomÚRandomr   rË   rv  Ú_generate_trialr€   rÖ   rD   rE   r)   r+   r×   )
r,   r   r-   Ú	generatorrs  r€   rƒ   rG   r$   Úother_statess
             r   Ú test_get_all_trials_state_optionz0StorageTestCase.test_get_all_trials_state_optionâ  sù  € Ø×+Ñ+¼×8OÑ8OÐ7PÐ+ÓQˆÜ—M‘M "Ó%ˆ	ô ×ÑÜ×ÑÜ×Ñð
ˆð ò 	AˆEÜ 	Ó*ˆAØˆAŒGØ !§¡¬:×+>Ñ+>Ò >˜‘uÀDˆAŒHØ×$Ñ$ X¸aÐ$Õ@ð		Að ×'Ñ'¨¸Ð'Ó>ˆÜ�6‹{˜aÒÐÐà×'Ñ'¨¼*×:MÑ:MÐ9OÐ'ÓPˆÜ�6‹{˜aÒÐÐÜÑB¸6ÔBÔBÐBÐBà×'Ñ'¨¼*×:MÑ:MÌz×O`ÑO`Ð9aÐ'ÓbˆÜ�6‹{˜aÒÐÐÜÑWÐPVÔWÔWÐWÐWà×'Ñ'¨¸Ð'Ó<ˆÜ�6‹{˜aÒÐÐô "ö
Ø Q¬*×*=Ñ*=Ó%=À!Äz×GXÑGXÓBXŠAð
ˆð 
ð "ò 	$ˆEØ×+Ñ+¨H¸e¸XÐ+ÓFˆFÜ�v“; !Ó#Ð#Ð#ñ	$ùò
s   Æ"HÆ;HÇHc                ó  — t        |ddd¬«      \  }}|j                  «       D �]å  }|j                  |d¬«      }t        j                  |«      }|D �]­  }|j
                  j                  «       sb|j                  |j                  ddt        dd	«      «       |j                  |j                  d
d«       |j                  |j                  dd«       |j
                  t        j                  k(  rÈ|j                  dz  dk(  r-|j                  |j                  t        j                   dg«       ŒÜ|j                  dz  d	k(  rK|j#                  |j                  dd«       |j                  |j                  t        j$                  dg«       �Œ9|j                  |j                  t        j&                  «       �Œe|j
                  t        j(                  k(  s�Œ„|j                  |j                  t        j                  «       �Œ° ||k(  r�ŒæJ ‚ y )Nr   rX  r^  ré   Frj  ÚparamXr   r   Ú	usr_attrXÚ	sys_attrXr¦   )rö   rï   rE   rl  rk  r€   rÜ   r  r”   r	   r<  rK  r   r�   r¡   rÛ   rË   r'  rv  ÚFAILrÙ   )r,   r   rÞ   rú   r-   rG   Údeepcopied_trialsr2  s           r   Ú test_get_all_trials_not_modifiedz0StorageTestCase.test_get_all_trials_not_modified  s¤  € Ü+¨G¸QÈÐQSÔTÑˆˆ?à'×,Ñ,Ó.ó 	/ˆHØ×+Ñ+¨H¸uÐ+ÓEˆFÜ $§¡¨fÓ 5Ðàó X�Ø—{‘{×.Ñ.Ô0Ø×+Ñ+¨E¯O©O¸XÀqÔJ[Ð\]Ð_`ÓJaÔbØ×/Ñ/°·±ÀÈaÔPØ×1Ñ1°%·/±/À;ÐPQÔRà—;‘;¤*×"4Ñ"4Ò4Ø—|‘| aÑ'¨1Ò,Ø×6Ñ6°u·±Ì
×H[ÑH[Ð^_Ð]`ÕaØŸ™¨Ñ)¨QÒ.Ø×<Ñ<¸U¿_¹_ÈaÐQRÔSØ×6Ñ6°u·±Ì
×HYÑHYÐ\]Ð[^Ö_à×6Ñ6°u·±Ì
ÏÉÖXØ—[‘[¤J×$6Ñ$6Ô6Ø×2Ñ2°5·?±?ÄJ×DVÑDVÖWðXð" Ð.Ô.Ð.Ð.ñ+	/r   c                ó  — t        |ddd¬«      \  }}|D ]  }|j                  |«      dk(  rŒJ ‚ t        |j                  «       «      dz   }t	        j
                  t        «      5  |j                  |«      sJ ‚	 d d d «       y # 1 sw Y   y xY w)Nr   é   é2   ré   r   )rö   Úget_n_trialsrÝ   rï   r<   r=   rF   )r,   r   Ústudy_id_to_frozen_studiesrÞ   r-   rZ   s         r   Útest_get_n_trialsz!StorageTestCase.test_get_n_trials!  s”   € Ü(6°wÈÐSTÐ[]Ô(^Ñ%Ð" AØ2ò 	7ˆHØ×'Ñ'¨Ó1°QÓ6Ð6Ð6ð	7ô !$Ð$>×$CÑ$CÓ$EÓ FÈÑ JÐÜ�]‰]œ8Ó$ñ 	?Ø×'Ñ'Ð(=Ô>Ð>Ñ>÷	?÷ 	?ñ 	?ús   Á%BÂBc                óÄ  — |j                  t        j                  f¬«      }t        j                  d«      }t
        j                  t
        j                  t
        j                  g}|D ]N  }t        |«      }||_	        |j                  t
        j                  k(  rdgnd |_
        |j                  ||¬«       ŒP |j                  |t
        j                  «      dk(  sJ ‚|j                  |t
        j                  «      dk(  sJ ‚t        D �cg c],  }|t
        j                  k7  sŒ|t
        j                  k7  sŒ+|‘Œ. }}|D ]  }|j                  ||«      dk(  rŒJ ‚ y c c}w )Nr   rr  rÔ   rÊ   r   r   r   )r&   r   rY   rw  rx  r   rË   rv  ry  r€   rÖ   rD   r‡  r×   )r,   r   r-   rz  rs  r$   rƒ   r{  s           r   Útest_get_n_trials_state_optionz.StorageTestCase.test_get_n_trials_state_option*  sN  € Ø×+Ñ+¼×8OÑ8OÐ7QÐ+ÓRˆÜ—M‘M "Ó%ˆ	ô ×ÑÜ×ÑÜ×Ñð
ˆð ò 	AˆAÜ 	Ó*ˆAØˆAŒGØ !§¡¬:×+>Ñ+>Ò >˜‘uÀDˆAŒHØ×$Ñ$ X¸aÐ$Õ@ð		Að ×#Ñ# H¬j×.AÑ.AÓBÀaÒGÐGÐGØ×#Ñ# H¬j×.?Ñ.?Ó@ÀAÒEÐEÐEô "ö
Ø Q¬*×*=Ñ*=Ó%=À!Äz×GXÑGXÓBXŠAð
ˆð 
ð ò 	:ˆAØ×'Ñ'¨°!Ó4¸Ó9Ð9Ð9ñ	:ùò
s   ÄEÄ$EÄ8ErR  rÖ   )rÔ   r   r  rÔ   r  r   z-infc                ój  — |j                  |g¬«      }t        j                  t        «      5  |j	                  |«       d d d «       t        j                  t
        «      5  |j	                  |dz   «       d d d «       t        j                  d«      }|D ]<  }t        |«      }t        j                  |_        ||_        |j                  ||¬«       Œ> |t        j                  k(  rt!        |«      n
t#        |«      }|j	                  |«      j                  |k(  sJ ‚y # 1 sw Y   ŒßxY w# 1 sw Y   ŒµxY w)Nr   r   rr  rÊ   )r&   r<   r=   r  Úget_best_trialrF   rw  rx  ry  r   rË   r€   r^   rD   r   rY   rÝ   Úmin)	r,   r   rR  rÖ   r-   rz  Úvr¿   Úexpected_values	            r   Útest_get_best_trialz#StorageTestCase.test_get_best_trialC  s  € ð ×+Ñ+¸	°{Ð+ÓCˆÜ�]‰]œ:Ó&ñ 	-Ø×"Ñ" 8Ô,÷	-ô �]‰]œ8Ó$ñ 	1Ø×"Ñ" 8¨a¡<Ô0÷	1ô —M‘M "Ó%ˆ	àò 	NˆAÜ,¨YÓ7ˆNÜ#-×#6Ñ#6ˆNÔ Ø#$ˆNÔ Ø×$Ñ$ X¸nÐ$ÕMð		Nð
 )2´^×5LÑ5LÒ(Lœ˜VœÔRUÐV\ÓR]ˆØ×%Ñ% hÓ/×5Ñ5¸ÒGÐGÑG÷	-ð 	-ú÷	1ð 	1ús   ­DÁ D)ÄD&Ä)D2c                óª  — |j                  t        j                  t        j                  f¬«      }t	        j
                  d«      }t        d«      D ]A  }t        |«      }t        j                  |_
        ||dz   g|_        |j                  ||¬«       ŒC t        j                  t        «      5  |j!                  |«       d d d «       y # 1 sw Y   y xY w)Nr   rr  r¦   r   rÊ   )r&   r   rY   r'   rw  rx  r£   ry  r   rË   r€   rÖ   rD   r<   r=   ÚRuntimeErrorr�  )r,   r   r-   rz  r¬   r¿   s         r   Ú4test_get_best_trial_for_multi_objective_optimizationzDStorageTestCase.test_get_best_trial_for_multi_objective_optimizationc  s¼   € Ø×+Ñ+Ü&×/Ñ/´×1HÑ1HÐIð ,ó 
ˆô —M‘M "Ó%ˆ	Ü�q“ò 	NˆAÜ,¨YÓ7ˆNÜ#-×#6Ñ#6ˆNÔ Ø%&¨¨A© JˆNÔ!Ø×$Ñ$ X¸nÐ$ÕMð		Nô �]‰]œ<Ó(ñ 	-Ø×"Ñ" 8Ô,÷	-÷ 	-ñ 	-ús   Â.C	Ã	Cc                ó0  — t        j                  t        «      5  |j                  dd¬«       d d d «       |j	                  t
        j                  g¬«      }t        j                  t        «      5  |j                  |d¬«       d d d «       |j                  |«      }||j                  |d¬«      k(  sJ ‚|j	                  t
        j                  g¬«      }|j                  |«      }||j                  |d¬«      k(  sJ ‚y # 1 sw Y   ŒÓxY w# 1 sw Y   Œ‰xY w)Nr   )r-   Útrial_numberr   )r–  )r<   r=   rF   Ú'get_trial_id_from_study_id_trial_numberr&   r   r'   rD   rÑ   s       r   Ú,test_get_trial_id_from_study_id_trial_numberz<StorageTestCase.test_get_trial_id_from_study_id_trial_numberr  s&  € Ü�]‰]œ8Ó$ñ 	XØ×;Ñ;ÀQÐUVÐ;ÔW÷	Xð ×+Ñ+¼×8OÑ8OÐ7PÐ+ÓQˆä�]‰]œ8Ó$ñ 	VØ×;Ñ;¸HÐSTÐ;ÔU÷	Vð ×+Ñ+¨HÓ5ˆà˜7×JÑJØ 1ð Kó 
ò 
ð 	
ð 
ð ×+Ñ+¼×8OÑ8OÐ7PÐ+ÓQˆà×+Ñ+¨HÓ5ˆà˜7×JÑJØ 1ð Kó 
ò 
ð 	
ñ 
÷)	Xð 	Xú÷
	Vð 	Vús   šD Á0DÄ D	ÄDc                ó$  — |j                  t        j                  g¬«      }|j                  |dd«       t	        j
                  t	        j                  |«      «      }|j                  |«      }|j                  |«      }||cxk(  rddik(  sJ ‚ J ‚y )Nr   r]   Úpickle)r&   r   r'   rt   rš  ÚloadsÚdumpsru   )r,   r   r-   Úrestored_storageÚstorage_system_attrsÚrestored_storage_system_attrss         r   Útest_pickle_storagez#StorageTestCase.test_pickle_storage‹  s‹   € Ø×+Ñ+¼×8OÑ8OÐ7PÐ+ÓQˆØ×%Ñ% h°°xÔ@ä!Ÿ<™<¬¯©°WÓ(=Ó>Ðà&×=Ñ=¸hÓGÐØ(8×(OÑ(OÐPXÓ(YÐ%Ø#Ð'DÔYÈÐPXÐHYÒYÐYÑYÐYÑYr   c                ó¼  — |j                  t        j                  g¬«      }|j                  |«      }|j	                  |t
        j                  «       |j	                  |t
        j                  «       t        j                  t        «      5  |j	                  |t
        j                  «       d d d «       t        j                  t        «      5  |j	                  |t
        j                  «       d d d «       t        j                  t        «      5  |j	                  |t
        j                  «       d d d «       y # 1 sw Y   ŒŽxY w# 1 sw Y   ŒXxY w# 1 sw Y   y xY wr2   )r&   r   r'   rD   Úcheck_trial_is_updatabler   r�   rÙ   r<   r=   r
   r�  rv  rË   rÑ   s       r   Útest_check_trial_is_updatablez-StorageTestCase.test_check_trial_is_updatable•  s  € Ø×+Ñ+¼×8OÑ8OÐ7PÐ+ÓQˆØ×+Ñ+¨HÓ5ˆà×(Ñ(¨´:×3EÑ3EÔFØ×(Ñ(¨´:×3EÑ3EÔFä�]‰]Ô3Ó4ñ 	HØ×,Ñ,¨X´z·±ÔG÷	Hô �]‰]Ô3Ó4ñ 	JØ×,Ñ,¨X´z×7HÑ7HÔI÷	Jô �]‰]Ô3Ó4ñ 	LØ×,Ñ,¨X´z×7JÑ7JÔK÷	Lð 	L÷	Hð 	Hú÷	Jð 	Jú÷	Lð 	Lús$   Â!D:Ã!EÄ!EÄ:EÅEÅEN©r   r   rW   rX   )r   r   rÌ   r   rÍ   r   rW   rX   )r   r   r`  z	list[str]rW   rX   )r   r   rR  r   rÖ   zlist[float]rW   rX   )7r:   Ú
__module__Ú__qualname__r0   r6   r9   rH   rL   rN   r[   rk   rp   rw   rz   r|   r­   r<   ÚmarkÚparametrizer   rª   rÎ   rÒ   rã   ræ   rþ   r  r  r$  r*  r-  r8  r@  rC  rH  rL  rO  rV  r\  r_  rg  rp  r|  rƒ  r‰  r‹  r   rY   r'   r!  r‘  r”  r˜  r   r£  r   r   r   r   r   %   st  „ ó_ó Nóbó+ó(
$ó9ó&Dó6Oó0?óQó0Aó>ó8Jðt ‡[�[×ÑØ"Ø
ˆ(�,‰,‹.˜,˜(Ÿ,™,›.Ð	)©H°T¸1¸aÓ,@Á(È4ÐQRÐTUÓBVÐ+WÐXóð4JØ"ð4JØ08ð4JØIQð4Jà	ò4Jó	ð4Jól;ó(ZóT9óDó,=Uó~:ó%_óN(EóT?ó
=ó!DóF>ó
?óFóB@óSó 5ó:ð ‡[�[×Ñ˜]¨c°3¨Z¸#¸s¸Ð,DÓEð<Ø"ð<Ø1:ð<à	ò<ó Fð<ó/ó#$óJ/ó4?ó:ð2 ‡[�[×Ñ˜[¨>×+BÑ+BÀN×D[ÑD[Ð*\Ó]Ø‡[�[×ÑØâØ‘%˜“, Ð$Ø‘%˜“- Ð%Ù�5‹\˜3 ¡U¨6£]Ð3Ù�5‹\ˆNÙ�6‹]ˆOð	
ó
ðHØ"ðHØ/=ðHØGRðHà	òHó
ó ^ðHó(-ó
ó2ZôLr   r   ÚMNISTr§   ÚimageÚclassification©r>  Útags)rd   ÚnoneÚjson_serializablezdict[str, JSONSerializable]rh   r  r&  )	ÚzeroÚpirÝ   znegative maxrŽ  znegative minr  znegative infr&  c                óŒ   — t        j                  | «      rt        j                  |«      S t        j                  |«      ry| |k(  S )NF)Úmathr)  )ra  rb  s     r   Úis_equal_floatsr´  »  s2   € Ü‡z�z�!„}Ü�z‰z˜!‹}ÐÜ‡z�z�!„}ØØ�‰6€Mr   c                óä   — | j                  t        j                  g¬«      }t        j	                  «       D ]8  \  }}| j                  |||«       t        | j                  |«      |   |«      rŒ8J ‚ y r2   )r&   r   r'   ÚFLOAT_ATTRSri   r`   r´  ra   ©r   r-   r]   r^   s       r   rn   rn   Ã  sp   € Ø×'Ñ'´N×4KÑ4KÐ3LÐ'ÓM€Hô "×'Ñ'Ó)ò S‰
ˆˆUØ×#Ñ# H¨c°5Ô9Ü˜w×;Ñ;¸HÓEÀcÑJÈEÕRÐRÐRñSr   c                óä   — | j                  t        j                  g¬«      }t        j	                  «       D ]8  \  }}| j                  |||«       t        | j                  |«      |   |«      rŒ8J ‚ y r2   )r&   r   r'   r¶  ri   rt   r´  ru   r·  s       r   ry   ry   Ì  sp   € Ø×'Ñ'´N×4KÑ4KÐ3LÐ'ÓM€Hô "×'Ñ'Ó)ò U‰
ˆˆUØ×%Ñ% h°°UÔ;Ü˜w×=Ñ=¸hÓGÈÑLÈeÕTÐTÐTñUr   c                óf  — | j                  t        j                  g¬«      }t        j	                  «       D ]y  }t        j                  |«      rŒ| j                  |«      }| j                  |t        j                  |f¬«       | j                  |«      j                  }|€J ‚t        ||«      rŒyJ ‚ y )Nr   rÕ   )r&   r   r'   r¶  rÖ   r³  r)  rD   rÛ   r   rË   rÚ   r^   r´  )r   r-   r^   r—   Ú	set_values        r   rå   rå   Õ  s¦   € Ø×'Ñ'´N×4KÑ4KÐ3LÐ'ÓM€HÜ×#Ñ#Ó%ò 1ˆÜ�:‰:�eÔØØ×+Ñ+¨HÓ5ˆØ×&Ñ& x´z×7JÑ7JÐTYÐS[Ð&Ô\Ø×%Ñ% hÓ/×5Ñ5ˆ	ØÐ$Ð$Ð$Ü˜y¨%Õ0Ð0Ð0ñ1r   c                ó0  — | j                  t        j                  g¬«      }| j                  |«      }t        j                  «       D ]Í  \  }}t        ||¬«      }t        |f¬«      }||fD ]§  }t        |t        «      rt        j                  |«      sŒ)|j                  j                  |z   }|j                  |«      }	| j                  |||	|«       t        | j!                  ||«      |	«      sJ ‚| j#                  |«      j$                  |   |k(  rŒ§J ‚ ŒÏ y )Nr   r  r  )r&   r   r'   rD   r¶  ri   r	   r   Ú
isinstancer³  ÚisfiniteÚ	__class__r:   rù   r  r´  rø   rÚ   rº   )
r   r-   r—   r]   r^   Úfloat_distributionÚcategorical_distributionrd  rf  Úinternal_reprs
             r   r  r  á  s  € Ø×'Ñ'´N×4KÑ4KÐ3LÐ'ÓM€HØ×'Ñ'¨Ó1€Hä!×'Ñ'Ó)ò Y‰
ˆˆUÜ.°5¸uÔEÐÜ#:ÀEÀ8Ô#LÐ Ø/Ð1IÐJò 	YˆLä˜,Ô(9Ô:Ä4Ç=Á=ÐQVÔCWØØ%×/Ñ/×8Ñ8¸3Ñ>ˆJØ(×9Ñ9¸%Ó@ˆMØ×#Ñ# H¨j¸-ÈÔVÜ" 7×#:Ñ#:¸8ÀZÓ#PÐR_Ô`Ð`Ð`Ø×$Ñ$ XÓ.×<Ñ<¸ZÑHÈLÓXÐXÐXñ	YñYr   c                ó,  — | j                  t        j                  g¬«      }| j                  |«      }t	        t
        j                  «       «      D ]B  \  }}| j                  |||«       t        | j                  |«      j                  |   |«      rŒBJ ‚ y r2   )r&   r   r'   rD   Ú	enumerater¶  rÖ   r'  r´  rÚ   r‹   )r   r-   r—   r¬   r^   s        r   r,  r,  ó  sŠ   € Ø×'Ñ'´N×4KÑ4KÐ3LÐ'ÓM€HØ×'Ñ'¨Ó1€HÜœk×0Ñ0Ó2Ó3ò Z‰ˆˆ5Ø×,Ñ,¨X°q¸%Ô@Ü˜w×0Ñ0°Ó:×NÑNÈqÑQÐSXÕYÐYÐYñZr   c                ó  — | j                  | j                  t        j                  g¬«      «      }t        j                  «       D ]8  \  }}| j                  |||«       t        | j                  |«      |   |«      rŒ8J ‚ y r2   )	rD   r&   r   r'   r¶  ri   r<  r´  r0  ©r   r—   r]   r^   s       r   rB  rB  û  s~   € Ø×'Ñ'Ø× Ñ ¬^×-DÑ-DÐ,EÐ ÓFó€Hô
 "×'Ñ'Ó)ò S‰
ˆˆUØ×#Ñ# H¨c°5Ô9Ü˜w×;Ñ;¸HÓEÀcÑJÈEÕRÐRÐRñSr   c                ó  — | j                  | j                  t        j                  g¬«      «      }t        j                  «       D ]8  \  }}| j                  |||«       t        | j                  |«      |   |«      rŒ8J ‚ y r2   )	rD   r&   r   r'   r¶  ri   rK  r´  rF  rÅ  s       r   rN  rN    s~   € Ø×'Ñ'Ø× Ñ ¬^×-DÑ-DÐ,EÐ ÓFó€Hô
 "×'Ñ'Ó)ò U‰
ˆˆUØ×%Ñ% h°°UÔ;Ü˜w×=Ñ=¸hÓGÈÑLÈeÕTÐTÐTñUr   c                ó  — t        j                  |«      }i }i }t        |«      D ]Ø  }d|› �}	|€/|j                  t        j
                  t        j                  g«      }| j                  |f|	¬«      }
| j                  |
d|«       | j                  |
d|«       i }t        |«      D ]:  }t        |«      }||_        | j                  |
|«      |_        |||j                  <   Œ< |||
<   t        |	|d|id|i|
¬«      ||
<   ŒÚ ||fS )Nztest-study-name-r8   Úur$   )r!   rR  rŽ   r‘   r-   )rw  rx  r£   Úchoicer   r'   rY   r&   r`   rt   ry  r¡   rD   r”   r   )r   rê   rë   rì   rR  rz  Ústudy_id_to_frozen_studyÚstudy_id_to_trialsr¬   r!   r-   rG   Újr2  s                 r   rö   rö     s-  € ô —‘˜dÓ#€IØ79ÐØ<>ÐÜ�7‹^ò 
ˆØ'¨ sÐ+ˆ
ØÐØ!×(Ñ(¬.×*AÑ*AÄ>×CZÑCZÐ)[Ó\ˆIØ×+Ñ+¸	°|ÐPZÐ+Ó[ˆØ×#Ñ# H¨c°1Ô5Ø×%Ñ% h°°QÔ7ØˆÜ�w“ò 	,ˆAÜ# IÓ.ˆEØˆEŒLØ%×6Ñ6°xÀÓGˆEŒOØ&+ˆF�5—?‘?Ò#ð		,ð
 (.Ð˜8Ñ$Ü-8Ø!ØØ˜Q�xØ˜q˜Øô.
Ð  Ò*ð
ð* $Ð%7Ð7Ð7r   c                ó  — | j                  dd«      t        dd«      f| j                  dd«      t        ddd¬«      f| j                  g d¢«      t        d«      f| j                  dd«      t        dd«      f| j                  dd	g«      t        d
«      fdœ}ddd dddgdœdœ}| j                  t        «      }i }i }i }i }i }|j                  «       D ]&  \  }	\  }
}| j                  ddg«      sŒ|
||	<   |||	<   Œ( |j                  «       D ]<  \  }	}
| j                  ddg«      r|
|d|	z   <   | j                  ddg«      sŒ5|
|d|	z   <   Œ> t        | j                  dd«      «      D ]+  }| j                  ddg«      sŒ| j                  dd«      ||<   Œ- t        d||t        j                  k(  r| j                  dd«      nd t        j                  «       |j                  «       rt        j                  «       nd |||||d¬«      S )Nr   r   r   T)r  )ÚCatAÚCatBÚCatCéýÿÿÿr  r   )r  r   )rõ   ÚparamBÚparamCÚparamDÚparamEÚvalueAr§   rª  r«  r¬  )ÚattrAÚattrBÚattrCÚattrDFÚusr_Úsys_é   r¶   iöÿÿÿ)r¡   r€   r^   r•   rœ   r‡   rº   rŽ   r‘   r‹   r—   )Úuniformr	   rÉ  r   r×   ri   r£   Úrandintr   r   rË   r   rª   rÜ   )rz  Úexample_paramsÚexample_attrsr€   r‡   rº   rŽ   r‘   r‹   r]   r^   Údistr¬   s                r   ry  ry  3  sB  € à×$Ñ$ Q¨Ó*Ô,=¸aÀÓ,CÐDØ×$Ñ$ Q¨Ó*Ô,=¸aÀÈÔ,MÐNà×ÑÒ5Ó6Ü#Ð$<Ó=ð
ð ×$Ñ$ R¨Ó+Ô->¸rÀ1Ó-EÐFØ×#Ñ# S¨# JÓ/Ô1HÈÓ1TÐUñ	€Nð ØØØ$)°GÐ=MÐ3NÑOñ	€Mð ×ÑœZÓ(€EØ€FØ€MØ€JØ#%€LØÐØ,×2Ñ2Ó4ò &Ñˆ‰]ˆe�TØ×Ñ˜T 5˜MÕ*ØˆF�3‰KØ!%ˆM˜#Òð&ð $×)Ñ)Ó+ò /‰
ˆˆUØ×Ñ˜T 5˜MÔ*Ø',ˆJ�v ‘|Ñ$Ø×Ñ˜T 5˜MÕ*Ø).ˆL˜ #™Ò&ð	/ô
 �9×$Ñ$ Q¨Ó+Ó,ò @ˆØ×Ñ˜T 5˜MÕ*Ø%.×%6Ñ%6°s¸BÓ%?Ð Ò"ð@ô ØØØ,1´Z×5HÑ5HÒ,Hˆi×Ñ  RÔ(ÈdÜ—|‘|“~Ø,1×,=Ñ,=Ô,?œ(Ÿ,™,œ.ÀTØØ#ØØ!Ø/Øôð r   )rW   r   )ra  r!  rb  r!  rW   Úboolr¤  r   )r   r   rê   r©   rë   r©   rì   r©   rR  zStudyDirection | NonerW   z@tuple[dict[int, FrozenStudy], dict[int, dict[int, FrozenTrial]]])rz  zrandom.RandomrW   r   ):Ú
__future__r   rl  r   r³  rš  rw  ÚsysÚtimer   Útypingr   r   Únumpyr(  r<   r>   Úoptuna.distributionsr   r	   Úoptuna.exceptionsr
   Úoptuna.storages._baser   Úoptuna.study._frozenr   Úoptuna.study._study_directionr   Úoptuna.trialr   r   Úoptuna._typingr   Úoptuna.storagesr   Úfixturer   r   rS   r×   rh   Ú__annotations__r±  Ú
float_inforÝ   rŽ  r!  r¶  r´  rn   ry   rå   r  r,  rB  rN  rö   ry  r   r   r   ú<module>rô     s‚  ðÞ "ã Ý Û Û Û Û 
Ý Ý Ý  ã Û ã Ý 8Ý 2Ý 6Ý ;Ý ,Ý 8Ý $Ý #ñ Ý/Ý+ð ‡�òó ð÷~Lñ ~LñB �*Ó€
ð ØØ,1¸GÐEUÐ;VÑWñ.€Ð*ó ð Ø
�'‰'Ø�>‰>×ÑØ—^‘^×'Ñ'Ð'Ø�>‰>×ÑØ—^‘^×'Ñ'Ð'Ù�‹<Ù˜5“\�MÙ�‹<ñ
€óóSóUó	1óYó$ZóSóUð  (,ð8Øð8àð8ð ð8ð ð	8ð
 %ð8ð Fó8ôD/r   