Ë
    âQ(hÆU  ã                   ój   — d dl Z d dlZd dlmZmZ d dlmZ d dlm	Z	 d„ Z
 G d„ d«      Z G d„ d	«      Zy)
é    N)Úassert_equalÚassert_allclose)Ústats)Ú	_survivalc                 óê  — dt         fdt        fg}t        j                  t	        | |«      D ��cg c]	  \  }}||f‘Œ c}}|¬«      }t        j
                  |d¬«      }|d   } t        j                  |d   «      }| j                  }t        j                  |dd«      }t        j                  ||z
  |z  «      }	t        j                  | d d d…   d¬	«      \  }
}| | d
z
     }|	| d
z
     }||fS c c}}w )NÚtimeÚcensored)Údtype)r   r	   )Úorderr   éÿÿÿÿT)Úreturn_indexé   )ÚfloatÚintÚnpÚarrayÚzipÚsortÚlogical_notÚsizeÚarangeÚcumprodÚunique)Útimesr	   r
   ÚtÚdÚdataÚdiedÚmÚnÚsfÚ_ÚindicesÚ	ref_timesÚref_sfs                 ú]/var/www/skyplay_api_hub/venv/lib/python3.12/site-packages/scipy/stats/tests/test_survival.pyÚ_kaplan_meier_referencer'      sí   € ð ”eˆ_˜z¬3Ð/Ð0€EÜ�8‰8¬¨E°8Ó(<×=¡  1�a˜’VÓ=ÀUÔK€DÜ�7‰7�4Ð3Ô4€DØ�‰L€EÜ�>‰>˜$˜zÑ*Ó+€Dà�
‰
€AÜ
�	‰	�!�Q˜Ó€AÜ	�‰�Q˜‘X ‘NÓ	#€Bô —‘˜5¡ 2 ™;°TÔ:�J€A€wØ�w�h ‘lÑ#€IØ��˜1‘Ñ€FØ�fÐÐùó >s   ¯C/
c            
       ó„  — e Zd Zed„ «       Zd„ Zd„ Zd„ Zd„ Zd„ Z	g d¢Z
g d¢Zg d	¢Zg d
¢Zg d¢Zg d¢Zg d¢Zg d¢Zg d¢Zg d¢Zg d¢Zg d¢Zg d¢Zg d¢Zg d¢Zej4                  j7                  de
eefeeefeeefeeefeeefg«      d„ «       Zej4                  j7                  dg d¢«      d„ «       Zd„ Zd„ Zd„ Z d„ Z!d„ Z"y )!ÚTestSurvivalc                 ór  — | j                  |«      }| j                  dd|«      j                  t        j                  «      }| j                  t        j                  ||«      «      }| j                  |j                  ¬«      | j                  «       kD  }t        j                  j                  ||«      }|||fS )Nr   é   ©r   )ÚrandomÚintegersÚastyper   Úint32ÚpermutedÚrepeatr   r   ÚCensoredDataÚright_censored)ÚrngÚn_uniqueÚunique_timesÚrepeatsr   r	   Úsamples          r&   Úget_random_samplezTestSurvival.get_random_sample(   s�   € ð —z‘z (Ó+ˆà—,‘,˜q ! XÓ.×5Ñ5´b·h±hÓ?ˆØ—‘œRŸY™Y |°WÓ=Ó>ˆØ—:‘: 5§:¡:�:Ó.°·±³Ñ=ˆÜ×#Ñ#×2Ñ2°5¸(ÓCˆØ�u˜hÐ&Ð&ó    c                 ó  — d}t        j                  t        |¬«      5  t        j                  dgg«       d d d «       t        j                  t        |¬«      5  t        j                  d«       d d d «       d}t        j                  t        |¬«      5  t        j                  t
        j                  g«       d d d «       d}t        j                  t        |¬«      5  t        j                  t        j                  j                  dgdg¬«      «       d d d «       d}t        j                  g d	¢«      }t        j                  t        |¬«      5  |j                  j                  d
