§
    fŠtjéE  ã                   ó¤  — d dl Z d dlZd dlmZ d dlmZ d dlmZm	Z	m
Z
mZmZmZmZmZ d dlmZmZ d dlmZ e j        j        ZegZ edd¦  «        d	„ ¦   «         Z ej        d d
h¬¦  «        d„ ¦   «         Zd„ Z eej        ¦  «         G d„ d¦  «        ¦   «         Z edd¦  «        d„ ¦   «         Z ee¦  «         G d„ d¦  «        ¦   «         ZdS )é    N)Ústats)Ú_xp_searchsorted)Úxp_default_dtypeÚis_numpyÚis_torchÚis_jaxÚmake_xp_test_caseÚSCIPY_ARRAY_APIÚxp_sizeÚxp_copy)Úxp_assert_closeÚxp_assert_equal)Ú_apply_over_batch)Úxé   )Úpr   c                 ó¦  — |dk    r| t          j        | ¦  «                  } t          j        |¦  «        }|                     ¦   «         }d||<   |dk    rL| j        dk    rt          j        || d         ¦  «        n$t
          j                             | |¦  «        j        }nP| 	                    d¦  «        r$t          t          j        | ¦  «        ||¦  «        }nt          j        | ||¬¦  «        }t          j        |¦  «        }|dk    r7t          j        t          j        | ¦  «        ¦  «        rt           j        |d d …<   t           j        ||<   |S )	NÚomitç      à?úharrell-davisr   r   Úround©ÚmethodÚ	propagate)ÚnpÚisnanÚcopyÚsizeÚ	full_liker   ÚmstatsÚhdquantilesÚdataÚ
startswithÚwinsor_reference_1dÚsortÚquantileÚasarrayÚanyÚnan)r   r   Ú
nan_policyr   Úp_maskÚress         ú]/var/www/html/CA-Chatbot/venv/lib/python3.11/site-packages/scipy/stats/tests/test_quantile.pyÚquantile_reference_last_axisr.      s"  € à�VÒÐØ�rŒx˜‰{Œ{ˆlŒOˆÝŒX�a‰[Œ[€FØ	�Š‰Œ€AØ€A€f�IØ�Ò Ð à()¬°!ª¨�rŒ|˜A˜q œtÑ$Ô$Ð$Ý”L×,Ò,¨Q°Ñ2Ô2Ô7ð 	ˆà	×	Ò	˜7Ñ	#Ô	#ð /Ý!¥"¤'¨!¡*¤*¨a°Ñ8Ô8ˆˆåŒk˜!˜Q vÐ.Ñ.Ô.ˆå
Œ*�S‰/Œ/€CØ�[Ò Ð ¥R¤V­B¬H°Q©K¬KÑ%8Ô%8Ð Ý”ˆˆAˆAˆA‰å”&€Cˆ�KØ€Jó    é   )Úexcludedc                 óð  — t          | ¦  «        }|dk    rEt          |dk     rt          j        ||z  ¦  «        nt          j        ||z  dz
  ¦  «        ¦  «        }n•|dk    rEt          |dk     rt          j        ||z  ¦  «        nt          j        ||z  dz
  ¦  «        ¦  «        }nJ|dk    rDt          |dk     rt          j        ||z  ¦  «        nt          j        ||z  dz
  ¦  «        ¦  «        }| |         S )NÚround_nearestr   r   Úround_outwardÚround_inward)ÚlenÚintr   r   ÚfloorÚceil)Úyr   r   ÚnÚjs        r-   r$   r$   /   sâ   € õ 	ˆA‰Œ€AØ�Ò Ð Ý 1 s¢7 7•”˜˜Q™‘”�µ´¸¸Q¹À¹Ñ0CÔ0CÑDÔDˆˆØ	�?Ò	"Ð	"Ý 1 s¢7 7•”˜˜Q™‘”�µ´¸¸A¹À¹	Ñ0BÔ0BÑCÔCˆˆØ	�>Ò	!Ð	!Ý ! c¢' '•”˜˜A™‘”�­r¬x¸¸A¹À¹	Ñ/BÔ/BÑCÔCˆØˆQŒ4€Kr/   c                ó  — t          j        | |d¦  «        t          j        t          j        |¦  «        |d¦  «        }} t          | |||¦  «        }t          j        |d|¦  «        }|st          j        ||¬¦  «        }|S )Néÿÿÿÿ©Úaxis)r   ÚmoveaxisÚ
