403Webshell
Server IP : 217.160.0.135  /  Your IP : 216.73.217.85
Web Server : Apache
System : Linux www 6.18.52-i1-ampere #1203 SMP Mon Sep 14 18:29:59 CEST 2026 aarch64
User : sws1074145052 ( 1074145052)
PHP Version : 8.3.32
Disable Function : NONE
MySQL : OFF  |  cURL : ON  |  WGET : ON  |  Perl : ON  |  Python : OFF  |  Sudo : OFF  |  Pkexec : OFF
Directory :  /lib/python3/dist-packages/scipy/stats/tests/__pycache__/

Upload File :
current_dir [ Writeable ] document_root [ Writeable ]

 

Command :


[ Back ]     

Current File : /lib/python3/dist-packages/scipy/stats/tests/__pycache__/test_morestats.cpython-311.pyc
�

d�c#���b�ddlZddlZddlZddlmZddlmZmZm	Z	m
Z
mZmZm
Z
mZddlZddlmZddlZddlmZddlmZddlmZdd	lmZd
dlmZmZddlmZdd
lm Z e!e ��Z 	ddl"Z"de"j#d<ddl$m%Z&dZ'n
#e($rdZ'YnwxYwgd�Z)gd�Z*gd�Z+gd�Z,gd�Z-gd�Z.gd�Z/gd�Z0gd�Z1gd�Z2d�Z3Gd�d��Z4Gd�d ��Z5Gd!�d"��Z6Gd#�d$��Z7Gd%�d&��Z8Gd'�d(��Z9Gd)�d*��Z:Gd+�d,��Z;Gd-�d.��Z<Gd/�d0��Z=Gd1�d2��Z>d3�Z?Gd4�d5��Z@Gd6�d7��ZAGd8�d9��ZBGd:�d;��ZCGd<�d=��ZDGd>�d?��ZEGd@�dA��ZFGdB�dC��ZGgdD�ZHGdE�dF��ZIGdG�dH��ZJGdI�dJ��ZKGdK�dL��ZLGdM�dN��ZMGdO�dP��ZNGdQ�dR��ZOGdS�dT��ZPGdU�dV��ZQdS)W�N)�RandomState)�assert_array_equal�assert_almost_equal�assert_array_less�assert_array_almost_equal�assert_�assert_allclose�assert_equal�suppress_warnings)�raises)�optimize)�stats)�
_abw_state�)�check_named_results�)�_get_wilcoxon_distr�_get_wilcoxon_distr2)�_binary_search_for_binom_tst)�distcont�Agg�backendTF)
�j�t��?�Zd;��?�V-����?��?gX9��v��?�-�����?��x�&1�?�+����?�+����?r)
rrrr灕C�l��?rrrr�V-���?)
�ʡE��?g/�$���?r!r�ףp=
��?r rrrr)
g�G�z�?rr rr$r rrrr)
rrg��C�l�?�G�z��?r�rh��|�?rrg\��(\�?r)
羟/�$�?gh��|?5�?r'r!r"rr$rg1�Zd�?r)
r%���Mb�?rgj�t��?rrg}?5^�I�?g)\��(�?rr)
rrrrrrrrrr)
rrrr$rr(r(rrr#)
r#r$r&r r!r!r#rr(r!c�R�tjtjj|i|����S�N)�np�logr�gamma�rvs)�args�kwargss  �B/usr/lib/python3/dist-packages/scipy/stats/tests/test_morestats.py�_old_loggamma_rvsr25s#��
�6�%�+�/�4�2�6�2�2�3�3�3�c� �eZdZd�Zd�Zd�ZdS)�
TestBayes_mvsc�J�gd�}tj|��\}}}t|jd��t	|jdd���t|jd��t	|jdd���t|jd	d
���t	|jdd���dS)
N���	���r<�
�"@�g4`�H#j@g�O�[��%@�+����=��rtol�$@�g��a�i	@g���Ӈu8@��&�.>�_紫�@���decimal�gY��s��?g���1O�@)r�	bayes_mvsr�	statisticr	�minmax��self�data�mean�var�stds     r1�
test_basiczTestBayes_mvs.test_basic:s���'�&�&�����.�.���c�3��D�N�C�0�0�0����%M�"�	$�	$�	$�	$�	�C�M�4�0�0�0���
�$K�"�	$�	$�	$�	$�	�C�M�+=�r�J�J�J�J���
�$L�"�	$�	$�	$�	$�	$�	$r3c�F�tttjg��dSr*)�
assert_raises�
ValueErrorrrK�rOs r1�test_empty_inputzTestBayes_mvs.test_empty_inputLs���j�%�/�2�6�6�6�6�6r3c��tjd��}d}tj|��}|D]}t	||���dS)N�)rLrM)r+�arangerrKr)rO�x�
attributes�res�is     r1�test_result_attributesz$TestBayes_mvs.test_result_attributesOsO���I�b�M�M��,�
��o�a� � ���	/�	/�A���:�.�.�.�.�	/�	/r3N)�__name__�
__module__�__qualname__rTrYra�r3r1r5r59sA������$�$�$�$7�7�7�/�/�/�/�/r3r5c�&�eZdZd�Zd�Zd�Zd�ZdS)�TestMvsdistc���gd�}tj|��\}}}t|���d��t	|�d��dd���t|���d��t	|�d��dd	���t|���d
d���t	|�d��d
d���dS)Nr7r>g�������?r?r@rArCrDrErFrGrHrJ)r�mvsdistrrQr	�intervalrNs     r1rTzTestMvsdist.test_basicYs��&�&�&����t�,�,���c�3��D�I�I�K�K��-�-�-���
�
�c�*�*�-A�GL�	N�	N�	N�	N�	�C�H�H�J�J��-�-�-�����S�)�)�,?�EJ�	L�	L�	L�	L�	�C�H�H�J�J�(:�B�G�G�G�G�����S�)�)�,@�FK�	M�	M�	M�	M�	M�	Mr3c�F�tttjg��dSr*�rVrWrrirXs r1rYzTestMvsdist.test_empty_inpuths���j�%�-��4�4�4�4�4r3c�L�dg}tttj|��dS�Nrrl�rOrPs  r1�test_bad_argzTestMvsdist.test_bad_argks#���s���j�%�-��6�6�6�6�6r3c��tj��5tjdt��d�t	jgd���D��d�t	jgd���D��ddd��dS#1swxYwYdS)N�errorc�6�g|]}|�����Sre�rQ��.0r]s  r1�
<listcomp>z*TestMvsdist.test_warns.<locals>.<listcomp>us ��8�8�8�!�Q�V�V�X�X�8�8�8r3�rr�c�6�g|]}|�����Srertrus  r1rwz*TestMvsdist.test_warns.<locals>.<listcomp>vs ��>�>�>�!�Q�V�V�X�X�>�>�>r3�rrry��)�warnings�catch_warnings�simplefilter�RuntimeWarningrrirXs r1�
test_warnszTestMvsdist.test_warnsps����
$�
&�
&�	?�	?��!�'�>�:�:�:�8�8�u�}�Y�Y�Y�7�7�8�8�8�8�>�>�u�}�_�_�_�=�=�>�>�>�>�	?�	?�	?�	?�	?�	?�	?�	?�	?�	?�	?�	?����	?�	?�	?�	?�	?�	?s�AA<�<B�BN)rbrcrdrTrYrpr�rer3r1rgrgXsS������
M�
M�
M�5�5�5�7�7�7�
?�?�?�?�?r3rgc�8�eZdZd�Zd�Zd�Zd�Zd�Zd�Zd�Z	dS)	�TestShapiroc�T�gd�}tj|��\}}tj|��}t|dd���t|jdd���t|dd���t|jdd���gd�}tj|��\}}tj|��}t|dd���t|jdd���t|dd	���t|jdd	���tj�d
d	dd�
��}tj|��\}}tj|��}t|dd���t|jdd���t|dd	���t|jdd	���gd�}d}d}	tj|��\}}tj|��}t||d���t|j|d���t||	d
���t|j|	d
���dS)N)�)\��(�?�{�G�z@�q=
ףp@�H�z�G$@�������@���Q�	@�q=
ףp�?���Q�@��z�G��?�@���Q��?�R���Q�?��Q���	@��(\���?��z�G��?��z�G�@��(\#@�333333@��G�z�?�p=
ף@�����?r8rH��Ɍ�?)��(\��?�=
ףp=�?�\��(\@�ffffff@�\��(\�?�(\���?��G�z@��(\����ףp=
�@皙�����?�)\��(�?�R���Q�r��p=
ף�?�������@�
ףp=
�?�{�G�z�?�\��(\
@�{�G�z@��Q���@�IddY��?�Dio���?ryr}�diNa���loc�scale�size�random_stateg��E�?g�Oٴ?)g�n���?gj�t��?gffffff�?g����Mb�?gj�t��?g;�O��n�?gj�t��?gX9��v�?g��S��?g�(\���?g�I+��?g��ʡE�?g���x�?g��|?5^�?r�g��C�l�?g1�Zd@g��S�@g�(\��	@g�G�z@g� �rh�@g�"�~j@g�Q���@g�t�V@g�ʡE�� @ghyܝ��?g��
0�M?r|)r�shapirorrL�pvalue�normr.)
rO�x1�w�pw�shapiro_test�x2�x3�x4�
W_expected�
p_expecteds
          r1rTzTestShapiro.test_basiczsb��8�8�8���
�b�!�!���2��}�R�(�(���A�2�A�>�>�>�>��L�2�4G�QR�S�S�S�S��B� 4�a�@�@�@�@��L�/�1E�q�Q�Q�Q�Q�&�&�&���
�b�!�!���2��}�R�(�(���A�y�!�4�4�4�4��L�2�I�q�I�I�I�I��B���3�3�3�3��L�/��!�D�D�D�D��Z�^�^����8�^�
L�
L���
�b�!�!���2��}�R�(�(���A�1�1�=�=�=�=��L�2�4F�PQ�R�R�R�R��B� 3�Q�?�?�?�?��L�/�1D�a�P�P�P�P�?�?�?���
��
��
�b�!�!���2��}�R�(�(���A�z�1�5�5�5�5��L�2�J��J�J�J�J��B�
�A�6�6�6�6��L�/��Q�G�G�G�G�G�Gr3c��gd�gd�g}tj|��\}}tj|��}t|dd���t|jdd���t|dd���t|jdd���gd�gd�g}tj|��\}}tj|��}t|d	d���t|jd	d���t|d
d���t|jd
d���dS)N)
r�r�r�r�r�r�r�r�r�r�)
r�r�r�r�r�r�r�r�r�r�r�r8rHr�)
r�r�r�r�r�r�r�r�r�r�)
r�r�r�r�r�r�r�r�r�r�r�r�ry)rr�rrLr�)rOr�r�r�r�r�s      r1�test_2dzTestShapiro.test_2d�sS�� � � �"8�"8�"8�9���
�b�!�!���2��}�R�(�(���A�2�A�>�>�>�>��L�2�4G�QR�S�S�S�S��B� 4�a�@�@�@�@��L�/�1E�q�Q�Q�Q�Q����&�&�&�'���
�b�!�!���2��}�R�(�(���A�y�!�4�4�4�4��L�2�I�q�I�I�I�I��B���3�3�3�3��L�/��!�D�D�D�D�D�Dr3c��tttjg��tttjgggg��dSr*�rVrWrr�rXs r1rYzTestShapiro.test_empty_input�s6���j�%�-��4�4�4��j�%�-�"�b�"��>�>�>�>�>r3c���tttjddg��tttjt	jgdggt�����dS)Nrr��dtype)rVrWrr�r+�array�objectrXs r1�test_not_enough_valuesz"TestShapiro.test_not_enough_values�sI���j�%�-�!�Q��8�8�8��j�%�-���2��s�)�6�1R�1R�1R�S�S�S�S�Sr3c�L�dg}tttj|��dSrnr��rOr]s  r1rpzTestShapiro.test_bad_arg�s#��
�C���j�%�-��3�3�3�3�3r3c�^�tjd��}tj|d<tj|��\}}tj|��}t|tj��t|jtj��t|d��t|jd��dS)NrCr9r)	r+r\�nanrr�r
rLrr�)rOr]r�r�r�s     r1�test_nan_inputzTestShapiro.test_nan_input�s����I�c�N�N���v��!���
�a� � ���2��}�Q�'�'���Q�������\�+�R�V�4�4�4��B��$�$�$��L�/��5�5�5�5�5r3c��tjgd���\}}tj|��}d}t||d���dS)N)i��i =i@�)gP��q~��?g�F���?��h㈵��>rA)r�boxcoxr�r	)rO�	trans_val�maxlogr_�refs     r1�test_gh14462zTestShapiro.test_gh14462�sP��"�L�)A�)A�)A�B�B��	�6��m�I�&�&��2����S�t�,�,�,�,�,�,r3N)
rbrcrdrTr�rYr�rpr�r�rer3r1r�r�ys�������)H�)H�)H�VE�E�E�*?�?�?�T�T�T�4�4�4�
	6�	6�	6�-�-�-�-�-r3r�c��eZdZd�Zd�Zd�Zd�Zd�Zd�Zd�Z	e