¬«       d d d «       t        j                  t        |¬«      5  |j                  j                  d¬«       d d d «       d}t        j                  t        |¬«      5  |j                  j                  d«       d d d «       t        j                  t        |¬«      5  |j                  j                  ddg«       d d d «       d}t        j                  t        |¬«      5  |j                  j                  «       }d d d «       d}t        j                  t        |¬«      5  j                  j                  «        d d d «       t        j                  t        |¬«      5  j                   j                  «        d d d «       y # 1 sw Y   �ŒÃxY w# 1 sw Y   �Œ—xY w# 1 sw Y   �ŒZx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   �ŒXx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)Nz,`sample` must be a one-dimensional sequence.©Úmatchr   z`sample` must not contain nanz5Currently, only uncensored and right-censored data...T)r	   zmethod` must be one of...)r   é   é   z	ekki-ekki©ÚmethodÚ	shrubberyz2confidence_level` must be a scalar between 0 and 1r   ç      à?ç333333ã?z:The confidence interval is undefined at some observations.z.Confidence interval bounds do not implement...)ÚpytestÚraisesÚ
ValueErrorr   Úecdfr   ÚnanÚNotImplementedErrorr3   Úleft_censoredÚcdfÚconfidence_intervalr!   ÚwarnsÚRuntimeWarningÚlowÚhigh)ÚselfÚmessageÚresÚcis       r&   Útest_input_validationz"TestSurvival.test_input_validation3   s¨  € Ø@ˆÜ�]‰]œ:¨WÔ5ñ 	Ü�J‰J˜˜�uÔ÷	ä�]‰]œ:¨WÔ5ñ 	Ü�J‰J�qŒM÷	ð 2ˆÜ�]‰]œ:¨WÔ5ñ 	!Ü�J‰JœŸ™�xÔ ÷	!ð JˆÜ�]‰]Ô.°gÔ>ñ 	OÜ�J‰J”u×)Ñ)×7Ñ7¸¸ÀtÀfÐ7ÓMÔN÷	Oð .ˆÜ�j‰jšÓ#ˆÜ�]‰]œ:¨WÔ5ñ 	<Ø�G‰G×'Ñ'¨{Ð'Ô;÷	<ä�]‰]œ:¨WÔ5ñ 	;Ø�F‰F×&Ñ&¨kÐ&Ô:÷	;ð GˆÜ�]‰]œ:¨WÔ5ñ 	,Ø�G‰G×'Ñ'¨Ô+÷	,ä�]‰]œ:¨WÔ5ñ 	3Ø�F‰F×&Ñ&¨¨S zÔ2÷	3ð OˆÜ�\‰\œ.°Ô8ñ 	/Ø—‘×,Ñ,Ó.ˆB÷	/ð CˆÜ�]‰]Ô.°gÔ>ñ 	)Ø�F‰F×&Ñ&Ô(÷	)ä�]‰]Ô.°gÔ>ñ 	*Ø�G‰G×'Ñ'Ô)÷	*ð 	*÷C	ñ 	ú÷	ñ 	ú÷	!ñ 	!ú÷	Oñ 	Oú÷
	<ñ 	<ú÷	;ñ 	;ú÷	,ñ 	,ú÷	3ñ 	3ú÷	/ð 	/ú÷	)ð 	)ú÷	*ð 	*úsƒ   žK7ÁLÂ%LÃ7LÅL+ÆL8ÇMÈMÉMÊM+ËM7Ë7LÌLÌLÌL(Ì+L5Ì8MÍMÍMÍM(Í+M4Í7N c                 ó^  — t        j                  g «      }t        |j                  j                  g «       t        |j                  j
                  g «       t        j                  dg«      }t        |j                  j                  dg«       t        |j                  j
                  dg«       y )Nr   )r   rI   r   rM   Ú	quantilesÚprobabilities)rS   rU   s     r&   Útest_edge_caseszTestSurvival.test_edge_casesY   sr   € Ü�j‰j˜‹nˆÜ�S—W‘W×&Ñ&¨Ô+Ü�S—W‘W×*Ñ*¨BÔ/ä�j‰j˜!˜‹oˆÜ�S—W‘W×&Ñ&¨¨Ô,Ü�S—W‘W×*Ñ*¨Q¨CÕ0r;   c                 óÂ  — g d¢}t        j                  |«      }t        j                  t        j                  |«      «      }t        j
                  dd«      dz  }d|z
  }t        |j                  j                  |«       t        |j                  j                  |«       t        |j                  j                  |«       t        |j                  j                  |«       y )N)gìQ¸…ë@gR¸…ëQ@g=