atleast_1dr.   Úsqueeze)r   r   r@   r*   Úkeepdimsr   r,   s          r-   Úquantile_referencerE   =   sx   € ÝŒ;�q˜$ Ñ#Ô#¥R¤[µ´¸qÑ1AÔ1AÀ4ÈÑ%LÔ%L€q€AÝ
& q¨!¨Z¸Ñ
@Ô
@€CÝ
Œ+�c˜2˜tÑ
$Ô
$€CØð )ÝŒj˜ 4Ð(Ñ(Ô(ˆØ€Jr/   c                   ó°  — e Zd Zd„ Zd„ Z edddg¬¦  «        ej                             dg d¢¦  «        ej                             d	g d
¢¦  «        ej                             dddg¦  «        d„ ¦   «         ¦   «         ¦   «         ¦   «         Z	ej         
                    d¦  «         eddddg¬¦  «        ej                             dddg¦  «        ej                             dddg¦  «        ej                             dg d¢¦  «        ej                             dddg¦  «        ej                             dg d¢¦  «        ej                             dddg¦  «        d„ ¦   «         ¦   «         ¦   «         ¦   «         ¦   «         ¦   «         ¦   «         ¦   «         Zd„ Zej                             dg dej        i fg d¢dddd ej        gej        dd!ej        ej        gi fg d¢g g i fej        d"ggdej        d"gdd#ifg g gd ej        d"ej        ¦  «        ddifg g gd ej        d$¦  «        ddd%œfg g gd ej        d&¦  «        ddd%œfg dd'g ej        d"ej        ¦  «        i f ej        dd(¦  «                             d)¦  «        dd*gggd+dd%œfdd"gd!d,ggg d-¢g d.¢gd+dd%œfg
¦  «        d/„ ¦   «         Zej                             dg d0¢¦  «        ej                             dddg¦  «        d1„ ¦   «         ¦   «         Zej                             dg d2¢¦  «        d3„ ¦   «         Zej                             d4ddg¦  «        d5„ ¦   «         Zej                             dg d6¢¦  «        d7„ ¦   «         Zej         
                    d¦  «        ej                             dg d6¢¦  «        ej                             d8d9d:g¦  «        d;„ ¦   «         ¦   «         ¦   «         Zd+S )<ÚTestQuantilec           	      ó2	  — |                      g d¢¦  «        }|                      d¦  «        }d}t          j        t          |¬¦  «        5  t	          j        |                      ddg¦  «        |¦  «         d d d ¦  «         n# 1 swxY w Y   t          j        t          ¦  «        5  t	          j        |                      ddg¦  «        |¦  «         d d d ¦  «         n# 1 swxY w Y   d	}t          j        t          |¬¦  «        5  t	          j        ||                      d
