Ë
    âQ(h¦$  ã                   ó”   — d dl Zd dlmZ ddlmZmZmZ ddlm	Z	 ddl
mZ ddlmZ d dlmZmZ  G d	„ d
«      Zd„ Zdd„Zd„ Z	 	 dd„Zy)é    N)Ústatsé   )Ú_get_pvalueÚ	_rankdataÚ_SimpleNormal)Ú
_morestats)Ú_broadcast_arrays)Ú_get_wilcoxon_distr)Ú
_lazywhereÚ_get_nanc                   óB   — e Zd Zd„ Zd„ Zd„ Zd„ Zd„ Zd„ Zd„ Z	d„ Z
d	„ Zy
)ÚWilcoxonDistributionc                 óÔ   — t        j                  |«      j                  t        d¬«      }|| _        t        j
                  |«      D �ci c]  }|t        |«      “Œ c}| _        y c c}w )NF©Úcopy)ÚnpÚasarrayÚastypeÚintÚnÚuniquer
   Ú_dists)Úselfr   Únis      úS/var/www/skyplay_api_hub/venv/lib/python3.12/site-packages/scipy/stats/_wilcoxon.pyÚ__init__zWilcoxonDistribution.__init__   sN   € Ü�J‰J�q‹M× Ñ ¤¨5Ð Ó1ˆØˆŒÜ=?¿Y¹YÀq»\ÖJ°r�rÔ.¨rÓ2Ñ2ÒJˆ�ùÒJs   Á	A%c                 óL   — | j                   |   }|d |dz    j                  «       S ©Nr   ©r   Úsum©r   Úkr   Úpmfss       r   Ú_cdf1zWilcoxonDistribution._cdf1   s(   € Ø�{‰{˜1‰~ˆØ�F�Q˜‘Uˆ|×ÑÓ!Ð!ó    c                 ó\   —  t        j                  | j                  t        g¬«      ||«      S ©N)Úotypes)r   Ú	vectorizer$   Úfloat©r   r"   r   s      r   Ú_cdfzWilcoxonDistribution._cdf   s"   € Ø7Œr�|‰|˜DŸJ™J´¨wÔ7¸¸1Ó=Ð=r%   c                 óF   — | j                   |   }||d  j                  «       S )Nr   r!   s       r   Ú_sf1zWilcoxonDistribution._sf1   s!   € Ø�{‰{˜1‰~ˆØ�A�Bˆx�|‰|‹~Ðr%   c                 ó\   —  t        j                  | j                  t        g¬«      ||«      S r'   )r   r)   r.   r*   r+   s      r   Ú_sfzWilcoxonDistribution._sf   s"   € Ø6Œr�|‰|˜DŸI™I¬u¨gÔ6°q¸!Ó<Ð<r%   c                 ó@   — | j                   | j                   dz   z  dz  S )Nr   é   )r   )r   s    r   ÚmeanzWilcoxonDistribution.mean    s   € Ø�v‰v˜Ÿ™ !™Ñ$ qÑ(Ð(r%   c                 óÞ   — t        j                  |«      j                  t        d¬«      }| j	                  «       }t        j
                  |j                  t         j                  ¬«      }|||fS )NFr   )Údtype)r   r   r   r   r3   ÚemptyÚshapeÚfloat64©r   r"   ÚmnÚouts       r   Ú_prepzWilcoxonDistribution._prep#   sN   € Ü�J‰J�q‹M× Ñ ¤¨5Ð Ó1ˆØ�Y‰Y‹[ˆÜ�h‰h�q—w‘w¤b§j¡jÔ1ˆØ�"�cˆzÐr%   c                 óŠ   ‡ — ‰ j                  |«      \  }}}t        ||k  |‰ j                  f‰ j                  ˆ fd„¬«      d   S )Nc                 ó4   •— d‰j                  | dz   |«      z
  S r   )r0   ©r"   r   r   s     €r   ú<lambda>z*WilcoxonDistribution.cdf.<locals>.<lambda>,   s   ø€ ¨!¨d¯h©h°q¸±s¸AÓ.>Ñ*>€ r%   ©Úf2© )r<   r   r   r,   r9   s   `   r   ÚcdfzWilcoxonDistribution.cdf)   sG   ø€ Ø—Z‘Z “]‰