×£p=@g®Gáz®@gÍÌÌÌÌÌ@r   é   é   )r   rI   r   r   r   r   r   rM   rY   rZ   r!   ©rS   r9   rU   Úref_xÚref_cdfr%   s         r&   Útest_uniquezTestSurvival.test_uniqueb   s™   € â/ˆÜ�j‰j˜Ó ˆÜ—‘œŸ	™	 &Ó)Ó*ˆÜ—)‘)˜A˜q“/ AÑ%ˆØ�W‘ˆÜ�S—W‘W×&Ñ&¨Ô.Ü�S—W‘W×*Ñ*¨GÔ4Ü�S—V‘V×%Ñ% uÔ-Ü�S—V‘V×)Ñ)¨6Õ2r;   c                 ó¾  — g d¢}t        j                  |«      }t        j                  t        j                  |«      «      }t        j
                  g d¢«      }d|z
  }t        |j                  j                  |«       t        |j                  j                  |«       t        |j                  j                  |«       t        |j                  j                  |«       y )N)r   r?   r   r?   r@   r+   )gUUUUUUÅ?gUUUUUUÕ?gUUUUUUå?g«ªªªªªê?r   r   )r   rI   r   r   r   r   r   rM   rY   rZ   r!   r_   s         r&   Útest_nonuniquezTestSurvival.test_nonuniquen   s”   € â#ˆÜ�j‰j˜Ó ˆÜ—‘œŸ	™	 &Ó)Ó*ˆÜ—(‘(Ò2Ó3ˆØ�W‘ˆÜ�S—W‘W×&Ñ&¨Ô.Ü�S—W‘W×*Ñ*¨GÔ4Ü�S—V‘V×%Ñ% uÔ-Ü�S—V‘V×)Ñ)¨6Õ2r;   c                 ó–  — t         j                  j                  d«      }| j                  |d«      \  }}}t	        j
                  |«      }|j                  j                  }|t        j                  ||d   dz   ¬«      dz  z   }t        |j                  j                  |«      |j                  j                  «       t        |j                  j                  |«      |j                  j                  «       t        |j                  j                  |d   dz
  «      d«       t        |j                  j                  t         j                   t         j                  g«      ddg«       t        |j                  j                  |«      |j                  j                  «       t        |j                  j                  |«      |j                  j                  «       t        |j                  j                  |d   dz
  «      d«       t        |j                  j                  t         j                   t         j                  g«      ddg«       y )Nl   ­I)û_ é   r   r   )Úappendr?   r   )r   r-   Údefault_rngr:   r   rI   rM   rY   Údiffr   ÚevaluaterZ   Úinfr!   )rS   r5   r9   r"   rU   ÚxÚxrs          r&   Útest_evaluate_methodsz"TestSurvival.test_evaluate_methodsz   sŒ  € ä�i‰i×#Ñ#Ð$7Ó8ˆØ×-Ñ-¨c°2Ó6‰ˆ��1Ü�j‰j˜Ó ˆØ�G‰G×ÑˆØ”—‘˜ 1 R¡5¨¡7Ô+¨AÑ-Ñ-ˆä�S—W‘W×%Ñ% aÓ(¨#¯'©'×*?Ñ*?Ô@Ü�S—W‘W×%Ñ% bÓ)¨3¯7©7×+@Ñ+@ÔAÜ�S—W‘W×%Ñ% a¨¡d¨1¡fÓ-¨qÔ1Ü�S—W‘W×%Ñ%¬¯© w´·±Ð&7Ó8¸1¸a¸&ÔAä�S—V‘V—_‘_ QÓ'¨¯©×)=Ñ)=Ô>Ü�S—V‘V—_‘_ RÓ(¨#¯&©&×*>Ñ*>Ô?Ü�S—V‘V—_‘_ Q q¡T¨!¡VÓ,¨aÔ0Ü�S—V‘V—_‘_¤r§v¡v g¬r¯v©vÐ%6Ó7¸!¸Q¸Õ@r;   )