dg¦  «        ¦  «         d d d ¦  «         n# 1 swxY w Y   d}t          j        t          |¬¦  «        5  t	          j        |||                     ||j        ¦  «        ¬¦  «         d d d ¦  «         n# 1 swxY w Y   d}t          j        t          |¬¦  «        5  t	          j        ||d¬¦  «         d d d ¦  «         n# 1 swxY w Y   t          j        t          |¬¦  «        5  t	          j        ||d¬¦  «         d d d ¦  «         n# 1 swxY w Y   d}t          j        t          |¬¦  «        5  t	          j        ||d¬¦  «         d d d ¦  «         n# 1 swxY w Y   t          |¦  «        sgd}t          j        t          |¬¦  «        5  t	          j        |                      |j	        ddg¦  «        |d¬¦  «         d d d ¦  «         n# 1 swxY w Y   d}t          j        t          |¬¦  «        5  t	          j        ||d¬¦  «         d d d ¦  «         n# 1 swxY w Y   d}t          j        t          |¬¦  «        5  t	          j        |||d¬¦  «         d d d ¦  «         n# 1 swxY w Y   d}t          j        t          |¬¦  «        5  t	          j        |||d¬¦  «         d d d ¦  «         n# 1 swxY w Y   d}t          j        t          |¬¦  «        5  t	          j        ||d¬¦  «         d d d ¦  «         n# 1 swxY w Y   d }t          j        t          |¬¦  «        5  t	          j        ||                      dd!g¦  «        d¬¦  «         d d d ¦  «         d S # 1 swxY w Y   d S )"N©r   r0   é   r   z`x` must have real dtype.)ÚmatchTFy      ð?      ð?r0   z"`p` must have real floating dtype.r   r   z`weights` must have real dtype.)Úweightsz"`axis` must be an integer or None.r?   )r   r>   z7`axis` is not compatible with the shapes of the inputs.zThe input contains nan valuesÚraise)r*   z`method` must be one of...za duckr   z4`method='harrell-davis'` does not support `weights`.r   ©rL   r   z4`method='round_nearest'` does not support `weights`.r3   z/If specified, `keepdims` must be True or False.é*   )rD   zD`keepdims` may be False only if the length of `p` along `axis` is 1.ç333333ã?)
r'   ÚpytestÚraisesÚ
ValueErrorr   r&   ÚastypeÚ	complex64r   r)   )ÚselfÚxpr   r   Úmessages        r-   Útest_input_validationz"TestQuantile.test_input_validationI   sÍ  € Ø�JŠJ�y�y�yÑ!Ô!ˆØ�JŠJ�s‰OŒOˆà-ˆÝŒ]�:¨WÐ5Ñ5Ô5ð 	9ð 	9ÝŒN˜2Ÿ:š: t¨U mÑ4Ô4°aÑ8Ô8Ð8ð	9ð 	9ð 	9ñ 	9ô 	9ð 	9ð 	9ð 	9ð 	9ð 	9ð 	9øøøð 	9ð 	9ð 	9ð 	9åŒ]�:Ñ&Ô&ð 	5ð 	5ÝŒN˜2Ÿ:š: t¨Q iÑ0Ô0°!Ñ4Ô4Ð4ð	5ð 	5ð 	5ñ 	5ô 	5ð 	5ð 	5ð 	5ð 	5ð 	5ð 	5øøøð 	5ð 	5ð 	5ð 	5ð 7ˆÝŒ]�:¨WÐ5Ñ5Ô5ð 	2ð 	2ÝŒN˜1˜bŸjšj¨!¨Q¨Ñ0Ô0Ñ1Ô1Ð1ð	2ð 	2ð 	2ñ 	2ô 	2ð 	2ð 	2ð 	2ð 	2ð 	2ð 	2øøøð 	2ð 	2ð 	2ð 	2ð 4ˆÝŒ]�:¨WÐ5Ñ5Ô5ð 	Eð 	EÝŒN˜1˜a¨¯ª°1°b´lÑ)CÔ)CÐDÑDÔDÐDð	Eð 	Eð 	Eñ 	Eô 	Eð 	Eð 	Eð 	Eð 	Eð 	Eð 	Eøøøð 	Eð 	Eð 	Eð 	Eð 7ˆÝŒ]�:¨WÐ5Ñ5Ô5ð 	+ð 	+ÝŒN˜1˜a cÐ*Ñ*Ô*Ð*ð	+ð 	+ð 	+ñ 	+ô 	+ð 	+ð 	+ð 	+ð 	+ð 	+ð 	+øøøð 	+ð 	+ð 	+ð 	+åŒ]�:¨WÐ5Ñ5Ô5ð 	/ð 	/ÝŒN˜1˜a gÐ.Ñ.Ô.Ð.ð	/ð 	/ð 	/ñ 	/ô 	/ð 	/ð 	/ð 	/ð 	/ð 	/ð 	/øøøð 	/ð 	/ð 	/ð 	/ð LˆÝŒ]�:¨WÐ5Ñ5Ô5ð 	)ð 	)ÝŒN˜1˜a aÐ(Ñ(Ô(Ð(ð	)ð 	)ð 	)ñ 	)ô 	)ð 	)ð 	)ð 	)ð 	)ð 	)ð 	)øøøð 	)ð 	)ð 	)ð 	)õ �b‰zŒzð 	RØ5ˆGÝ”�z°Ð9Ñ9Ô9ð Rð RÝ”˜rŸzšz¨2¬6°1°a¨.Ñ9Ô9¸1ÈÐQÑQÔQÐQðRð Rð Rñ Rô Rð Rð Rð Rð Rð Rð Røøøð Rð Rð Rð Rð /ˆÝŒ]�:¨WÐ5Ñ5Ô5ð 	2ð 	2ÝŒN˜1˜a¨Ð1Ñ1Ô1Ð1ð	2ð 	2ð 	2ñ 	2ô 	2ð 	2ð 	2ð 	2ð 	2ð 	2ð 	2øøøð 	2ð 	2ð 	2ð 	2ð IˆÝŒ]�:¨WÐ5Ñ5Ô5ð 	Dð 	DÝŒN˜1˜a¨°?ÐCÑCÔCÐCð	Dð 	Dð 	Dñ 	Dô 	Dð 	Dð 	Dð 	Dð 	Dð 	Dð 	Døøøð 	Dð 	Dð 	Dð 	Dð IˆÝŒ]�:¨WÐ5Ñ5Ô5ð 	Dð 	DÝŒN˜1˜a¨°?ÐCÑCÔCÐCð	Dð 	Dð 	Dñ 	Dô 	Dð 	Dð 	Dð 	Dð 	Dð 	Dð 	Døøøð 	Dð 	Dð 	Dð 	Dð DˆÝŒ]�:¨WÐ5Ñ5Ô5ð 	.ð 	.ÝŒN˜1˜a¨"Ð-Ñ-Ô-Ð-ð	.ð 	.ð 	.ñ 	.ô 	.ð 	.ð 	.ð 	.ð 	.ð 	.ð 	.øøøð 	.ð 	.ð 	.ð 	.ð YˆÝŒ]�:¨WÐ5Ñ5Ô5ð 	Fð 	FÝŒN˜1˜bŸjšj¨#¨s¨Ñ4Ô4¸uÐEÑEÔEÐEð	Fð 	Fð 	Fñ 	Fô 	Fð 	Fð 	Fð 	Fð 	Fð 	Fð 	Fð 	Føøøð 	Fð 	Fð 	Fð 	Fð 	Fð 	Fsê   Á
+BÂBÂBÂ%+CÃC Ã#C Ä+D;Ä;D?ÅD?Å#1F Æ F$Æ'F$ÇG,Ç,G0Ç3G0ÈH6È6H:È=H:ÉJÊJÊ	JÊ93K8Ë8K<Ë?K<Ì MÍMÍMÍ,NÎNÎNÎ9OÏO"Ï%O"ÐP*Ð*P.Ð1P.Ñ-RÒRÒRc                 ó¾  — t          j        ||d¦  «        }|j        }t          j        |¦  «        }|                     d|j        d         t           j        ¬¦  «        }g }g }|D ]T}|                     |¦  «        }	|                     t          j	        ||	¦  «        ¦  «         |                     |	¦  «         ŒUt          j
        |¦  «        t          j
        |¦  «        }}|dk     r,t          j        |d¬¦  «        t          j        |d¬¦  «        }}t          j        |d|¦  «        t          j        |d|¦  «        }}t          j        ||j        ¬¦  «        }||fS )Nr>   é
   )r   Údtyper0   r   r?   ©r\   )r   ÚswapaxesÚndimÚ