j�dgd	���d
���Z
dS)�TestAndersonc��td��}|�d���}|�d���}tj|��\}}}t|dd�|��tj|��\}}}t||dd���t
jd��}d|d<tj|��\}}}t|d��dS)	N���I�2�r��������
rg�����	@)	r�standard_exponential�standard_normalr�andersonrr+�onesr	)rO�rsr�r��A�crit�sig�vs        r1�test_normalzTestAnderson.test_normal�s���
��
$�
$��
�
$�
$�"�
$�
-�
-��
�
�
�R�
�
(�
(���~�b�)�)���4���$�s��s�)�Q�'�'�'��~�b�)�)���4���!�T�"�#�#�Y�'�'�'��G�B�K�K����!���~�a�(�(���4��	��8�$�$�$�$�$r3c��td��}|�d���}|�d���}tj|d��\}}}t||dd���t
jd���5tj|d��\}}}ddd��n#1swxYwYt||dk��dS)	Nr�r�r��exponr��ignore)�allr�)	rr�r�rr�rr+�errstater)rOr�r�r�r�r�r�s       r1�
test_exponzTestAnderson.test_expon�s��
��
$�
$��
�
$�
$�"�
$�
-�
-��
�
�
�R�
�
(�
(���~�b�'�2�2���4���!�T�"�#�#�Y�'�'�'�
�[�X�
&�
&�
&�	7�	7� �>�"�g�6�6�L�A�t�S�	7�	7�	7�	7�	7�	7�	7�	7�	7�	7�	7����	7�	7�	7�	7���D��H������s�B(�(B,�/B,c	��tjd��}d|d<tj|d��\}}}t	|��}tj�|��\}}tj�|||��}tj�|||��}	tj	d|dz��}
|tj
d|
zdz
||	ddd�zz��z
}t||��dS)Nr��r�gumbelrrr�)r+r�rr��len�gumbel_l�fit�logcdf�logsfr\rQr	)rOr��a2r�r��n�xbar�sr�r�r`�expected_a2s            r1�test_gumbelzTestAnderson.test_gumbels���
�G�C�L�L����!����q�(�3�3�
��D�#���F�F���.�$�$�Q�'�'���a���&�&�q�$��2�2����$�$�Q��a�0�0���I�a��1������b�2�7�A�a�C�!�G���t�t��t��0D�#E�F�F�F����K�(�(�(�(�(r3c�L�tttjdgd���dS�Nr�plate_of_shrimp��dist)rVrWrr�rXs r1rpzTestAnderson.test_bad_args$���j�%�.�1�#�<M�N�N�N�N�N�Nr3c��td��}|�d���}tj|��}d}t	||��dS)Nr�r�r��rL�critical_values�significance_level)rr�rr�r)rOr�r]r_r^s     r1raz#TestAnderson.test_result_attributessO��
��
$�
$���#�#��#�,�,���n�Q����K�
��C��,�,�,�,�,r3c���td��}|�d���}tj|d��\}}}tj|d��\}}}t	||��dS)Nr�r�r�r�r�)rr�rr�r	)	rOr�r]�A1�crit1�sig1�A2�crit2�sig2s	         r1�
test_gumbel_lzTestAnderson.test_gumbel_l sh����
$�
$���I�I�3�I�����.��H�5�5���E�4��.��J�7�7���E�4���B�����r3c�H�td��}|�d���}tjd��}d|d<t	j|d��\}}}t	j|d��\}}}	t
||dd���t||dk��dS)	Nr�r�r�rr�gumbel_rr�r�)rr�r+r�rr�rr)
rOr�r�r�r
rrr
rrs
          r1�
test_gumbel_rzTestAnderson.test_gumbel_r*s�����
$�
$��
�Y�Y�C�Y�
 �
 ��
�W�S�\�\����1���.��Z�8�8���E�4��.��Z�8�8���E�4��"�e�B�C�C�j�)�)�)���U�2�Y������r3�distname)r�r�r��extreme1r�r�logisticc��tj�d��}|dvrdn|}tt|��}t
|}|j|d|d��}t	j||��}|jj	sJ�dS)Nl'Da�h�$>r�rr����r�r�)
r+�random�default_rng�getattrrrr.r��
fit_result�success)rOr�rng�
real_distnamer�paramsr]r_s        r1�test_anderson_fit_paramsz%TestAnderson.test_anderson_fit_params9s���
�i�#�#�$6�7�7��'/�3I�'I�'I���&�	��u�m�,�,���-�(���D�H�f�4�c�:�:�:���n�Q��)�)���~�%�%�%�%�%�%r3N)rbrcrdr�r�r�rprarr�pytest�mark�parametrizer"rer3r1r�r��s�������%�%�%�.���)�)�)� O�O�O�-�-�-� � � �
 �
 �
 ��[���Z�@�@�@�A�A�	&�	&�A�A�	&�	&�	&r3r�c�D�eZdZd�Zd�Zd�Zd�Zd�Zd�Zd�Z	d�Z
d	�Zd
S)�TestAndersonKSampc�J�gd�}tjgd���}tjgd���}tjgd���}tj||||fd���\}}}t	|dd��tgd	�|d
d�d��t
|d
d���dS)N�g�����YC@g�D@gfffff�E@g@F@g�F@gG@g������G@gM@�g������C@gfffff�C@g������C@g33333�D@gfffff�D@g33333sE@gfffff�E@gfffff�F@��A@g�A@g�C@gD@��E@r-�F@g�F@�r,�ffffffA@r0g33333�A@g������B@gfffff�B@g������D@gffffffE@F��midrankg�����@ry�g��C�l��?gvq
�-�?g,Ԛ���?g�l��@g�Pk��	@rr}r|g�J�4a?���Mb0?��atol�r+r�r�anderson_ksamprrr	�rO�t1�t2�t3�t4�Tk�tm�ps        r1�test_example1az TestAndersonKSamp.test_example1aIs���
>�
=�
=��
�X�F�F�F�
G�
G��
�X�F�F�F�
G�
G��
�X�F�F�F�
G�
G���(�"�b�"�b�)9�5�I�I�I�	��B���B��q�)�)�)�!�"J�"J�"J�"$�Q�q�S�'�1�	.�	.�	.���6��0�0�0�0�0�0r3c�n�tjgd���}tjgd���}tjgd���}tjgd���}tj||||fd���\}}}t	|dd��tgd	�|d
d�d��t
|d
d���dS)Nr)r*r+r/Tr1g�Q���@ryr3rr}r|g����Mb`?r4r5r7r9s        r1�test_example1bz TestAndersonKSamp.test_example1bZs���
�X�F�F�F�
G�
G��
�X�F�F�F�
G�
G��
�X�F�F�F�
G�
G��
�X�F�F�F�
G�
G���(�"�b�"�b�)9�4�H�H�H�	��B���B��q�)�)�)�!�"J�"J�"J�"$�Q�q�S�'�1�	.�	.�	.���6��0�0�0�0�0�0r3c�B�gd�}gd�}gd�}gd�}gd�}gd�}gd�}gd�}gd	�}	gd
�}
gd�}gd�}gd
�}
gd�}tj|||||||||	|
|||
|fd���\}}}t|dd��tgd�|dd�d��t	|dd���dS)N���r[�)��!��
i�rG�:�%r��Ar9�i���$���v�
�"��rW�C�9�>r;�rU��Zr��<��=�1rG��8��O�T�,�;rHrS��i6�L�rg�rZ�����F�erp��JrY�0rHi�r:rq�rHi�rh���7i@rc�h�����/���rI�rmi�Wr;�xrGrZr}���Gr~rv�*rdr}r:r��ryrGr�r�rGr��r]rr��4�_��a�3r�r|�rW��D�M�Prr��j���R�6rV���.�o�'�?rW�rW�rb�r�rg�f�Hr[r�ryr[���re�Xr�r}r}rQr[���r��rmr=rG�	�gr9r:ii[ryrzri��r��r}i�#r:�i�rWr�r;�br}�U�[�+��ryro�r���rGrYr�� rXrh��rwrGr��Br`rUFr1g���M
@ry�g^�I+�?g���~�:�?g��+e�?g�H.�!�@gc�ZB>h@rr}r|�	�^)�p?r4r5�rr8rrr	�rOr:r;r<r=�t5�t6�t7�t8�t9�t10�t11�t12�t13�t14r>r?r@s                  r1�test_example2az TestAndersonKSamp.test_example2ajs^��(�
'�
'��
G�
G�
G��
4�
4�
4��5�5�5��
%�
%�
%��
?�
?�
?��
@�
@�
@��I�I�I��B�B�B��1�1�1��4�4�4��'�'�'��?�?�?�������(�"�b�"�b�"�b�"�b�*,�c�3��S�#�*G�16�8�8�8�	��B��	�B��q�)�)�)�!�"J�"J�"J�"$�Q�q�S�'�1�	.�	.�	.���6��0�0�0�0�0�0r3c�B�gd�}gd�}gd�}gd�}gd�}gd�}gd�}gd�}gd	�}	gd
�}
gd�}gd�}gd
�}
gd�}tj|||||||||	|
|||
|fd���\}}}t|dd��tgd�|dd�d��t	|dd���dS)NrErKrTr\rnrsrxr�r�r�r�r�r�r�Tr1g'1�Z
@ryr�rr}r|r�r4r5r�r�s                  r1�test_example2bz TestAndersonKSamp.test_example2b�s^��(�
'�
'��
G�
G�
G��
4�
4�
4��5�5�5��
%�
%�
%��
?�
?�
?��
@�
@�
@��I�I�I��B�B�B��1�1�1��4�4�4��'�'�'��?�?�?�������(�"�b�"�b�"�b�"�b�*,�c�3��S�#�*G�15�7�7�7�	��B��	�B��q�)�)�)�!�"J�"J�"J�"$�Q�q�S�'�1�	.�	.�	.���6��0�0�0�0�0�0r3c�N�tjddd��}t��5}|�td���tj||dzgd���\}}}ddd��n#1swxYwYt|dd	��t|d
��t��5}|�td���tj||dzg��\}}}ddd��n#1swxYwYt|dd	��t|d
��t��5}|�td���tj||d
zgd���\}}}ddd��n#1swxYwYt|dd��t|d��t��5}|�td���tj||d
zg��\}}}ddd��n#1swxYwYt|dd��t|d��tj||dzgd���\}}}t|dd��t|ddd���tj||dzg��\}}}t|dd��t|ddd���tj||dzgd���\}}}t|dd��t|ddd���tj||dzgd���\}}}t|d d��t|d!d"d���dS)#Nrr�zp-value floored��messageg@D@Fr1g=
ףp�D@ry���MbP?g�G�z�D@zp-value capped��?g?�ܵ��r|��?g?�ܵ��@g�!��u�?gףp=
׳?g{�G�zt?r�r6rBr8g�B�i�q�?g�鷯�?�'@g��:M@g46<�Rv?g����Mb@?g+@g6�>W[1@g��1ZGUS?�-C��6?)
r+�linspacer�filter�UserWarningrr8rr
r	)rOr��supr��_r@s      r1�test_R_kSamplesz!TestAndersonKSamp.test_R_kSamples�sY��B�[��C��
%�
%���
 �
 �	K�C��J�J�{�,=�J�>�>�>��*�B��T�	�?�E�J�J�J�G�A�q�!�	K�	K�	K�	K�	K�	K�	K�	K�	K�	K�	K����	K�	K�	K�	K�	�A�v�q�)�)�)��Q�����
�
 �
 �	<�C��J�J�{�,=�J�>�>�>��*�B��T�	�?�;�;�G�A�q�!�	<�	<�	<�	<�	<�	<�	<�	<�	<�	<�	<����	<�	<�	<�	<�	�A�v�q�)�)�)��Q������
 �
 �	I�C��J�J�{�,<�J�=�=�=��*�B��R��=�%�H�H�H�G�A�q�!�	I�	I�	I�	I�	I�	I�	I�	I�	I�	I�	I����	I�	I�	I�	I�	�A�w��*�*�*��Q�����
�
 �
 �	:�C��J�J�{�,<�J�=�=�=��*�B��R��=�9�9�G�A�q�!�	:�	:�	:�	:�	:�	:�	:�	:�	:�	:�	:����	:�	:�	:�	:�	�A�w��*�*�*��Q������&��B��H�~�u�E�E�E���1�a��A�v�q�)�)�)���6��A�6�6�6�6��&��B��F�|�4�4���1�a��A�v�q�)�)�)���6��A�6�6�6�6��&��B��I���F�F�F���1�a��A�v�q�)�)�)���7��a�8�8�8�8��&��B��I���F�F�F���1�a��A�v�q�)�)�)���7��a�8�8�8�8�8�8sG�<A-�-A1�4A1�':C-�-C1�4C1�'<E/�/E3�6E3�):G/�/G3�6G3c�j�tttjt	jd����dS�Nr}�rVrWrr8r+r�rXs r1�test_not_enough_samplesz)TestAndersonKSamp.test_not_enough_samples�s%���j�%�"6����
�
�C�C�C�C�Cr3c��tttjt	jd��t	jd��f��dSr�r�rXs r1�test_no_distinct_observationsz/TestAndersonKSamp.test_no_distinct_observations�s=���j�%�"6��w�q�z�z�2�7�1�:�:�.�	0�	0�	0�	0�	0r3c�n�tttjt	jd��gf��dSr�r�rXs r1�test_empty_samplez#TestAndersonKSamp.test_empty_samples*���j�%�"6������R�8H�I�I�I�I�Ir3c���gd�}tjgd���}tj||fd���}d}t	||��t|j|j��dS)Nr)r*Fr1r)r+r�rr8rr
rr�)rOr:r;r_r^s     r1raz(TestAndersonKSamp.test_result_attributessl��
=�
=�
=��
�X�F�F�F�
G�
G���"�B��8�U�;�;�;��K�
��C��,�,�,��S�+�S�Z�8�8�8�8�8r3N)rbrcrdrArCr�r�r�r�r�r�rarer3r1r'r'Hs�������1�1�1�"1�1�1� 1�1�1�@1�1�1�@N9�N9�N9�`D�D�D�0�0�0�J�J�J�	9�	9�	9�	9�	9r3r'c���eZdZd�Zd�Zd�Zd�Zd�Zd�Zd�Z	e
j�dgd	�gd
�ddfgd	�gd
�d
dfgd�gd�ddfgd�gd�d
dfgd�gd�ddfgd�gd�d
dfg��d���Z
d�ZdS)�
TestAnsaric��gd�}gd�}t��5}|�td��tj||��\}}ddd��n#1swxYwYt|dd��t|dd��dS)N�rrryryr|�ryrr8rr8rr|r�%Ties preclude use of exact statistic.g�7@r�g8�#�oG�?)rr�r�r�ansarir)rOr]�yr��W�pvals      r1�
test_smallzTestAnsari.test_smalls����O�O��$�$�$��
�
 �
 �	)�C��J�J�{�$K�L�L�L��l�1�a�(�(�G�A�t�	)�	)�	)�	)�	)�	)�	)�	)�	)�	)�	)����	)�	)�	)�	)�	�A�t�R�(�(�(��D�"5�r�:�:�:�:�:s�4A�A�Ac�L�tjd��}tjd��}t��5}|�td��tj||��\}}ddd��n#1swxYwYt|dd��t|dd��dS)N�r��kr��cr�r��m�lrzr�rr�`r�r�r��q�tr��nr��r�r�r�r��i�gr�r�rzr�r�r�r�rz�rr�r�r�r�r�r�g0g@r�g�|��:�?)r+r�rr�r�rr�r)rO�ramsay�parekhr�r�r�s      r1�test_approxzTestAnsari.test_approxs�����J�K�K����$�%�%���
 �
 �	3�C��J�J�{�$K�L�L�L��l�6�6�2�2�G�A�t�	3�	3�	3�	3�	3�	3�	3�	3�	3�	3�	3����	3�	3�	3�	3�	�A�u�b�)�)�)��D�"5�r�:�:�:�:�:s�4A7�7A;�>A;c��tjgd�gd���\}}t|dd��t|dd��dS)N�rrryr|)r[r}rdr<r�r:rCr�g�?r;)rr�r)rOr�r�s   r1�
test_exactzTestAnsari.test_exact)sM���,�|�|�|�-C�-C�-C�D�D���4��A�t�R�(�(�(��D�"6��:�:�:�:�:r3c��tttjgdg��tttjdgg��dSrn�rVrWrr�rXs r1rpzTestAnsari.test_bad_arg.s8���j�%�,��Q�C�8�8�8��j�%�,���R�8�8�8�8�8r3c���gd�}gd�}t��5}|�td��tj||��}ddd��n#1swxYwYd}t||��dS)Nr�r�r��rLr�)rr�r�rr�r)rOr]r�r�r_r^s      r1raz!TestAnsari.test_result_attributes2s����O�O��$�$�$��
�
 �
 �	%�C��J�J�{�$K�L�L�L��,�q�!�$�$�C�	%�	%�	%�	%�	%�	%�	%�	%�	%�	%�	%����	%�	%�	%�	%�-�
��C��,�,�,�,�,s�1A�A�Ac��gd�}gd�}d}tt|���5tj||d���ddd��dS#1swxYwYdS)Nr��r}r8r;r<z!'alternative' must be 'two-sided'��match�foo��alternativer)rOr�r�rs    r1�test_bad_alternativezTestAnsari.test_bad_alternative;s���
�\�\��
�\�\��3��
�:�U�
3�
3�
3�	4�	4��L��R�U�3�3�3�3�	4�	4�	4�	4�	4�	4�	4�	4�	4�	4�	4�	4����	4�	4�	4�	4�	4�	4s�A�A
�
A
c�V�gd�}gd�}tj||��\}}tj||d���j}tj||d���j}|dksJ�|dksJ�tj|t|��t|����}t
||zd|zd	�
��t
||dzd	�
��t
|d|z|dzz
d	�
��tj||d���j}tj||d���j}	|dksJ�|	dksJ�dS)N)���rr}r�r[rdri)r�g!@�#@g%@r�g)@�lessr
�greater�ffffff�?皙�����?r��-���q=r5r)rr�r�r�pmfr�r	)
rOr�r�rLr��pval_l�pval_g�prob�pval_l_reverse�pval_g_reverses
          r1�test_alternative_exactz!TestAnsari.test_alternative_exactCsQ��
'�
'�
'��
.�
.�
.�� �,�r�2�.�.��	�4���b�"�&�9�9�9�@����b�"�)�<�<�<�C����}�}�}�}���}�}�}�}��~�i��R���#�b�'�'�:�:�������T���>�>�>�>�	���Q��U�3�3�3�3����$��t�A�v�
�E�:�:�:�:���b�"�&�A�A�A�H����b�"�)�D�D�D�K����$�$�$�$���$�$�$�$�$�$r3zx, y, alternative, expectedr�rrg�@�A�?rrx)r|r}r8r;r<g�G�$I��?g�%I�$I�?r{)r8r;r<c�b�tj|||���j}t||d���dS)Nr
rr5)rr�r�r	)rOr]r�r�expectedr�s      r1�test_alternative_exact_with_Rz(TestAnsari.test_alternative_exact_with_R^s8��8�|�A�q�k�:�:�:�A����h�U�3�3�3�3�3�3r3c���tj�dddd���}tj�dddd���}tj||d���j}tj||d	���j}t|d
d���t|d
d���tj�dddd���}tj�dddd���}tj||��j}tj||d���j}tj||d	���j}t||dzd���t|d|dzz
d���dS)Nrr}r��{rrrr
rrrr5r�r^g�?r)rr�r.r�r�r	)rOr�r�rrr�s      r1�test_alternative_approxz"TestAnsari.test_alternative_approx}sP��
�Z�^�^�A�q�s��^�
=�
=��
�Z�^�^�A�q�s��^�
=�
=����b�"�&�9�9�9�@����b�"�)�<�<�<�C�����%�0�0�0�0����%�0�0�0�0��Z�^�^�A�q�r��^�
<�
<��
�Z�^�^�A�s��#�^�
>�
>���|�B��#�#�*����b�"�&�9�9�9�@����b�"�)�<�<�<�C�����Q��U�3�3�3�3����$�q�&��u�5�5�5�5�5�5r3N)rbrcrdr�r�rrprarrr#r$r%rr!rer3r1r�r�s?������;�;�;�;�;�;�;�;�;�
9�9�9�-�-�-�4�4�4�%�%�%�6�[���%��,�,����f�o�	>�
�,�,����i��	A�
�)�)�_�_�_�f�o�	>�
�)�)�_�_�_�i��	A�
�/�/�9�9�9�f�o�	>�
�/�/�9�9�9�i��	A�	C���4�4���4�&6�6�6�6�6r3r�c�,�eZdZd�Zd�Zd�Zd�Zd�ZdS)�TestBartlettc
���tttttt
ttttg
}tj|�\}}t|dd��t|dd��dS)NgzU@/�4@r;gG���?)
�g1�g2�g3�g4�g5�g6�g7�g8�g9�g10r�bartlettr)rOr/�Tr�s    r1�	test_datazTestBartlett.test_data�sW���B��B��B��B��C�8���.�$�'���4��A�0�!�4�4�4��D�/�1�5�5�5�5�5r3c�H�tttjdg��dSrn)rVrWrr/rXs r1rpzTestBartlett.test_bad_arg�s���j�%�.�1�#�6�6�6�6�6r3c
���tttttt
ttttg
}tj|�}d}t||��dS�Nr)
r%r&r'r(r)r*r+r,r-r.rr/r�rOr/r_r^s    r1raz#TestBartlett.test_result_attributes�sC���B��B��B��B��C�8���n�d�#��,�
��C��,�,�,�,�,r3c���tttttt
ttttgf}ttjtjftj|���dSr*)r%r&r'r(r)r*r+r,r-r.r
r+r�rr/)rOr/s  r1�test_empty_argzTestBartlett.test_empty_arg�sD���B��B��B��B��C��<���b�f�b�f�%�u�~�t�'<�=�=�=�=�=r3c��tjddgddgg��}tttjt|��dS�Nrrryr|)r+r�rVrWrr/r%r�s  r1�
test_1d_inputzTestBartlett.test_1d_input�s:���H�q�!�f�q�!�f�%�&�&���j�%�.�"�a�8�8�8�8�8r3N)rbrcrdr1rprar7r:rer3r1r#r#�s_������6�6�6�7�7�7�-�-�-�>�>�>�
9�9�9�9�9r3r#c�D�eZdZd�Zd�Zd�Zd�Zd�Zd�Zd�Z	d�Z
d	�Zd
S)�
TestLevenec
���tttttt
ttttg
}tj|�\}}t|dd��t|dd��dS)Ng���YpK�?r;g�I�{�]�?)
r%r&r'r(r)r*r+r,r-r.r�levener)rOr/r�r�s    r1r1zTestLevene.test_data�sW���B��B��B��B��C�8���,��%���4��A�1�1�5�5�5��D�/�1�5�5�5�5�5r3c���tjtttd���\}}tjtttdd���\}}t||��t||��dS)NrQ��center�trimmedr��rA�proportiontocut)rr>r%r&r'r)rO�W1�pval1�W2�pval2s     r1�
test_trimmed1zTestLevene.test_trimmed1�sk���L��R��F�;�;�;�	��E��L��R��I�14�6�6�6�	��E��B��#�#�#��E�5�)�)�)�)�)r3c��gd�}gd�}tj�d��tj�|��}t	j||dd���\}}t	j||dd���\}}t	j|dd�|dd�d	�
��\}}	t
||��t
||��t
||	��dS)N�g333333�?�@�@�@�@�@�@gY@�r�rM�@rNg@rOg@gi@��rB��?rCrr�rQr@)r+r�seed�permutationrr>r)
rOr]r�r��W0�pval0rErFrGrHs
          r1�
test_trimmed2zTestLevene.test_trimmed2�s���6�6�6��6�6�6��
�	���t����
�Y�
"�
"�1�
%�
%���L��A�i�16�8�8�8�	��E��L��Q�y�16�8�8�8�	��E��L��1�R�4��!�A�b�D�'�&�A�A�A�	��E��B��#�#�#��B��#�#�#��E�5�)�)�)�)�)r3c�`�tjddd��}tj�d��tj�|��}|dz}tj||d���\}}tj||d���\}}t||��t||��dS)	Nr�rrvrTryrQr@�median)r+r�rrVrWrr>r)rOr]r�r�rErFrGrHs        r1�test_equal_mean_medianz!TestLevene.test_equal_mean_median�s����K��A�r�"�"��
�	���t����
�Y�
"�
"�1�
%�
%��
�q�D���L��A�f�5�5�5�	��E��L��Q�x�8�8�8�	��E��B��#�#�#��E�5�)�)�)�)�)r3c�x�tjddd��}tttj||d���dS�Nr�rrv皙�����?)�portiontocut)r+r�rV�	TypeErrorrr>r�s  r1�test_bad_keywordzTestLevene.test_bad_keyword�s6���K��A�r�"�"���i���q�!�#�F�F�F�F�F�Fr3c�x�tjddd��}tttj||d���dS�Nr�rrv�trimr@)r+r�rVrWrr>r�s  r1�test_bad_center_valuez TestLevene.test_bad_center_value�s6���K��A�r�"�"���j�%�,��1�V�D�D�D�D�D�Dr3c�H�tttjdg��dSrn)rVrWrr>rXs r1�test_too_few_argszTestLevene.test_too_few_args�s���j�%�,���4�4�4�4�4r3c
���tttttt
ttttg
}tj|�}d}t||��dSr4)
r%r&r'r(r)r*r+r,r-r.rr>rr5s    r1raz!TestLevene.test_result_attributes�sC���B��B��B��B��C�8���l�D�!��,�
��C��,�,�,�,�,r3c��tjddgddgg��}tttjt|��dSr9)r+r�rVrWrr>r%r�s  r1r:zTestLevene.test_1d_input�s:���H�q�!�f�q�!�f�%�&�&���j�%�,��A�6�6�6�6�6r3N)rbrcrdr1rIrZr]rcrgrirar:rer3r1r<r<�s�������6�6�6�*�*�*�*�*�*�$*�*�*�G�G�G�E�E�E�5�5�5�-�-�-�7�7�7�7�7r3r<c��eZdZdZedd���Zd�Zd�Zd�Zd	�Z	d
�Z
ej�
ejdkd�
��d���ZdS)�TestBinomTestPzU
    Tests for stats.binomtest as a replacement for deprecated stats.binom_test.
    Nr��	two-sidedc��tj|���tj��}t	|��dkr|d|dz}|d}nVt	|��dkr4|d}|�||krtd���tj|��}ntd���t
j||||���}|jS)Nrrrzn must be >= xzIncorrect length for x.�r@r)	r+�
atleast_1d�astype�int_r�rWr�	binomtestr�)r]r�r@r�results     r1�binom_test_funczTestBinomTestP.binom_test_func�s���
�M�!���#�#�B�G�,�,���q�6�6�Q�;�;��!��q��t��A��!��A�A�
��V�V�q�[�[��!��A��y�A��E�E� �!1�2�2�2����
�
�A�A��6�7�7�7����A���D�D�D���}�r3c��|�dd��}t|dd��|�dd��}t|dd��|�dd	gd
���}t|dd��dS)
Nr��g��-�"�^?r�rRi�g`���w�?i�����?�r@g���z�?�rvr)rOr�s  r1r1zTestBinomTestP.test_datas����#�#�C��-�-���D�"7��<�<�<��#�#�C��-�-���D�"5�r�:�:�:��#�#�S�#�J�#�#�6�6���D�"5�r�:�:�:�:�:r3c�@�tt|jgd���dS�Nrx�rVrWrvrXs r1�test_bad_len_xzTestBinomTestP.test_bad_len_xs!���j�$�"6�	�	�	�B�B�B�B�Br3c�z�tt|jdg��tt|jdgd���dS)Nr�r��r�rrXs r1�
test_bad_nzTestBinomTestP.test_bad_ns=��	�j�$�"6���>�>�>��j�$�"6����D�D�D�D�D�Dr3c�D�tt|jddgd���dS)Nr�rLr{rrXs r1�
test_bad_pzTestBinomTestP.test_bad_ps3���j��*�R��H��	=�	=�	=�	=�	=�	=r3c��|�dddd���}t|d��|�dddd���}t|d��|�dddd	���}t|d
��dS)Nr���gUUUUUU�?rrpg-$�Ǻl�?rgs0�,j.�?rng�JfJ"f�?r|�rOr_s  r1�test_alternativesz TestBinomTestP.test_alternatives"s����"�"�2�s�c�v�"�F�F���C�!2�3�3�3��"�"�2�s�c�y�"�I�I���C�!4�5�5�5��"�"�2�s�c�{�"�K�K���C�!3�4�4�4�4�4r3lz32-bit does not overflow��reasonc�`�tt|jddtjj���dS)NrOr8r{)rV�
OverflowErrorrv�sys�
float_info�minrXs r1�test_boost_overflow_raisesz)TestBinomTestP.test_boost_overflow_raises,s+��	�m�T�%9�3��S�^�EW�X�X�X�X�X�Xr3)Nr�rn)rbrcrd�__doc__�staticmethodrvr1r�r�r�r�r#r$�skipifr��maxsizer�rer3r1rmrm�s��������������\��";�;�;�C�C�C�E�E�E�=�=�=�5�5�5��[�����u�,�5O��P�P�Y�Y�Q�P�Y�Y�Yr3rmc��eZdZdZd�Zd�Zd�Zd�Zej	�
dgd���d���Zej	�
dgd	���d
���Zej	�
dgd���d
���Z
ej	�
dgd���d���Zej	�
dgd�gd�gd�gd�gd�gd�gd�gd�gd�gd�gd�gd�gd�gd�gd �gd!�gd"�gd#�gd$�gd%�gd&�gd'�gd(�gd)�g��d*���Zd+�Zej	�
d,d-d.g��d/���Zd0�Zd1�Zd2�Zd3�Zd4S)5�
TestBinomTestzTests for stats.binomtest.c�r�d}tjddd��}t|jd|���tjddd��}t|jd|���tjd	dd��}t|jd
|���tjddd��}t|jdd
���dS)N绽���|�=i�Ιi@o@���Q��?rrAi�ΙgX7�����?i	ϙg��=����?iϙg��ybw��?rE�rrtr	r��rOrBr_s   r1�test_two_sided_pvalues1z%TestBinomTest.test_two_sided_pvalues19s������o�h��$�7�7����
�C�d�3�3�3�3��o�h��$�7�7����
�O�$�?�?�?�?��o�h��$�7�7����
�O�$�?�?�?�?��o�h��$�7�7����
�O�$�?�?�?�?�?�?r3c��d}tjddd���}t|jd|���tjddd��}t|jd	|���tjd
dd��}t|jd|���tjddd��}t|jd
|���tjddd��}t|jd|���tjddd��}t|jd|���tjddd��}t|jd|���dS)Nr�r9rvr��r�r@gX��-h�?rAr|g�㤶y��?r�g	��y�}�?r;g+����,�?ryr�r�g�?rg�������?g{�G�z�?g333333�?g��<,��?r�r�s   r1�test_two_sided_pvalues2z%TestBinomTest.test_two_sided_pvalues2Gs7�����o�a�2��.�.�.����
�O�$�?�?�?�?��o�a��T�*�*����
�$5�D�A�A�A�A��o�b�"�d�+�+����
�O�$�?�?�?�?��o�a��T�*�*����
�O�$�?�?�?�?��o�a��R�(�(����
�G�$�7�7�7�7��o�a��B�'�'����
�D�t�4�4�4�4��o�a��B�'�'����
�F��6�6�6�6�6�6r3c�&�d}tjddd��}t|jd|���tjddd	��}t|jd|���tjd
dd��}t|jd|���tjd
dd��}t|jd|���tjdd
d��}t|jd|���tjddd��}t|jd|���dS)Nr�i�i�r�rrAryr}gAL� &W�?r=r�g!Y�B�?r[rgg�E]t��?r;r8r�r�r�s   r1�test_edge_caseszTestBinomTest.test_edge_casesXs	�����o�c�3��,�,����
�A�D�1�1�1�1��o�a��T�*�*����
�A�D�1�1�1�1��o�b�"�e�,�,����
�A�D�1�1�1�1��o�b�"�e�,�,����
�A�D�1�1�1�1��o�a��S�)�)����
�A�D�1�1�1�1��o�a��S�)�)����
�A�D�1�1�1�1�1�1r3c	����d�d�d}tjtj��z���dz��}tj�|����}tjtj�|����|kd���}t��fd�|tj��z�����}�|z
t|tj�|����k��z}t||d�	��d
}tjtj
��z��dz��}tj�|����}tjtj�|����|kd���}t��fd�|dtj
��z����}|dz}t||d�	��dS)Nr�r�ryrr��axisc�H��tj�|����Sr*�r�binomr�r�r�r@s ��r1�<lambda>z>TestBinomTest.test_binary_srch_for_binom_tst.<locals>.<lambda>ts ���+0�;�?�?�2�q�!�+D�+D�*D�r3rErAr;c�F��tj�|����Sr*r�r�s ��r1r�z>TestBinomTest.test_binary_srch_for_binom_tst.<locals>.<lambda>�s���*/�+�/�/�"�a��*C�*C�r3)r+r\�ceilrr�r�sumr�intr	�floor)	rO�kr`�d�y1�ix�y2r�r@s	       @@r1�test_binary_srch_for_binom_tstz,TestBinomTest.test_binary_srch_for_binom_tstgs�����
����
���I�b�g�a�!�e�n�n�a��c�*�*���K�O�O�A�q�!�$�$��
�V�E�K�O�O�A�q�!�,�,��1��
:�
:�
:��
)�+E�+E�+E�+E�+E�+,�"�b�g�a�!�e�n�n�a�A�A����V�c�!�u�{���r�1�a�8�8�8�9�9�
9����B�T�*�*�*�*�
���I�b�h�q�1�u�o�o��)�*�*���K�O�O�A�q�!�$�$��
�V�E�K�O�O�A�q�!�,�,��1��
:�
:�
:��
)�+D�+D�+D�+D�+D�*+�Q����Q����A�A���!�V����B�T�*�*�*�*�*�*r3z"alternative, pval, ci_low, ci_high))rg�$W��?r�g����?)rg}���?g�GI~�?r)rng�}�G��?g�>�?6�?g4�M����?c��tjddd|���}t|j|d���t	|jd��|�d�	��}t|j|jf||fd���dS)
Nrdr�r��r�r@rrrA皙�����?r��confidence_level�	rrtr	r�r
rL�
proportion_ci�low�high�rOrr��ci_low�ci_highr_�cis       r1�test_confidence_intervals1z(TestBinomTest.test_confidence_intervals1�s����o�b�C�4�[�I�I�I����
�D�u�5�5�5�5��S�]�C�(�(�(�
�
�
��
�
5�
5�������)�F�G�+<�5�I�I�I�I�I�Ir3))rga&�+w?r�g�ny��?)rg5!Rx��?g���5�%�?r)rng�~th�?gSu�R&|?g|KnG��?c��tjddd|���}t|j|d���t	|jd��|�d�	��}t|j|jf||fd���dS)
Nryr�r�r����ư>rAg���Q��?r%r�r�r�s       r1�test_confidence_intervals2z(TestBinomTest.test_confidence_intervals2�s����o�a�2��+�F�F�F����
�D�t�4�4�4�4��S�]�D�)�)�)�
�
�
��
�
5�
5�������)�F�G�+<�4�H�H�H�H�H�Hr3zalternative, pval, ci_high))rg�‹�լ?g̓A��?)rrr)rng�E�ow�?g;Kڞj��?c���tjddd|���}t|j|d���|�d���}t|jd	��t|j|d���dS)
Nrr�r�rpr�rArr�r�)rrtr	r�r�r
r�r�)rOrr�r�r_r�s      r1�!test_confidence_interval_exact_k0z/TestBinomTest.test_confidence_interval_exact_k0�sx���o�a��t��E�E�E����
�D�t�4�4�4�4�
�
�
��
�
5�
5���R�V�S�!�!�!�����t�4�4�4�4�4�4r3zalternative, pval, ci_low))rrr�)r�6�����>g@>}_��?)rnr�gcڒ�� �?c���tjddd|���}t|j|d���|�d���}t|jd��t|j|d���dS)	Nr�r�rpr�rArr�r)rrtr	r�r�r
r�r�)rOrr�r�r_r�s      r1�%test_confidence_interval_exact_k_is_nz3TestBinomTest.test_confidence_interval_exact_k_is_n�sx���o�b�"��+�F�F�F����
�D�t�4�4�4�4�
�
�
��
�
5�
5���R�W�c�"�"�"�����T�2�2�2�2�2�2r3z+k, alternative, corr, conf, ci_low, ci_high)ryrnTrg`����?g�����?)ryrnTr%grt�!�?g����/��?)ryrnFrgVK�5��?g5d��M�?)ryrnFr%g�XCVu^�?g*�r�[��?)ryrTrr�g�P-��V�?)ryrTr%r�g^'����?)ryrFrr�gS5�a���?)ryrFr%r�g�����?)ryrTrg!Շ`ⰸ?r)ryrTr%g��$x"�?r)ryrFrg�W�~=�?r)ryrFr%g	��L��?r)rrnTrr�g�X����?)rrnFrr�g�N����?)rrTrr�g/�<7#9�?)rrFrr�gA�A�?)rrTrr�r)rrFrr�r)r�rnTrg��7���?r)r�rnFrg�؍�s�?r)r�rTrr�r)r�rFrr�r)r�rTrgh�adn��?r)r�rFrg��AB�/�?rc��tj|dd|���}|rd}nd}|�||���}	t|	j|	jf||fd���dS)	Nr�r`r��wilsoncc�wilson)r��methodr�rA)rrtr�r	r�r�)
rOr�r�corr�confr�r�r_r�r�s
          r1�test_ci_wilson_methodz#TestBinomTest.test_ci_wilson_method�sq��>�o�a�2��+�F�F�F���	��F�F��F�
�
�
��V�
�
D�
D�������)�F�G�+<�4�H�H�H�H�H�Hr3c��tjddd��}t|jd��t|jd��dS)Nr|r�r�r)rrtr
rLr�r�s  r1�&test_estimate_equals_hypothesized_propz4TestBinomTest.test_estimate_equals_hypothesized_prop�sA���o�a��T�*�*���S�]�D�)�)�)��S�Z��%�%�%�%�%r3zk, n)rr)r�rc��tjtd���5tj||��ddd��dS#1swxYwYdS)Nz must be an integer not less thanr�r#rrWrrt)rOr�r�s   r1�test_invalid_k_nzTestBinomTest.test_invalid_k_n�s���
�]�:�!C�E�E�E�	"�	"��O�A�q�!�!�!�	"�	"�	"�	"�	"�	"�	"�	"�	"�	"�	"�	"����	"�	"�	"�	"�	"�	"s�?�A�Ac��tjtd���5tjddd��ddd��dS#1swxYwYdS)Nzk must not be greater than nrr�r�r�r�rXs r1�test_invalid_k_too_bigz$TestBinomTest.test_invalid_k_too_big�s���
�]�:�!?�A�A�A�	*�	*��O�B��D�)�)�)�	*�	*�	*�	*�	*�	*�	*�	*�	*�	*�	*�	*����	*�	*�	*�	*�	*�	*s�A�A�Ac���tjddd���}tjtd���5|�d���ddd��dS#1swxYwYdS)	Nryr�r`r�zmust be in the intervalrr�r��rrtr#rrWr�r�s  r1�test_invalid_confidence_levelz+TestBinomTest.test_invalid_confidence_level�s����o�a�2��-�-�-��
�]�:�-F�
G�
G�
G�	3�	3����r��2�2�2�	3�	3�	3�	3�	3�	3�	3�	3�	3�	3�	3�	3����	3�	3�	3�	3�	3�	3��A�A�Ac���tjddd���}tjtd���5|�d���ddd��dS#1swxYwYdS)	Nryr�r`r�zmethod must berzplate of shrimp�r�r�r�s  r1�test_invalid_ci_methodz$TestBinomTest.test_invalid_ci_method�s����o�a�2��-�-�-��
�]�:�-=�
>�
>�
>�	8�	8����%6��7�7�7�	8�	8�	8�	8�	8�	8�	8�	8�	8�	8�	8�	8����	8�	8�	8�	8�	8�	8r�c�h�tjddd���}t|j|j��dS)Nryr�r`r�)rrtr
�proportion_estimaterLr�s  r1�
test_aliaszTestBinomTest.test_aliass3���o�a�2��-�-�-���S�,�c�m�<�<�<�<�<r3N)rbrcrdr�r�r�r�r�r#r$r%r�r�r�r�r�r�r�r�r�r�r�rer3r1r�r�2s������$�$�@�@�@�7�7�7�"
2�
2�
2�+�+�+�@�[���A�H�H�H�I�I�J�J�I�I�J��[���A�F�F�F�G�G�I�I�G�G�I��[���9�D�D�D�E�E�5�5�	E�E�5��[���8�F�F�F�G�G�3�3�	G�G�3��[���5�	=�	=�	=�	;�	;�	;�	<�	<�	<�	=�	=�	=�	0�	0�	0�	0�	0�	0�	1�	1�	1�	1�	1�	1�	4�	4�	4�	4�	4�	4�	4�	4�	4�	5�	5�	5�	5�	5�	5�	6�	6�	6�	0�	0�	0�	0�	0�	0�	-�	-�	-�	.�	.�	.�	6�	6�	6�	7�	7�	7�	+�	+�	+�	,�	,�	,�	4�	4�	4�	4�	4�	4�3	6���:I�I�;��:I�&�&�&��[���V�f�g�%6�7�7�"�"�8�7�"�
*�*�*�
3�3�3�
8�8�8�
=�=�=�=�=r3r�c�8�eZdZd�Zd�Zd�Zd�Zd�Zd�Zd�Z	dS)	�TestFlignerc�|�tjd��}ttj||dz��dd��dS)Nr}r)g�n��f�	@g��Dq��?r�)r+r\rr�fligner)rOr�s  r1r1zTestFligner.test_data
sC��
�Y�q�\�\��!�%�-��B��E�":�":�"L�"$�	&�	&�	&�	&�	&r3c�^�	�tj�d���	�	fd�}|t��}|t��}|t
��}t
j|||d���\}}t
j|||dd���\}}t||��t||��dS)Nr c���tj|��d��t|����zz���S)Nr�)r+�asarray�randnr��tolist)�gr�s �r1r�z+TestFligner.test_trimmed1.<locals>.<lambda>s7���b�j��m�m�e�B�H�H�S��V�V�4D�4D�.D�D�L�L�N�N�r3rQr@rBr�rC)	r+rrr%r&r'rr�r)
rO�_perturb�g1_�g2_�g3_�Xsq1rF�Xsq2rHr�s
         @r1rIzTestFligner.test_trimmed1s�����Y�
"�
"�3�
'�
'��N�N�N�N���h�r�l�l���h�r�l�l���h�r�l�l���m�C��c�&�A�A�A���e��m�C��c�)�47�9�9�9���e��D�$�'�'�'��E�5�)�)�)�)�)r3c���gd�}gd�}tj||dd���\}}tj|dd�|dd�d�	��\}}t||��t||��dS)
NrKrRrBrUrCrr�rQr@)rr�r)rOr]r�r�rFr�rHs       r1rZzTestFligner.test_trimmed2!s���6�6�6��6�6�6���m�A�q��49�;�;�;���e��m�A�a��d�G�Q�q��t�W�V�D�D�D���e��D�$�'�'�'��E�5�)�)�)�)�)r3c�x�tjddd��}tttj||d���dSr_)r+r�rVrbrr�r�s  r1rczTestFligner.test_bad_keyword;s6���K��A�r�"�"���i����1�3�G�G�G�G�G�Gr3c�x�tjddd��}tttj||d���dSre)r+r�rVrWrr�r�s  r1rgz!TestFligner.test_bad_center_value?s6���K��A�r�"�"���j�%�-��A�f�E�E�E�E�E�Er3c�H�tttjdg��dSrn)rVrWrr�rXs r1�test_bad_num_argszTestFligner.test_bad_num_argsCs���j�%�-�!��5�5�5�5�5r3c��tjd��}ttjtjft	j||dzg����dS)Nr}r)r+r\r
r�rr�r�s  r1r7zTestFligner.test_empty_argGs@���I�a�L�L���b�f�b�f�%�u�}�Q��1��b�'A�'A�B�B�B�B�Br3N)
rbrcrdr1rIrZrcrgr�r7rer3r1r�r�s�������&�&�&�*�*�*� 
*�
*�
*�4H�H�H�F�F�F�6�6�6�C�C�C�C�Cr3r�c#�K�gd�}gd�}t|��D]�\}}tj�|��}|�d��}|�ddd���}|�ddd���}t||��D])\}}	t
|dz||	z��D]
}
||||
<��*|�|��tj|d	��\}}||d
g||�RV���dS)N))g]Y��C��gQ����ɣ?)g�m�e�5�gϵ���5�?)g��;j�g��H��?)iF�ei�HcMi%�:r�rr�r})r�r�r�rrr)	�	enumerater+rr�integers�zip�range�shuffle�split)
�expected_results�seeds�sirVr�xy�tie_ind�num_ties_per_indr`r��jr]r�s
             r1�mood_cases_with_tiesrLs;����@�@�@��
.�-�-�E��e�$�$�2�2���D��i�#�#�D�)�)��
�Z�Z��_�_���,�,�1�2�A�,�6�6���<�<�A�A�A�<�>�>����!1�2�2�	�	�D�A�q��1�q�5�!�a�%�(�(�
�
���1���1���
�	���B�����x��A�����1���F�1�-�b�1�1�1�1�1�1�1�2�2r3c�"�eZdZej�de����d���Zej�dgd���d���Zd�Z	d�Z
d�Zd	�Zd
�Z
d�Zej�dgd
���d���ZdS)�TestMoodz$x,y,alternative,stat_expect,p_expectc��tj|||���\}}t||d���t||d���dS)am
        Example code used to generate SAS output:
        DATA myData;
        INPUT X Y;
        CARDS;
        1 0
        1 1
        1 2
        1 3
        1 4
        2 0
        2 1
        2 4
        2 9
        2 16
        ods graphics on;
        proc npar1way mood data=myData ;
           class X;
            ods output  MoodTest=mt;
        proc contents data=mt;
        proc print data=mt;
          format     Prob1 17.16 Prob2 17.16 Statistic 17.16 Z 17.16 ;
            title "Mood Two-Sample Test";
        proc print data=myData;
            title "Data for above results";
          run;
        r
g��ؗ�Ҝ<r5N�r�moodr	)rOr]r�r�stat_expect�p_expectrLr�s        r1�test_against_SASzTestMood.test_against_SASdsO��<"�J�q�!��E�E�E��	�6���Y�U�;�;�;�;���&�u�5�5�5�5�5�5r3zalternative, expected))rn)�3oX�Q�?g�p��L��?)r)rgԣ�l�?)r)rg�p��L��?c�d�gd�}gd�}tj|||���}t||��dS)Nr�r�r
r)rOrrr]r�r_s      r1�test_against_SAS_2zTestMood.test_against_SAS_2�sP��
<�
<�
<��
=�
=�
=���j��A�;�7�7�7����X�&�&�&�&�&r3c�N�tj�d��tj�dd��}tj�dd��}t	j||��\}}t	j||��\}}t
||g||g��dS)NrTr�rr[)r+rrVr�rr
r)rOr�r��z1�p1�z2�p2s       r1�test_mood_order_of_argsz TestMood.test_mood_order_of_args�s���	�	���t����
�Y�_�_�R��
#�
#��
�Y�_�_�R��
#�
#����B��#�#���B���B��#�#���B�!�2�r�(�b�S�"�I�6�6�6�6�6r3c���gd�}gd�}tj|��}tj|��}d|_d|_tt	j||d���ddg��dS)N)g����g���۟��?gy�<7x��g1�)E��?g9c�ç�?g�
��FA�g<=ӟ	2�?g�{�![��?g�_�l�?g�ڻ{�ӿg�GSsA0�?g��1��?g��3��g�����gL���?g�_l����g�,*.���gː5��3�?gvN��qG�?g��W=�?)gd�"���g�`Y#��?g�^���g�?g�N���g�B�
a���gp5_�%�?g��(����?gv�]�l�οg��g��?g�����g�r�6ʺ�?g��̄j�?g��K��!ٿg���{��g�&����?g*h��}�g��Z��?g�y�BU�?g .�ۋ4�?g��|w;��?g�&ck��@gR�j�2�g�}+w(o�?g����@F�?g�}�_�9t?g �=��g	�S��?gx�
(�gg7��Y�?g�{ih��?)r�r)r[rr�g�4���g^�}�E	�?)r+r��shaperrr
)rOr�r�s   r1�test_mood_with_axis_nonez!TestMood.test_mood_with_axis_none�s���5�5�5��	H�	H�	H���X�b�\�\��
�X�b�\�\��������!�%�*�R��$�"?�"?�"?�#1�=�"A�	C�	C�	C�	C�	Cr3c���d}tj�d��tj�d|��}tj�d|��}t	j||��\}}t
|��D]G}t||||gt	j|dd�|f|dd�|f�����H|���}|���}t	j||d���\}}t
|��D]G}t||||gt	j||dd�f||dd�f�����HdS)Nr}rTr�r[rr�)	r+rrVr�rr
r�r�	transpose)rO�nyr�r��	z_vectest�pval_vectestrr`s        r1�test_mood_2dzTestMood.test_mood_2d�sp����
�	���t����
�Y�_�_�R��
$�
$��
�Y�_�_�R��
$�
$��"'�*�R��"4�"4��	�<��r���	F�	F�A�%�y��|�\�!�_�&E�&+�j��A�A�A�q�D��2�a�a�a��d�8�&D�&D�
F�
F�
F�
F��\�\�^�^��
�\�\�^�^��"'�*�R��!�"<�"<�"<��	�<��r���	F�	F�A�%�y��|�\�!�_�&E�&+�j��A�q�q�q�D��2�a����d�8�&D�&D�
F�
F�
F�
F�	F�	Fr3c
��d}tj�d��tjj|�}tjj|�}t	d��D�]}tj|||���\}}ddgddgddgf}t	|||d��D]�}t	|||d��D]�}	|dkr|dd�||	f}
|dd�||	f}n;|dkr||dd�|	f}
||dd�|	f}n|||	dd�f}
|||	dd�f}t|||	f|||	fgtj|
|����������dS)N)r�r}r8rTryr�rrr)r+rrVr�r�rr
r)rOrr�r�r�r r!�axes_idxr`r�slice1�slice2s            r1�test_mood_3dzTestMood.test_mood_3d�s�����
�	���t����
�Y�_�e�
$��
�Y�_�e�
$���!�H�H�	J�	J�D�&+�j��R�d�&C�&C�&C�#�I�|��A���A���A��/�H��5��$���!2�3�4�4�
J�
J���u�X�d�^�A�%6�7�8�8�
J�
J�A��q�y�y�!#�A�A�A�q�!�G���!#�A�A�A�q�!�G��������!#�A�q�q�q�!�G���!#�A�q�q�q�!�G����!#�A�q�!�!�!�G���!#�A�q�!�!�!�G���-�y��A���/;�A�q�D�/A�/C�.3�j���.H�.H�J�J�J�J�
J�
J�
	J�	Jr3c�J�tttjdgg��dSrn)rVrWrr
rXs r1�test_mood_bad_argzTestMood.test_mood_bad_arg�s"��	�j�%�*�q�c�2�6�6�6�6�6r3c��tj�d��tj�dd���}tj�dd���}tj||d���\}}tj||d���\}}tj||d	���\}}||cxkr|ksnJ�t|dd
���t||dz��t|d
|dzz
��tj	td���5tj||d���ddd��dS#1swxYwYdS)Nrrzr�)r�r�g�?rnr
rr�H�����z>r5rrzalternative must be...rz	ekki-ekki)r+rrVrr�r.r
r	r#rrW)	rOr]r��stat1r�stat2r�stat3�p3s	         r1�test_mood_alternativezTestMood.test_mood_alternative�s���
�	���q�����J�N�N��C�N�0�0���J�N�N��C�N�0�0���J�q�!��=�=�=�	��r��J�q�!��8�8�8�	��r��J�q�!��;�;�;�	��r���&�&�&�&��&�&�&�&�&�&���A�D�)�)�)�)���B�q�D�!�!�!���A��1��H�%�%�%�
�]�:�-E�
F�
F�
F�	6�	6��J�q�!��5�5�5�5�	6�	6�	6�	6�	6�	6�	6�	6�	6�	6�	6�	6����	6�	6�	6�	6�	6�	6s�D<�<E�Er)rnrrc��tj�d��}|�d��}|�d��}t	j|||���}t
|j|jf|��dS)Nl	��hTt�f�U�6H~�)r�r)r[rr
)	r+rrr�rr
r
rLr�)rOrrr�r�r_s      r1�test_resultzTestMood.test_resultss���i�#�#�$K�L�L��
�
 �
 ��
)�
)��
�
 �
 ��
)�
)���j��R�[�9�9�9���c�m�S�Z�0�#�6�6�6�6�6r3N)rbrcrdr#r$r%rrrrrr"r'r)r0r2rer3r1r
r
cs4�������[���C�1�1�3�3�5�5�6�6�5�5�6�@�[���4�@�@�@�A�A�'�'�A�A�'�7�7�7�C�C�C�8F�F�F�.J�J�J�87�7�7�
6�6�6�$�[���]�,L�,L�,L�M�M�7�7�N�M�7�7�7r3r
c�~�eZdZd�Zd�Zd�Zej�e	d���d���Z
d�Zd�Zd	�Z
d
S)�TestProbplotc�@�tj�dd���}tj|d���\}}gd�}t	|tj|����t	||��tj|d���\}}gd�}t	||��dS)	Nrd�90rF�r�)g���+�/��g�Y��3��g��t���g�5Z�0�gF�j�g�!�2��gsR���{ܿg�7��Կg>�h���ǿgc�z隇��gc�z隇�?g>�h����?g�7���?gsR���{�?g�!�2��?gF�j�?g�5Z�0�?g��t���?g�Y��3�?g���+�/�?T)gƝV����?g�����?g�à��?)rr�r.�probplotr	r+�sort)rOr]�osm�osr�osm_expectedr_�res_fit�res_fit_expecteds        r1rTzTestProbplot.test_basics����J�N�N���N�7�7���>�!��/�/�/���S�G�G�G��
	��R�W�Q�Z�Z�(�(�(���\�*�*�*��~�a�T�2�2�2���W�?�?�?����!1�2�2�2�2�2r3c��tj�dd���}tj|dd���\}}tj|dd���\}}tj|dd���\}}t	||��t	||��t	||��t	||��tj|dd���\}}	dS)Nr�i@�rF��sparamsr�rre)rr�r.r8r	)
rOr]�osm1�osr1�osm2�osr2�osm3�osr3r:r;s
          r1�test_sparams_keywordz!TestProbplot.test_sparams_keyword!s����J�N�N��&�N�9�9���^�A�t��?�?�?�
��d��^�A�q�e�<�<�<�
��d��^�A�r�u�=�=�=�
��d���d�#�#�#���d�#�#�#���d�#�#�#���d�#�#�#��>�!�R�U�;�;�;���S�S�Sr3c�b�tj�dd���}tj|ddd���\}}tj|dtjd���\}}t||��t||��t
ttj|d�	��t
ttj|g�	��Gd
�d��}tj|dd�
��\}}tj||��d���\}}t||��t||��dS)Nrdr6rF�t)ry)r�rrAzwrong-dist-namerc��eZdZdZd�ZdS)�3TestProbplot.test_dist_keyword.<locals>.custom_distz6Some class that looks just enough like a distribution.c�D�tj�|d���S)Nr)r�)rr��ppf)rO�qs  r1rNz7TestProbplot.test_dist_keyword.<locals>.custom_dist.ppf;s���z�~�~�a�Q�~�/�/�/r3N)rbrcrdr�rNrer3r1�custom_distrL9s)������H�H�
0�
0�
0�
0�
0r3rP)rr@)rr�)	rr�r.r8rJr	rVrW�AttributeError)rOr]rBrCrDrErPs       r1�test_dist_keywordzTestProbplot.test_dist_keyword/s0���J�N�N���N�7�7���^�A�5�s�D�I�I�I�
��d��^�A�5�u�w��M�M�M�
��d���d�#�#�#���d�#�#�#��j�%�.�!�:K�L�L�L�L��n�e�n�a�b�A�A�A�A�	0�	0�	0�	0�	0�	0�	0�	0�
�^�A�t��?�?�?�
��d��^�A�K�K�M�M�u�E�E�E�
��d���d�#�#�#���d�#�#�#�#�#r3�
no matplotlibr�c��tj��}|�d��tj�ddd���}tj|t���\}}tj��tj|d���\}}tj|dt���}tj��tj|dd���}tt|��t|��cxko-t|��cxkot|��cxkod	knc��t||��t||��t||��t||��tj��}|�d��}	tj|d|	���tj��dS)
Nr�ryr���tr��plotF)r�rWr)�plt�figure�add_subplotrrJr.r8�closerr�r	)
rO�figr]�res1�fitres1�res2�fitres2�res3�res4�axs
          r1�test_plot_kwargzTestProbplot.test_plot_kwargCs����j�l�l����������G�K�K���'�K�:�:����q�s�3�3�3�
��g��	������q�t�4�4�4�
��g��~�a�U��5�5�5���	�����~�a�U��6�6�6��	��D�	�	�S��Y�Y�E�E�E�E�#�d�)�)�E�E�E�E�s�4�y�y�E�E�E�E�A�E�E�E�E�F�F�F���d�#�#�#���d�#�#�#���d�#�#�#����)�)�)��j�l�l��
�_�_�S�
!�
!��
��q�e�"�-�-�-�-��	�����r3c�L�tttjdgd���dSr)rVrWrr8rXs r1�test_probplot_bad_argsz#TestProbplot.test_probplot_bad_args\s$���j�%�.�1�#�<M�N�N�N�N�N�Nr3c�`�ttjgd���tjg��tjg��f��ttjgd���tjg��tjg��ftjtjdff��dS)NFr7Tr�)r
rr8r+r�r�rXs r1�
test_emptyzTestProbplot.test_empty`s����U�^�B�E�2�2�2��h�r�l�l�B�H�R�L�L�1�	3�	3�	3��U�^�B�D�1�1�1��x��|�|�R�X�b�\�\�2��v�r�v�s�+�-�	.�	.�	.�	.�	.r3c�.�tjd���5ttjdgd���tjdg��tjdg��ftjtjdff��ddd��dS#1swxYwYdS)Nr�)�invalidrTr7r�)r+r�r
rr8r�r�rXs r1�test_array_of_size_onez#TestProbplot.test_array_of_size_onegs���
�[��
*�
*�
*�	2�	2�������6�6�6��8�R�D�>�>�2�8�Q�C�=�=�9��6�2�6�3�/�1�
2�
2�
2�	2�	2�	2�	2�	2�	2�	2�	2�	2�	2�	2�	2����	2�	2�	2�	2�	2�	2s�A'B
�
B�BN)rbrcrdrTrHrRr#r$r��have_matplotlibrdrfrhrkrer3r1r4r4s�������
3�
3�
3�<�<�<�$�$�$�(�[���O�+�O��D�D���E�D��0O�O�O�.�.�.�2�2�2�2�2r3r4c��eZdZd�Zd�Zd�Zd�Zd�Zd�Zd�Z	d�Z
d	�Zd
�Zd�Z
ej�dgd
�gd�gd�g��d���Zd�ZdS)�TestWilcoxonc�4�tttjdgddg��tttjddgddgd��tttjddgddgd���tttjdgdzd���dS)Nrr�dummyr
r��xyz��mode)rVrWr�wilcoxonrXs r1�test_wilcoxon_bad_argz"TestWilcoxon.test_wilcoxon_bad_argos���	�j�%�.�1�#��1�v�>�>�>��j�%�.�1�a�&�1�a�&�'�J�J�J��j�%�.�1�a�&�1�a�&�")�	+�	+�	+�	+��j�%�.�1�#�b�&�u�E�E�E�E�E�Er3c�
�tjd��}tttj||dd���tttj||dd���t
t	j||dd���d��dS)Nrd�wilcox�approxrr�pratt�zsplit)g@Z@r)r+r\rVrWrrtr
r�s  r1�test_zero_diffzTestWilcoxon.test_zero_diffxs����I�b�M�M���j�%�.�!�Q��#�	%�	%�	%�	%��j�%�.�!�Q��#�	%�	%�	%�	%�	�U�^�A�q�(��B�B�B�#�	%�	%�	%�	%�	%r3c���gd�}gd�}t��5}|�td���tj||dd���}ddd��n#1swxYwYt|d��dS)	Nr�)rrryr}�Sample size too smallr�ryrx��zero_methodrs)r�g�|��N�?)rr�r�rrtr	)rOr]r�r�r_s     r1�
test_prattzTestWilcoxon.test_pratt�s���
�L�L���L�L��
�
 �
 �	K�C��J�J�{�,C�J�D�D�D��.��A�7��J�J�J�C�	K�	K�	K�	K�	K�	K�	K�	K�	K�	K�	K����	K�	K�	K�	K�	��7�8�8�8�8�8s�5A�A�Ac��gd�}tj|dd���}tj|dd���}tj|dd���}dS)N)rrryrr�ryrrrrrryrxr~rzrw)rrt)rO�arrr�s   r1�test_wilcoxon_arg_typez#TestWilcoxon.test_wilcoxon_arg_type�sU��1�0�0���N�3�G�(�C�C�C���N�3�H�8�D�D�D���N�3�H�8�D�D�D���r3c�f�gd�}tdd��}tjd�t||��D����}tj|j��}t
j||dd���\}}t|d��t|d	��t
j||d
d���\}}t|d��t|d��t
j||d
d���\}}t|d��t|d��tj	gd���}tj	gd���}t
j||dd���\}}t|d��t|dd���t
j||dd���\}}t|d��t|dd���dS)N)	rr|r�r[r<r|r}rr���r}c� �g|]\}}|g|z��Srere)rv�ur�s   r1rwz7TestWilcoxon.test_accuracy_wilcoxon.<locals>.<listcomp>�s"��@�@�@���1�Q�C�!�G�@�@�@r3ryrxrri�g.����i?rzi�gFYv�]Uj?rwiGg%��^Ez?�r�r�rJr����yr��r���p����w�r�����r��sr�r��F��
correctionrsrUg^��҂<�?r�rATg:�b�+�?)r�r+�concatenater��zerosr�rrtr	r�r
)rO�freq�numsr]r�r0r@s       r1�test_accuracy_wilcoxonz#TestWilcoxon.test_accuracy_wilcoxon�s���,�,�,���R��|�|���N�@�@��D�$���@�@�@�A�A���H�Q�V�����~�a��G�(�;�;�;���1���3������0�1�1�1��~�a��H�8�<�<�<���1���3������0�1�1�1��~�a��H�8�<�<�<���1���3������+�,�,�,�
�H�Q�Q�Q�R�R���H�Q�Q�Q�R�R���~�a��u�8�D�D�D���1��Q�������9�4�0�0�0�0��~�a��t�(�C�C�C���1��Q�������9�4�0�0�0�0�0�0r3c��tjgd���}tjgd���}tj||dd���}d}t	||��dS)Nr�r�Frxr�r)r+r�rrtr)rOr]r�r_r^s     r1�test_wilcoxon_result_attributesz,TestWilcoxon.test_wilcoxon_result_attributes�sa���H�Q�Q�Q�R�R���H�Q�Q�Q�R�R���n�Q��e�(�C�C�C��,�
��C��,�,�,�,�,r3c���tj�d��}|�d��|�d��}}tj||d���}tj�|jdz��}t|j	|��tj||d���}t|d��rJ�tj||��}t|d��rJ�dS)Nl�	m$Sr[rxrrr�exact�
zstatistic)r+rrrrtr�rNr�r	r��hasattr)rOrr]r�r_r�s      r1�test_wilcoxon_has_zstatisticz)TestWilcoxon.test_wilcoxon_has_zstatistic�s����i�#�#�K�0�0���z�z�"�~�~�s�z�z�"�~�~�1���n�Q���1�1�1���j�n�n�S�Z��\�*�*������,�,�,��n�Q���0�0�0���3��-�-�-�-�-��n�Q��"�"���3��-�-�-�-�-�-�-r3c��tjdgdzd���\}}d}t|d��t||d���tjdgdzd	d�
��\}}d}t|d��t||d���dS)Nr`r�rxrrg4I�ťY?rr�rATr�gc��2_?)rrtr
r	)rO�statr@�
expected_ps    r1�test_wilcoxon_tiezTestWilcoxon.test_wilcoxon_tie�s����.�#����(�;�;�;���a� �
��T�1������:�D�1�1�1�1��.�#�����8�L�L�L���a� �
��T�1������:�D�1�1�1�1�1�1r3c��gd�}gd�}t��5}|�td���tj||dd���\}}ddd��n#1swxYwYt|d��t
|d	d
���t��5}|�td���tj||ddd�
��\}}ddd��n#1swxYwYt|d��t
|dd
���t��5}|�td���tj||dd���\}}ddd��n#1swxYwYt|d��t
|dd
���t��5}|�td���tj||ddd�
��\}}ddd��n#1swxYwYt|d��t
|dd
���dS)N)
�}r�ro�r�r�r�r�r��)
r��zr�r�r��|r �r��r}r�rrx�rrsrwg��13}��?r8rHT)rr�rsg7�a��%�?rg����?g�ڏoJT�?)rr�r�rrtr
r)rOr]r�r�r�r@s      r1�
test_onesidedzTestWilcoxon.test_onesided�s2��
?�>�>��>�>�>��
�
 �
 �	K�C��J�J�{�,C�J�D�D�D��>�!�Q�F��J�J�J�D�A�q�	K�	K�	K�	K�	K�	K�	K�	K�	K�	K�	K����	K�	K�	K�	K�	�Q������A�y�!�4�4�4�4�
�
 �
 �	1�C��J�J�{�,C�J�D�D�D��>�!�Q�F�t�'/�1�1�1�D�A�q�	1�	1�	1�	1�	1�	1�	1�	1�	1�	1�	1����	1�	1�	1�	1�	�Q������A�y�!�4�4�4�4�
�
 �
 �	N�C��J�J�{�,C�J�D�D�D��>�!�Q�I�H�M�M�M�D�A�q�	N�	N�	N�	N�	N�	N�	N�	N�	N�	N�	N����	N�	N�	N�	N�	�Q������A�y�!�4�4�4�4�
�
 �
 �	1�C��J�J�{�,C�J�D�D�D��>�!�Q�I�$�'/�1�1�1�D�A�q�	1�	1�	1�	1�	1�	1�	1�	1�	1�	1�	1����	1�	1�	1�	1�	�Q������A�y�!�4�4�4�4�4�4sG�8A�A�"A�9C�C�"C�8E�E�!E�9G�G�!Gc��tdd��D]v}t|��}t|��}t||dzzdzdzt	|����tt|��d��t
||���wdS)Nrr�r)r�rrr
r�r�r)rOr��pmf1�pmf2s    r1�test_exact_basiczTestWilcoxon.test_exact_basics����q�"���	2�	2�A�&�q�)�)�D�'��*�*�D���A�a�C����Q���D�	�	�2�2�2���T���A�&�&�&�%�d�D�1�1�1�1�	2�	2r3c���tjgd���}tjgd���}tj||dd���\}}t	|dd���tj||d	d���\}}t	|d
d���tj||dd���\}}t	|dd���tjd
d��dz}tjdd
d��}tj||dd���\}}t	|dd���tj||d	d���\}}t	|dd���tj||dd���\}}t	|dd���dS)N)
g�(\���?g=
ףp=�?g�(\���?g���Q�޿g�Q����?g{�G�z�?g�p=
ף�gq=
ףp�?g�g�Q���?)
r�g�������?g������ɿg333333�?皙�����g������ܿg�z�G��g���Q�οg�(\���gR���Q�rnr�r�g֔���?r8rHrgYJ����?rg��*�?rrdr�r�g�(���?g�(���?g�S���?)r+r�rrtrr\)rOr]r�r�r@s     r1�test_exact_pvalzTestWilcoxon.test_exact_pval	s����H�#�#�#�
$�
$���H�$�$�$�
%�
%���~�a���'�J�J�J���1��A�y�!�4�4�4�4��~�a���W�E�E�E���1��A�y�!�4�4�4�4��~�a��	��H�H�H���1��A�z�1�5�5�5�5��I�a����s�"���I�b�!�R� � ���~�a���'�J�J�J���1��A�y�!�4�4�4�4��~�a���W�E�E�E���1��A�y�!�4�4�4�4��~�a��	��H�H�H���1��A�y�!�4�4�4�4�4�4r3r])r�r�ry)r�r���r�r})r�r�ryr�ri����r;r<c���tj|��\}}tj|��}||dk���}t||��t|d��dS�Nrr)rrtr+r�r�r
)rOr]r�r@�wtrues     r1�test_exact_p_1zTestWilcoxon.test_exact_p_1%s_���~�a� � ���1��H�Q�K�K���!�a�%��������Q������Q������r3c��tjdd��dz}tjddd��}ttj||��tj||d�����tjdd��}t��5}|�td�	��tj|��\}}ddd��n#1swxYwYttj|d
���||f��tjdd��}ttj|��tj|d
�����dS)
Nrrir�r�r�rrr=zExact p-value calculationr�rxrr�)r+r\r
rrtrr�r�)rOr]r�r�r�r�r@s       r1�	test_autozTestWilcoxon.test_auto/sa���I�a����s�"���I�b�!�R� � ���U�^�A�q�)�)��^�A�q�w�7�7�7�	9�	9�	9�
�I�a�����
�
 �
 �	%�C��J�J�{�,G�J�H�H�H��>�!�$�$�D�A�q�	%�	%�	%�	%�	%�	%�	%�	%�	%�	%�	%����	%�	%�	%�	%�	�U�^�A�H�5�5�5��1�v�>�>�>�
�I�a������U�^�A�&�&���q�x�(H�(H�(H�I�I�I�I�Is�
4C
�
C�CN)rbrcrdrur{r�r�r�r�r�r�r�r�r�r#r$r%r�r�rer3r1rnrnns(������F�F�F�	%�	%�	%�9�9�9�E�E�E�1�1�1�:-�-�-�.�.�.�2�2�2�&$5�$5�$5�L2�2�2�5�5�5�8�[���S�;�;�;�#5�#5�#5�#@�#@�#@�#B�C�C���C�C��J�J�J�J�Jr3rnc�&�eZdZd�Zd�Zd�Zd�ZdS)�	TestKstatc���tj�d��tj�d����fd�dD��}gd�}t	||d���tj�d�	��}tj�d
�	��}tj�d�	��}t	|||f|dd�d
d���dS)Ni�}r6c�:��g|]}tj�|����Sre)r�kstat)rvr�rPs  �r1rwz>TestKstat.test_moments_normal_distribution.<locals>.<listcomp>Fs%���>�>�>�A�5�;�t�Q�'�'�>�>�>r3r�)g�H��Q,�?glZ)rI�?g88�}���?gc��JN�?r�rAr)�momentrryr�g{�G�z�?�{�G�z�?r�)r+rrVr�r	rr�)rO�momentsr�m1�m2�m3rPs      @r1� test_moments_normal_distributionz*TestKstat.test_moments_normal_distributionCs����
�	���u�����y���u�%�%��>�>�>�>��>�>�>��>�>�>������5�5�5�5��\�$�q�
)�
)�
)��
�\�$�q�
)�
)�
)��
�\�$�q�
)�
)�
)����R���h�s��s�m�$�T�J�J�J�J�J�Jr3c�F�tttjg��dSr*)rVrWrr�rXs r1rYzTestKstat.test_empty_inputQs���j�%�+�r�2�2�2�2�2r3c��tjd��}tj|d<tt	j|��tj��dS�NrCr8�r+r\r�r
rr�ros  r1r�zTestKstat.test_nan_inputT�<���y��~�~���&��Q���U�[��&�&���/�/�/�/�/r3c�|�tjd��}dD]$}tttj||����%dS)Nr�)rg/�$@r�)r+r\rVrWrr��rOrPr�s   r1�test_kstat_bad_argzTestKstat.test_kstat_bad_argZsD���y��}�}���	>�	>�A��*�e�k�4�1�=�=�=�=�=�	>�	>r3N)rbrcrdr�rYr�r�rer3r1r�r�BsS������K�K�K�3�3�3�0�0�0�>�>�>�>�>r3r�c� �eZdZd�Zd�Zd�ZdS)�TestKstatVarc�F�tttjg��dSr*�rVrWr�kstatvarrXs r1rYzTestKstatVar.test_empty_inputbs���j�%�.�"�5�5�5�5�5r3c��tjd��}tj|d<tt	j|��tj��dSr�r�ros  r1r�zTestKstatVar.test_nan_inputer�r3c�T�dg}d}tttj||���dS)Nrr�r�r�r�s   r1rpzTestKstatVar.test_bad_argks-���s�����j�%�.�$�!�<�<�<�<�<�<r3N)rbrcrdrYr�rprer3r1r�r�asA������6�6�6�0�0�0�=�=�=�=�=r3r�c�x�eZdZd�Zd�Zd�Zej�e	d���d���Z
d�Zd�Zd	S)
�TestPpccPlotc�:�tddd���dz|_dS�Nr}��rUr�r2r]rXs r1�setup_methodzTestPpccPlot.setup_methods�!��"�1�3�W�E�E�E��I����r3c��d}tj|jdd|���\}}gd�}t|t	jdd|�����t||��dS)Nr}���r���N)g�9%�	�?gx�M� _�?g";9���?g�®X�Z�?g�����?��num)r�	ppcc_plotr]r	r+r�)rOr��svals�ppcc�
ppcc_expecteds     r1rTzTestPpccPlot.test_basicvsm��
���o�d�f�c�2��;�;�;���t�%�%�%�
���r�{�3���:�:�:�;�;�;���m�,�,�,�,�,r3c�f�tj|jddd���\}}tj|jddtj���\}}t	||d���t	||d���tj|jdd��\}}t	||d���t	||d���dS)Nr�r��tukeylambdarg#B����;rA)rr�r]r�r	)rO�svals1�ppcc1�svals2�ppcc2�svals3�ppcc3s       r1�	test_distzTestPpccPlot.test_dist~s��������R�m�L�L�L�
��������R�-2�->�@�@�@�
������U�3�3�3�3���u�5�1�1�1�1������R�8�8�
������U�3�3�3�3���u�5�1�1�1�1�1�1r3rSr�c�N�tj��}|�d��}tj|jddt���|�|��|�d��}tj|jdd|���tj��dS�Nr�i��rdrV)rXrYrZrr�r]�delaxesr[�rOr\rcs   r1rdzTestPpccPlot.test_plot_kwarg�s����j�l�l��
�_�_�S�
!�
!��
�����R�c�2�2�2�2����B�����_�_�S�
!�
!��
�����R�b�1�1�1�1��	�����r3c��tttj|jdd��tttjgd�ddd���dS)Nrrrxrr)rVrWrr�r]rXs r1�test_invalid_inputsz TestPpccPlot.test_invalid_inputs�sR���j�%�/�4�6�1�a�@�@�@�	�j�%�/�9�9�9�a��,�	.�	.�	.�	.�	.�	.r3c���tjgdd��\}}t|tjddd�����t|tjdt�����dS)Nrrr�r�r�)rr�r	r+r�r��float)rOr�r�s   r1rhzTestPpccPlot.test_empty�sa���o�b�!�Q�/�/���t���r�{�1�a�R�8�8�8�9�9�9���b�h�r��7�7�7�8�8�8�8�8r3N)
rbrcrdr�rTr�r#r$r�rlrdr�rhrer3r1r�r�rs�������J�J�J�-�-�-�
2�
2�
2��[���O�+�O��D�D�
�
�E�D�
�.�.�.�9�9�9�9�9r3r�c�&�eZdZd�Zd�Zd�Zd�ZdS)�TestPpccMaxc�P�dg}tttj|d���dSr)rVrWr�ppcc_maxros  r1�test_ppcc_max_bad_argz!TestPpccMax.test_ppcc_max_bad_arg�s)���s���j�%�.�$�=N�O�O�O�O�O�Or3c��tj�ddddd���dz}ttj|��dd	�
��dS)N�ffffff�rr��'��r����@��{���r;rH)rr�r.rrr�s  r1�test_ppcc_max_basiczTestPpccMax.test_ppcc_max_basic�sX����!�!�$�A�S�u�/6�
"�
8�
8�:=�
>���E�N�1�-�-�/C�Q�O�O�O�O�O�Or3c�T�tj�ddddd���dz}tj|d�	��}tj|tj�	��}t	|d
d���t	|d
d���tj|��}t	|d
d���dS)
Nrrr�rrr�r	r�rr
r}rH)rr�r.rr)rOr]�max1�max2�max3s     r1r�zTestPpccMax.test_dist�s�����!�!�$�A�S�u�/6�
"�
8�
8�:=�
>���~�a�m�4�4�4���~�a�e�&7�8�8�8���D�"6��B�B�B�B��D�"6��B�B�B�B��~�a� � ���D�"6��B�B�B�B�B�Br3c�0�tj�ddddd���dz}tttj|d�	��t
tj|d
�	��dd�
��t
tj|d�	��dd�
��dS)Nrrr�rrr�r	)r�rr�)�brack)rrr
r;rH)r�r)rr�r.rVrWrrr�s  r1�
test_brackzTestPpccMax.test_brack�s�����!�!�$�A�S�u�/6�
"�
8�
8�:=�
>���j�%�.�!�?�K�K�K�K��E�N�1�F�;�;�;�0�!�	=�	=�	=�	=�	�E�N�1�G�<�<�<�0�!�	=�	=�	=�	=�	=�	=r3N)rbrcrdrrr�rrer3r1rr�sY������P�P�P�
P�P�P�
C�C�C�	=�	=�	=�	=�	=r3rc�,�eZdZd�Zd�Zd�Zd�Zd�ZdS)�TestBoxcox_llfc�&�tj�ddd���}d}tj||��}|jdztjtj|���dz����z}t||��dS)Nrr��1��r�r�r�rrLr)
rr�r.�
boxcox_llfr�r+r,r�rSr	)rOr]�lmbda�llf�llf_expecteds     r1rTzTestBoxcox_llf.test_basic�sz���J�N�N��2�E�N�B�B������u�a�(�(����w��|�b�f�R�V�A�E�E�G�G�Q�J�-?�-?�&@�&@�@����\�*�*�*�*�*r3c���tj�ddd���}d}tj||��}tj|t	|����}t||d���dS�Nr�r�rrrrrA)rr�r.r�listr	�rOr]rr�llf2s     r1�test_array_likezTestBoxcox_llf.test_array_like�se���J�N�N���%�N�@�@������u�a�(�(�����t�A�w�w�/�/����T��.�.�.�.�.�.r3c��tj�ddd���}d}tj||��}tj|t	j||g��j��}t||g|d���dSr)rr�r.rr+�vstackr0r	rs     r1�
test_2d_inputzTestBoxcox_llf.test_2d_input�sv��
�J�N�N���%�N�@�@������u�a�(�(�����r�y�!�Q��'8�'8�':�;�;����c�
�D�u�5�5�5�5�5�5r3c�n�ttjtjdg������dSrn)rr+�isnanrrrXs r1rhzTestBoxcox_llf.test_empty�s-������)�!�R�0�0�1�1�2�2�2�2�2r3c�X�gd�}tjd|��}t|d��dS)N)g�h@� m@r(g�x@i����g#:�x�1�)rrr	)rOrPrs   r1�test_gh_6873zTestBoxcox_llf.test_gh_6873�s8��,�+�+����r�4�(�(����/�0�0�0�0�0r3N)rbrcrdrTr!r$rhr)rer3r1rr�s_������+�+�+�/�/�/�6�6�6�3�3�3�1�1�1�1�1r3r(�iU>iϵi��i_�
i7�i��is�iv�i+�i�*i-qir:i�i��i�i�i�_i4�i��
iD�i�i�,iE�i\�i�i�|ig�iMiӢi[�i!�iif6iP�i�ii�Ii/�iF&i�i�i�`#i�i�i�&ii��i�Qi�ai��
iXZiw�i_�i�miDi�-iB�i:Ci��	i�'i�i %i�xi�iz_i��Qi!#i�=i� ixi�
iU7i_i1kiW9i&�ix�i�Ii��iT�i
�i�i��irvim�i͟i�iO�iv�i)�i�P
i4�i=ii�i�Ui��i�ig>ix	i`i�|i�0i�`iM	i�'i�[i8i�i3[i�i)�
i`Ni
�	i��i�[i:�+ip�i	%i
�i�Vi[$iX�i_�ib�ieZi��i`?	it�i��i��i�Ei�iTiy�i��imvi<i�Zi�!i��i(�ir
i��i�iɪi|�i��i>ii�i�i*
i�#ixi�
i&i�i#�i1gi�Pi�)iۆi�{%i�?i۽i�|i�i�&i��-iư)i+ci7�iQi	oi�&i|_i�y
i6i��i��i��i:Ji#ji8�i��i%�ii%Wi(�i`�i:��inai�i��i�%i$i4�iA�(i�qi��i�"i�yi�XZi��izri��i�$i;�is�igYi��i�7i�p	ip�iL6"i�DHi��i!3i�iI�(i|�Ciшi��i�miSi��i{8i��i�i��i�Pi2Li��(i@is�iu�i2i_�i�ji)mig�Zi�_iu�	i�9ik�i';i�ii�iK�i܃i��i=iv�i=�i�
ia ir�i��i�i}Wi*�i��i�Jii�i2�i�UFiNj	iV,"i�piA�i%�i��iv�9iI�5i��i�T-i� i8~i?e`i1�
iV�i/Xi=
i�4i�iحi^�i��i/�igi(i�i|B;i�pi�ic�iE'i�Ei$
i�i�{i�ki:�i�IiA_i��iPioPi6_i�iܲ
iO	iA7iL�i�Ai�iV�iwmi�iZhi|�ViԇCiw�
i&�i~i��i'�i!�i�i��
i�}i�ei3
i��i��i_�i�i@�)ižiV�ii�iF�i��?i�&
i"Yi�wi��i�HiSirdi�bi@�i�KiA�;iz|id+i��	i�
8i"�i�	iOi�_i$�li!9iii�TiuLiQ'%iqpiC�i]K@i��!i�i��iIp|i��i?	Bi�i�io�0i��i
�
i��i�5%i��i�i�*i�i�iv0i��*i�*iwi�ivei$i��i-�iD~i-�i��i,4i�bi��i�iP0i)�i�mi�	i�Ri�Qi
�i��i�c���eZdZd�Zd�Zd�Zd�Zd�Zd�Ze	j
�dgd���d	���Zd
�Z
e	j
�dgd���d
���Zd�ZdS)�
TestBoxcoxc�f�tddd���dz}tj|d���}t||dz
��tj|d���}t|dd|zz
��tj|d���}t|t	j|����tjt
|��d���}t|t	j|����tjt	jd	��d
��}t|t	j	d	����dS)Nr}r�r6rr�rr�rr�r)
r2rr�r	r+r,rr�r
r�)rOr]�xts   r1�test_fixed_lmbdazTestBoxcox.test_fixed_lmbda)s���a�b�u�=�=�=��A��
�\�!�1�
%�
%�
%����A��E�"�"�"�
�\�!�2�
&�
&�
&����A��!��G�$�$�$�
�\�!�1�
%�
%�
%����B�F�1�I�I�&�&�&��\�$�q�'�'��
+�
+�
+����B�F�1�I�I�&�&�&��\�"�'�"�+�+�q�
)�
)���R���"���&�&�&�&�&r3c��d}tj�ddd���}||zdz|z}tj|��\}}t	|d|zd�	��dS)
Ng@r�iP�i�)r�r�r�rr�rrH)rr�r.r�r)rOrr]�x_invr.r�s      r1�test_lmbda_NonezTestBoxcox.test_lmbda_None;sh�����J�N�N�r��D�N�A�A���U��Q��5�&�)���\�%�(�(�
��F��F�B��J��:�:�:�:�:�:r3c��tj�d��}tdd|���dz}t	j|d���\}}}t
|ddg��t	j|d	���\}}}t
|d
dg��tdd
|���dz}t	j|d���\}}}t
|ddg��t	j|d���\}}}t
|ddg��dS)NrTr}r�rrz)�alphag�x��@gv�[%�@rgxˉE�k�?g8B�hk @r;r�r[r�g���\��?gx��̡�&@rg��7�(U@g�p:sY@)r+rrr2rr�r	)rOrr]r�rjs     r1�
test_alphazTestBoxcox.test_alphaEs���i�#�#�D�)�)���a�b�s�;�;�;�a�?����a�t�4�4�4���1�h���#4�6G�"H�I�I�I���a�t�4�4�4���1�h���#5�7H�"I�J�J�J�
�a�c��<�<�<�r�A����a�u�5�5�5���1�h���9�k�":�;�;�;���a�u�5�5�5���1�h���:�z�":�;�;�;�;�;r3c�D�tjddg��}tttj|��tttjtjdg����tttjtjdgdgg����dS)Nr�rr)r+r�rVrWrr�r�s  r1�test_boxcox_bad_argzTestBoxcox.test_boxcox_bad_argVss���H�b�!�W�����j�%�,��2�2�2��j�%�,���!��
�
�>�>�>��j�%�,���1�#��s��0D�0D�E�E�E�E�Er3c�Z�ttjg��jdk��dS�N)r)rrr�rrXs r1rhzTestBoxcox.test_empty_s(�����R� � �&�$�.�/�/�/�/�/r3c�b�tjt��\}}t|dd���dS)NgsHj�dr��r�rA�rr��_boxcox_datar	)rOr��lams   r1r)zTestBoxcox.test_gh_6873bs2����l�+�+���3�	��Y�T�2�2�2�2�2�2r3�bounds�)r�r)r�r)r�r�c����fd�}tjtd|���\}}�d|cxkr�dksnJ�dS)Nc�2��tj|�d���S�N�bounded)r>r��r
�minimize_scalar��funr>s �r1�	optimizerzBTestBoxcox.test_bounded_optimizer_within_bounds.<locals>.optimizerm�$����+�C��3<�>�>�>�
>r3�rrHrr)rr�r<)rOr>rHr�rs `   r1�$test_bounded_optimizer_within_boundsz/TestBoxcox.test_bounded_optimizer_within_boundsjsk���	>�	>�	>�	>�	>��<��D�I�N�N�N���5��a�y�5�,�,�,�,�6�!�9�,�,�,�,�,�,�,�,r3c�����tjtd���\}}|dz|dzf�ddi���fd�}tjtd|���\}}||ksJ�t|�d��dS)	Nr-r`r�xatolrc�4��tj|�d����S)NrC)r>r��optionsrD)rGr>rOs ��r1rHzPTestBoxcox.test_bounded_optimizer_against_unbounded_optimizer.<locals>.optimizers*����+�C��3<�g�O�O�O�
Or3rJrr;)rOr�rrH�
lmbda_boundedr>rOs     @@r1�2test_bounded_optimizer_against_unbounded_optimizerz=TestBoxcox.test_bounded_optimizer_against_unbounded_optimizerts�����
�<��D�9�9�9���5��#�+�u�q�y�)���E�"��	O�	O�	O�	O�	O�	O�
!�<��D�2;�=�=�=���=���%�%�%�%��
�v�a�y�1�1�1�1�1r3rH)�str)rrr`c��tjtd���5tjt
d|���ddd��dS#1swxYwYdS)Nz`optimizer` must be a callablerrJ�r#rrWrr�r<�rOrHs  r1�$test_bad_optimizer_type_raises_errorz/TestBoxcox.test_bad_optimizer_type_raises_error�s����]�:�-M�
N�
N�
N�	H�	H��L��T�Y�G�G�G�G�	H�	H�	H�	H�	H�	H�	H�	H�	H�	H�	H�	H����	H�	H�	H�	H�	H�	Hs�A�A
�
A
c��d�}d}tjt|���5tjt
d|���ddd��dS#1swxYwYdS)Nc��dSrnre)rGs r1rHzCTestBoxcox.test_bad_optimizer_value_raises_error.<locals>.optimizer�s���1r3z;`optimizer` must return an object containing the optimal...rrJrT)rOrHr�s   r1�%test_bad_optimizer_value_raises_errorz0TestBoxcox.test_bad_optimizer_value_raises_error�s���
	�	�	�P��
�]�:�W�
5�
5�
5�	H�	H��L��T�Y�G�G�G�G�	H�	H�	H�	H�	H�	H�	H�	H�	H�	H�	H�	H����	H�	H�	H�	H�	H�	Hs�A�A�AN)rbrcrdr/r2r5r7rhr)r#r$r%rKrQrVrYrer3r1r+r+'s������'�'�'�$;�;�;�<�<�<�"F�F�F�0�0�0�3�3�3��[���X�'F�'F�'F�G�G�-�-�H�G�-�2�2�2�*�[���[�*>�*>�*>�?�?�H�H�@�?�H�

H�
H�
H�
H�
Hr3r+c���eZdZd�Zd�Zd�Zd�Zej�	dgd���ej�	dgd���d	�����Z
d
�Zd�ZdS)
�TestBoxcoxNormmaxc�:�tddd���dz|_dS�Nr}r�r6rr�rXs r1r�zTestBoxcoxNormmax.setup_method��!��"�1�2�E�B�B�B�Q�F����r3c�\�tj|j��}t|dd���dS)N�m��?r�rA�r�boxcox_normmaxr]r	�rOr�s  r1�
test_pearsonrzTestBoxcoxNormmax.test_pearsonr�s/���%�d�f�-�-�����t�4�4�4�4�4�4r3c��tj|jd���}t|dd���tj|j��\}}t||��dS)N�mler���	��.!�?r�rA)rrbr]r	r�)rOr�r��
maxlog_boxcoxs    r1�test_mlezTestBoxcoxNormmax.test_mle�s\���%�d�f�U�;�;�;�����t�4�4�4�4�!�<���/�/���=��
�v�.�.�.�.�.r3c�d�tj|jd���}t|ddgd���dS)Nr�r�r`rgr�rAra)rO�
maxlog_alls  r1�test_allzTestBoxcoxNormmax.test_all�s9���)�$�&��?�?�?�
��
�X�x�$8�t�D�D�D�D�D�Dr3r�)rf�pearsonrr�r>r?c�����fd�}tj|j||���}tj�d|k��sJ�tj|�dk��sJ�dS)Nc�2��tj|�d���SrBrDrFs �r1rHzITestBoxcoxNormmax.test_bounded_optimizer_within_bounds.<locals>.optimizer�rIr3)r�rHrr)rrbr]r+r�)rOr�r>rHr�s  `  r1rKz6TestBoxcoxNormmax.test_bounded_optimizer_within_bounds�s����	>�	>�	>�	>�	>��%�d�f�V�09�;�;�;���v�f�Q�i�&�(�)�)�)�)�)��v�f�v�a�y�(�)�)�)�)�)�)�)r3c�,���tj|j��}tj|d��}tj|dz
|dzd���Gd�d�����fd�}tj|j|���}||ksJ�t
||d��dS)	Nr}r�i�c��eZdZdS)�?TestBoxcoxNormmax.test_user_defined_optimizer.<locals>.MyResultN)rbrcrdrer3r1�MyResultrr�s�������Dr3rsc���g}�D] }|�||�����!���}�tj|��|_|Sr*)�appendr+�argminr])rG�objsrr_rs�lmbda_ranges    ��r1rHz@TestBoxcoxNormmax.test_user_defined_optimizer.<locals>.optimizer�sX����D�$�
(�
(�����C�C��J�J�'�'�'�'��(�*�*�C���	�$���0�C�E��Jr3)rHr�)rrbr]r+�roundr�r	)rOr�
lmbda_roundedrH�lmbda2rsrxs     @@r1�test_user_defined_optimizerz-TestBoxcoxNormmax.test_user_defined_optimizer�s������$�T�V�,�,������*�*�
��k�-��"4�m�D�6H�$�O�O��	�	�	�	�	�	�	�	�	�	�	�	�	�	��%�d�f�	�B�B�B�����������t�,�,�,�,�,r3c��tj}tj|jd|���tjtd���5tj|jd|���ddd��dS#1swxYwYdS)N)rrHz,`brack` must be None if `optimizer` is givenr)g�rL)r
rErrbr]r#rrWrUs  r1�2test_user_defined_optimizer_and_brack_raises_errorzDTestBoxcoxNormmax.test_user_defined_optimizer_and_brack_raises_error�s����,�	�	��T�V�4�9�E�E�E�E�
�]�:�.D�E�E�E�	6�	6�
� ���{�+4�
6�
6�
6�
6�	6�	6�	6�	6�	6�	6�	6�	6�	6�	6�	6�	6����	6�	6�	6�	6�	6�	6s�A.�.A2�5A2N)
rbrcrdr�rdrirlr#r$r%rKr|r~rer3r1r[r[�s�������G�G�G�5�5�5�/�/�/�E�E�E��[���X�'A�'A�'A�B�B��[���X�'F�'F�'F�G�G�	*�	*�H�G�C�B�	*�-�-�-�,6�6�6�6�6r3r[c�r�eZdZd�Zd�Zej�ed���d���Z	d�Z
d�ZdS)	�TestBoxcoxNormplotc�:�tddd���dz|_dSr�r�rXs r1r�zTestBoxcoxNormplot.setup_method�r�r3c��d}tj|jdd|���\}}gd�}t|t	jdd|�����t||��dS)Nr}r�r�r�)g!�X4�}�?g��΁i��?g�-�	15�?gW��$��?g\ܱ�{��?r�)r�boxcox_normplotr]r	r+r�)rOr��lmbdasr�r�s     r1rTzTestBoxcoxNormplot.test_basic�sn��
���,�T�V�S�"��B�B�B����%�%�%�
�����C��� ;� ;� ;�<�<�<���m�,�,�,�,�,r3rSr�c�N�tj��}|�d��}tj|jddt���|�|��|�d��}tj|jdd|���tj��dSr�)rXrYrZrr�r]r�r[r�s   r1rdz"TestBoxcoxNormplot.test_plot_kwarg�s����j�l�l��
�_�_�S�
!�
!��
��d�f�c�2�C�8�8�8�8����B�����_�_�S�
!�
!��
��d�f�c�2�B�7�7�7�7��	�����r3c��tttj|jdd��tttjddgdd��dS)Nrrr�)rVrWrr�r]rXs r1r�z&TestBoxcoxNormplot.test_invalid_inputs�s@���j�%�"7����A�F�F�F��j�%�"7�"�a��!�Q�G�G�G�G�Gr3c�^�ttjgdd��jdk��dSr�)rrr�r�rXs r1rhzTestBoxcoxNormplot.test_emptys-����%�b�!�Q�/�/�4��9�:�:�:�:�:r3N)rbrcrdr�rTr#r$r�rlrdr�rhrer3r1r�r��s�������J�J�J�-�-�-��[���O�+�O��D�D�
�
�E�D�
�H�H�H�;�;�;�;�;r3r�c� �eZdZd�Zd�Zd�ZdS)�TestYeojohnson_llfc���tj�ddd���}d}tj||��}tj|t	|����}t||d���dS)Nr�rrrrrrA)rr�r.�yeojohnson_llfrr	rs     r1r!z"TestYeojohnson_llf.test_array_like	se���J�N�N����N�?�?�����"�5�!�,�,���#�E�4��7�7�3�3����T��.�.�.�.�.�.r3c��tj�ddd���}d}tj||��}tj|t	j||g��j��}t||g|d���dSr)rr�r.r�r+r#r0r	rs     r1r$z TestYeojohnson_llf.test_2d_inputst���J�N�N���%�N�@�@�����"�5�!�,�,���#�E�2�9�a��V�+<�+<�+>�?�?����c�
�D�u�5�5�5�5�5�5r3c�n�ttjtjdg������dSrn)rr+r&rr�rXs r1rhzTestYeojohnson_llf.test_emptys-������-�a��4�4�5�5�6�6�6�6�6r3N)rbrcrdr!r$rhrer3r1r�r�sA������/�/�/�6�6�6�7�7�7�7�7r3r�c�J�eZdZd�Zej�dgd���d���Zd�Zd�Z	ej�de
je
jg��d���Z
ej�de
je
je
je
jg��d	���Zd
�ZdS)�TestYeojohnsonc�p�tj�d��}tdd|���dz}tj|dk��sJ�tj|d���}t||��tj|d���}t|dd|dzzz
��tj|d���}t|tj|dz����tj|d���}t||��tdd|���dz
}tj|dk��sJ�tj|d	���}t|tj|dz����tj|d���}t||��tj|d
���}t|d|dzzdz
��tdd|���d	z
}tj|dk��rJ�tj|dk��rJ�|dk}tj|d���}t||||��tj|d���}t||dd||dzzz
��tj|d���}t||tj||dz����tj|d���}t||||��|}tj|d	���}t||tj||dz����tj|d���}t||||��tj|d
���}t||d||dzzdz
��dS)Nr6r}r�rrrr-r�rry)	r+rrr2r�r�
yeojohnsonr	r,)rOrr]r.�pos�negs      r1r/zTestYeojohnson.test_fixed_lmbdas����i�#�#�E�*�*��
�a�b�s�;�;�;�a�?���v�a�!�e�}�}����
�
�a�q�
)�
)�
)����A����
�
�a�r�
*�
*�
*����A��Q��U��O�,�,�,�
�
�a�q�
)�
)�
)����B�F�1�q�5�M�M�*�*�*�
�
�a�q�
)�
)�
)����A����
�a�b�s�;�;�;�a�?���v�a�!�e�}�}����
�
�a�q�
)�
)�
)����R�V�Q�B��F�^�^�O�,�,�,�
�
�a�q�
)�
)�
)����A����
�
�a�q�
)�
)�
)����A�!��a��L�1�,�-�-�-�
�a�b�s�;�;�;�a�?���6�!�a�%�=�=� � � ��6�!�q�&�>�>�!�!�!��1�f��
�
�a�q�
)�
)�
)����3���3��(�(�(�
�
�a�r�
*�
*�
*����3���Q�!�C�&�1�*�%5�!5�6�6�6�
�
�a�q�
)�
)�
)����3�����#���
�!3�!3�4�4�4�
�
�a�q�
)�
)�
)����3���3��(�(�(��d��
�
�a�q�
)�
)�
)����3��"�&�!�C�&��1��"5�"5�!5�6�6�6�
�
�a�q�
)�
)�
)����3���3��(�(�(�
�
�a�q�
)�
)�
)����3���q��v�g��k�!2�Q�!6�7�7�7�7�7r3r)rr`r�rc��d�}d}tj�d��tj�dd|���}|||��}t	j|��\}}t
||d���tdtj�	||z
��|zd	�
��td|�
��d�
��td|���d�
��dS)Nc�0�tj|j|j���}|dk}t	|��tjd��kr!tj||��dz
||<n*tj|||zdzd|z��dz
||<t	|dz
��tjd��kr4dtjd|z
||zdzdd|z
z��z
||<n#dtj||��z
||<|S)Nr�rrrr)r+r�rr��abs�spacing�exp�power)r]rr1r�s    r1�_inverse_transformz:TestYeojohnson.test_lmbda_None.<locals>._inverse_transformQs���H�Q�W�A�G�4�4�4�E��q�&�C��5�z�z�B�J�r�N�N�*�*��V�A�c�F�^�^�a�/��c�
�
��X�a��f�u�n�q�&8�!�e�)�D�D�q�H��c�
��5�1�9�~�~��
�2���.�.��"�(�Q��Y�<�!�S�D�'�+A�A�+E�+,��E�	�?�#<�#<�<��s�d��� �"�&�!�S�D�'��"2�"2�2��s�d���Lr3i Nrrr)r�r�r�r�r5rrH)r+rrV�normalrr�r	r�linalgr�rQrS)rOrr��	n_samplesr]r1r.r�s        r1r2zTestYeojohnson.test_lmbda_NoneLs���
	�	�	�&�	�
�	���w�����I����!�9��>�>��"�"�1�e�,�,���%�e�,�,�
��F����D�1�1�1�1��A�r�y�~�~�a�"�f�5�5�	�A�1�M�M�M�M��A�r�w�w�y�y�!�4�4�4�4��A�r�v�v�x�x��3�3�3�3�3�3r3c�Z�ttjg��jdk��dSr9)rrr�rrXs r1rhzTestYeojohnson.test_emptyqs)���� ��$�$�*�d�2�3�3�3�3�3r3c���tj�ddd���}tj|��\}}tjt	|����\}}t||d���dS)Nr�rrrrrA)rr�r.r�rr	)rOr]�xt1r��xt2s     r1r!zTestYeojohnson.test_array_liketsd���J�N�N����N�?�?���!�!�$�$���Q��!�$�q�'�'�*�*���Q���S�u�-�-�-�-�-�-r3r�c���tjd|���}d}tjt|���5tj|��ddd��dS#1swxYwYdS)Nr8r�z>Yeo-Johnson transformation is not defined for complex numbers.r)r+r\r#rrWrr�)rOr�r]�err_msgs    r1�test_input_dtype_complexz'TestYeojohnson.test_input_dtype_complexzs����I�a�u�%�%�%����
�]�:�W�
5�
5�
5�	 �	 ���Q����	 �	 �	 �	 �	 �	 �	 �	 �	 �	 �	 �	 ����	 �	 �	 �	 �	 �	 s�A�A�Ac��tjd|���}tjdtj���}tj|��\}}tj|��\}}t||d���t||d���dS)Nr<r�r+rA)r+r\�float64rr�r	)rOr��x_int�x_float�xt_int�	lmbda_int�xt_float�lmbda_floats        r1�test_input_dtype_integerz'TestYeojohnson.test_input_dtype_integer�s����	�!�5�)�)�)���)�A�R�Z�0�0�0��!�,�U�3�3���	� %� 0�� 9� 9���+����t�4�4�4�4��	�;�T�:�:�:�:�:�:r3c���tjgd���}tj|��\}}tj|dz��\}}t||d���t||d���dS)N)��(���HA��G�.�"A� 5fAg�(��@Ag��V�AgH�z��uoAg)\�¿�+Ar�g�(\�^@Ag��RAr�g��#Ag\�µ�=Ag
ף��NAr�g�p=�o�HAg 6bAg{�c�sAg)\�tmhAg����/BAg'|�Ag�(\w�ipAg ݃hAg�Q��lBAgףp�@Arr�rA)r+r�rr�r�r	)rOr]�xt_yeo�lam_yeo�xt_box�lam_boxs      r1�test_input_high_variancez'TestYeojohnson.test_input_high_variance�s���H�"�"�"�
#�
#�� �*�1�-�-�����,�q�1�u�-�-�������T�2�2�2�2����t�4�4�4�4�4�4r3N)rbrcrdr/r#r$r%r2rhr!r+�	complex64�
complex128r��int8�uint8�int16�int32r�r�rer3r1r�r�s�������-8�-8�-8�^�[���W�n�n�n�5�5�"4�"4�6�5�"4�H4�4�4�.�.�.��[���W�r�|�R�]�&C�D�D� � �E�D� ��[���W�r�w���"�(�B�H�&M�N�N�;�;�O�N�;�5�5�5�5�5r3r�c� �eZdZd�Zd�Zd�ZdS)�TestYeojohnsonNormmaxc�:�tddd���dz|_dSr]r�rXs r1r�z"TestYeojohnsonNormmax.setup_method�r^r3c�\�tj|j��}t|dd���dS)Ng�����?r�rA)r�yeojohnson_normmaxr]r	rcs  r1rizTestYeojohnsonNormmax.test_mle�s/���)�$�&�1�1�����t�4�4�4�4�4�4r3c�h�gd�}tj|��}tj|dd���sJ�dS)N)gffffff@g������ �rrLgffffff�?g333333@rSgffffff@g������?g������@rQrMg������"@r�g�g�z�G��?r�r5)rr�r+�allclose)rOr]rs   r1�test_darwin_examplez)TestYeojohnsonNormmax.test_darwin_example�sF��
�
�
���(��+�+���{�5�%�d�3�3�3�3�3�3�3�3r3N)rbrcrdr�rir�rer3r1r�r��sD������G�G�G�5�5�5�4�4�4�4�4r3r�c�4�eZdZej�dejdfejdfej	dfg��d���Z
d�Zej�dejej
fejejfej	ejfg��d���Zd	�Zd
�Zd�Zej�dejdfejdfej	dfg��d���Zej�d
ejejej	g��d���Zej�d
ejejej	g��d���Zej�dejejddd�fejejddd�fej	ejddd�fg��d���Zej�dejdejgd���ejddg��d�fejdejgd���ejdzdzzejdd g��d�fej	d!ejgd"���ejd#d$g��d�fg��d%���Zej�dejdfejdfej	dfg��d&���Zej�d
ejejej	g��d'���Zej�d
ejejej	g��d(���Zej�d)d*d+dd,d-d.ejgejd*d+dd,d-d.ejgd/d0d1d2ejd3d4gg��g��ej�d
ejejej	g��d5�����Zej�d)d*d+dd,d-d.ejgejd*d+dd,d-d.ejgd/d0d1d2ejd3d4gg��g��ej�d
ejejej	g��d6�����Z d7�Z!d8�Z"d9�Z#d:S);�
TestCircFuncsztest_func,expectedgA:w��v�?g@��pz?gE�|�2@c�l�tjgd���}t||d���|d���dS�N��cr}rr�r��^�h�r�r+rA�r+r�r	�rO�	test_funcrr]s    r1�test_circfuncszTestCircFuncs.test_circfuncs�sD��

�H�.�.�.�/�/���	�	�!�#�.�.�.��t�D�D�D�D�D�Dr3c���tjgd���}|���}tj|d���}t||d���|tjzdz���}|dz}tj|d���}t||d���|�	��}tj
|d���}t||d���dS)	N)rdrvr[rW�g�4@g3333333@r�r�r�rAr�rLr�)r+r�rQr�circmeanr	�pirR�circvarrS�circstd)rOr]�M1�M2�V1�V2�S1�S2s        r1�test_circfuncs_smallz"TestCircFuncs.test_circfuncs_small�s����H�5�5�5�6�6��
�V�V�X�X��
�^�A�C�
(�
(�
(����B�T�*�*�*�*����g�c�k�
�
�
 �
 ���"�W��
�]�1�3�
'�
'�
'����B�T�*�*�*�*�
�U�U�W�W��
�]�1�3�
'�
'�
'����B�T�*�*�*�*�*�*r3ztest_func, numpy_funcc��tjdgdzdgdzz��}||��}||��}t||d���dS)Ng˛ɦv9�?r�g��u�v9�?r�g:�0�yU>r5r�)rOr��
numpy_funcr]�circstatr�s      r1�test_circfuncs_closez"TestCircFuncs.test_circfuncs_close�s_��
�H�)�*�R�/�3F�2G�#�2M�M�N�N���9�Q�<�<����A������&�t�4�4�4�4�4�4r3c���tjgd�gd�gd�g���tj�d���}tj����d���}t||d���tj�dd�	��}�fd
�t
�jd��D��}t||d���tj�dd�	��}�fd�t
�jd��D��}t||d���dS)
Nr���_r;r|�`r9�]�ier9r<ifr|idr�r�r@rAr�r�r�c�H��g|]}tj�|d�����S�r�r��rr��rvr`r]s  �r1rwz4TestCircFuncs.test_circmean_axis.<locals>.<listcomp>�s,���
H�
H�
H��e�n�Q�q�T��,�,�,�
H�
H�
Hr3rc�P��g|]"}tj�dd�|fd�����#S�Nr�r�r�r�s  �r1rwz4TestCircFuncs.test_circmean_axis.<locals>.<listcomp>�s4���
K�
K�
K�A�e�n�Q�q�q�q�!�t�W�3�/�/�/�
K�
K�
Kr3)r+r�rr��ravelr	r�r)rOr�r�r]s   @r1�test_circmean_axisz TestCircFuncs.test_circmean_axis�s%����H�/�/�/�.�.�.�.�.�.�0�
1�
1���^�A�C�
(�
(�
(��
�^�A�G�G�I�I�C�
0�
0�
0����B�U�+�+�+�+�
�^�A�C�a�
0�
0�
0��
H�
H�
H�
H�e�A�G�A�J�6G�6G�
H�
H�
H����B�U�+�+�+�+�
�^�A�C�a�
0�
0�
0��
K�
K�
K�
K��q�w�q�z�9J�9J�
K�
K�
K����B�U�+�+�+�+�+�+r3c���tjgd�gd�gd�g���tj�d���}tj����d���}t||d���tj�dd�	��}�fd
�t
�jd��D��}t||d���tj�dd�	��}�fd�t
�jd��D��}t||d���dS)
Nr�r�r�r�r��dy���=rArr�c�H��g|]}tj�|d�����Sr��rr�r�s  �r1rwz3TestCircFuncs.test_circvar_axis.<locals>.<listcomp>��,���
G�
G�
G��e�m�A�a�D�s�+�+�+�
G�
G�
Gr3rc�P��g|]"}tj�dd�|fd�����#Sr�r�r�s  �r1rwz3TestCircFuncs.test_circvar_axis.<locals>.<listcomp>��4���
J�
J�
J�1�e�m�A�a�a�a��d�G�#�.�.�.�
J�
J�
Jr3)r+r�rr�r�r	r�r)rOr�r�r]s   @r1�test_circvar_axiszTestCircFuncs.test_circvar_axis��%����H�/�/�/�.�.�.�.�.�.�0�
1�
1���]�1�3�
'�
'�
'��
�]�1�7�7�9�9�3�
/�
/�
/����B�U�+�+�+�+�
�]�1�3�Q�
/�
/�
/��
G�
G�
G�
G�U�1�7�1�:�5F�5F�
G�
G�
G����B�U�+�+�+�+�
�]�1�3�Q�
/�
/�
/��
J�
J�
J�
J��a�g�a�j�8I�8I�
J�
J�
J����B�U�+�+�+�+�+�+r3c���tjgd�gd�gd�g���tj�d���}tj����d���}t||d���tj�dd�	��}�fd
�t
�jd��D��}t||d���tj�dd�	��}�fd�t
�jd��D��}t||d���dS)
Nr�r�r�r�r�r�rArr�c�H��g|]}tj�|d�����Sr��rr�r�s  �r1rwz3TestCircFuncs.test_circstd_axis.<locals>.<listcomp>	r�r3rc�P��g|]"}tj�dd�|fd�����#Sr�r�r�s  �r1rwz3TestCircFuncs.test_circstd_axis.<locals>.<listcomp>	r�r3)r+r�rr�r�r	r�r)rOr�r�r]s   @r1�test_circstd_axiszTestCircFuncs.test_circstd_axis�r�r3c�H�gd�}t||d���|d���dSr�)r	r�s    r1�test_circfuncs_array_likez'TestCircFuncs.test_circfuncs_array_like	s9��

&�%�%���	�	�!�#�.�.�.��t�D�D�D�D�D�Dr3r�c�Z�ttj|g������dSr*)rr+r&)rOr�s  r1rhzTestCircFuncs.test_empty	s*��	�����2���'�'�(�(�(�(�(r3c��ddddddtjg}ttj||d�������dS)	Nr�r}rr�r�r�r�r��r+r�rr&�rOr�r]s   r1�test_nan_propagatez TestCircFuncs.test_nan_propagate	sG���!�Q��R��b�f�-�������1�3�/�/�/�0�0�1�1�1�1�1r3g��5�:v@gf�Pb�?r�ge�e?g�OQ�O�v?g���#�q@g�auy�+@c�$�tjgd�ddddddtjgdtjtjtjtjtjtjgg��}|���D]�}||d	|�
��}|�"t	tj|�����4t
|d||d�
��t	tj|dd����������dS)N)r�r}rr�r�r�rr�r;r|r�r9r�rr�r�rr+rA)r+r�r��keysrr&r	r��rOr�rr]r��outs      r1�test_nan_propagate_arrayz&TestCircFuncs.test_nan_propagate_array	s���
�H�2�2�2��A�q�#�q�#�r�v�6��"�&�"�&�"�&�"�&�"�&�"�&�I�K�
L�
L���M�M�O�O�	1�	1�D��)�A�C�d�3�3�3�C��|�����
�
�&�&�&�&���A�����T�B�B�B�B�����Q�R�R��)�)�-�-�/�/�0�0�0�0�
	1�	1r3gr�Q�vv@)gv@rPrMg8v@rg�u@g5@��v�?g2>�B�jv@g���G2�?)gaF�+��?��z���?rgA ��~@碲��?r
r�rg��X�҄?gJPm�$�@)g���j@竃���?rg)h;@�{��?rgun��2@g+�f��a @c�"�tjddddddtjgddd	d
ddtjgtjtjtjtjtjtjtjgg��}|���D]s}||d
d|���}|�t	|||d����,t	|dd�||d���ttj|d�����tdS)Nr�r}rr�r�r�r�r;r|r�r9r�r��omit)r��
nan_policyr�r+rAr�)r+r�r�rr	rr&r	s      r1�test_nan_omit_arrayz!TestCircFuncs.test_nan_omit_array0	s��(
�H�s�A�q�#�r�3���7��A�q�#�q�#�r�v�6��v�r�v�r�v�r�v�r�v�r�v�r�v�N�P�
Q�
Q���M�M�O�O�	+�	+�D��)�A�C�F��F�F�F�C��|���X�d�^�$�?�?�?�?�?���C�R�C��(�4�.�t�D�D�D�D�����R��)�)�*�*�*�*�
	+�	+r3c�h�ddddddtjg}t||dd�	��|d