é%   é+   é/   é8   é<   é>   éG   éM   éP   éQ   )
r   r   r   r   r   r   r   r   r   r   )
r   r   ç      ì?ç      è?rz   rz   rz   rD   g      Ð?r   )
é   é   é   é   é   é   r{   é    é   é(   )
r   r   r   r   r   r   r   r   r   r   )	çÍÌÌÌÌÌì?gžï§ÆK7é?çš™™™™™å?r…   gHáz®Gá?gìQ¸…ëÙ?çHáz®GÑ?r†   r†   )
é!   é   é)   é0   rŠ   é   ro   rŠ   r‹   rp   )
r   r   r   r   r   r   r   r   r   r   )r   ry   rz   rz   rE   rE   rE   )é   r@   é   é   rŒ   é   r~   r?   é   é   rŒ   r   r|   r   é
   é   r]   r^   é	   r‘   )r   r   r   r   r   r   r   r   r   r   r   r   r   r   r   r   r   r   r   r   )çffffffî?r•   gNbX9´ì?ç5^ºIë?r–   r–   r–   r–   r–   r–   gR¸…ëQè?çoƒÀÊ¡å?r—   r—   r—   çÓMbX9à?r˜   )r@   r^   r{   r’   r^   r^   r{   r|   rf   r~   r?   r�   r’   r”   r|   r^   r{   r�   )r   r   r   r   r   r   r   r   r   r   r   r   r   r   r   r   r   r   )
gh‘í|?5î?g¦›Ä °rì?g´Èv¾Ÿç?çòÒMbXá?r™   r™   g´Èv¾Ÿ×?ç^ºI+Ç?rš   rš   Úcasec                 ó:  — |\  }}}t         j                  j                  |t        j                  |«      «      }t        j
                  |«      }t        |j                  j                  |d¬«       t        |j                  j                  t        j                  t        j                  |«      «      «       t        |t        j                  |«      «      }t        |d   t        j                  t        j                  |«      «      «       t        |d   |d¬«       y )Nçü©ñÒMbP?©Úatolr   r   )r   r3   r4   r   r   rI   r   r!   rZ   r   rY   r   r   r'   )rS   r›   r   r   Úrefr9   rU   s          r&   Ú$test_right_censored_against_examplesz1TestSurvival.test_right_censored_against_examples¥   sÂ   € ð  Ñˆˆt�SÜ×#Ñ#×2Ñ2°5¼"¿.¹.ÈÓ:NÓOˆÜ�j‰j˜Ó ˆÜ˜Ÿ™×,Ñ,¨c¸Õ=Ü�S—V‘V×%Ñ%¤r§w¡w¬r¯y©y¸Ó/?Ó'@ÔAô & e¬R¯^©^¸DÓ-AÓBˆÜ�S˜‘VœRŸW™W¤R§Y¡Y¨uÓ%5Ó6Ô7Ü˜˜A™ ¨$Ö/r;   Úseed)l   €-øjï|Il   #}|4ÌIÝQl   ˆ>5Gnó?l   òe“ˆ5"c                 óº  — t         j                  j                  |«      }|j                  dd«      }| j	                  ||«      \  }}}t        j                  |«      }t        ||«      }t        |j                  j                  |d   «       t        |j                  j                  |d   «       t        j                  |¬«      }t        j                  |«      }t        j                  |«      }t        |d   |j                  j                  «       t        |d   |j                   j                  d¬«       t        |d   |j                  j                  d¬«       y )	Nr’   éd   r   r   )Ú
uncensoredç›+¡†›„=©Úrtolr?   )r   r-   rh   r.   r:   r   rI   r'   r   r!   rY   rZ   r3   r   Ú_ecdf_right_censoredr   rM   )	rS   r¢   r5   r6   r9   r   r	   rU   r    s	            r&   Ú4test_right_censored_against_reference_implementationzATestSurvival.test_right_censored_against_reference_implementation´   s  € ô �i‰i×#Ñ# DÓ)ˆØ—<‘<  CÓ(ˆØ"&×"8Ñ"8¸¸hÓ"GÑˆ��xÜ�j‰j˜Ó ˆÜ% e¨XÓ6ˆÜ˜Ÿ™×(Ñ(¨#¨a©&Ô1Ü˜Ÿ™×,Ñ,¨c°!©fÔ5ô ×#Ñ#¨uÔ5ˆÜ×,Ñ,¨VÓ4ˆÜ�j‰j˜ÓˆÜ�S˜‘V˜SŸV™V×-Ñ-Ô.Ü˜˜A™ §¡× 5Ñ 5¸EÕBÜ˜˜A™ §¡× 4Ñ 4¸5ÖAr;   c                 óÚ  — | j                   | j                  }}t        j                  j	                  |t        j                  |«      «      }t        j                  |«      }g d¢}|j                  j                  «       }|j                  j                  «       }|j                  j                  |j                  j                  z
  }t        ||d¬«       t        |j                  j                  t        j                  |j                  j                  |z
  dd«      «       t        |j                  j                  t        j                  |j                  j                  |z   dd«      «       t        |j                  j                  t        j                  |j                  j                  |z
  dd«      «       t        |j                  j                  t        j                  |j                  j                  |z   dd«      «       g d¢}	g d¢}