atleast_2dÚintegersÚshapeÚint32ÚpermutedÚappendÚrepeatÚstackrC   r'   r\   )
rV   r   r@   Úrngr_   ÚcountsÚx_reprL   Úx_Úcounts_s
             r-   Ú_get_weights_x_repzTestQuantile._get_weights_x_rep   s9  € ÝŒK˜˜4 Ñ$Ô$ˆØŒvˆÝŒM˜!ÑÔˆØ—’˜b q¤w¨r¤{½"¼(�ÑCÔCˆØˆØˆØð 	$ð 	$ˆBØ—l’l 6Ñ*Ô*ˆGØ�LŠL�œ 2 wÑ/Ô/Ñ0Ô0Ð0Ø�NŠN˜7Ñ#Ô#Ð#Ð#Ýœ %™œ­"¬(°7Ñ*;Ô*;ˆwˆØ�!Š8ˆ8ÝœZ¨°AÐ6Ñ6Ô6½¼
À7ÐQRÐ8SÑ8SÔ8S�7ˆEÝœ U¨B°Ñ5Ô5µr´{À7ÈBÐPTÑ7UÔ7UˆwˆÝ”*˜W¨A¬GÐ4Ñ4Ô4ˆØ˜ˆ~Ðr/   TzPyTorch doesn't have `betainc`.Úcupy)Úcpu_onlyÚreasonÚ
exceptionsr   )Úinverted_cdfÚaveraged_inverted_cdfÚclosest_observationÚhazenÚinterpolated_inverted_cdfÚlinearÚmedian_unbiasedÚnormal_unbiasedÚweibullr   r3   r4   r5   Ú_lowerÚ_higherÚ	_midpointÚ_nearestzshape_x, shape_p, axis))r[   Nr>   )r[   r[   r>   )r[   )r0   rJ   r>   ))r[   r0   Nr   rL   Fc           	      ó  — |rD|                      d¦  «        s|                      d¦  «        s|dk    rt          j        d¦  «         t          |¦  «        }t          j                             d¦  «        }|                     |¬¦  «        }	|                     |¬¦  «        }
|r|                      |	||¦  «        \  }}nd |	}}t          ||
|                      d¦  «        r
|dd …         n||d|d u¬	¦  «        }| 	                    |	|¬
¦  «        | 	                    |
|¬
¦  «        }
}	|€|n| 	                    ||¬
¦  «        }t          j        |	|
|||¬¦  «        }t          || 	                    ||¬
¦  «        ¦  «         d S )NÚ_r   r   z4`weights` not supported by private (legacy) methods.ì   ìlv_š©r   r   r   )r   r@   r*   rD   r]   )r   rL   r@   )r#   rQ   Úskipr   r   ÚrandomÚdefault_rngrm   rE   r'   r   r&   r   )rV   r   Úshape_xÚshape_pr@   rL   rW   r\   rh   r   r   rj   Úrefr,   s                 r-   Útest_against_referencez#TestQuantile.test_against_reference’   s�  € ð ð 	P˜×)Ò)¨#Ñ.Ô.ð 	P°&×2CÒ2CÀGÑ2LÔ2Lð 	PØ! ?Ò2Ð2ÝŒKÐNÑOÔOÐOÝ  Ñ$Ô$ˆÝŒi×#Ò#Ð$5Ñ6Ô6ˆØ�JŠJ˜GˆJÑ$Ô$ˆØ�JŠJ˜GˆJÑ$Ô$ˆàð 	%Ø!×4Ò4°Q¸¸cÑBÔB‰NˆG�U�Uà! 1�UˆGå Ø�1¨6×+<Ò+<¸SÑ+AÔ+AÐM˜V A B BœZ˜ZÀvØ +¸ÀtÐ8KðMñ Mô Mˆð �zŠz˜! 5ˆzÑ)Ô)¨2¯:ª:°a¸u¨:Ñ+EÔ+Eˆ1ˆØ$˜_�'�'°"·*²*¸WÈE°*Ñ2RÔ2RˆÝŒn˜Q ¨&¸'ÈÐMÑMÔMˆå˜˜RŸZšZ¨°5˜ZÑ9Ô9Ñ:Ô:Ð:Ð:Ð:r/   z%ignore:torch.searchsorted:UserWarningz	jax.numpyr@   r   r   rD   r*   )r   r   Úmarrayr\   Úfloat32Úfloat64)rw   r   r3   c                 ó   — t          |¦  «        r|dk    rt          j        d¦  «         |r|dv rt          j        d¦  «         t          j                             d¦  «        }d}	|                     |	¬¦  «                             |¦  «        }
|                     |	¬¦  «                             |¦  «        }|                     |	¬¦  «        dk    }t          j        |¦  «        sJ ‚t          j        |
|<   |st          j	        ||d	¬
¦  «        }t          |¦  «        r|                     ¦   «         }d|d<   d|d<   t          ||¦  «        }|r4|                      |
||¦  «        \  }}|€|n|                     |¦  «        }nd |
}}|dk    �r t          st          j        d¦  «         |�t          j        d¦  «         t          j        d¦  «        }t#          |||¬¦  «        }|                     |¦  «        }|                     |
|¬¦  «        }|€|n|                     |¦  «        }t'          j        ||                     |¦  «        fd|i|¤Ž}t+          ||fddi|¤Ž}t-          |j        |                     ||¬¦  «        ¦  «         d S t#          ||||¬¦  «        }t'          j        |                     |
¦  «        |                     |¦  «        fd|i|¤Ž}t+          ||fi |¤Ž}t-          ||                     ||¬¦  «        ¦  «         d S )NrŠ   z(`marray` currently incompatible with JAX>   r3   r   z'These methods don't yet support weightsr�   )é   é   r‚   gš™™™™™é?T©r@   rD   ç        r   ç      ð?éþÿÿÿz-MArray is only available if SCIPY_ARRAY_API=1z)MArray is not yet compatible with weights)r@   rD   r   )ÚmaskrL   r*   r   r]   )r@   rD   r*   r   )r   rQ   rƒ   r   r„   r…   rT   r(   r)   Úmeanr   ÚravelÚgetattrrm   r'   r
   ÚimportorskipÚdictÚ_get_namespacer   r&   rE   r   r"   )rV   r@   rD   r*   r\   r   rL   rW   rh   rb   r   r   r”   Úp0rj   rŠ   ÚkwargsÚmxpÚx_mpr,   rˆ   s                        r-   Útest_against_reference_2z%TestQuantile.test_against_reference_2¶   s!  € õ �"‰:Œ:ð 	D˜*¨Ò0Ð0ÝŒKÐBÑCÔCÐCØð 	C�vÐ!CÐCÐCÝŒKÐAÑBÔBÐBÝŒi×#Ò#Ð$5Ñ6Ô6ˆØˆØ�JŠJ˜EˆJÑ"Ô"×)Ò)¨%Ñ0Ô0ˆØ�JŠJ˜EˆJÑ"Ô"×)Ò)¨%Ñ0Ô0ˆØ�zŠz˜uˆzÑ%Ô%¨Ò+ˆÝŒv�d‰|Œ|ÐÐˆ|Ý”&ˆˆ$‰Øð 	5Ý”˜ ¨tÐ4Ñ4Ô4ˆAõ
 �B‰<Œ<ð 	Ø—’‘”ˆBØˆBˆq‰EØˆBˆr‰Få˜˜EÑ"Ô"ˆàð 	%Ø!×4Ò4°Q¸¸cÑBÔB‰NˆG�UØ!( �g�g°b·j²jÀÑ6IÔ6IˆGˆGà! 1�UˆGà˜Ò!Ñ!Ý"ð MÝ”ÐKÑLÔLÐLØÐ"Ý”ÐGÑHÔHÐHÝÔ(¨Ñ2Ô2ˆFÝ˜t¨h¸vÐFÑFÔFˆFØ×'Ò'¨Ñ+Ô+ˆCØ—;’;˜q t�;Ñ,Ô,ˆDØ!( �g�g°c·k²kÀ'Ñ6JÔ6JˆGÝ”.  s§{¢{°1¡~¤~ÐQÐQ¸wÐQÈ&ÐQÐQˆCÝ$ U¨AÐKÐK¸&ÐKÀFÐKÐKˆCÝ˜CœH b§j¢j°¸E jÑ&BÔ&BÑCÔCÐCØˆFå˜4¨(Ø!+°Fð<ñ <ô <ˆåŒn˜RŸZšZ¨™]œ]¨B¯JªJ°q©M¬MÐUÐUÀ7ÐUÈfÐUÐUˆÝ  ¨Ð4Ð4¨VÐ4Ð4ˆÝ˜˜RŸZšZ¨°5˜ZÑ9Ô9Ñ:Ô:Ð:Ð:Ð:r/   c                 ó˜   — t          j        |                     d|j        ¬¦  «        d¦  «        }|j        t          |¦  «        k    sJ ‚d S )Nr[   r]   r   )r   r&   ÚarangeÚint64r\   r   )rV   rW   r,   s      r-   Útest_integer_input_output_dtypez,TestQuantile.test_integer_input_output_dtypeõ   sG   € ÝŒn˜RŸYšY r°´˜YÑ:Ô:¸CÑ@Ô@ˆØŒyÕ,¨RÑ0Ô0Ò0Ð0Ð0Ð0Ð0Ð0r/   zx, p, ref, kwargsr   rI   r>   g      ø?rJ   r0   r   )r   r�   )r   r   rP   é   )rJ   rJ   rJ   g      ,@Né   )g      Ð?r   g      è?)g      ü?g      @g      
@c                 óì   — t          |¦  «        }|                     |¦  «        |                     |¦  «        |                     ||¬¦  «        }}}t          j        ||fi |¤Ž}t	          ||¦  «         d S )Nr]   )r   r'   r   r&   r   )rV   r   r   rˆ   rœ   rW   Údefault_dtyper,   s           r-   Útest_edge_caseszTestQuantile.test_edge_casesù   sq   € õ$ )¨Ñ,Ô,ˆØ—J’J˜q‘M”M 2§:¢:¨a¡=¤=°"·*²*¸SÈ°*Ñ2VÔ2Vˆcˆ1ˆÝŒn˜Q Ð,Ð, VÐ,Ð,ˆÝ˜˜SÑ!Ô!Ð!Ð!Ð!r/   )r   r   r0   c                 ó  — g d¢}|                      ¦   «         }|rd||<   n|                     |¦  «         t          j        |                     t          |¦  «        ¦  «        d||¬¦  «        }|j        t          |¦  «        k    sJ ‚d S )N)rJ   r¥   r   r   r   r�   )r   Úpopr   r&   ÚzerosÚtuplerb   )rV   r@   rD   rW   rb   Ú	out_shaper,   s          r-   Útest_size_0zTestQuantile.test_size_0  sŠ   € ð �	�	ˆØ—J’J‘L”Lˆ	Øð 	 ØˆI�d‰OˆOà�MŠM˜$ÑÔÐÝŒn˜RŸXšX¥e¨E¡l¤lÑ3Ô3°S¸tÈhÐWÑWÔWˆØŒy�E )Ñ,Ô,Ò,Ð,Ð,Ð,Ð,Ð,r/   )rr   rs   rt   r{   r|   r}   r~   c                 ó   — |dk    r$t           j        dk     rt          j        d¦  «         t          j        dt           j        ¬¦  «        }t          j        ddd¦  «        }t          j        |                     |¦  «        |                     |¦  «        |¬	¦  «        }t          j        ||| 	                    d
¦  «        r
|dd …         n|¬	¦  «        }t          ||                     ||j        ¬¦  «        ¦  «         d S )Nrt   z2.0.1z5Bug in np.quantile (numpy/numpy#26656) fixed in 2.0.1g       @r]   r   g     €ð?g       ?r   r€   r   )r   Ú__version__rQ   rƒ   r¡   rŒ   r   r&   r'   r#   r   )rV   r   rW   r   r   r,   rˆ   s          r-   Útest_transitionzTestQuantile.test_transition  sÞ   € ð Ð*Ò*Ð*­r¬~ÀÒ/GÐ/GÝŒKÐOÑPÔPÐPÝŒI�b¥¤
Ð+Ñ+Ô+ˆÝŒI�a˜ 'Ñ*Ô*ˆÝŒn˜RŸZšZ¨™]œ]¨B¯JªJ°q©M¬MÀ&ÐIÑIÔIˆÝŒk˜!˜Q°V×5FÒ5FÀsÑ5KÔ5KÐ'W v¨a¨b¨b¤z zÐQWÐXÑXÔXˆÝ˜˜RŸZšZ¨°2´:˜ZÑ>Ô>Ñ?Ô?Ð?Ð?Ð?r/   Úzero_weightsc           	      ó„  — t          |¦  «        r|j        dk     rt          j        d¦  «         t	          |¦  «        }t
          j                             d¦  «        }d}|                     d¬¦  «        }|                     d¬¦  «        }|r	d||dk     <   t          j        d	d