���dS)Nr�r}rr�r�r�r�r�r�rr+rA)r+r�r	r�s    r1�
test_nan_omitzTestCircFuncs.test_nan_omitO	sS��
�!�Q��R��b�f�-���	�	�!�#�&�A�A�A� �t�	-�	-�	-�	-�	-�	-r3c���tjtjtjtjtjg}ttj||d�������dS)Nr�rrrs   r1�test_nan_omit_allzTestCircFuncs.test_nan_omit_allX	sK���V�R�V�R�V�R�V�R�V�4�������1��8�8�8�9�9�:�:�:�:�:r3c���tjtjtjtjtjtjgtjtjtjtjtjgg��}||dd���}ttj|�������tt
|��dk��dS)Nrr)rr�r)r+r�r�rr&r�r�)rOr�r]r
s    r1�test_nan_omit_all_axisz$TestCircFuncs.test_nan_omit_all_axis^	s���
�H�r�v�r�v�r�v�r�v�r�v�>��v�r�v�r�v�r�v�r�v�>�@�
A�
A���i��f�1�5�5�5������
�
�!�!�#�#�$�$�$���C���A�
�����r3r]r�r}r�r�r�r�r;r|r�r9r�c�8�tt||dd���dS)Nr��raiser�rVrWrs   r1�test_nan_raisezTestCircFuncs.test_nan_raiseg	s#��	�j�)�Q�S�W�M�M�M�M�M�Mr3c�8�tt||dd���dS)Nr��foobarrrrs   r1�test_bad_nan_policyz!TestCircFuncs.test_bad_nan_policyp	s#��	�j�)�Q�S�X�N�N�N�N�N�Nr3c�Z�d}|}tj|��}t||d���dS)Nrr�rA)rr�r	)rOr]r�r�s    r1�test_circmean_scalarz"TestCircFuncs.test_circmean_scalary	s6����
��
�^�A�
�
����B�T�*�*�*�*�*�*r3c��tjtjddd��tjtj��}t|tjk��t|tjk��dS)Nrrr`)rr�r+r\r�r)rO�ms  r1�test_circmean_rangez!TestCircFuncs.test_circmean_range	s[��
�N�2�9�Q��3�/�/������?�?����B�E�	������R�U�F�
�����r3c��tjddgd���}ttj|d���d��ttj|d���dd	�
��ttj|d���dd	�
��dS)N�r�r�r�r�r�g@e@gLw�qA��?r+rAg)#p_�4@)r+r�r
rr�r	r�r�r�s  r1�test_circfuncs_uint8z"TestCircFuncs.test_circfuncs_uint8�	s���
�H�c�2�Y�g�.�.�.���U�^�A�C�0�0�0�%�8�8�8���
�a�c�2�2�2�O�$�O�O�O�O���
�a�c�2�2�2�K�d�K�K�K�K�K�Kr3N)$rbrcrdr#r$r%rr�r�r�r�r�r+rQrRrSr�r�r�r�rrhrr�rr�r�rrrrrr"r$r'r*rer3r1r�r��sj�������[���1�$�~�{�;�$�}�.B�C�$�}�k�:�<�=�=�E�E�	=�=�E�+�+�+�"�[���4�$�~�r�w�7�$�}�b�f�5�$�}�b�f�5�7�8�8�5�5�	8�8�5�,�,�,� ,�,�,�",�,�,�"�[���1�$�~�{�;�$�}�.B�C�$�}�k�:�<�=�=�E�E�	=�=�E��[���[�5�>�5�=�+0�=�+:�;�;�)�)�;�;�)��[���[�5�>�5�=�+0�=�+:�;�;�2�2�;�;�2��[���1�$�~�&(�f��*�M�M�O�$�}�&(�f�#7�#7� 9� 9�:� %�}�&(�f��
�K�K�M�
N�O�O�
1�
1�O�O�
1��[���1�$�~�&7�#+�2�8�-4�-4�-4�$5�$5�#+�2�8�Z��,F�#G�#G� I� I�J�
 %�}�&:�#+�2�8�-N�-N�-N�$.�$.�/1�u�S�y�1�n�$=�$,�2�8�-A�-@�-B�$C�$C�	 D� D�E� %�}�&7�#+�2�8�-9�-9�-9�$:�$:�$,�2�8�Z��,D�#E�#E�	 G� G�H�I�J�J�&
+�
+�'J�J�&
+��[���1�$�~�{�;�$�}�.B�C�$�}�k�:�<�=�=�-�-�	=�=�-�
�[���[�5�>�5�=�+0�=�+:�;�;�;�;�;�;�;��[���[�5�>�5�=�+0�=�+:�;�;���;�;���[���S�"�A�q�#�r�3���?�&�b�h��a��C��S�"�&�(I�),�a��C����C�(H�(J�K�K�L�M�M��[���[�5�>�5�=�+0�=�+:�;�;�N�N�;�;�	M�M�N��[���S�"�A�q�#�r�3���?�&�b�h��a��C��S�"�&�(I�),�a��C����C�(H�(J�K�K�L�M�M��[���[�5�>�5�=�+0�=�+:�;�;�O�O�;�;�	M�M�O�+�+�+����L�L�L�L�Lr3r�c��eZdZd�Zd�Zd�Zd�Zd�Zd�Zd�Z	d�Z
d	�Zd
�Zd�Z
ej�dd
dg��d���ZdS)�TestMedianTestc�J�tttjgd���dSr~�rVrWr�median_testrXs r1�test_bad_n_samplesz!TestMedianTest.test_bad_n_samples�	s!���j�%�"3�Y�Y�Y�?�?�?�?�?r3c�L�tttjggd���dSr~r.rXs r1r�z TestMedianTest.test_empty_sample�	s#���j�%�"3�R����C�C�C�C�Cr3c�Z�tttjgd�gd�ddgd���dS)N)rrrr)rrrrrr���tiesr.rXs r1�test_empty_when_ties_ignoredz+TestMedianTest.test_empty_when_ties_ignored�	sG��
	�j�%�"3�"�l�l�I�I�I��1�v�H�	F�	F�	F�	F�	F�	Fr3c��tttjgd�gd���tttjgd�gd�d���dS)N)rrr�abover3r.rXs r1�test_empty_contingency_rowz)TestMedianTest.test_empty_contingency_row�	s\��
	�j�%�"3�Y�Y�Y�	�	�	�J�J�J�	�j�%�"3�Y�Y�Y�	�	�	�"�	$�	$�	$�	$�	$�	$r3c�T�tttjgd�ddgd���dS)Nrxr|r}r	r3r.rXs r1�
test_bad_tieszTestMedianTest.test_bad_ties�	s9���j�%�"3�Y�Y�Y��A�� �	"�	"�	"�	"�	"�	"r3c�T�tttjgd�ddgd���dS)Nrxr|r}r!rr.rXs r1r"z"TestMedianTest.test_bad_nan_policy�	s-���j�%�"3�Y�Y�Y��A��S[�\�\�\�\�\�\r3c�T�tttjgd�ddgd���dS)Nrxr|r}r	)r	)rVrbrr/rXs r1rczTestMedianTest.test_bad_keyword�	s9���i��!2�I�I�I��1�v��	!�	!�	!�	!�	!�	!r3c���gd�}gd�}tj||��\}}}}t|d��t|ddgddgg��t|d��t|d��dS)NrxrLrrr)rr/r
r)rOr]r�r�r@�med�tbls       r1�test_simplezTestMedianTest.test_simple�	s����I�I���I�I��!�-�a��3�3���a��c�	�S�#�����3�!�Q��!�Q�� 0�1�1�1�	�T�1�����Q������r3c��gd�}ddg}gd�}tj|||��\}}}}t|d��t|gd�gd�g��tj|||d���\}}}}t|d��t|gd�gd	�g��tj|||d
���\}}}}t|d��t|gd�gd	�g��dS)Nr�r}r8)r;r<r9)rrry)r|rrr�r3)r|rrr7)rrry)rr/r
)rOr]r��zr�r@r&r?s        r1�test_ties_optionsz TestMedianTest.test_ties_options�	s
���L�L��
��F���I�I�� �+�A�q�!�4�4���a��C��Q������S�9�9�9�i�i�i�0�1�1�1��+�A�q�!�(�C�C�C���a��C��Q������S�9�9�9�i�i�i�0�1�1�1��+�A�q�!�'�B�B�B���a��C��Q������S�9�9�9�i�i�i�0�1�1�1�1�1r3c���ddtjg}gd�}tj||d���}tj||d���\}}}}t	|tjtjtjdf��t|d��t|d��t	|d	��t	|tjd
dgddgg����tttj||d���dS)Nrr)r|r}r8�	propagaterrg�?gإva�o�?rNrr)	r+r�rr/r
r	r�rVrW)rOr]r��mt1r�r@r&rJs        r1�test_nan_policy_optionsz&TestMedianTest.test_nan_policy_options�	s���
��2�6�N���I�I�����1��=�=�=���&�q�!��?�?�?�
��1�a���S�2�6�2�6�2�6�4�8�9�9�9���.�/�/�/���.�/�/�/��Q������Q���1�a�&�!�Q���1�1�2�2�2��j�%�"3�Q��g�N�N�N�N�N�Nr3c��gd�}gd�}tj||��\}}}}t|d��t|ddgddgg��tj|��\}}}	}
t	||��t	||��tj||d���\}}}}t|d��t|ddgddgg��tj|d���\}}}	}
t	||��t	||��tj||d�	��\}}}}t|d��t|ddgddgg��tj|d�	��\}}}	}
t	||��t	||��dS)
Nr{)rr|r8r<r|rrr)�lambda_F�r�)rr/r
�chi2_contingencyr	)rOr]r�r�r@r&r?�exp_stat�exp_p�dof�es           r1rTzTestMedianTest.test_basic�	s���
�O�O���L�L���+�A�q�1�1���a��C��Q������S�A�q�6�A�q�6�*�+�+�+�"'�"8��"=�"=���%��a���h�'�'�'���5�!�!�!��+�A�q�!�<�<�<���a��C��Q������S�A�q�6�A�q�6�*�+�+�+�"'�"8��a�"H�"H�"H���%��a���h�'�'�'���5�!�!�!��+�A�q�U�C�C�C���a��C��Q������S�A�q�6�A�q�6�*�+�+�+�"'�"8���"O�"O�"O���%��a���h�'�'�'���5�!�!�!�!�!r3r�FTc��gd�}gd�}tj|||���}t|j|j|j|jf|��dS)NrxrJ)rr/r
rLr�r\�table)rOr�r]r�r_s     r1r2zTestMedianTest.test_result
sR���I�I���I�I�����1��<�<�<���c�m�S�Z���S�Y�G��M�M�M�M�Mr3N)rbrcrdr0r�r5r8r:r"rcr@rCrGrTr#r$r%r2rer3r1r,r,�	s�������@�@�@�D�D�D�F�F�F�
$�
$�
$�"�"�"�]�]�]�!�!�!�
�
�
�2�2�2�(O�O�O�"�"�"�>�[���\�E�4�=�9�9�N�N�:�9�N�N�Nr3r,c��eZdZd�Zej�dejdzejdzgdfddejzgdfg��d���Z	d�Z
d	�Zd
�Zd�Z
d�Zd
S)�TestDirectionalStatsc�H�tjtjgd�����}tjtjgd�����}tjtj|��tj|��ztj|��tj|��ztj|��fd���}t
j|��}|j}tj	|d��}tjgd���}t||��dS)N)	g33333su@gO@g33333sB@g;@gpv@g������@g333333I@g�����Yv@r.)	gfffff�P@g�����,Q@gfffff�Q@gfffff�T@g�S@g@R@g33333SQ@gffffffM@g33333�I@rr�r|)g�HP��?g&S��:��g[���<�)r+�deg2radr��stack�cos�sinr�directional_stats�mean_directionryr	)rO�decl�inclrP�dirstats�directional_mean�mean_rounded�reference_means        r1�"test_directional_stats_correctnessz7TestDirectionalStats.test_directional_stats_correctness
s���
�2�8�%<�%<�%<�=�=�>�>�>���
�2�8�%<�%<�%<�=�=�>�>�>���x��������t���4��������t���4������'�� � � ��
�*�4�0�0��#�2���x� 0�!�4�4����"<�"<�"<�=�=����n�5�5�5�5�5r3zangles, refrrrr�c�
�t|��r||��}tjtj|��tj|��gd���}dtj|��jz
}t||��dS)Nrr�)	�callabler+rVrWrXrrY�mean_resultant_lengthr	)rO�anglesr�rPr_s     r1�'test_directional_stats_2d_special_casesz<TestDirectionalStats.test_directional_stats_2d_special_cases&
sv��
�C�=�=�	��#�f�+�+�C��x�����������8�q�A�A�A���%�)�$�/�/�E�E����S�!�!�!�!�!r3c�F�tj�d��}dtjz|�d��z}tjtj|��tj|��fd���}tj|��}|j	}tj
|d|d��}|dtjzz}tj|��}t||��d|j
z
}tj|��}	t||	��dS)Nl	}�7�Q�h�0P4u&4M�r)rrr�r)r+rrr�rVrWrXrrYrZ�arctan2r�r	rdr�)
rOr�testdata�testdata_vectorr]r^�directional_mean_angler��directional_var�circular_vars
          r1�test_directional_stats_2dz.TestDirectionalStats.test_directional_stats_2d1
s���i�#�#�$F�G�G���r�u�9�s�z�z�(�3�3�3���(�B�F�8�$4�$4�$&�F�8�$4�$4�$6�()�+�+�+���*�?�;�;��#�2��!#��,<�Q�,?�,<�Q�,?�"A�"A��!7�1�R�U�7�!C���>�(�+�+����"8�9�9�9��h�<�<���}�X�.�.�����6�6�6�6�6r3c��tjgd�gd�g��}tj|d��}tjgd�gd�ggd�gd�gg��}tj|d���}t||j��dS)N��+P�z��?r�r�)rq��r�)rrrr)rr�r�rr�)r+r��tilerrYr	rZ)rOrP�
full_arrayrr]s     r1� test_directional_mean_higher_dimz5TestDirectionalStats.test_directional_mean_higher_dimE
s����x�-�-�-�.�.�.�0�1�1���W�T�<�0�0�
��8�l�l�l�*�l�l�,�*�l�l�*�l�l�,�-�.�.���*�:�A�>�>�>����(�"9�:�:�:�:�:r3c��gd�gd�g}tj|��}tj|��}tj|��}t	|j|j��t	|j|j��dS)Nrp)rqrrr)r+r�rrYr	rZrd)rOrP�
data_arrayr_r�s     r1�)test_directional_stats_list_ndarray_inputz>TestDirectionalStats.test_directional_stats_list_ndarray_inputR
s���$�$�$�&:�&:�&:�;���Z��%�%�
��%�d�+�+���%�j�1�1����*�C�,>�?�?�?���1��1�	3�	3�	3�	3�	3r3c���tjd��}d}tjttj|�����5tj|��ddd��dS#1swxYwYdS)N)r}zIsamples must at least be two-dimensional. Instead samples has shape: (5,)r)	r+r�r#rrW�re�escaperrY)rOrPr�s   r1�test_directional_stats_1d_errorz4TestDirectionalStats.test_directional_stats_1d_error\
s����w�u�~�~��6��
�]�:�R�Y�w�-?�-?�
@�
@�
@�	*�	*��#�D�)�)�)�	*�	*�	*�	*�	*�	*�	*�	*�	*�	*�	*�	*����	*�	*�	*�	*�	*�	*s�A&�&A*�-A*c�D�tjgd�gd�g��}tj|d���}|tj�|dd���z}tj|d���}t
|j|j��t
|j|j��dS)Nrp)g�+P�z��?g�r�T)�	normalizer�)r��keepdimsF)	r+r�rrYr�r�r	rZrd)rOrPr_�normalized_datar�s     r1� test_directional_stats_normalizez5TestDirectionalStats.test_directional_stats_normalized
s����x�-�-�-�-�-�-�/�0�0���%�d�d�;�;�;��������2�9=�"0�"?�"?�?���%�o�05�7�7�7����*�C�,>�?�?�?���1��1�	3�	3�	3�	3�	3r3N)rbrcrdrar#r$r%r+r�rfrnrurxr|r�rer3r1rSrS
s�������6�6�6�(�[���]�
�5�&��(�B�E�!�G�	�b�!�
�Q�r�u�W��r��-���"�"�	��"�7�7�7�(;�;�;�3�3�3�*�*�*�
3�
3�
3�
3�
3r3rS)Rr~r��numpyr+�numpy.randomr�
numpy.testingrrrrrr	r
rr#rrVrz�scipyr
r�scipy.stats._morestatsr�common_testsr�
_hypotestsrr�scipy.stats._binomtestr�scipy.stats._distr_paramsr�dict�
matplotlib�rcParams�matplotlib.pyplot�pyplotrXrl�	Exceptionr%r&r'r(r)r*r+r,r-r.r2r5rgr�r�r'r�r#r<rmr�r�rr
r4rnr�r�r�rrr<r+r[r�r�r�r�r�r,rSrer3r1�<module>r�s��
����
�
�
�
�����$�$�$�$�$�$�?�?�?�?�?�?�?�?�?�?�?�?�?�?�?�?�?�?�?�?��
�
�
�*�*�*�*�*�*�	�	�	�	�������������-�-�-�-�-�-�-�-�-�-�-�-�B�B�B�B�B�B�B�B�?�?�?�?�?�?�.�.�.�.�.�.��4��>�>�������%*�J��	�"�#�#�#�#�#�#��O�O�������O�O�O�����L�K�K��K�K�K��K�K�K��K�K�K��K�K�K��K�K�K��K�K�K��K�K�K��K�K�K��L�L�L��
4�4�4�/�/�/�/�/�/�/�/�>?�?�?�?�?�?�?�?�Bh-�h-�h-�h-�h-�h-�h-�h-�Va&�a&�a&�a&�a&�a&�a&�a&�HE9�E9�E9�E9�E9�E9�E9�E9�P@6�@6�@6�@6�@6�@6�@6�@6�F9�9�9�9�9�9�9�9�:B7�B7�B7�B7�B7�B7�B7�B7�J:Y�:Y�:Y�:Y�:Y�:Y�:Y�:Y�zS=�S=�S=�S=�S=�S=�S=�S=�lAC�AC�AC�AC�AC�AC�AC�AC�H2�2�2�.j7�j7�j7�j7�j7�j7�j7�j7�Z[2�[2�[2�[2�[2�[2�[2�[2�|QJ�QJ�QJ�QJ�QJ�QJ�QJ�QJ�h>�>�>�>�>�>�>�>�>=�=�=�=�=�=�=�=�"39�39�39�39�39�39�39�39�l"=�"=�"=�"=�"=�"=�"=�"=�J#1�#1�#1�#1�#1�#1�#1�#1�P0�0�0��drH�rH�rH�rH�rH�rH�rH�rH�jE6�E6�E6�E6�E6�E6�E6�E6�P ;� ;� ;� ;� ;� ;� ;� ;�F7�7�7�7�7�7�7�7�(|5�|5�|5�|5�|5�|5�|5�|5�~4�4�4�4�4�4�4�4�"aL�aL�aL�aL�aL�aL�aL�aL�H~N�~N�~N�~N�~N�~N�~N�~N�Ba3�a3�a3�a3�a3�a3�a3�a3�a3�a3s�-B�B�
B

Youez - 2016 - github.com/yon3zu
LinuXploit