+
    ,i                     F    ^ RI t ^ RIt^ RIHtHt ^ RIHt  ! R R4      tR# )    N)assert_equalassert_array_equal)CensoredDatac                      a  ] tR t^	t o R tR tR tR tR tR t	R t
R t]P                  P                  R	]P                   ]P"                  .4      R
 4       tR tR tRtV tR# )TestCensoredDatac                   ^.p^ .p^^.p^^..p\        WVVR7      p\        VP                  V4       \        VP                  V4       \        VP                  V4       \        VP
                  V4       VP                  4       p\        V\        P                  ! WV\        P                  ! V^R7      34      4       R# )   )leftrightinterval)axisN)
r   r   _uncensored_left_right	_interval	_uncensornpconcatenatemean)self
uncensoredr
   r   r   dataudatas   &      F/usr/lib/python3/dist-packages/scipy/stats/tests/test_censored_data.py
test_basicTestCensoredData.test_basic   s    S
sAF8J%-/T%%z2TZZ&T[[%(T^^X. UBNNJe,.GGH1,E,G H 	I    c                   \         P                  ! . RO4      p\         P                  ! . RO\        R7      p\        P                  ! W4      p\        VP                  W( ,          4       \        VP                  W,          4       \        VP                  . 4       \        VP                  \         P                  ! R4      4       R# r   )dtypeN)r            @)r   r	   r   r      )r   arrayboolr   right_censoredr   r   r   r   r   emptyr   xis_censoredr   s   &   r   test_right_censored$TestCensoredData.test_right_censored   sy    HH[!hhy5**1:T%%q7T[[!.1TZZ$T^^RXXf%56r   c                   \         P                  ! . RO4      p\         P                  ! . RO\        R7      p\        P                  ! W4      p\        VP                  W( ,          4       \        VP                  W,          4       \        VP                  . 4       \        VP                  \         P                  ! R4      4       R# r   )r   r%   r&   r   left_censoredr   r   r   r   r   r(   r)   s   &   r   test_left_censored#TestCensoredData.test_left_censored$   sy    HH[!hhy5))!9T%%q7TZZ0T[["%T^^RXXf%56r   c                j   . ROp. ROp\         P                  ! WR7      p\        VP                  \        P
                  ! \        \        W4      4      4      4       VP                  P                  R8X  g   Q hVP                  P                  R8X  g   Q hVP                  P                  R8X  g   Q hR# )      ?lowhighN)r3          @      @g      @)      ?r"   g      @g      @r   )r   interval_censoredr   r   r   r%   listzipr   shaper   r   r   abr   s   &   r   test_interval_censored_basic-TestCensoredData.test_interval_censored_basic-   s      --!<4>>288DQO+DE%%---zz4'''{{  D(((r   c                t   R \         P                  ) R	RRRR
.pR R\         P                  RRR\         P                  .p\        P                  ! WR7      p\	        VP
                  RR.RR..4       \	        VP                  R R.4       \	        VP                  R.4       \	        VP                  R	R
.4       R# )r3   r7   r9   g      $@g     @r8   g      &@r4   Ng      *g      )	r   infr   r;   r   r   r   r   r   r?   s   &   r   test_interval_censored_mixed-TestCensoredData.test_interval_censored_mixed6   s    
 266'5#sD$7&"&&#sD"&&9--!<4>>S#Jt+EF4++c3Z84::x04;;6r   c           
        \         P                  ! ^ ^.^^.^^.^	\         P                  .^\         P                  .\         P                  ) ^ .^^..4      p\        VR7      p\	        VP
                  ^^.4       \	        VP                  ^ .4       \	        VP                  ^	^.4       \	        VP                  ^ ^.^^..4       R# )r   r   N)	r   r%   rE   r   r   r   r   r   r   )r   r   r   s   &  r   test_interval_to_other_types-TestCensoredData.test_interval_to_other_typesC   s    
 88aVVV[[!vvgq\V% & X.T%%1v.TZZ!%T[[1a&)T^^q!fq!f%56r   c                0   \        . . . . R 7      pVP                  P                  R8X  g   Q hVP                  P                  R8X  g   Q hVP                  P                  R8X  g   Q hVP
                  P                  R8X  g   Q h\        V4      ^ 8X  g   Q hR# ))r   r
   r   r   Nr:   r#   )r   r   r>   r   r   r   len)r   r   s   & r   test_empty_arrays"TestCensoredData.test_empty_arraysU   s    r"rJ%%---zz4'''{{  D(((~~##v---4yA~~r   c                   \         P                  ! \        R R7      ;_uu_ 4        \        . RO.R7       RRR4       \         P                  ! \        R R7      ;_uu_ 4        \        . RO.R7       RRR4       \         P                  ! \        R R7      ;_uu_ 4        \        . RO.R7       RRR4       \         P                  ! \        RR7      ;_uu_ 4        \        . RO.R7       RRR4       \         P                  ! \        RR7      ;_uu_ 4        \        ^\        P
                  ^.R7       RRR4       \         P                  ! \        RR7      ;_uu_ 4        \        ^\        P
                  ^.R7       RRR4       \         P                  ! \        RR7      ;_uu_ 4        \        ^\        P
                  ^.R7       RRR4       \         P                  ! \        RR7      ;_uu_ 4        \        ^\        P
                  .^^..R7       RRR4       \         P                  ! \        R	R7      ;_uu_ 4        \        ^^.^^	.\        P                  \        P                  ..R7       RRR4       \         P                  ! \        R
R7      ;_uu_ 4        \        ^^ .^^..R7       RRR4       R#   + '       g   i     EL; i  + '       g   i     ELg; i  + '       g   i     EL?; i  + '       g   i     EL; i  + '       g   i     EL; i  + '       g   i     EL; i  + '       g   i     ELu; i  + '       g   i     EL<; i  + '       g   i     L; i  + '       g   i     R# ; i)zmust be a one-dimensionalmatch)r   N)r
   )r   zmust be a two-dimensionalrI   zmust not contain nanz both values must not be infinitez$left value must not exceed the rightr	   r$   r!   )pytestraises
ValueErrorr   r   nanrE   r   s   &r   test_invalid_constructor_args.TestCensoredData.test_invalid_constructor_args]   s0   ]]:-HIIYK0 J]]:-HIIyk* J]]:-HII	{+ J]]:-HII9+. J ]]:-CDDQN3 E]]:-CDDq"&&!n- E]]:-CDD2661~. E]]:-CDDArvv;A"78 E ]]:!CE EAq6Aq6BFFBFF3C"DEE ]]:!GI IAq6Aq6"23I I- JIIIIIIIIIII EDDDDDDDDDDDE EI I Isw   J;!KK#K7LL)L33!M 2MM.;K	K 	#K4	7L	L	L0	3M	M	M+	.M?	funcc                   \         P                  ! \        R R7      ;_uu_ 4        V! . RO.. RO4       RRR4       \         P                  ! \        RR7      ;_uu_ 4        V! . RO. RO.4       RRR4       \         P                  ! \        RR7      ;_uu_ 4        V! ^^\        P                  .. RO4       RRR4       \         P                  ! \        RR7      ;_uu_ 4        V! . RO. RO4       RRR4       R#   + '       g   i     L; i  + '       g   i     L; i  + '       g   i     Lr; i  + '       g   i     R# ; i)	z`x` must be one-dimensionalrQ   Nz"`censored` must be one-dimensionalz`x` must not containmust have the same lengthrS   r   r	   r	   r   r   r	   r	   )rT   rU   rV   r   rW   )r   r[   s   &&r   %test_invalid_left_right_censored_args6TestCensoredData.test_invalid_left_right_censored_argsx   s     ]]:!>@ @)i(@ ]]:!EG GYK(G ]]:-CDD!Q+ E]]:-HIIL) JI@ @G G EDIIIs/   C< DD"$D5<D	D	"D2	5E	c                   \         P                  ! \        R R7      ;_uu_ 4        \        P                  ! ^..^^.R7       RRR4       \         P                  ! \        RR7      ;_uu_ 4        \        P                  ! ^.^^..R7       RRR4       \         P                  ! \        RR7      ;_uu_ 4        \        P                  ! ^^\
        P                  .. RO4       RRR4       \         P                  ! \        RR7      ;_uu_ 4        \        P                  ! . RO. R	O4       RRR4       R#   + '       g   i     L; i  + '       g   i     L; i  + '       g   i     L; i  + '       g   i     R# ; i)