d¦  «        }t          j
        |                     ||¬¦  «        |                     ||¬¦  «        ||                     ||¬¦  «        ¬¦  «        }	t          j
        ||||¬¦  «        }
t          |	|                     |
|¬¦  «        ¦  «         d S )Nz2.0z'`weights` not supported by NumPy < 2.0.ì   ¢z-\"}	rr   éd   r‚   r   r   r‘   r’   i,  r]   )r   rL   )r   r°   rQ   rƒ   r   r   r„   r…   Úlinspacer   r&   r'   r   )rV   r²   rW   r\   rh   r   r   rL   r   r,   rˆ   s              r-   Útest_weights_against_numpyz'TestQuantile.test_weights_against_numpy*  s5  € å�B‰<Œ<ð 	C˜BœN¨UÒ2Ð2ÝŒKÐAÑBÔBÐBÝ  Ñ$Ô$ˆÝŒi×#Ò#Ð$5Ñ6Ô6ˆØˆØ�JŠJ˜CˆJÑ Ô ˆØ—*’* #�*Ñ&Ô&ˆØð 	'Ø%&ˆG�G˜c’MÑ"ÝŒK˜˜B Ñ$Ô$ˆÝŒn˜RŸZšZ¨°˜ZÑ7Ô7¸¿ºÀAÈU¸Ñ9SÔ9SØ$*°B·J²J¸wÈe°JÑ4TÔ4TðVñ Vô VˆåŒk˜!˜Q v°wÐ?Ñ?Ô?ˆÝ˜˜RŸZšZ¨°5˜ZÑ9Ô9Ñ:Ô:Ð:Ð:Ð:r/   )	rr   rs   rt   ru   rv   rw   rx   ry   rz   c                 ó  — t           j                             d¦  «        }d}|                     |                     |¬¦  «        ¦  «        }t	          |¦  «        }|                     |                     |¬¦  «        ¦  «        }|                     |                     |¬¦  «        dk     ¦  «        }|                     |                     |¬¦  «        ¦  «        }	|                     |d|	¦  «        }	t          j        |||	|¬¦  «        }
t          j        ||          ||	|          |¬¦  «        }t          |
|¦  «         t          ||¦  «         d}|                     |                     |¬¦  «        ¦  «        }t	          |¦  «        }|                     |                     |¬¦  «        ¦  «        }|                     |                     |¬¦  «        dk     ¦  «        }|                     |                     |¬¦  «        ¦  «        }	|                     ||j
        |¦  «        }|                     |d|	¦  «        }t          j        ||||d¬	¦  «        }