ˆˆ2ˆsÜ˜!˜r™' A t§v¡v ;°·	±	Û>ô@Ø@BñDð 	Dr%   c                 óŠ   ‡ — ‰ j                  |«      \  }}}t        ||k  |‰ j                  f‰ j                  ˆ fd„¬«      d   S )Nc                 ó4   •— d‰j                  | dz
  |«      z
  S r   )r,   r?   s     €r   r@   z)WilcoxonDistribution.sf.<locals>.<lambda>1   s   ø€ ¨!¨d¯i©i¸¸!¹¸QÓ.?Ñ*?€ r%   rA   rC   )r<   r   r   r0   r9   s   `   r   ÚsfzWilcoxonDistribution.sf.   sG   ø€ Ø—Z‘Z “]‰
ˆˆ2ˆsÜ˜!˜r™' A t§v¡v ;°·±Û?ôAØACñEð 	Er%   N)Ú__name__Ú
__module__Ú__qualname__r   r$   r,   r.   r0   r3   r<   rD   rG   rC   r%   r   r   r      s2   „ òKò
"ò>òò=ò)òòDó
Er%   r   c                 óþ  — t        j                  |«      d   }d}t        j                  |j                  t         j                  «      r|j
                  dk7  rt        |«      ‚d}	 |€t        j                  | «      } | }nt        | |f|¬«      \  } }| |z
  }t        j                  ||d«      }d}|�*| j                  |   |j                  |   k7  rt        |«      ‚d}t        j                  |j                  t         j                  «      r|j                  t         j                  «      }t        j                  |j                  t         j                  «      st        |«      ‚t        |«      j                  «       }h d	£}
d
|
› d�}||
vrt        |«      ‚ddh}d|› d�}||vrt        |«      ‚t        |«      j                  «       }h d£}d|› d�}||vrt        |«      ‚t!        |t"        j$                  «      sh d£}d|› d�}||vrt        |«      ‚|dk(  rdnd}t        j&                  |dk(  «      }|dk(  r|j                  d   dkD  rd}||||||||fS # t         j                  $ r}	t        |«      |	‚d }	~	ww xY w)NrC   z`axis` must be an integer.r   z<`axis` must be compatible with the shape(s) of `x` (and `y`)©Úaxiséÿÿÿÿz3`x` and `y` must have the same length along `axis`.z<`x` (and `y`, if provided) must be an array of real numbers.>   ÚprattÚwilcoxÚzsplitz`zero_method` must be one of ú.TFz`correction` must be one of >   ÚlessÚgreaterú	two-sidedz`alternative` must be one of >   ÚautoÚexactÚ
asymptoticz`method` must be one of z- or an instance of `stats.PermutationMethod`.rX   rV   é2   )r   r   Ú
issubdtyper5   ÚintegerÚndimÚ
ValueErrorr	   ÚmoveaxisÚ	AxisErrorr7   r   r8   ÚfloatingÚstrÚlowerÚ
isinstancer   ÚPermutationMethodr    )ÚxÚyÚzero_methodÚ
correctionÚalternativeÚmethodrM   ÚmessageÚdÚeÚzero_methodsÚcorrectionsÚalternativesÚmethodsÚoutput_zÚn_zeros                   r   Ú_wilcoxon_ivrt   4   sa  € ä�:‰:�dÓ˜BÑ€DØ*€GÜ�=‰=˜Ÿ™¤R§Z¡ZÔ0°D·I±IÀ²NÜ˜Ó!Ð!àL€Gð	)Øˆ9Ü—
‘
˜1“ˆAØ‰Aä$ a¨ V°$Ô7‰DˆAˆqØ�A‘ˆAÜ�K‰K˜˜4 Ó$ˆð D€GØ€}˜Ÿ™ ™¨!¯'©'°$©-Ò7Ü˜Ó!Ð!àL€GÜ	‡}�}�Q—W‘WœbŸj™jÔ)Ø�H‰H”R—Z‘ZÓ ˆÜ�=‰=˜Ÿ™¤"§+¡+Ô.Ü˜Ó!Ð!ä�kÓ"×(Ñ(Ó*€KÚ0€LØ-¨l¨^¸1Ð=€GØ˜,Ñ&Ü˜Ó!Ð!à˜�-€KØ,¨[¨M¸Ð;€GØ˜Ñ$Ü˜Ó!Ð!ä�kÓ"×(Ñ(Ó*€KÚ3€LØ-¨l¨^¸1Ð=€GØ˜,Ñ&Ü˜Ó!Ð!ä�fœe×5Ñ5Ô6Ú1ˆØ-¨g¨Yð 7?ð ?ˆà˜Ñ Ü˜WÓ%Ð%Ø Ò-‰t°5€Hô
 �V‰V�A˜‘F‹^€FØ�Ò˜AŸG™G B™K¨"Ò,Øˆàˆk˜: {°F¸DÀ(ÈFÐRÐRøô[ �<‰<ò )Ü˜Ó! qÐ(ûð)ús   Á&AI ÉI<É+I7É7I<c                 ó®  — | dk(  }|dk(  r2| j                   d   s| j                  «       } t        j                  | |<   t        j                  | «      }t        j
                  |d¬«      }| j                  d   |z
  }t        t        | «      dd¬«      \  }}t        j
                  | dkD  |z  d¬«      }	t        j
                  | dk  |z  d¬«      }
|dk(  j                  «       }|d	k(  r't        j
                  ||z  d¬«      d
z  }|	|z  }	|
|z  }
||dz   z  dz  }||dz   z  d|z  dz   z  }|dk(  rK|j                  d¬«      }|||dz   z  dz  z  }|||dz   z  d|z  dz   z  z  }d||j                  d¬«      df<   |dz  |z
  j                  d¬«      }||d
z  z  }t        j                  |dz  «      }|dv r	|	|z
  |z  }nt        j                  }|	|
||||fS )Nr   rP   Ú	WRITEABLErN   rL   ÚaverageT)Úreturn_tiesrQ   é   g      ð?g      Ð?g       @rO   é   é   )rX   rV   )Úflagsr   r   ÚnanÚisnanr    r7   r   ÚabsÚanyÚsqrt)rl   rj   rg   Úi_zerosÚi_nanÚn_nanÚcountÚrÚtÚr_plusÚr_minusÚhas_tiesÚr_zero_2r:   Úsers   Útie_correctÚzs                     r   Ú_wilcoxon_statisticr�   t   sþ  € à�A‰v€Gà�hÒð �w‰w�{Ò#Ø—‘“ˆAÜ—V‘Vˆˆ'‰
ä�H‰H�Q‹K€EÜ�F‰F�5˜rÔ"€EØ�G‰G�B‰K˜%Ñ€Eä”S˜“V˜Y°DÔ9�D€A€qä�V‰V�Q˜‘U˜a‘K bÔ)€FÜ�f‰f�a˜!‘e˜q‘[ rÔ*€Gà�Q‘�|‰|‹~€Hà�hÒô —6‘6˜' A™+¨BÔ/°!Ñ3ˆØ�(ÑˆØ�8Ñˆà	�%˜"‘*Ñ	 Ñ	$€BØ	�%˜"‘*Ñ	  e¡¨b¡Ñ	1€Bà�gÒð —‘ "�Ó%ˆØ
ˆf˜ ™Ñ$ tÑ+Ñ+ˆØ
ˆf˜ ™Ñ$¨¨V©°bÑ(8Ñ9Ñ9ˆð &'ˆˆ'�+‰+˜2ˆ+Ó
 Ð
!Ñ"à�a‘4˜!‘8—.‘. b�.Ó)€KØˆ+�a‰-Ñ€BÜ	�‰��b‘Ó	€Bð Ð'Ñ'Ø�b‰[˜BÑ‰ä�F‰Fˆà�7˜B  5¨(Ð2Ð2r%   c                 óD   — |dk(  ry|dk(  ryt        j                  | «      S )NrT   r   rS   rN   )r   Úsign)rŽ   ri   s     r   Ú_correction_signr’   ³   s&   € Ø�iÒØØ	˜Ò	Øä�w‰w�q‹zÐr%   c           	      óÀ  ‡‡— t        | |‰||‰|«      }|\  }Š}}Š}}	}
|j                  dk(  r0t        |«      }t        j                  ||¬«      }‰dk(  r||_        |S t        |‰‰«      \  }}}}}}‰dk(  r3|s|
dkD  sdŠn)|j                  d   dk  rt        j                  «       ŠndŠ‰dk(  r6|rt        ||«      }||dz  |z  z  }t        |t        «       |t        ¬	«      }�n‰dk(  rÔt        |«      }|d
k(  r%|j                  t        j                   |«      «      }nÚ|dk(  r%|j#                  t        j$                  |«      «      }n°dt        j&                  |j#                  t        j$                  |«      «      |j                  t        j                   |«      «      «      z  }t        j(                  |dd«      }n;t        j*                  |fˆˆfd„fddi‰j-                  «       ¤|ddœ¤Žj.                  }|dk(  rt        j&                  ||«      n|}|dk(  r‰dk(  rt        j0                  |«       n|}t        j                  ||d   ¬«      }|	r
|d   |_        |S )Nr   )Ú	statisticÚpvaluerX   rV   rW   rN   é   g      à?)ÚxprS   rT   ry   r   c                 ó$   •— t        | ‰‰«      d   S )Nr   )r�   )rl   rj   rg   s    €€r   r@   z_wilcoxon_nd.<locals>.<lambda>ø   s   ø€ Ô/°°6¸;ÓGÈÑJ€ r%   Úpermutation_typeÚsamples)ri   rM   rU   rC   )rt   Úsizer   r   ÚWilcoxonResultÚ
zstatisticr�   r7   r   rd   r’   r   r   r   r   rD   ÚceilrG   ÚfloorÚminimumÚclipÚpermutation_testÚ_asdictr•   r   )re   rf   rg   rh   ri   rj   rM   Útemprl   rr   rs   ÚNaNÚresrˆ   r‰   rŒ   rŽ   r…   rŠ   r‘   ÚpÚdistr”   s     `  `                 r   Ú_wilcoxon_ndr©   ¼   sL  ù€ ô ˜˜1˜k¨:°{ÀFÈDÓQ€DØNRÑK€A€{�J ¨V°T¸8ÀVà‡v�v�‚{Ü�q‹kˆÜ×'Ñ'°#¸cÔBˆØ�\Ò!Ø ˆCŒNØˆ
ä.AØ	ˆ6�;ó/Ñ+€FˆG�R˜˜E 8ð �ÒÙ˜F QšJØ‰FØ�W‰W�R‰[˜BÒô
 ×,Ñ,Ó.‰Fð "ˆFà�ÒÙÜ# A {Ó3ˆDØ�˜‘˜b‘Ñ ˆAÜ˜œ=›?¨K¼BÔ?ŠØ	�7Ò	Ü# EÓ*ˆð ˜&Ò Ø—‘œŸ™ ›Ó)‰AØ˜IÒ%Ø—‘œŸ™ Ó(Ó)‰Aà”B—J‘J˜tŸw™w¤r§x¡x°Ó'7Ó8Ø#Ÿx™x¬¯©°«Ó8ó:ñ :ˆAä—‘˜˜1˜aÓ ‰Aä×"Ñ"ØˆDÔJñ.à&ð.à*0¯.©.Ó*:ð.ð $¨"ó.÷ /5©fð 	
ð 0;¸KÒ/G”—
‘
˜6 7Ô+ÈV€IØ" kÒ1°fÀÒ6LŒ�‰�‹‰
ÐST€Aä
×
#Ñ
#¨iÀÀ"ÁÔ
F€CÙØ˜2™ˆŒØ€Jr%   )rP   )NrP   TrU   rV   r   )Únumpyr   Úscipyr   Ú	_stats_pyr   r   r   Ú r   Ú_axis_nan_policyr	   Ú
_hypotestsr
   Úscipy._lib._utilr   r   r   rt   r�   r’   r©   rC   r%   r   ú<module>r±      sK   ðÛ å ß <Ñ <Ý Ý /Ý +ß 1÷&Eñ &EòR=Só@<3ò~ð >BØ>?ôGr%   