|j                  j                  d¬	«      }t        |j                  j                  |	d
¬«       t        |j                  j                  |
d
¬«       y )N)çú~j¼t“¸?r¬   gHáz®GÁ?ç#Ûù~j¼Ä?r­   r­   r­   r­   r­   r­   g1¬ZdË?ç°rh‘í|Ï?r®   r®   r®   çmçû©ñÒÕ?r¯   r�   rž   r   r   )çiÿ¬U;æ?r°   gLÅÆ¼Ž¸ä?çSX© ¢êâ?r±   r±   r±   r±   r±   r±   g•·#œ¼Ý?çvß1<ö³×?r²   r²   r²   ç</OçŠÄ?r³   )ç–è,³Åï?r´   gXÇñC%ï?çd–=	lNî?rµ   rµ   rµ   rµ   rµ   rµ   gGÿËµhí?çyÊjºžhë?r¶   r¶   r¶   çò•@JìÚè?r·   úlog-logrA   g�íµ ÷Æ°>)Út4Úd4r   r3   r4   r   r   rI   r!   rN   rM   rZ   rQ   r   ÚcliprR   )rS   r   r   r9   rU   Úref_allowanceÚsf_ciÚcdf_ciÚ	allowanceÚref_lowÚref_highs              r&   Útest_right_censored_ciz#TestSurvival.test_right_censored_ciÉ   s¾  € à—g‘g˜tŸw™wˆtˆÜ×#Ñ#×2Ñ2°5¼"¿.¹.ÈÓ:NÓOˆÜ�j‰j˜Ó ˆò.ˆð —‘×*Ñ*Ó,ˆØ—‘×,Ñ,Ó.ˆØ—F‘F×(Ñ(¨5¯9©9×+BÑ+BÑBˆ	ä˜	 =°tÕ<Ü˜Ÿ	™	×/Ñ/ÜŸ™ §¡× 4Ñ 4°yÑ @À!ÀQÓGô	Iä˜Ÿ
™
×0Ñ0ÜŸ™ §¡× 4Ñ 4°yÑ @À!ÀQÓGô	Iä˜Ÿ
™
×0Ñ0ÜŸ™ §¡× 5Ñ 5¸	Ñ AÀ1ÀaÓHô	Jä˜Ÿ™×1Ñ1ÜŸ™ §¡× 5Ñ 5¸	Ñ AÀ1ÀaÓHô	JòEˆòFˆð —‘×*Ñ*°)Ð*Ó<ˆÜ˜Ÿ	™	×/Ñ/°¸tÕDÜ˜Ÿ
™
×0Ñ0°(ÀÖFr;   c                 ó>  — | j                   | j                  }}t        j                  j	                  |t        j                  |«      «      }t        j                  |«      }t        j                  g d¢«      }t        j                  g d¢«      }|j                  j                  d¬«      }|j                  j                  d¬«      }t        |j                  j                  |d¬«       t        |j                  j                  |d¬«       t        |j                  j                  d|z
  d¬«       t        |j                  j                  d|z
  d¬«       g d¢}	g d	¢}