t          j        |||	d
|d¬¦  «        }t          |
|¦  «         t          ||¦  «         d S )Nr´   rµ   r‚   çš™™™™™¹?r‘   rN   )rŽ   rµ   r>   )rL   r   r@   r   )rL   r*   r   r@   )r   r„   r…   r'   r   Úwherer   r&   r   r   r)   )rV   r   rW   rh   r;   r   Úx0r   Úi_zerorL   r,   rˆ   rb   Úx_nannedÚweights_zeroeds                  r-   Útest_zero_weightszTestQuantile.test_zero_weights;  sD  € õ
 Œi×#Ò#Ð$5Ñ6Ô6ˆð ˆØ�JŠJ�s—z’z q�zÑ)Ô)Ñ*Ô*ˆÝ�Q‰ZŒZˆØ�JŠJ�s—z’z q�zÑ)Ô)Ñ*Ô*ˆØ—’˜CŸJšJ¨A˜JÑ.Ô.°Ò4Ñ5Ô5ˆØ—*’*˜SŸZšZ¨Q˜ZÑ/Ô/Ñ0Ô0ˆØ—(’(˜6 2 wÑ/Ô/ˆÝŒn˜Q ¨7¸6ÐBÑBÔBˆÝŒn˜Q ˜wœZ¨°G¸V¸GÔ4DÈVÐTÑTÔTˆÝ˜˜SÑ!Ô!Ð!Ý˜˜2ÑÔÐð ˆØ�JŠJ�s—z’z u�zÑ-Ô-Ñ.Ô.ˆÝ�Q‰ZŒZˆØ�JŠJ�s—z’z u�zÑ-Ô-Ñ.Ô.ˆØ—’˜CŸJšJ¨E˜JÑ2Ô2°SÒ8Ñ9Ô9ˆØ—*’*˜SŸZšZ¨U˜ZÑ3Ô3Ñ4Ô4ˆØ—8’8˜F B¤F¨AÑ.Ô.ˆØŸš &¨"¨gÑ6Ô6ˆÝŒn˜Q ¨>À&ÈrÐRÑRÔRˆÝŒn˜X q°'Ø(.°vÀBðHñ Hô Hˆå˜˜SÑ!Ô!Ð!Ý˜˜2ÑÔÐÐÐr/   rb   é2   )rÀ   rJ   c                 ó�  — t           j                             d¦  «        }|                     |                     |¬¦  «        ¦  «        }|                     |                     |¬¦  «        ¦  «        }|                     |¦  «        }t          j        ||||¬¦  «        }t          j        |||¬¦  «        }	t          ||	¦  «         d S )Nl   ¢.ý,+r‚   rN   r   )r   r„   r…   r'   Ú	ones_liker   r&   r   )