z`low` must be a one-dimensionalrQ   r4   Nz `high` must be a one-dimensionalz`low` must not containr]   r^   rS   r_   )rT   rU   rV   r   r;   r   rW   rX   s   &r   test_invalid_censored_args+TestCensoredData.test_invalid_censored_args   s    ]]:!BD D**uAq6BD ]]:!CE E**s1a&BE ]]:-EFF**Aq"&&>9E G]]:-HII**9lC JID DE E GFIIIs/   D4.E7+EE-4E	E	E*	-E>	c                    . ROp\        V4      pVP                  4       ^ 8X  g   Q h\        R.^
.^ ^..R7      pVP                  4       ^8X  g   Q hR# )r	   r"   )r   r
   r   NrS   )r   num_censored)r   r*   data1data2s   &   r   test_count_censored$TestCensoredData.test_count_censored   sV    Q!!#q(((RDQF8L!!#q(((r    N)__name__
__module____qualname____firstlineno__r   r,   r0   rB   rF   rJ   rN   rY   rT   markparametrizer   r/   r'   r`   rc   ri   __static_attributes____classdictcell__)__classdict__s   @r   r   r   	   s|     I 77)77$46 [[Vl&@&@&2&A&A&C D
*D
*
D) )r   r   )	rT   numpyr   numpy.testingr   r   scipy.statsr   r   rk   r   r   <module>rx      s      : $O) O)r   