|j                  j                  dd
¬«      }t        |j                  j                  |	«       t        |j                  j                  |
«       g d¢}	g d¢}
|j                  j                  d
¬«      }t        |j                  j                  |	«       t        |j                  j                  |
«       y )N)
gÂ4Så?g n À;ùã?g–Î‡g	2Ý?çœoD÷¬kÒ?rÄ   rÄ   g½§rÚSrÀ?ç?T1³Ÿ?rÅ   rÅ   )
gÕæ!S¾ï?gn‹2dï?g Ã|yöë?çU¯²¶©ç?rÆ   rÆ   gƒmÄ“ÝLã?ç^ô¤‹Û?rÇ   rÇ   r¸   rA   gñhãˆµøä>rž   r   )
g”‹eÀÌç?gûõÔÀCòå?g’W�ä0à?çJµ’É|Õ?rÈ   rÈ   g c£a—vÄ?ç#êè£	§?rÉ   rÉ   )
g—–x‰ ¦ï?g]Õ^L"Øî?gD%V
§dë?ç×Ô’ˆÑæ?rÊ   rÊ   g[œ<\é+â?ç¾7Ô§xáØ?rË   rË   r„   )rB   Úconfidence_level)
g7A�˜caë?g|X›þ§‹è?gí1�ç�á?çä£/”Ö?rÍ   rÍ   gRï¼¯ŒÂ?ç        rÎ   rÎ   )
ç      ð?rÏ   g¡±¨Œüªì?ç8Ù’`§ç?rÐ   rÐ   gñ©à×Eyâ?ç,ËªÞ~O×?rÑ   rÑ   )rÌ   )Út5Úd5r   r3   r4   r   r   rI   r   r!   rN   rM   r   rQ   rZ   rR   )rS   r   r   r9   rU   ÚlowerÚupperr½   r¾   rQ   rR   s              r&   Ú test_right_censored_ci_example_5z-TestSurvival.test_right_censored_ci_example_5ó   s£  € à—g‘g˜tŸw™wˆtˆÜ×#Ñ#×2Ñ2°5¼"¿.¹.ÈÓ:NÓOˆÜ�j‰j˜Ó ˆÜ—‘ò Ló Mˆä—‘ò Ló Mˆð —‘×*Ñ*°)Ð*Ó<ˆØ—‘×,Ñ,°IÐ,Ó>ˆä˜Ÿ	™	×/Ñ/°¸TÕBÜ˜Ÿ
™
×0Ñ0°%¸dÕCÜ˜Ÿ
™
×0Ñ0°!°E±'ÀÕEÜ˜Ÿ™×1Ñ1°1°U±7ÀÕFò$ˆò$ˆð —‘×*Ñ*°)Ø<?ð +ó Aˆä˜Ÿ	™	×/Ñ/°Ô5Ü˜Ÿ
™
×0Ñ0°$Ô7ò/ˆòLˆð —‘×*Ñ*¸CÐ*Ó@ˆÜ˜Ÿ	™	×/Ñ/°Ô5Ü˜Ÿ
™
×0Ñ0°$Õ7r;   c           
      óä  — | j                   | j                  }}t        j                  j	                  |t        j                  |«      «      }t        j                  |«      }g d¢}t
        j                  ddddt
        j                  g}t
        j                  ddddt
        j                  g}t        j                  |j                  j                  |«      }d	}	t        j                  t        |	¬
«      5  |j                  j                  «       }
d d d «       t!        
j"                  j$                  |   dd  |dd  «       t!        |
j&                  j$                  |   dd  |dd  «       t
        j                  ddddt
        j                  g}t
        j                  ddddt
        j                  g}t        j                  |j                  j                  |«      }t        j                  t        |	¬
«      5  |j(                  j                  «       }
d d d «       t!        |
j"                  j$                  |   dd  |dd  «       t!        |
j&                  j$                  |   dd  |dd  «       dddddddddt
        j                  g
}dddddddddt
        j                  g
}t!        |
j"                  j$                  |«       t!        |
j&                  j$                  |«       t        j                  t        |	¬
«      5  |j(                  j                  d¬«      }
d d d «       t
        j                  t
        j                  dddddddt
        j                  g
}t
        j                  t
        j                  ddddddd t
        j                  g
}t!        |
j"                  j$                  |«       t!        |
j&                  j$                  |«       y # 1 sw Y   �Œ¿xY w# 1 sw Y   �ŒÊxY w# 1 sw Y   ŒÐxY w)!N)ro   rq   rr   rv   rw   rx   r   gµ!E±¯ûª?gÜ™{²k›Õ?gÕuOÂªÖ?gY®ešá?ga§çEPî?rÏ   z9The confidence interval is undefined at some observationsr=   r   g|EØžªä?g¡U<ÏÝËÜ?gs‘C±¯ûª?gå­ëEPî?g3Â&J2å?gÀgEØžªä?gG<ÏÝËÜ?g !E±¯ûª?rÎ   gæ­ëEPî?g3Â&J2å?r¸   rA   gökô´›ÄØ?gutz¼Ì%Ô?g€¼
àš´?gKuËÏ€…?gò/Ÿ8’gï?g(ÈyDëÉí?go¡â(Óqê?g9–'Ôòüä?)Út1Úd1r   r3   r4   r   r   rI   rJ   ÚsearchsortedrM   rY   rF   rO   rP   rN   r   rQ   rZ   rR   r!   )rS   r   r   r9   rU   rl   ÚfloÚfupÚirT   rV   rQ   rR   s                r&   Útest_right_censored_ci_nansz(TestSurvival.test_right_censored_ci_nans%  s3  € à—g‘g˜tŸw™wˆtˆÜ×#Ñ#×2Ñ2°5¼"¿.¹.ÈÓ:NÓOˆÜ�j‰j˜Ó ˆò %ˆÜ�v‰v�q˜!Ð.Ð0AÄ2Ç6Á6ÐJˆÜ�v‰v�} l°LÀ#ÄrÇvÁvÐNˆÜ�O‰O˜CŸG™G×-Ñ-¨qÓ1ˆàMˆÜ�\‰\œ.°Ô8ñ 	/Ø—‘×,Ñ,Ó.ˆB÷	/ô 	˜Ÿ™×,Ñ,¨QÑ/°°Ð3°S¸¸°WÔ=Ü˜Ÿ™×-Ñ-¨aÑ0°°Ð4°c¸!¸"°gÔ>ô �v‰v�} n°mÀQÌÏÉÐOˆÜ�v‰v�s˜CÐ!2Ð4EÄrÇvÁvÐNˆÜ�O‰O˜CŸG™G×-Ñ-¨qÓ1ˆä�\‰\œ.°Ô8ñ 	.Ø—‘×+Ñ+Ó-ˆB÷	.ô 	˜Ÿ™×,Ñ,¨QÑ/°°Ð3°S¸¸°WÔ=Ü˜Ÿ™×-Ñ-¨aÑ0°°Ð4°c¸!¸"°gÔ>ð �2Ð*Ð,?Ø"Ð$7Ð9LØ" B¬¯©ð0ˆð �B˜˜B  B¨Ð,>Ø"¤B§F¡Fð,ˆä˜Ÿ™×,Ñ,¨cÔ2Ü˜Ÿ™×-Ñ-¨tÔ4ô �\‰\œ.°Ô8ñ 	>Ø—‘×+Ñ+°9Ð+Ó=ˆB÷	>ä�v‰v”r—v‘vÐ2Ð4GØ"Ð$7Ð9LØ"Ð$7¼¿¹ðAˆô —‘œŸ™Ð 2Ð4FØ"Ð$6Ð8JØ"Ð$6¼¿¹ð@ˆô 	˜Ÿ™×,Ñ,¨cÔ2Ü˜Ÿ™×-Ñ-¨tÕ4÷e	/ñ 	/ú÷	.ñ 	.ú÷4	>ð 	>ús$   Ã4OÇ6OË:O&ÏOÏO#Ï&O/c                 ó  — t         j                  j                  d«      }|j                  ddd¬«      }t        j                  |«      }d|t        j
                  |«      <   t        j                  |«      }t        j                  t        j                  j                  ||«      «      }t        |j                  j                  |j                  j                  «       t        |j                  j                  |j                  j                  «       t        |j                  j                  d d |j                  j                  d d «       t        |j                  j                   d d |j                  j                   d d d¬	«       y )
Nl   m"ê%ª< r’   r¤   iè  r,   Tr   r¦   r§   )r   r-   rh   r.   Ú
zeros_likeÚargmaxr   rI   r3   r4   r   r!   rY   Ú_nÚ_dr   Ú_sf)rS   r5   r9   r	   rU   r    s         r&   Ú&test_right_censored_against_uncensoredz3TestSurvival.test_right_censored_against_uncensoredk  s  € Ü�i‰i×#Ñ#Ð$7Ó8ˆØ—‘˜b #¨D�Ó1ˆÜ—=‘= Ó(ˆØ&*ˆ”—‘˜6Ó"Ñ#Ü�j‰j˜Ó ˆÜ�j‰jœ×+Ñ+×:Ñ:¸6À8ÓLÓMˆÜ�S—V‘V×%Ñ% s§v¡v×'7Ñ'7Ô8Ü�S—V‘V—Y‘Y §¡§	¡	Ô*Ü�S—V‘V—Y‘Y˜s �^ S§V¡V§Y¡Y¨s° ^Ô4Ü˜Ÿ™Ÿ
™
 3 B˜¨¯©¯©°C°R¨¸uÖEr;   c                 óÂ  — t         j                  j                  d«      }|j                  dd«      }| j	                  ||«      \  }}}t        j                  |«      }	 dd lm} |j                  j                  «        y # t        t        f$ rP d}t        j                  t        |¬«      5  |j                  j                  «        d d d «       Y y # 1 sw Y   Y y xY ww xY w)Nl   ‘u!u°\xD r’   r¤   r   z2matplotlib must be installed to use method `plot`.r=   )r   r-   rh   r.   r:   r   rI   Úmatplotlib.pyplotÚpyplotr!   ÚplotÚModuleNotFoundErrorÚImportErrorrF   rG   )rS   r5   r6   r9   r"   rU   ÚpltrT   s           r&   Útest_plot_ivzTestSurvival.test_plot_ivw  s¬   € Ü�i‰i×#Ñ#Ð$7Ó8ˆØ—<‘<  CÓ(ˆØ×-Ñ-¨c°8Ó<‰ˆ��1Ü�j‰j˜Ó ˆð	Ý+Ø�F‰F�K‰K�MøÜ#¤[Ð1ò 	ØKˆGÜ—‘Ô2¸'ÔBñ Ø—‘—‘”÷÷ ó úð	ús*   Á A? Á?,CÂ+CÃCÃC	ÃCÃCN)#Ú__name__Ú
__module__Ú__qualname__Ústaticmethodr:   rW   r[   rb   rd   rn   rØ   rÙ   Úr1Út2Úd2Úr2Út3Úd3Úr3r¹   rº   Úr4rÒ   rÓ   Úr5rF   ÚmarkÚparametrizer¡   rª   rÂ   rÖ   rÞ   rå   rí   © r;   r&   r)   r)   &   s(  „ àñ'ó ð'ò$*òL1ò
3ò
3òAò& 
2€BÚ	'€BÚ	<€Bò 
0€BÚ	'€BÚ	B€BÚ	1€BÚ	'€BÚ	.€Bò
.€Bâ	E€Bò
A€Bò 
H€BÚ	?€BÚ	O€Bà‡[�[×Ñ˜V r¨2¨r l°R¸¸R°LÀ2ÀrÈ2À,Ø')¨2¨r l°R¸¸R°Lð&Bó Cñ0óCð0ð ‡[�[×Ñ˜Vò &Nó OñBóOðBò&(GòT08òdD5òL
Fór;   r)   c            
       óÈ   — e Zd Zej                  j                  dg d¢g d¢gg d¢g d¢gdg d¢fg d¢g d	¢gg d
¢g d¢gdg d¢fg d¢g d¢gg d¢g d¢gdg d¢fg«      d„ «       Zd„ Zy)ÚTestLogRankzx, y, statistic, pvalue)r{   r|   r}   r~   r   r€   )r{   r�   r‚   rƒ   )r‡   rˆ   r‰   )rŠ   rŠ   r‹   ro   rŠ   r‹   rp   gK±è÷©@)gÜ‡í~�?gÆØá%Ýï?g �í~q?)rŽ   r]   r^   r+   )r‚   rŽ   r‘   r~   )é   r   é   )r   rf   r�   r�   r^   g8v\¸ê?)g®ìÊÓ×?g‰hG:ê?gÝ]âÊÓÇ?)r]   r�   r   é   ro   é&   é1   é2   é?   éO   éV   éb   éÊ   éÛ   )é   rq   rw   éR   r  é•   )r’   r’   r|   r�   r~   rf   r   r‘   r�   r‚   rŒ   rŒ   r‹   rˆ   r  r‡   é#   ro   rƒ   rƒ   é.   rŠ   éL   rx   r  é[   ép   éµ   )é"   rƒ   éF   gð….ƒü@)gžJVáÚQy?g8j^áÚQi?g–¡%®æï?c                 ó>  — t        j                  |d   |d   ¬«      }t        j                  |d   |d   ¬«      }t        g d¢«      D ]S  \  }}t        j                  |||¬«      }t	        |j
                  dz  |d¬«       t	        |j                  ||   d¬«       ŒU y )	Nr   r   )r¥   Úright)z	two-sidedÚlessÚgreater)rl   ÚyÚalternativer?   g»½×Ùß|Û=rž   )r   r3   Ú	enumerateÚlogrankr   Ú	statisticÚpvalue)rS   rl   r  r  r   rÝ   r  rU   s           r&   Útest_log_rankzTestLogRank.test_log_rankˆ  s�   € ôj ×Ñ¨!¨A©$°a¸±dÔ;ˆÜ×Ñ¨!¨A©$°a¸±dÔ;ˆä'Ò(HÓIò 	?‰NˆAˆ{Ü—-‘- ! q°kÔBˆCô
 ˜CŸM™M¨1Ñ,¨i¸eÕDÜ˜CŸJ™J¨¨q©	¸Ö>ñ	?r;   c                 ób  — t        j                  ddg«      }d}t        j                  t        |¬«      5  t        j
                  |ddgg¬«       d d d «       d}t        j                  t        |¬«      5  t        j
                  ddgg|¬«       d d d «       y # 1 sw Y   ŒJxY w# 1 sw Y   y xY w)Nr   r?   z`y` must ber=   )rl   r  z`x` must be)r   r3   rF   rG   rH   r  )rS   r9   Úmsgs      r&   Útest_raiseszTestLogRank.test_raisesÉ  s•   € Ü×#Ñ# Q¨ FÓ+ˆàˆÜ�]‰]œ:¨SÔ1ñ 	0Ü�M‰M˜F¨¨1 v hÕ/÷	0ð ˆÜ�]‰]œ:¨SÔ1ñ 	0Ü�M‰M˜a ˜V˜H¨Õ/÷	0ð 	0÷		0ð 	0ú÷	0ð 	0ús   µBÁ5B%ÂB"Â%B.N)rî   rï   rð   rF   rû   rü   r!  r$  rý   r;   r&   rÿ   rÿ   †  s™   „ à‡[�[×ÑØ!ò4 'ªÐ8ÚÒ9Ð:àÚHð
ò Ò.Ð/ÚÒ/Ð0ØÚDð
ò IÚ(ð*òFâðð ÚHð

ð!	ó13ñh?ói3ðh?ó	0r;   rÿ   )rF   Únumpyr   Únumpy.testingr   r   Úscipyr   Úscipy.statsr   r'   r)   rÿ   rý   r;   r&   ú<module>r)     s2   ðÛ Û ß 7Ý Ý !ò÷<]ñ ]÷@L0ò L0r;   