rV   r   rb   rW   rh   r   r   rL   r,   rˆ   s
             r-   Útest_unity_weightszTestQuantile.test_unity_weights^  s«   € õ Œi×#Ò#Ð$5Ñ6Ô6ˆØ�JŠJ�s—z’z u�zÑ-Ô-Ñ.Ô.ˆØ�JŠJ�s—z’z u�zÑ-Ô-Ñ.Ô.ˆØ—,’,˜q‘/”/ˆÝŒn˜Q ¨7¸6ÐBÑBÔBˆÝŒn˜Q ¨&Ð1Ñ1Ô1ˆÝ˜˜SÑ!Ô!Ð!Ð!Ð!r/   )Ú__name__Ú
__module__Ú__qualname__rY   rm   Úskip_xp_backendsrQ   ÚmarkÚparametrizer‰   ÚfilterwarningsrŸ   r£   r   r)   Úfullr«   r¡   Úreshaper¨   r®   r±   r·   r¿   rÃ   © r/   r-   rG   rG   F   sQ  € € € € € ð3Fð 3Fð 3Fðlð ð ð& Ð˜tÐ,MØ"( ð+ñ +ô +à„[×Ò˜Xð
8ð 
8ð 
8ñ9ô 9ð „[×ÒÐ5ØLÐLÐLñNô Nà„[×Ò˜Y¨°¨Ñ6Ô6ð;ð ;ñ 7Ô6ñNô Nñ9ô 9ñ+ô +ð;ð2 „[×ÒÐ GÑHÔHØÐ˜tÐ,MØ"(¨+Ð!6ð8ñ 8ô 8à„[×Ò˜V a¨ VÑ,Ô,Ø„[×Ò˜Z¨%°¨Ñ7Ô7Ø„[×Ò˜\Ð+JÐ+JÐ+JÑKÔKØ„[×Ò˜W y°)Ð&<Ñ=Ô=Ø„[×Ò˜XÐ'SÐ'SÐ'SÑTÔTØ„[×Ò˜Y¨°¨Ñ6Ô6ð4;ð 4;ñ 7Ô6ñ UÔTñ >Ô=ñ LÔKñ 8Ô7ñ -Ô,ñ8ô 8ñ IÔHð4;ðl1ð 1ð 1ð „[×ÒÐ0Ø
ˆc�2”6˜2Ð	Ø
ˆ)ˆ)�b˜!˜Q  R¤VÐ,¨r¬v°q¸!¸R¼VÀRÄVÐ.LÈbÐ	QØ
ˆ)ˆ)�R˜˜RÐ	 ØŒF�Aˆ;ˆ-˜˜rœv q˜k¨L¸&Ð+AÐ	BØˆrˆ(�C˜˜œ  B¤FÑ+Ô+¨f°b¨\Ð	:Øˆrˆ(�C˜˜œ $™œ°!ÀÐ)GÐ)GÐ	HØˆrˆ(�C˜˜œ &Ñ)Ô)°AÀ4Ð+HÐ+HÐ	IØ
ˆs�Cˆj˜'˜"œ' ! R¤VÑ,Ô,¨bÐ	1Ø
ˆ"Œ)�A�rÑ
Ô
×
"Ò
" 9Ñ
-Ô
-¨s°s°e°W°IØ TÐ
*Ð
*ð
,àˆqˆ6�A�q�6Ð
Ð-Ð-Ð-Ð0AÐ0AÐ0AÐ/BØ TÐ
*Ð
*ð
,ð	ñô ð""ð "ñ#ô ð""ð „[×Ò˜V Y Y YÑ/Ô/Ø„[×Ò˜Z¨%°¨Ñ7Ô7ð-ð -ñ 8Ô7ñ 0Ô/ð-ð „[×Ò˜Xð	7ð 	7ð 	7ñ8ô 8ð	@ð 	@ñ8ô 8ð	@ð „[×Ò˜^¨e°T¨]Ñ;Ô;ð;ð ;ñ <Ô;ð;ð  „[×Ò˜Xð	ð 	ð 	ñô ðð ñ	ô ðð> „[×ÒÐ GÑHÔHØ„[×Ò˜Xð	ð 	ð 	ñô ð „[×Ò˜W r¨7 mÑ4Ô4ð"ð "ñ 5Ô4ñ	ô ñ IÔHð"ð "ð "r/   rG   )Úar   )Úvr   c                 ó0   — t          j        | ||¬¦  «        S )N©Úside)r   Úsearchsorted)rÎ   rÏ   rÒ   s      r-   Únp_searchsortedrÔ   o  s   € åŒ?˜1˜a dÐ+Ñ+Ô+Ð+r/   c            	       ór  — e Zd Zej                             dddg¦  «        ej                             dddg¦  «        ej                             dg d¢¦  «        ej                             d	ddg¦  «        ej                             d
ddg¦  «        d„ ¦   «         ¦   «         ¦   «         ¦   «         ¦   «         ZdS )ÚTest_XPSearchsortedrÒ   ÚleftÚrightÚtiesFTrb   )r   r   r0   r[   é   iè  i'  )r0   r   )r   r0   )r0   r[   )r0   rJ   rÚ   Únans_xÚinfs_xc                 óD  — |r#t          |¦  «        rt          j        d¦  «         t          j                             d¦  «        }|r|                     d|¬¦  «        }n|                     |¦  «        }t          j        |t          j        ¬¦  «        }t          j	        |t          j
        ¦  «        }	t          j	        |t          j
         ¦  «        }
t          j        t          j
         t          j
        t          j        g¦  «        }t          j        ||j        d d…         dz   ¦  «        }|                     t          j        |
||	|fd¬¦  «        d¬¦  «        }|r(|                     |¦  «        d	k     }t          j        ||<   |rQ|                     |¦  «        d	k     }t          j
         ||<   |                     |¦  «        d
k    }t          j
        ||<   t          j        |d¬¦  «        }t          j        |t          j        ¬¦  «        t          j        |t          j        ¬¦  «        }}|                     d¦  «        j        }t'          |¦  «        dk    rP|j        dk    rE|j        d         dk    r4|                     |j        d d…         |j        d         fz   |¬¦  «        }n'|                     t-          |||¬¦  «        |¬¦  «        }|                     |¦  «        |                     |¦  «        }}t/          |||¬¦  «        }t1          ||¦  «         d S )Nztorch sorts NaNs differentlyl   %Dègº óhrŽ   r‚   r]   r>   )rJ   r?   r¹   gÍÌÌÌÌÌì?r   r   rÑ   )r   rQ   rƒ   r   r„   r…   ra   r'   r‹   Ú	nextafterÚinfr)   Úbroadcast_torb   rd   Úconcatenater%   rŒ   r\   r   r_   ÚemptyrÔ   r   r   )rV   rÒ   rÙ   rb   rÛ   rÜ   rW   rh   r   ÚxrÚxlrk   r:   r”   Úxp_default_intrˆ   r,   s                    r-   Útest_ndzTest_XPSearchsorted.test_ndv  s   € ð ð 	8•h˜r‘l”lð 	8ÝŒKÐ6Ñ7Ô7Ð7ÝŒi×#Ò#Ð$6Ñ7Ô7ˆØð 	"Ø—’˜Q U�Ñ+Ô+ˆAˆAà—
’
˜5Ñ!Ô!ˆAåŒJ�q¥¤
Ð+Ñ+Ô+ˆÝŒ\˜!�RœVÑ$Ô$ˆÝŒ\˜!�bœf˜WÑ%Ô%ˆÝŒZ�"œ&˜¥"¤&­"¬&Ð1Ñ2Ô2ˆÝŒ_˜R ¤¨¨"¨¤°Ñ!4Ñ5Ô5ˆØ�LŠL�œ¨¨Q°°B¨¸bÐAÑAÔAÈˆLÑKÔKˆØð 	Ø—:’:˜eÑ$Ô$ sÒ*ˆDÝ”fˆAˆd‰GØð 	Ø—:’:˜eÑ$Ô$ sÒ*ˆDÝ”v�gˆAˆd‰GØ—:’:˜eÑ$Ô$ sÒ*ˆDÝ”fˆAˆd‰GÝŒG�A˜BÐÑÔˆÝŒz˜!¥2¤:Ð.Ñ.Ô.µ´
¸1ÅBÄJÐ0OÑ0OÔ0Oˆ1ˆØŸš A™œÔ,ˆÝ�1‰:Œ:˜Š?ˆ?˜qœv¨šz˜z¨a¬g°b¬k¸QÒ.>Ð.>Ø—(’(˜1œ7 3 B 3œ<¨1¬7°2¬;¨.Ñ8À�(ÑOÔOˆCˆCà—*’*�_¨Q°¸Ð=Ñ=Ô=À^�*ÑTÔTˆCØ�zŠz˜!‰}Œ}˜bŸjšj¨™mœmˆ1ˆÝ˜q !¨$Ð/Ñ/Ô/ˆÝ˜˜SÑ!Ô!Ð!Ð!Ð!r/   N)rÄ   rÅ   rÆ   rQ   rÈ   rÉ   ræ   rÍ   r/   r-   rÖ   rÖ   t  sß   € € € € € à„[×Ò˜V f¨gÐ%6Ñ7Ô7Ø„[×Ò˜V e¨T ]Ñ3Ô3Ø„[×Ò˜Wð 'Lð 'Lð 'Lñ Mô Mà„[×Ò˜X¨¨t }Ñ5Ô5Ø„[×Ò˜X¨¨t }Ñ5Ô5ð "ð  "ñ 6Ô5ñ 6Ô5ñMô Mñ 4Ô3ñ 8Ô7ð "ð  "ð  "r/   rÖ   ) rQ   Únumpyr   Úscipyr   Úscipy.stats._quantiler   Úscipy._lib._array_apir   r   r   r   r	   r
   r   r   Úscipy._lib._array_api_no_0dr   r   Úscipy._lib._utilr   rÈ   rÇ   Úlazy_xp_modulesr.   Ú	vectorizer$   rE   r&   rG   rÔ   rÖ   rÍ   r/   r-   ú<module>rï      s  ðØ €€€Ø Ð Ð Ð à Ð Ð Ð Ð Ð Ø 2Ð 2Ð 2Ð 2Ð 2Ð 2ð	ð 	ð 	ð 	ð 	ð 	ð 	ð 	ð 	ð 	ð 	ð 	ð 	ð 	ð 	ð 	ð 	ð 	ð 	ð 	ð IÐ HÐ HÐ HÐ HÐ HÐ HÐ HØ .Ð .Ð .Ð .Ð .Ð .à”;Ô/Ð à�'€àÐ�8˜XÑ&Ô&ðð ñ 'Ô&ðð. €„˜˜1�vÐÑÔð
ð 
ñ Ôð
ðð ð ð Ð�5”>Ñ"Ô"ðe"ð e"ð e"ð e"ð e"ñ e"ô e"ñ #Ô"ðe"ðP	 Ð�8˜XÑ&Ô&ð,ð ,ñ 'Ô&ð,ð ÐÐ#Ñ$Ô$ð'"ð '"ð '"ð '"ð '"ñ '"ô '"ñ %Ô$ð'"ð '"ð '"r/   