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_distributions.cpython-311.pyc
�

d�c���d�dZddlZddlZddlZddlZddlmZddlZddlZddl	Z	ddl
mZmZm
Z
mZmZmZmZmZmZmZddlZddlmZddlZddlZddlmZmZddlmZddlmZdd	l m!Z!dd
l"m#Z#m$Z$m%Z%m&Z&ddl'm(Z(ddl)m*Z*ddl+Zddl,m-Z-m.Z.m/Z/dd
l0m1Z1m2Z2ddl3m4Z4m5Z5ddl6m7Z7m8Z8ddl9m:Z:m;Z;ddl<m=Z=ej>j?dkZ@ej	dkoe	jA��dkZBddgZC�dld�ZDd�ZEd�ZFd�ZGd�ZHd�ZId�ZJd�ZKejL�Mdgd ���d!���ZNejL�Md"gd#���d$���ZOd%�ZPejL�Md&gd'���d(���ZQd)�ZRd*�ZSd+�ZTejL�Md,gd-���d.���ZUGd/�d0��ZVGd1�d2��ZWGd3�d4��ZXGd5�d6��ZYGd7�d8��ZZGd9�d:��Z[Gd;�d<��Z\Gd=�d>��Z]Gd?�d@��Z^GdA�dB��Z_GdC�dD��Z`GdE�dF��ZaGdG�dH��ZbGdI�dJ��ZcGdK�dL��ZdGdM�dN��ZeGdO�dP��ZfGdQ�dR��ZgGdS�dT��ZhGdU�dV��ZiGdW�dX��ZjGdY�dZ��ZkGd[�d\��ZlGd]�d^��ZmGd_�d`��ZnGda�db��ZoGdc�dd��ZpGde�df��ZqGdg�dh��ZrGdi�dj��ZsGdk�dl��ZtGdm�dn��ZuGdo�dpev��ZwGdq�dr��ZxGds�dt��Zydu�Zzdv�Z{dw�Z|ejL�Mdxgdy���ejL�Mdzddgddgddggd{�gd|�gddgddggdgdggg��d}�����Z}ejL�Mdzddgddgddggd{�gd|�gddgddggdgdggg��d~���Z~Gd�d���ZGd��d���Z�d�d�ej�d�d�ej�zz��zdd�ej�d�d�ej�zz��zd�d�ej�d�d�ej�zz��zd�d�ej�d�d�ej�zz��zd�d�ej�d�d�ej�zz��zg	fd�ej�d�d�ej�zz��dd�ej�d�d�ej�zz��zd�d�ej�d�d�ej�zz��zd�zd�d�ej�d�d�ej�zz��zd�zd�d�ej�d�d�ej�zz��zd�zg	fd�d�ej�d�ej�z��zdd�d�ej�d�ej�z��zzd�d�d�ej�d�ej�z��zzd�d�d�ej�d�ej�z��zzd�d�d�ej�d�ej�z��zzd�d�d�ej�d�ej�z��zzd�d�d�ej�d�ej�z��zzd�gfgZ�Gd��d���Z�Gd��d���Z�Gd��d���Z�Gd��d���Z�Gd��d���Z�Gd��d���Z�Gd��d���Z�Gd��d���Z�Gd��d���Z�Gd��d���Z�Gd��d���Z�Gd��d���Z�Gd��d���Z�Gd��d���Z�Gd��d¦�Z�GdÄdĦ�Z�GdńdƦ�Z�GdDŽdȦ�Z�dɄZ�Gdʄd˦�Z�Gd̄dͦ�Z�Gd΄dϦ�Z�GdЄdѦ�Z�Gd҄dӦ�Z�GdԄdզ�Z�Gdքdצ�Z�Gd؄d٦�Z�Gdڄdۦ�Z�Gd܄dݦ�Z�Gdބdߦ�Z�Gd�d�ev��Z�Gd�d��Z�Gd�d��Z�Gd�d��Z�Gd�d��Z�Gd�d��Z�Gd�d��Z�d�Z�d�Z�d�Z�d�Z�ejL��e@d���d���Z�d��Z�d��Z�d��Z�d��Z�d��Z�d��Z�d��Z�d��Z�d��Z�d��Z�d��Z��d�Z��d�Z��d�Z��d�Z��d�Z�ejL�M�dg�d����d���Z��d�Z��d	�Z��d
�Zd�Z�ejL���d���d
���ZŐd�Z�ejL�M�d�dej�d�dg��f�dej�d�dg��f�dej�d�dg��f�dej�d�dg��fg���d���Zǐd�Z�ejL���d���d���Z�ejL���d���d���Zʐd �Zːd!�Z̐d"�Z�G�d#��d$e(j���Z�G�d%��d&e(j���Z�G�d'��d(e(j���Z�G�d)��d*e(j���Z�G�d+��d,��Z�G�d-��d.��Z�ejL��e@�d/����d0���ZՐd1�Z֐d2�Zאd3�Zؐd4�Zِd5�Zڐd6�Zېd7�Zܐd8�Zݐd9�Zސd:�Zߐd;�Z�d<�Z�d=�Z�ejL�M�d>g�d?����d@���Z�ejL�M�dA�dB�dCg���dD���Z�ejL�M�dEd�d�g�dF�f�dG�dHg�dF�f�dI�dJg�dK�fg���dL���Z�dM�Z�dN�Z�ejL��ej��dOk�dP����dQ���Z�G�dR��dS��Z�dT�Z�G�dU��dV��Z�G�dW��dX��Z�G�dY��dZ��Z�G�d[��d\��Z�d]�Z�ejL�M�d^e5e2z���d_���Z�d`�Z�da�Z�ejL�M�db�dc�ddg���de���Z�ejL�M�dfg�dg����dh���Z��di�Z��dj�Z��dk�Z�dS(mz!
Test functions for stats module
�N)�Path)
�assert_equal�assert_array_equal�assert_almost_equal�assert_array_almost_equal�assert_allclose�assert_�assert_warns�assert_array_less�suppress_warnings�IS_PYPY)�raises)�	typecodes�array)�rec_append_fields)�special)�check_random_state)�IntegrationWarning�quad�	trapezoid�cumulative_trapezoid)�
argsreduce)�xlogy�	polygamma�entr)�distcont�invdistcont�)�distdiscrete�invdistdiscrete)�FitDataError�
_argus_phi)�root�fmin)�product�darwin�x86_64�tukeylambda�pearson3c�V�|�|�d|��}tt||��|���dS)Nz does not have attribute ��msg)r	�hasattr)�a�br,s   �F/usr/lib/python3/dist-packages/scipy/stats/tests/test_distributions.py�_assert_hasattrr14s8��
�{�12���A�A�6���G�A�q�M�M�s�#�#�#�#�#�#�c�D�ttjjd��dS)N�f_gen)r1�scipy�stats�
distributions�r2r0�test_api_regressionr9:s���E�K�-�w�7�7�7�7�7r2c���tj|||���}t|�|��|�|dtjz|zz����dS)N��loc�scale�)r6�vonmisesr�pdf�numpy�pi��k�L�s�x�vms     r0�check_vonmises_pdf_periodicrI?sS��	���q��	*�	*�	*�B�����q�	�	�2�6�6�!�q���z�!�|�*<�#=�#=�>�>�>�>�>r2c���tj|||���}t|�|��dz|�|dtjz|zz��dz��dS)Nr;rr>)r6r?r�cdfrArBrCs     r0�check_vonmises_cdf_periodicrLDs\��	���q��	*�	*�	*�B�����q�	�	�A�
�r�v�v�a�1�U�X�:�a�<�.@�'A�'A�A�'E�F�F�F�F�Fr2c��ttjjj��}d�t
D��}d�tD��}gd�}||z|z}ttd�|����}|�d��||ksJ�dS)Nc��g|]
}|d��S�rr8��.0�dists  r0�
<listcomp>z0test_distributions_submodule.<locals>.<listcomp>Ks��/�/�/�d�$�q�'�/�/�/r2c��g|]
}|d��SrOr8rPs  r0rSz0test_distributions_submodule.<locals>.<listcomp>Ls��1�1�1�D��Q��1�1�1r2)�rv_discrete�
rv_continuous�rv_histogram�entropy�trapzc�H�t|���d��S)N�<)�str�
startswith�rFs r0�<lambda>z.test_distributions_submodule.<locals>.<lambda>Ss���A���(9�(9�#�(>�(>�$>�r2�gilbrat)	�setr5r6r7�__all__rr�filter�remove)�actual�
continuous�discrete�other�expecteds     r0�test_distributions_submodulerjIs���
���*�2�
3�
3�F�/�/�h�/�/�/�J�1�1�L�1�1�1�H�
!�
!�
!�E��H�$�u�,�H��6�>�>��I�I�J�J�H��M�M�)�����X������r2c��dD]�}ddtjddfD]n}t|dd|��t|dd|��t|dd|��t|dd|��t|dd|��t|dd|���o��dS)N)皙�����?r�err�
�d)rArBrIrL)rDrGs  r0�test_vonmises_pdf_periodicrp[s���
�5�5���Q���"�c�*�	5�	5�A�'��1�a��3�3�3�'��1�a��3�3�3�'��1�b�!�4�4�4�'��1�a��3�3�3�'��1�a��3�3�3�'��1�b�!�4�4�4�4�	5�5�5r2c��ttjjtj��ttjjtj��dS�N)rr6�
vonmises_liner.�nprBr/r8r2r0�test_vonmises_line_supportrugs:����$�&����/�/�/���$�&���.�.�.�.�.r2c�t�tjd��}t|�d��d��dS)N� r��?)r6r?rrK)rHs r0�test_vonmises_numericalryls1��	���	�	�B�����q�	�	�3�'�'�'�'�'r2zx, kappa, expected_pdf))rl�{�G�z�?g��|65��?)rl�9@g��?�U�?)rlrwg"����?)�@rzg�D��fI�?)r|r{g���1��<)r|rw�c�j�tj�||��}t||d���dS�N�V瞯�<��rtol)r6r?r@r)rG�kappa�expected_pdfr@s    r0�test_vonmises_pdfr�{s6���.�
�
�Q��
&�
&�C��C��E�2�2�2�2�2�2r2zkappa, expected_entropy))rg�A��	�?)�g,�eޞ�?)rogBb9d22�)��g���
G�)��g��
�c�h�tj�|��}t||d���dS)N�vIh�%<=r�)r6r?rXr)r��expected_entropyrXs   r0�test_vonmises_entropyr��s5���n�$�$�U�+�+�G��G�-�E�:�:�:�:�:�:r2c��ttd����}tj�|��}tj�|��}tj�|��}tjddd����|���}tjddtjzd����|���}tjdddtjzt|��zdz����|���}t||d���t||d���dS)	N�
von_mises_rvsrrr;)�random_stater>r���atol)
�abs�hashrt�random�default_rngr6r?�rvsrBr)�seed�rng1�rng2�rng3�rvs1�rvs2�rvs3s       r0�test_vonmises_rvs_gh4598r��s���t�O�$�$�%�%�D�
�9� � ��&�&�D�
�9� � ��&�&�D�
�9� � ��&�&�D��>�!��!�,�,�,�0�0�d�0�C�C�D��>�!��2�5���2�2�2�6�6�D�6�I�I�D��>�!��!"�2�5���T���!2�1�!4�7�7�7�7:�s��s�7M�7M�	��D�$�U�+�+�+�+��D�$�U�+�+�+�+�+�+r2zx, kappa, expected_logpdf))rlrzg�E�J?��)rlr{go�ι���?)rlrwg��0�,��)r|rzgdt�y��)r|r{gʹ�w�[A�)r|rwg������c�j�tj�||��}t||d���dSr)r6r?�logpdfr)rGr��expected_logpdfr�s    r0�test_vonmises_logpdfr��s6���^�
"�
"�1�e�
,�
,�F��F�O�%�8�8�8�8�8�8r2c�:�tj�d��}|�d��dz\}}}tj||����d���}t
|d��tj|jtj	��sJ�||dtj
zzf}tj||���jd�g|�R�}t
|d��tj|jtj	��sJ�||dtj
zzf}tj||���jd	�g|�Rd
di�}t
tj|��|dtj
zz��tj|jtj��sJ�dS)a
    Test that the vonmises expectation values are
    computed correctly.  This test checks that the
    numeric integration estimates the correct normalization
    (1) and mean angle (loc).  These expectations are
    independent of the chosen 2pi interval.
    lkD+xN�n�rn)r<r�c��dS�Nrr8�rGs r0r_z&test_vonmises_expect.<locals>.<lambda>�����r2rr>c��dSr�r8r�s r0r_z&test_vonmises_expect.<locals>.<lambda>�r�r2c�0�tjd|z��S)Ny�?)rt�expr�s r0r_z&test_vonmises_expect.<locals>.<lambda>�s����r�!�t���r2�complex_funcN)
rtr�r�r6r?�expectr�
issubdtype�dtype�floatingrB�angle�complexfloating)�rngr<r��lb�res�boundss      r0�test_vonmises_expectr��s���
�)�
�
� 3�
4�
4�C��Z�Z��]�]�R�'�N�C���
�.�S��
.�
.�
.�
5�
5�k�k�
B�
B�C��C�����
�=���B�K�0�0�0�0�0�
��a�"�%�i��
�F�
5�%�.�S��
.�
.�
.�
5�k�k�
K�F�
K�
K�
K�C��C�����
�=���B�K�0�0�0�0�0�
��a�"�%�i��
�F�
5�%�.�S��
.�
.�
.�
5�6L�6L�O�7=�O�O�O�LM�O�O�C��B�H�S�M�M�3�!�B�E�'�?�3�3�3�
�=���B�$6�7�7�7�7�7�7�7r2c���|j|fi|��}tt|��|��j|fi|��}|||��}|||��}||kstj||d���sJ�dSdS)aS
    This utility function checks that the log-likelihood (computed by
    func) of the result computed using dist.fit() is less than or equal
    to the result computed using the generic fit method.  Because of
    normal numerical imprecision, the "equality" check is made using
    `np.allclose` with a relative tolerance of 1e-15.
    r�r�N)�fit�super�typert�allclose)rR�data�func�kwds�mle_analytical�
numerical_opt�ll_mle_analytical�ll_numerical_opts        r0�_assert_less_or_close_logliker��s����T�X�d�+�+�d�+�+�N�/�E�$�t�*�*�d�+�+�/��=�=��=�=�M���^�T�2�2���t�M�4�0�0���!1�1�1��K�)�+;�%�H�H�H�
2�1�1�1�1�1�1r2c
��ddg}|jr6t|j�d����}|gd�d|�z
}tt	|tjt|��������}gd�}tjtd���5|j
|fi|��ddd��n#1swxYwYtjtd���5|�
t
jg��ddd��n#1swxYwYtjtd���5|�
t
j
g��ddd��n#1swxYwYtjtd	���5|�
|d
���ddd��n#1swxYwYtjtd���5|j
|gd
gt|��d
z
z�R�ddd��dS#1swxYwYdS)N�floc�fscale�,)�f0�f1�f2�rr>r�z3All parameters fixed. There is nothing to optimize.��matchz#The data contains non-finite valueszUnknown keyword arguments:r>)�
extra_keywordzToo many positional arguments.r)�shapes�len�split�dict�ziprt�arange�pytestr�RuntimeErrorr��
ValueError�nan�inf�	TypeError)rR�param�nshapes�	all_fixedr�s     r0�assert_fit_warningsr��sL��
�X��E��{�.��d�k�'�'��,�,�-�-��
�#�#�#�H�W�H�-�-���S���	�#�e�*�*� 5� 5�6�6�7�7�I��9�9�D�	��|�&�
'�
'�
'�$�$�	����#�#��#�#�#�$�$�$�$�$�$�$�$�$�$�$����$�$�$�$�
��z�B�
D�
D�
D������"�&�����������������������
��z�B�
D�
D�
D������"�&�����������������������
��y�(D�	E�	E�	E�(�(�����Q��'�'�'�(�(�(�(�(�(�(�(�(�(�(����(�(�(�(�	��y�(H�	I�	I�	I�.�.�����-���S��Z�Z�!�^�,�-�-�-�-�.�.�.�.�.�.�.�.�.�.�.�.����.�.�.�.�.�.sZ�B8�8B<�?B<�!D�D�D�1!E�E"�%E"�F(�(F,�/F,�#G>�>H�HrR)�alpha�	betaprime�fatiguelife�invgamma�invgauss�
invweibull�	johnsonsb�levy�levy_l�lognorm�gibrat�powerlognorm�rayleigh�waldc��tt��}||}tt|��}t	|j|jg|�R�d��t|j|jg|�R�tj
��t	|j|jg|�R�d��t|j|jg|�R�tj
��dS)zgh-6235rN)r�r�getattrr6rr@r.rr�rtr�r/)rR�dct�argss   r0�test_supportr��s����x�.�.�C��t�9�D��5�$���D�������/�$�/�/�/��3�3�3�����T�V�+�d�+�+�+�b�f�W�5�5�5�������/�$�/�/�/��3�3�3�����T�V�+�d�+�+�+�b�f�W�5�5�5�5�5r2c�&�eZdZd�Zd�Zd�Zd�ZdS)�TestRandIntc�D�tj�d��dS�N���rtr�r���selfs r0�setup_methodzTestRandInt.setup_method���
�	���t�����r2c�v�tj�ddd���}tt	j|dk��t	j|dk��z��tt
|��dk��tj�ddd���}tt	j|��dk��t|jj	tdv��tj�dd��}t|dk|dkz��tt|tj��tt|�����	��tjdd���d
��}t|jj	tdv��dS)Nr��ro��size�r>�2�
AllInteger��.r+r�)r6�randintr�r	rA�allr��shaper��charr�
isinstance�
ScalarType�reprr��r��vals�vals   r0�test_rvszTestRandInt.test_rvssZ���}� � ��B�S� �1�1����	�$��)�$�$�u�y����';�';�;�<�<�<���D�	�	�S� �!�!�!��}� � ��B�W� �5�5�����D�!�!�W�,�-�-�-���
��9�\�#:�:�;�;�;��m����B�'�'������s�R�x�(�)�)�)��
�3�� 0�1�1�t�D��I�I���G�G�G�G��m�B��#�#�'�'��*�*����	��)�L�"9�9�:�:�:�:�:r2c���tjdd�}tj|dk|dkzdd��}tj�|dd��}t
||��dS)Nr�$r�r��{�G�z�?)rA�r_�wherer6r�pmfr)r�rD�outrs    r0�test_pdfzTestRandInt.test_pdfs`���H�Q�r�T�N���k�1��6�a�"�f�-�z�1�=�=���}� � ��A�r�*�*��!�$��,�,�,�,�,r2c��tjddd��}tj|��}tj|dk|dkgd|dz
dzd	zgd��}t
j�|dd��}t||d
���dS)Nrrror�r���?�@rr{���decimal)	rt�linspacerA�floor�selectr6rrKr)r�rGrDrrs     r0�test_cdfzTestRandInt.test_cdf$s����K��2�s�#�#���K��N�N���l�A��G�Q�!�V�,�s�Q�s�U�1�W�v�4F�.G��K�K���}� � ��A�r�*�*��!�$��R�8�8�8�8�8�8r2N)�__name__�
__module__�__qualname__r�rrr"r8r2r0r�r�
sP���������;�;�;�-�-�-�9�9�9�9�9r2r�c�,�eZdZd�Zd�Zd�Zd�Zd�ZdS)�	TestBinomc�D�tj�d��dSr�r�r�s r0r�zTestBinom.setup_method-r�r2c���tj�ddd���}tt	j|dk��t	j|dk��z��tt	j|��dk��t|jjtdv��tj�dd��}tt|t����tjdd���d��}tt|tj����t|jjtdv��dS�Nrn��?rr�rrr�)
r6�binomr�r	rArrr�r	rr
�int�ndarrayr
s   r0rzTestBinom.test_rvs0s���{���r�4�g��6�6����	�$�!�)�$�$�u�y����'<�'<�<�=�=�=����D�!�!�W�,�-�-�-���
��9�\�#:�:�;�;�;��k�o�o�b�$�'�'���
�3��$�$�%�%�%��k�"�d�#�#�'�'��*�*���
�3��
�.�.�/�/�/���	��)�L�"9�9�:�:�:�:�:r2c���tj�ddd��}tj�ddd��}t|ddd���t|ddd���dS)Nrorrrr��r�r�)r6r,rr)r��vals1�vals2s   r0�test_pmfzTestBinom.test_pmf;sd�������S�!�,�,�������3��*�*����s��Q�7�7�7�7���s��Q�7�7�7�7�7�7r2c���tjdd��}tjgd���}t	t||����}|���}t||��tjdd��}|���}t|d��tjdd��}|���}t|d��dS)Nr>rx)��?rxr5r}r)	r6r,rtr�sumrrXrr)r�r/�
expected_p�
expected_h�hs     r0�test_entropyzTestBinom.test_entropyBs����K��3�����X�/�/�/�0�0�
��%�
�J�7�7�8�8�8�
�
�I�I�K�K����:�&�&�&��K��3����
�I�I�K�K���Q������K��3����
�I�I�K�K���Q������r2c�j�tj��5tjdt��t	tjdd������d��t	tjdd������d��ddd��dS#1swxYwYdS)N�errorr>r��n�p)	�warnings�catch_warnings�simplefilter�RuntimeWarningrr6r,�mean�stdr�s r0�
test_warns_p0zTestBinom.test_warns_p0Rs���
�
$�
&�
&�	9�	9��!�'�>�:�:�:����q�A�.�.�.�3�3�5�5�q�9�9�9����q�A�.�.�.�2�2�4�4�a�8�8�8�	9�	9�	9�	9�	9�	9�	9�	9�	9�	9�	9�	9����	9�	9�	9�	9�	9�	9s�BB(�(B,�/B,N)r#r$r%r�rr3r:rFr8r2r0r'r',s_���������	;�	;�	;�8�8�8���� 9�9�9�9�9r2r'c��eZdZd�ZdS)�TestArcsinec��tj�ddg��}t|tjtjg��dS�Nrr)r6�arcsiner@rrtr��r�r?s  r0�test_endpointszTestArcsine.test_endpoints\s<��
�M���q�!�f�%�%���Q�����(�)�)�)�)�)r2N)r#r$r%rMr8r2r0rHrHZs#������*�*�*�*�*r2rHc� �eZdZd�Zd�Zd�ZdS)�
TestBernoullic�D�tj�d��dSr�r�r�s r0r�zTestBernoulli.setup_methoddr�r2c���tj�dd���}tt	j|dk��t	j|dk��z��tt	j|��dk��t|jjtdv��tj�d��}tt|t����tjd���d��}tt|tj����t|jjtdv��dS)Nr+rr�rrrr�)
r6�	bernoullir�r	rArrr�r	rr
r-r.r
s   r0rzTestBernoulli.test_rvsgs
����"�"�4�g�"�6�6����	�$�!�)�$�$�u�y����';�';�;�<�<�<����D�!�!�W�,�-�-�-���
��9�\�#:�:�;�;�;��o�!�!�$�'�'���
�3��$�$�%�%�%��o�d�#�#�'�'��*�*���
�3��
�.�.�/�/�/���	��)�L�"9�9�:�:�:�:�:r2c��tjd��}dtjd��zdtjd��zz
}|���}t||��tjd��}|���}t
|d��tjd��}|���}t
|d��dS)Nr5�пr+r}r)r6rRrt�logrXrr)r�r/r8r9s    r0r:zTestBernoulli.test_entropyrs����O�D�!�!���2�6�$�<�<�'�$�r�v�d�|�|�*;�;�
�
�I�I�K�K����:�&�&�&��O�C� � ��
�I�I�K�K���Q������O�C� � ��
�I�I�K�K���Q������r2N)r#r$r%r�rr:r8r2r0rOrOcsA���������	;�	;�	;�
�
�
�
�
r2rOc��eZdZd�ZdS)�TestBradfordc���d}tjdd��}tj�||��}tj�||��}t
||��dS)Nrl������)rt�logspacer6�bradfordrK�ppfr)r��crG�q�xxs     r0�test_cdf_ppfzTestBradford.test_cdf_ppf�s\�����K��R� � ���N���q�!�$�$��
�^�
�
��1�
%�
%����2�����r2N)r#r$r%rar8r2r0rWrW�s#����������r2rWc�(�eZdZdZdZd�Zd�Zd�ZdS)�TestChig�nT��9�;g2 >h�?@c�t�tj�dd��}t||jd���dS)Nrn�r�r�)r6�chi�sfr�CHI_SF_10_4�r�rFs  r0�test_sfzTestChi.test_sf�s5���I�L�L��Q������4�+�%�8�8�8�8�8�8r2c�t�tj�|jd��}t	|dd���dS)Nrernr�r�)r6rf�isfrhr�r�rGs  r0�test_isfzTestChi.test_isf�s5���I�M�M�$�*�A�.�.����2�E�*�*�*�*�*�*r2c�t�tj�d���}t||jd���dS)Nr�)�df��-���q=r�)r6rfrDr�
CHI_MEAN_1000rms  r0�	test_meanzTestChi.test_mean�s5���I�N�N�d�N�#�#����4�-�E�:�:�:�:�:�:r2N)r#r$r%rhrrrjrnrsr8r2r0rcrc�sO������'�K�$�M�9�9�9�+�+�+�;�;�;�;�;r2rcc�&�eZdZd�Zd�Zd�Zd�ZdS)�
TestNBinomc�D�tj�d��dSr�r�r�s r0r�zTestNBinom.setup_method�r�r2c��tj�ddd���}tt	j|dk����tt	j|��dk��t|jjtdv��tj�dd��}tt|t����tjdd���d��}tt|tj����t|jjtdv��dSr*)
r6�nbinomr�r	rArrr�r	rr
r-r.r
s   r0rzTestNBinom.test_rvs�s����|����D�w��7�7����	�$�!�)�$�$�%�%�%����D�!�!�W�,�-�-�-���
��9�\�#:�:�;�;�;��l���r�4�(�(���
�3��$�$�%�%�%��l�2�t�$�$�(�(��+�+���
�3��
�.�.�/�/�/���	��)�L�"9�9�:�:�:�:�:r2c	�2�ttjtj�ddd����tj�ddd����tjj�ddd��}t|d��dS)Ni�i�g�p=
ף�?rr)	rrtr�r6rx�logpmfrr5r)r�rs  r0r3zTestNBinom.test_pmf�s�����u�|�2�2�3��T�B�B�C�C���(�(��c�4�8�8�	:�	:�	:��k� �'�'��1�a�0�0���S�!�����r2c���tj�gd�dd���}tjtj�gd�dd�����}t
||��dS)N)rr�rr��333333@��������?r=)r6rx�logcdfrtrUrKr)r�r�refs   r0�test_logcdf_gh16159zTestNBinom.test_logcdf_gh16159�sb���|�"�"�<�<�<�3�$�"�?�?���f�U�\�%�%�l�l�l�c�T�%�B�B�C�C����c�"�"�"�"�"r2N)r#r$r%r�rr3r�r8r2r0ruru�sP���������	;�	;�	;����#�#�#�#�#r2ruc���eZdZd�Zejjd���Zejjd���Zejjd���Z	ejjd���Z
d�Zd�Zd�Z
d	�Zd
S)�TestGenInvGaussc�D�tj�d��dSr�r�r�s r0r�zTestGenInvGauss.setup_method�r�r2c��tjdd��}tj|�dd���|j��\}}t|dkd��dS)N�ffffff@��?��r��rr�皙�����?T�r6�geninvgauss�kstestr�rKr�r��gig�_r?s    r0�test_rvs_with_mode_shiftz(TestGenInvGauss.test_rvs_with_mode_shift��W�����S�)�)���|�C�G�G��D�G�A�A�3�7�K�K���1��Q��X�t�$�$�$�$�$r2c��tjdd��}tj|�dd���|j��\}}t|dkd��dS)N��������?r+r�r�r�r�Tr�r�s    r0�test_rvs_without_mode_shiftz+TestGenInvGauss.test_rvs_without_mode_shift�sW�����T�*�*���|�C�G�G��D�G�A�A�3�7�K�K���1��Q��X�t�$�$�$�$�$r2c��tjdd��}tj|�dd���|j��\}}t|dkd��dS)Nrl皙�����?r�r�r�r�Tr�r�s    r0�test_rvs_new_methodz#TestGenInvGauss.test_rvs_new_method�r�r2c��d�}t|dd��d��t|dd��d��t|dd��d��dS)Nc��tj||��}|�dd���}tj||j��ddkS)Nr�r�r�rr�)r6r�r�r�rK)r?r/r�r�s    r0�my_ks_checkz4TestGenInvGauss.test_rvs_p_zero.<locals>.my_ks_check�sG���#�A�q�)�)�C��'�'�t�$�'�7�7�C��<��S�W�-�-�a�0�4�7�7r2rr�Tr�r�)r)r�r�s  r0�test_rvs_p_zerozTestGenInvGauss.test_rvs_p_zero�sm��	8�	8�	8�
	�[�[��C�(�(�$�/�/�/��[�[��C�(�(�$�/�/�/��[�[��C�(�(�$�/�/�/�/�/r2c���ttjdd���dd���dtjdd���dd���z��dS)N���r>rnr�r�rr�)rr6r�r�r�s r0�test_rvs_negative_pz#TestGenInvGauss.test_rvs_negative_p�sn����!�$��*�*�.�.�B�T�.�J�J��E�%�c�1�-�-�1�1�r��1�M�M�M�	O�	O�	O�	O�	Or2c�H�tj�dddd���}ttj|ddg���ddkd	��d
tjddd��}}tj�|dd|z|�
��}t|tj	|���|����tj�
|dd|z|�
��}t|tj	|���
|����dS)Nr���rr�)rr?r/r�r��r��333333�?Trorzrn)r?r/r=)r6r�r�rr�rtrr@rr�rK)r��ig�murG�pdf_ig�cdf_igs      r0�
test_invgausszTestGenInvGauss.test_invgauss�s��
�
�
"�
"�����
"�
M�
M���U�\�"�j��s�;�;�;�A�>��E�t�L�L�L��R�[��q�"�-�-�A���"�&�&�q�D�A��F�"�&�E�E������r� 2� 2� 6� 6�q� 9� 9�:�:�:��"�&�&�q�D�A��F�"�&�E�E������r� 2� 2� 6� 6�q� 9� 9�:�:�:�:�:r2c��tjgd���}tjddd��}t|tj�|dd����dS)N)
g�����;g�8y8��?g��\}Z��?g�hܲ<�?g&K�hdd�?g7�!�L�?g	C��f��?g_��诱?g>�0��|�?g��Rq�?rzr�rnrxr)rtrrrr6r�r@)r��vals_RrGs   r0�
test_pdf_RzTestGenInvGauss.test_pdf_R�sd����,�,�,�-�-��
�K��a��$�$����� 1� 5� 5�a��a� @� @�A�A�A�A�Ar2c���ttj�ddd��d��ttj�ddd��d��dS)Nrrxg��>Arr>)rr6r�r@r�s r0�
test_pdf_zerozTestGenInvGauss.test_pdf_zerosT���U�&�*�*�1�c�3�7�7��;�;�;�	�U�&�*�*�3��A�6�6��:�:�:�:�:r2N)r#r$r%r�r��mark�slowr�r�r�r�r�r�r�r�r8r2r0r�r��s�����������[��%�%���%��[��%�%���%��[��%�%���%��[��0�0���0�O�O�O�	;�	;�	;�	B�	B�	B�;�;�;�;�;r2r�c�D�eZdZd�Zd�Zd�Zd�Zd�Zd�Zd�Z	d�Z
d	�Zd
S)�TestGenHyperbolicc�D�tj�d��dSr�r�r�s r0r�zTestGenHyperbolic.setup_method	r�r2c��tjgd���}d\}}}d\}}|||z||zf}tj|||d��}tjddd��}	t|�|	��|dd�	��dS)
N)
g�F^�T=g{X�'�=gNunƢsy>g7!�t�?g�ɫz�I�?g��̸���?g����?gq�
6.�?gmP��Zz?g��:Pg�N?�r>r>r�rxr�r;���rnrr��r�r�)rtrr6�
genhyperbolicrrr@�
r�r��lmbdar��betar��deltar��ghrGs
          r0�
test_pdf_rzTestGenHyperbolic.test_pdf_rs�����������%���u�d��	��E��u�U�{�D��J�/��
�
 �$�B�e�
<�
<�
<���K��R��$�$������q�	�	�6���>�>�>�>�>�>r2c��tjgd���}d\}}}d\}}|||z||zf}tj|||d��}tjddd��}	t|�|	��|dd�	��dS)
N)
g=R0�W�<=g�EB����=g�#N�b>g_�P��G�>g!�� 1Or?g�U���M�?g� �u./�?gf��y�?gl5��\��?g�k��V��?r�r�r;r�rnr���ư>r�)rtrr6r�rrrKr�s
          r0�
test_cdf_rzTestGenHyperbolic.test_cdf_r"s�����������%���u�d��	��E��u�U�{�D��J�/��
�
 �$�B�e�
<�
<�
<���K��R��$�$������q�	�	�6���=�=�=�=�=�=r2c�����gd�}d\}}}d\��||�z|�zf����fd�tdd��D��}t||dd�	��dS)
N)g�@쐧�@g�@쐧� @g��܊�B@gvG�ă�i@r�r�c�Z��g|]'}tj���d���|����(S)r;)r6r��moment)rQ�ir�r�r�s  ���r0rSz4TestGenHyperbolic.test_moments_r.<locals>.<listcomp>HsI�������
���2�U�;�;�;�B�B�1�E�E���r2rr�rr�r�)�ranger)	r�r�r�r�r��vals_usr�r�r�s	      @@@r0�test_moments_rz TestGenHyperbolic.test_moments_r8s������5�5�5��%���u�d��	��E��u�U�{�D��J�/���������1�a�[�[�����
	���a�e�<�<�<�<�<�<r2c���d\}}}d\}}|||z||zf}tj|||d��}tj|�dd���|j��\}}	t|	dkd��dS)	Nr�r�r;r�r�r�r�T)r6r�r�r�rKr)
r�r�r�r�r�r�r�r�r�r?s
          r0rzTestGenHyperbolic.test_rvsOs���%���u�d��	��E��u�U�{�D��J�/��
�
 �$�B�e�
<�
<�
<���|�B�F�F��4�F�@�@�"�&�I�I���1��Q��X�t�$�$�$�$�$r2c�2�tjddd��}tj|d��tjtj��jzd}}dtj|��}}|dz||f}tj|||d��}tj|�	d��|�	d��d	��dd�tj
f}t|�|��tj
�||��dd
���dS)Nrr�rnr>rr;rz�G�z��?rr�r�)rtr�float_power�finfo�float32�eps�sqrtr6r�r]�newaxisrr@�t)	r�rpr�r�r�r�r�r�rGs	         r0�
test_pdf_tzTestGenHyperbolic.test_pdf_t\s���
�[��B��
#�
#���n�R��+�+�B�H�R�Z�,@�,@�,D�D�a�t���r�w�r�{�{�E����A��u�d�#��
�
 �$�B�e�
<�
<�
<���K����t���b�f�f�T�l�l�B�7�7����2�:�
�F����F�F�1�I�I�u�w�{�{�1�b�)�)���	�	�	�	�	�	r2c��dtjtj��jd}}}d\}}|||f}t	j|||d��}tj|�d��|�d��d��dd�tjf}t|�
|��tj�
|��dd�	��dS)
Nr�r)rrr;rzr�rr�r�)rtr�r�r�r6r�rr]r�rr@�cauchy)	r�r�r�r�r�r�r�r�rGs	         r0�test_pdf_cauchyz!TestGenHyperbolic.test_pdf_cauchyns���
"�2�8�B�J�#7�#7�#;�Q�d�u���	��E��u�d�#��
�
 �$�B�e�
<�
<�
<���K����t���b�f�f�T�l�l�B�7�7����2�:�
�F����F�F�1�I�I�u�|�'�'��*�*���	�	�	�	�	�	r2c��tjddd��}tjtj��j}d\}}}|||z||zf}tj|||d��}tjddd��dd�tjf}t|�	|��t
j
�	||d���d	d
���dS)Nr�rn)rrrr;rY�rrr�dy���=r�)rtrr�r�r�r6r�r�rr@�laplace)	r�r<r�r�r�r�r�r�rGs	         r0�test_pdf_laplacez"TestGenHyperbolic.test_pdf_laplaces����k�#�r�2�&�&������$�$�(��$���u�d��u�U�{�D��J�/���
 �$�C�u�
=�
=�
=���K��R��$�$�Q�Q�Q��
�]�3����F�F�1�I�I�u�}�(�(���1�(�=�=���	�	�	�	�	�	r2c
�p�tjddd��tjddd��tjdtd����ztjddd��tjddd��f\}}}}d	}|||z||zf}t	j|||d
��}tj|�d��|�d��d
��dd�tjf}t|�	|��tj
�	|||||���dd���dS)Nrr�rnr�������ror�r;rzr�r)r.r/r<r=r�r�)rtrr�r�r6r�r]r�rr@�norminvgauss)	r�r�r�r�r�r�r�r�rGs	         r0�test_pdf_norminvgaussz'TestGenHyperbolic.test_pdf_norminvgauss�s/����A�r�2�&�&���A�r�2�&�&�r�~�b�%��)�)�'D�'D�D���A�q�"�%�%���D�#�r�*�*�	"���t�U�B����u�u�}�d�U�l�3��
�
 �$�B�e�
<�
<�
<���K����t���b�f�f�T�l�l�B�7�7����2�:�
�F����F�F�1�I�I�u�)�-�-��U�d��%�.�9�9���	�	�	�	�	�	r2N)r#r$r%r�r�r�r�rr�r�r�r�r8r2r0r�r�s����������?�?�?�,>�>�>�,=�=�=�.%�%�%����$���"���*����r2r�c��eZdZd�Zd�Zd�Zej�dgd���d���Z	d�Z
d�Zd	�Zd
�Z
dS)�TestNormInvGaussc�D�tj�d��dSr�r�r�s r0r�zTestNormInvGauss.setup_method�r�r2c���tjgd���}tjgd���}tj�|dd���}t||d���dS)N)gp~٧�>g @�X�>g}��e�?g{�)߳��?gw��H��?�i�������r�rrrx�r.r/��&�.>r�)rtrr6r�rKr)r��r_cdf�x_test�vals_cdfs    r0�
test_cdf_RzTestNormInvGauss.test_cdf_R�sr����<�<�<�=�=����,�,�,�-�-���%�)�)�&�A��)�=�=����%�d�3�3�3�3�3�3r2c���tjgd���}tjgd���}tj�|dd���}t||d���dS)N)g��p϶>g\H�55$?geСt�&�?go N��iH?g�]�u���>r�rrxr�r�r�)rtrr6r�r@r)r��r_pdfr��vals_pdfs    r0r�zTestNormInvGauss.test_pdf_R�sp����<�<�<�=�=����,�,�,�-�-���%�)�)�&�A��)�=�=����%�d�3�3�3�3�3�3r2zx, a, b, sf, rtol))r�rr��!7��?r�)�rr�����?=�-C��6?)rr�r�g��y5�`?rq)rnr�r�g�M�LC�9r�c���tj�|||��}t|||���tj�|||��}t|||���dS�Nr��r6r�rgrrl)r�rGr.r/rgr�rFr�s        r0�test_sf_isf_mpmathz#TestNormInvGauss.test_sf_isf_mpmath�si��L
��!�!�!�Q��*�*����2�D�)�)�)�)���"�"�2�q�!�,�,����1�4�(�(�(�(�(�(r2c���ddg}ddg}d}ddg}tj�|||��}t||dd�	��tj�|||��}t||d
���dS)Nr�r�rrr�r�r�缉ؗ�Ҝ<r0r�r�r�)r�rGr.r/rgrFr�s       r0�test_sf_isf_mpmath_vectorizedz.TestNormInvGauss.test_sf_isf_mpmath_vectorized�s���
��H��
��F��
�� �"8�
9����!�!�!�Q��*�*����2�E��6�6�6�6���"�"�2�q�!�,�,����1�4�(�(�(�(�(�(r2c���tjdd��}tjddd��}|�|��}|�|��}t
||��dS)Nrrr�r>)r6r�rtr�rgrlr)r��dstrGrgrls     r0�test_gh8718zTestNormInvGauss.test_gh8718�s[��� ��A�&�&���I�a��Q����
�V�V�A�Y�Y���g�g�b�k�k����Q�����r2c�.�d\}}tj|dz|dzz
��}||z|dz|dzzd|z|tj|��zzddd|dzz|dzzzz|zf}t|tj�||d�����dS)	N�rrxr>r��@rre�mvsk��moments)rtr�rr6r�)r�r.r/�gamma�v_statss     r0�
test_statszTestNormInvGauss.test_statss������1����1��q�!�t��$�$���u�9�a��d�U�A�X�o�s�Q�w�!�b�g�e�n�n�:L�/M��!�a�!�Q�$�h��A��o�-�.��6�8���W�e�0�6�6�q�!�V�6�L�L�M�M�M�M�Mr2c���d\}}tjgd���}tj�|||��}t|tj�|||����dS)Nr����MbP?rx�+����?)rtrr6r�r]rrK)r�r.r/r�rs     r0�test_ppfzTestNormInvGauss.test_ppfsf�����1���-�-�-�.�.���!�%�%�f�a��3�3����� 2� 6� 6�t�Q�� B� B�C�C�C�C�Cr2N)r#r$r%r�r�r�r�r��parametrizer�r�r�rrr8r2r0r�r��s����������
4�
4�
4�4�4�4��[���0�J�J�J�K�K�
$)�$)�K�K�
$)�L	)�	)�	)� � � �N�N�N�D�D�D�D�Dr2r�c�D�eZdZd�Zd�Zd�Zd�Zd�Zd�Zd�Z	d�Z
d	�Zd
S)�TestGeomc�D�tj�d��dSr�r�r�s r0r�zTestGeom.setup_methodr�r2c��tj�dd���}tt	j|dk����tt	j|��dk��t|jjtdv��tj�d��}tt|t����tjd���d��}tt|tj����t|jjtdv��dS)Nr+rr�rrr�)
r6�geomr�r	rArrr�r	rr
r-r.r
s   r0rzTestGeom.test_rvss����z�~�~�d��~�1�1����	�$�!�)�$�$�%�%�%����D�!�!�W�,�-�-�-���
��9�\�#:�:�;�;�;��j�n�n�T�"�"���
�3��$�$�%�%�%��j����"�"�1�%�%���
�3��
�.�.�/�/�/���	��)�L�"9�9�:�:�:�:�:r2c�P�tj�d��}tj�tjd��d|���}|jtjksJ�tj	|tj
tj��jk��sJ�dS)Nl��A�:i���r�r�)
rtr�r�r6rr�r�r��int64r�iinfo�int32�max)r�r�r�s   r0�
test_rvs_9313zTestGeom.test_rvs_9313s���i�#�#�O�4�4���j�n�n�R�V�C�[�[�q�s�n�C�C���y�B�H�$�$�$�$��v�c�B�H�R�X�.�.�2�2�3�3�3�3�3�3�3r2c�n�tj�gd�d��}t|gd���dS)Nr�rx)rxr5��?)r6rrr�r�rs  r0r3zTestGeom.test_pmf's8���z�~�~�i�i�i��-�-��!�$�(:�(:�(:�;�;�;�;�;r2c�8�tjtj�gd�d����}tj�gd�d��}t
||dd���tj�dd��}t|d��dS)Nr�rxr�rr0rr})rtrUr6rrrzrr)r�r1r2rs    r0�test_logpmfzTestGeom.test_logpmf+s�����u�z�~�~�i�i�i��5�5�6�6���
�!�!�)�)�)�S�1�1����u�5�q�9�9�9�9��j����1�%�%���S�#�����r2c��tj�gd�d��}tj�gd�d��}t	gd���}t||��t|d|z
��dS)Nr�rx�rxr+��?r)r6rrKrgrr�r�r�vals_sfris    r0�test_cdf_sfzTestGeom.test_cdf_sf5sr���z�~�~�i�i�i��-�-���*�-�-�	�	�	�3�/�/���+�+�+�,�,��!�$��1�1�1�!�'�1�X�:�6�6�6�6�6r2c�:�tj�gd�d��}tj�gd�d��}t	gd���}t|t
j|����t|t
j|����dS)Nr�rxr)	r6rr~�logsfrrrtrU�log1pr!s    r0�test_logcdf_logsfzTestGeom.test_logcdf_logsf<s����z� � ����C�0�0���*�"�"�9�9�9�c�2�2���+�+�+�,�,��!�$���x�(8�(8�9�9�9�!�'�2�8�X�I�+>�+>�?�?�?�?�?r2c��tj�gd�d��}tgd���}t	||��dS)Nrrx)rr|r)r6rr]rr�r�rris   r0rzTestGeom.test_ppfCsE���z�~�~�0�0�0�#�6�6������)�)��!�$��1�1�1�1�1r2c�f�ttj�dd��dd���dS)N�#B����;r�+����=r�)rr6rr]r�s r0�test_ppf_underflowzTestGeom.test_ppf_underflowHs-����
���u�e�4�4�c��F�F�F�F�F�Fr2N)r#r$r%r�rrr3rr#r'rr-r8r2r0rrs����������	;�	;�	;�4�4�4�<�<�<����7�7�7�@�@�@�2�2�2�
G�G�G�G�Gr2rc� �eZdZd�Zd�Zd�ZdS)�
TestPlanckc�D�tj�d��dSr�r�r�s r0r�zTestPlanck.setup_methodNr�r2c��tj�gd�d��}tgd���}t	||��dS)Nr�r)g�|d�y�?g8�'�\��>ggr�U��!>)r6�planckrgrrr)s   r0rjzTestPlanck.test_sfQsQ���|���y�y�y�"�-�-���2�2�2�3�3��	"�$��1�1�1�1�1r2c��tj�gd�d��}tgd���}t	||��dS)N)�@�@�@�@�p�@r4)gP�.�gh�>�gT�F�)r6r2r%rrr)s   r0�
test_logsfzTestPlanck.test_logsfXsJ���|�!�!�"7�"7�"7��?�?���:�:�:�;�;��!�$��1�1�1�1�1r2N)r#r$r%r�rjr7r8r2r0r/r/MsA���������2�2�2�2�2�2�2�2r2r/c�&�eZdZd�Zd�Zd�Zd�ZdS)�TestGennormc��gd�}tj�|d��}tj�|��}t	||��dS�Nr�r)r6�gennormr@r�r�r��points�pdf1�pdf2s    r0�test_laplacezTestGennorm.test_laplace_sM�������}� � ���+�+���}� � ��(�(���D�$�'�'�'�'�'r2c��gd�}tj�|d��}tj�|d���}t	||��dS�Nr�r>g�;f���?�r=)r6r<r@�normrr=s    r0�	test_normzTestGennorm.test_normfsO�������}� � ���+�+���z�~�~�f�F�~�3�3���D�$�'�'�'�'�'r2c��tj�d��tjd��}|�d���}tj||j��jdksJ�tjd��}|�d���}tj	�d���}tj
||��jdksJ�tjd��}|�d���}tj�dd�	��}tj
||��jdksJ�dS)
Nrrxr�r�rlrr>g�;f���?)r=r)rtr�r�r6r<r�r�rK�pvaluer��ks_2samprE)r�rRr��rvs_laplace�rvs_norms     r0rzTestGennorm.test_rvsms��
�	���q�����}�S�!�!���h�h�D�h�!�!���|�C���*�*�1�C�7�7�7�7��}�Q�����h�h�D�h�!�!���m�'�'�T�'�2�2���~�c�;�/�/�6��<�<�<�<��}�Q�����h�h�D�h�!�!���:�>�>��t�>�<�<���~�c�8�,�,�3�c�9�9�9�9�9�9r2c��tj�d��tjddgddgg��}|�gd����}tj|dd�ddftjd��j��dd	ksJ�tj|dd�ddftjd��j��dd	ksJ�tj|dd�ddftjd��j��dd	ksJ�tj|dd�ddftjd��j��dd	ksJ�dS)
Nrrxrr|r)r�r>r>r�rrl)rtr�r�r6r<r�r�rK)r�rRr�s   r0�test_rvs_broadcastingz!TestGennorm.test_rvs_broadcasting~sC��
�	���q�����}�s�B�i�"�b��2�3�3���h�h�L�L�L�h�)�)���|�C����1�a��L�%�-��*<�*<�*@�A�A�!�D�s�J�J�J�J��|�C����1�a��L�%�-��*<�*<�*@�A�A�!�D�s�J�J�J�J��|�C����1�a��L�%�-��*<�*<�*@�A�A�!�D�s�J�J�J�J��|�C����1�a��L�%�-��*<�*<�*@�A�A�!�D�s�J�J�J�J�J�Jr2N)r#r$r%rArFrrMr8r2r0r9r9^sU������(�(�(�(�(�(�:�:�:�"K�K�K�K�Kr2r9c� �eZdZd�Zd�Zd�ZdS)�TestHalfgennormc��gd�}tj�|d��}tj�|��}t	||��dSr;)r6�halfgennormr@�exponrr=s    r0�
test_exponzTestHalfgennorm.test_expon�sL������� �$�$�V�Q�/�/���{���v�&�&���D�$�'�'�'�'�'r2c��gd�}tj�|d��}tj�|d���}t	||��dSrC)r6rQr@�halfnormrr=s    r0�
test_halfnormzTestHalfgennorm.test_halfnorm�sS������� �$�$�V�Q�/�/���~�!�!�&��!�7�7���D�$�'�'�'�'�'r2c��gd�}tj�|d��}tj�|d��}t	|d|z��dS)Nr�g��
(��?r>)r6rQr@r<rr=s    r0�test_gennormzTestHalfgennorm.test_gennorm�sT������� �$�$�V�W�5�5���}� � ���1�1���D�!�D�&�)�)�)�)�)r2N)r#r$r%rSrVrXr8r2r0rOrO�sA������(�(�(�(�(�(�*�*�*�*�*r2rOc� �eZdZd�Zd�Zd�ZdS)�TestLaplaceasymmetricc���tjgd���}tj�|d��}tj�|��}t
||��dSr;)rtrr6�laplace_asymmetricr@r�rr=s    r0rAz"TestLaplaceasymmetric.test_laplace�sX����)�)�)�$�$���'�+�+�F�A�6�6���}� � ��(�(����d�#�#�#�#�#r2c���tjgd���}d}d|z}tj�||��}tj�||dzz|��}t||��dS)Nr�r>r)rtrr6r\r@r)r�r>r��kapinvr?r@s      r0�test_asymmetric_laplace_pdfz1TestLaplaceasymmetric.test_asymmetric_laplace_pdf�sr����)�)�)�$�$�����5����'�+�+�F�E�:�:���'�+�+�F�E�1�H�,=�v�F�F����d�#�#�#�#�#r2c
���tjtjd��tjd��g��}d}tj�||��}tj�||��}tj�||��}tjddg��}tjddg��}tjdd	g��}tj�||��}	|}
tj�	||��}|}ttj||||	|f��tj||||
|f����dS)
N�rnr>rlg����Mbp?r�gV-����?皙�����?g����Mb`?)rtrrUr6r\r@rKrgr]rlr�concatenate)
r�r>r�r?�cdf1�sf1r@�cdf2�sf2�ppf1�ppf2�isf1�isf2s
             r0�!test_asymmetric_laplace_log_10_16z7TestLaplaceasymmetric.test_asymmetric_laplace_log_10_16�s9����B�F�2�J�J�;���r�
�
�3�4�4�����'�+�+�F�E�:�:���'�+�+�F�E�:�:���&�)�)�&�%�8�8���x��u�
�&�&���x��i�(�)�)���h��U�|�$�$���'�+�+�D�%�8�8�����'�+�+�C��7�7��������d�C��t�'D�E�E����d�C��t�'D�E�E�	G�	G�	G�	G�	Gr2N)r#r$r%rAr_rlr8r2r0rZrZ�sF������$�$�$�$�$�$�G�G�G�G�Gr2rZc�,�eZdZd�Zd�Zd�Zd�Zd�Zd�Zd�Z	d�Z
d!d
�Zgd�gd�gd
�gd�dej
ddddgej
dddddggd�gd�gd�g	Zeje��Zej�de��d���Zd�Zd�Zd�Zd�Zd�Zd S)"�
TestTruncnormc�D�tj�d��dSr�r�r�s r0r�zTestTruncnorm.setup_method�r�r2c	���tj�gd�dddgdzd���}tjtjdd	dd
dtjg��}t
||��dS)N�r�rr�rxg��H.��?rr>��rr��r>r;r�f���O�?���l�@r�)r6�	truncnormr]rtrr�rr)s   r0�test_ppf_ticket1131z!TestTruncnorm.test_ppf_ticket1131��n����"�"�#E�#E�#E�s�B�()�s�1�u�A�#�7�7���8�R�V�Q�
�A�z�1�b�f�M�N�N��!�$��1�1�1�1�1r2c	���tj�gd�dddgdzd���}tjtjdd	dd
dtjg��}t
||��dS)Nrqrrrr�rsr>r;r�rurtr)r6rvrlrtrr�rr)s   r0�test_isf_ticket1131z!TestTruncnorm.test_isf_ticket1131�rxr2c���d\}}tj�||ddd���}t||���cxko|���cxko|knc��d\}}tj�||ddd���}t||���cxko|���cxko|knc��dS)N)i����r�rrrnr�)rn��r6rvr�r	�minr�r��low�highrGs    r0�test_gh_2477_small_valuesz'TestTruncnorm.test_gh_2477_small_values�s����	��T��O����T�1�a�b��9�9����a�e�e�g�g�.�.�.�.������.�.�.�.�$�.�.�.�.�/�/�/��	��T��O����T�1�a�b��9�9����a�e�e�g�g�.�.�.�.������.�.�.�.�$�.�.�.�.�/�/�/�/�/r2c��d\}}tj�||ddd���}t||���cxko|���cxko|knc��t
|||g��fd\}}tj�||ddd���}t||���cxko|���cxko|knc��d\}}tj�||ddd���}t||���cxko|���cxko|knc��d\}}tj�||ddd���}t||���cxko|���cxko|knc��dS)	N)rormrrrnr�)r���)�'i')i�������)r6rvr�r	r~rr\rs    r0�test_gh_2477_large_valuesz'TestTruncnorm.test_gh_2477_large_values�s����	��T��O����T�1�a�b��9�9����q�u�u�w�w�1�1�1�1�!�%�%�'�'�1�1�1�1�T�1�1�1�1�2�2�C��d�A��4G�4G�G�G��	��T��O����T�1�a�b��9�9����a�e�e�g�g�.�.�.�.������.�.�.�.�$�.�.�.�.�/�/�/� �	��T��O����T�1�a�b��9�9����a�e�e�g�g�.�.�.�.������.�.�.�.�$�.�.�.�.�/�/�/�"�	��T��O����T�1�a�b��9�9����a�e�e�g�g�.�.�.�.������.�.�.�.�$�.�.�.�.�/�/�/�/�/r2c���ddgddgfD�]�\}}tjtj||tjg��}||zdz}tj�|||��}tj�|||��}tj�|||��}tjgd���}tjgd���}	tjgd���}
|d	krtjgd
���}
t||��t||	��t||
��ttj	|
d|
dz��|d
z��tjgd���}tj�
|||��}tj|tj|��dz|g��}
t||
��|d	kr_ttj�|||��d��ttj�|||��d��n^ttj�|||��d��ttj�|||��d��tj�|||��}ttj	||
dz��|dzdz�����dS)Nr�rerZ���r|�rrrr�rrr}r})r�d._MT�
@�B�K�g��?rr)rr�r�rrr>rx�rrxrg�Ėy�	@g�d�v*�?g,mj%V��?r5)rtrr�r6rvrKrgr@rrUr]�sign)r�r�r��xvals�xmid�cdfs�sfs�pdfs�
expected_cdfs�expected_sfs�
expected_pdfs�pvals�ppfs�
expected_ppfsr@s               r0�test_gh_9403_nontail_valuesz)TestTruncnorm.test_gh_9403_nontail_values�s����a�&�2�r�(�+� 	M� 	M�I�C���H�r�v�g�s�D�"�&�9�:�:�E���H�c�>�D��?�&�&�u�c�4�8�8�D��/�$�$�U�C��6�6�C��?�&�&�u�c�4�8�8�D��H�\�\�\�2�2�M��8�$8�$8�$8�9�9�L��H�%A�%A�%A�B�B�M��Q�w�w� "��)E�)E�)E� F� F�
���m�4�4�4���\�2�2�2���m�4�4�4����}�Q�'7�
�a�8H�'H� I� I� #�c�	�
+�
+�
+��H�]�]�]�+�+�E��?�&�&�u�c�4�8�8�D��H�c�2�7�3�<�<�	�+A�4�%H�I�I�M���m�4�4�4��Q�w�w�#�E�O�$6�$6�t�S�$�$G�$G�$6�8�8�8�#�E�O�$7�$7��c�4�$H�$H�$6�8�8�8�8�$�E�O�$7�$7��c�4�$H�$H�$6�8�8�8�#�E�O�$6�$6�t�S�$�$G�$G�$6�8�8�8��/�%�%�d�C��6�6�C����s�=��+;�';� <� <�t�D�y�!�m�L�L�L�L�A 	M� 	Mr2c	��ddgddgfD�]F\}}tjtj||tjg��}||zdz}tj�|||��}tj�|||��}tj�|||��}tjgd���}tjgd���}	tjgd���}
|d	krtjgd
���}
t||��t||	��t||
��ttj	|
d|
dz��|d
z��tjgd���}tj�
|||��}tj|tj|��dz|g��}
t||
��tj�|||��}t||��|d	kr_ttj�|||��d��ttj�|||��d��n^ttj�|||��d��ttj�|||��d��tj�|||��}ttj	||
dz��|dzdz��tj||d��}|ddd�}ttj�|||��tj�|||��ddd���ttj�|||��tj�|||��ddd���ttj�|||��tj�|||��ddd�����HdS)N�'�(����i���r|r�r�)r��pG�C@�F�f��<rr)rr�r�rrr>rxr�g���E�C@gi����?gspX�io)>r5r|r�)
rtrr�r6rvrKrgr@rrUr]r�r)r�r�r�r�r�r�r�r�r�r�r�r�r�r�r@�xvals2s                r0�test_gh_9403_medium_tail_valuesz-TestTruncnorm.test_gh_9403_medium_tail_valuess���r�(�S�#�J�/�,	P�,	P�I�C���H�r�v�g�s�D�"�&�9�:�:�E���H�c�>�D��?�&�&�u�c�4�8�8�D��/�$�$�U�C��6�6�C��?�&�&�u�c�4�8�8�D��H�\�\�\�2�2�M��8�$8�$8�$8�9�9�L��H�%K�%K�%K�L�L�M��Q�w�w� "��*=�*=�*=�!>�!>�
���m�4�4�4���\�2�2�2���m�4�4�4����}�Q�'7�
�a�8H�'H� I� I� #�c�	�
+�
+�
+��H�]�]�]�+�+�E��?�&�&�u�c�4�8�8�D��H�c�2�7�3�<�<��+C�T�%J�K�K�M���m�4�4�4��?�&�&�t�S�$�7�7�D���e�,�,�,��Q�w�w�#�E�O�$6�$6�t�S�$�$G�$G�$6�8�8�8�#�E�O�$7�$7��c�4�$H�$H�$9�;�;�;�;�$�E�O�$7�$7��c�4�$H�$H�$6�8�8�8�#�E�O�$6�$6�t�S�$�$G�$G�$9�;�;�;��/�%�%�d�C��6�6�C����s�=��+;�';� <� <�t�D�y�!�m�L�L�L��K��T�2�.�.�E��D�D�b�D�k�\�F���� 3� 3�E�3�� E� E� %�� 2� 2�6�D�5�3�$� G� G���"�� M�
O�
O�
O���� 2� 2�5�#�t� D� D� %�� 3� 3�F�T�E�C�4� H� H���2�� N�
P�
P�
P���� 3� 3�E�3�� E� E� %�� 3� 3�F�T�E�C�4� H� H���2�� N�
P�
P�
P�
P�W,	P�,	Pr2c���ttjdd���d��d��ttjdtj���d��d��dS)Ng*@g.@�,@g��T���?r�g������ @g�����R�?)rr6rvrKrtr�r�s r0�test_cdf_tail_15110_14753z'TestTruncnorm.test_cdf_tail_15110_14753Asr��
	����S�1�1�5�5�c�:�:�*�	,�	,�	,�����2�6�2�2�6�6�s�;�;�*�	,�	,�	,�	,�	,r2�H�����z>c��|dd�\}}}}tj�||d���\}	}
}}t|	|��t|
|��t|||���t|||���dS)Nrerrr��r6rvr)
r�r.r/rir��m0�v0�s0�k0�m�vrFrDs
             r0�_test_moments_one_rangez%TestTruncnorm._test_moments_one_rangeKs���!�"�1�"����B��B��_�*�*�1�a��*�@�@�
��1�a����2������2������2�D�)�)�)�)���2�D�)�)�)�)�)�)r2)���r�r}rr}r})r�rnr}rr}g�L�Fu�)r�r�r}gMFmz�%�?r}g���ſ)���r>r}gh�I}��?r}gHN�M�rgQ6�3E��?g�nl$�A�?gr�1"D��?gK8��L��?gQ6�3E��gr�1"D��)r�r�g��U�*�?�b�n��?g�n��A�?�?c3�TXʿ)r�rg��U�*ҿr�g�n��A�r�)r����g"$��7"�gO��xr�?gp�7�`��g��'>K@�casec��|\}}}}}}tj�||d���\}}	}
}t||	|
|g||||gd���dS)Nrr��FF�g<r�r�)r�r�r.r/r�r�r�r�r�r�rFrDs            r0�test_momentszTestTruncnorm.test_moments|sd��#���1�b�"�b�"��_�*�*�1�a��*�@�@�
��1�a����A�q�!��r�2�r�2�&6�U�C�C�C�C�C�Cr2c��tj�dtjd���\}}t	|d��t	|d��dS)Nr�mvrg�e3E��?g�:�&�A�?)r6rvrtr�r)r�r�r�s   r0�test_9902_momentszTestTruncnorm.test_9902_moments�sJ����$�$�Q����$�=�=���1��A�z�*�*�*��A�z�*�*�*�*�*r2c���d\}}tj�||ddd���}t||���cxko|���cxko|knc��dS)N)rnrrrrnr�r}rs    r0�test_gh_1489_trac_962_rvsz'TestTruncnorm.test_gh_1489_trac_962_rvs�sy���	��T��O����T�1�a�b��9�9����a�e�e�g�g�.�.�.�.������.�.�.�.�$�.�.�.�.�/�/�/�/�/r2c�P�ddtjdtjtjdddddg}dddtjddddd	tjtjg}tj�||dt|��f�
��}tj|��dt|��fksJ�ttj||�	d���k����ttj|�
d���|k����dS)
Nr�rnr�i���r�r|r�r��-r�r)�axis)rtr�r6rvr�r�rr	rr~rrs    r0�test_gh_11299_rvszTestTruncnorm.test_gh_11299_rvs�s����B�����b�f�W�r�v�g�s�C��S�"�M���B��2�6�2�s�B��R�����H���O����T��C��H�H�
��>�>���x��{�{�q�#�c�(�(�m�+�+�+�+����s�a�e�e��e�m�m�+�,�,�-�-�-����q�u�u�!�u�}�}��,�-�-�.�.�.�.�.r2c��ttjd��rAtj�dddtj������dSdS)Nr�r�r�r�r�)r-rtr�r6rvr�r�r�s r0�test_rvs_Generatorz TestTruncnorm.test_rvs_Generator�sh���2�9�m�,�,�	F��O����R�a�-/�Y�-B�-B�-D�-D�
 �
F�
F�
F�
F�
F�	F�	Fr2c��tjtjtjdtjdg��}tjtjtjddtjg��}tjgd���}gd�}tt	j||���|��|��tt	j||���|��|��dS)N���rnr�)rn�@r��	r�)g
�O�ul"�gځ�=��!�g�L���!�g>EG	f��g�^�K[X�)rtrr�rr6rvr~r%)r�r.r/rGris     r0�test_logcdf_gh17064z!TestTruncnorm.test_logcdf_gh17064�s����H�r�v�g���w��R�V�G�R�8�9�9���H�b�f�b�f�a��R�V�4�5�5���H�*�*�*�+�+��+�+�+��	����1�-�-�4�4�Q�7�7��B�B�B������Q�B�/�/�5�5�q�b�9�9�8�D�D�D�D�Dr2N)r�)r#r$r%r�rwrzr�r�r�r�r�r�rtr��_truncnorm_stats_datarr�r�r
r�r�r�r�r�r�r8r2r0rnrn�s����������2�2�2�2�2�2�0�0�0�0�0�0�&!M�!M�!M�F-P�-P�-P�^,�,�,�*�*�*�*�	�	�	�	1�	1�	1�	=�	=�	=�	<�	<�	<�	
�B�F�	�	�	�	�		�

�&��!�	�	�	�	�		�
	�	�	�
	�	�	�
	�	�	�;"��F%�B�H�%:�;�;���[���V�%:�;�;�D�D�<�;�D�
+�+�+�
0�0�0�/�/�/�F�F�F�	E�	E�	E�	E�	Er2rnc�Z�eZdZej�dgd���d���ZdS)�TestGenLogistic�x, expected))i���g4��`n��)i����g4�	ncg�)rgX��
=��)rogh@��X�)r�g
h���<��c�n�d}tj�||��}t||d���dS)Nr�r�r�)r6�genlogisticr�r)r�rGrir^�logps     r0�test_logpdfzTestGenLogistic.test_logpdf�s<��
��� �'�'��1�-�-����h�U�3�3�3�3�3�3r2N)r#r$r%r�r�r
r�r8r2r0r�r��s\�������[���]�-I�-I�-I�J�J�
4�4�J�J�
4�4�4r2r�c�D�eZdZd�Zd�Zd�Zd�Zd�Zd�Zd�Z	d�Z
d	�Zd
S)�
TestHypergeomc�D�tj�d��dSr�r�r�s r0r�zTestHypergeom.setup_method�r�r2c���tj�dddd���}tt	j|dk��t	j|dk��z��tt	j|��dk��t|jjtdv��tj�ddd��}tt|t����tjddd���d��}tt|tj����t|jjtdv��dS)Nr�rnr�rr�rr)
r6�	hypergeomr�r	rArrr�r	rr
r-r.r
s   r0rzTestHypergeom.test_rvs�s����"�"�2�r�1�7�"�;�;����	�$�!�)�$�$��	�$�!�)�$�$�%�	&�	&�	&����D�!�!�W�,�-�-�-���
��9�\�#:�:�;�;�;��o�!�!�"�a��,�,���
�3��$�$�%�%�%��o�b�!�R�(�(�,�,�Q�/�/���
�3��
�.�.�/�/�/���	��)�L�"9�9�:�:�:�:�:r2c��d}d}d}|}|}tj�d|||��}t|dd��dS)Ni�	r��r>g�k�ㅒP?r|)r6r�rr)r��Mr>�N�tot�good�hgpmfs       r0�test_precisionzTestHypergeom.test_precision�sO��������������#�#�A�s�D�!�4�4���E�#8�"�=�=�=�=�=r2c��ttj�dddd��dd��ttj�dddd��dd��ttj�dddd��dd��ttj�dddd��dd��dS)Nrr>rrr|r})rr6r�rr�s r0�	test_argszTestHypergeom.test_args�s���	�E�O�/�/��1�a��;�;�S�"�E�E�E��E�O�/�/��1�a��;�;�S�"�E�E�E��E�O�/�/��1�a��;�;�S�"�E�E�E��E�O�/�/��1�a��;�;�S�"�E�E�E�E�Er2c	��tdtj�dddd��cxkodknc��dS)Nrr�i�o�iif0r)r	r6r�rKr�s r0�test_cdf_above_onez TestHypergeom.test_cdf_above_one�sH����U�_�(�(��X�t�U�C�C�J�J�J�J�s�J�J�J�J�K�K�K�K�Kr2c�<���	�d�d�tjgd���dz}d�	���	fd�|D��}tjgd���}t||dd	�
��gd�}tj�|��z�d��}gd
�}t||dd	�
��dS)Ng�+�@g�@)r��ffffff@�333333@regffffff@g������@r����@���@c�Z��g|]'}tj����z�|����(Sr8)r6r�rg)rQ�eaten�oranges�pears�quantiles  ���r0rSz1TestHypergeom.test_precision2.<locals>.<listcomp>�sC���*�*�*����!�!�(�G�e�O�W�e�L�L�*�*�*r2)rgp��R(g�㘍R2gJ�ޱp9g?�Mf��=��
�G���?rrg���Ơ>r�)g��@r�g��@g��@g��@)rr�g���)9g���"��1)rtrrr6r�rg)
r��fruits_eatenr�ri�	quantiles�res2�	expected2r�r�r�s
       @@@r0�test_precision2zTestHypergeom.test_precision2�s�����������x� =� =� =�>�>��D����*�*�*�*�*�*�(�*�*�*���8�9�9�9�:�:����X�A�D�9�9�9�9�0�/�/�	���!�!�)�W�u�_�g�u�M�M��>�>�>�	���i�a�d�;�;�;�;�;�;r2c�`�tjddd��}|���}tjddg��}tjt
||����}t||��tjddd��}|���}t|d��dS)Nrerr+r5r})	r6r�rXrtrr6rrr)r��hgr9r7r8s     r0r:zTestHypergeom.test_entropy�s���
�_�Q��1�
%�
%���J�J�L�L���X�t�T�l�+�+�
��f�U�:�z�:�:�;�;�;�
���:�&�&�&�
�_�Q��1�
%�
%���J�J�L�L���Q������r2c��d}d}d}d}tj�||||��}d}t||d���d}d	}d
}d}tj�||||��}d}t||d
���dS)Nr���cA���.A�j�@go�����r�rr�@�X�,g_7�	j$�r)r6r�r%r�r�rDr�r>r��resultris       r0r7zTestHypergeom.test_logsf�s���
����������&�&�q�!�Q��2�2�����F�H�a�8�8�8�8�
����������&�&�q�!�Q��2�2�� ���F�H�b�9�9�9�9�9�9r2c�D�d}d}d}d}tj�||||��}d}t||d���d}d	}d
}d}tj�||||��}d}t||d
���d}d	}d}d}tj�||||��}d}t||d
���t	jgd���}d	}d
}d}tj�||||��}t	jdd��}t||d
���dS)Nrr�r�r�g)\��U���r�rr�r�rr�g*�@�V�r�}�r�g���ר��)r�r�r�)r6r�r~rrtr�fullr�s       r0�test_logcdfzTestHypergeom.test_logcdfsM��
����������'�'��1�a��3�3�����F�H�a�8�8�8�8�
����������'�'��1�a��3�3��&���F�H�b�9�9�9�9�����������'�'��1�a��3�3�� ���F�H�b�9�9�9�9�

�H�\�\�\�"�"����������'�'��1�a��3�3���7�1�1�2�2���F�H�b�9�9�9�9�9�9r2N)r#r$r%r�rr�r�r�r�r:r7rr8r2r0r�r��s����������
;�
;�
;�>�>�>�F�F�F�L�L�L�<�<�<�&
�
�
�:�:�:�0,:�,:�,:�,:�,:r2r�c��eZdZej�dddg��d���Zd�Zd�Zej�ddd	g��d
���Z	dS)�TestLoggammazx, c, sf)rer�gQOu�3;)�rog\Z{���0c���tj�||��}t||d���tj�||��}t||d���dS)Nrqr�)r6�loggammargrrl)r�rGr^rgrF�ys      r0�test_sf_isfzTestLoggamma.test_sf_isfKsb��
�N���a��#�#����2�E�*�*�*�*��N���q�!�$�$����1�5�)�)�)�)�)�)r2c�j�tj�dd��}t|dd���dS)Ni���r>g@��r,r�)r6rr�r)r��lps  r0r�zTestLoggamma.test_logpdfSs6��
�^�
"�
"�4��
+�
+����G�%�0�0�0�0�0�0r2c���tjgd����dd��}|D]?\}}}}}tj�|d���}t|||||gd����@dS)N)rxg�"�~j��g46<�@go�ŏ���@rg�	h"lx�g����Q�?g�Zd;��333333@g(@g{�/L��@g�?�?gгY���ҿgh��|?5�?r�r��msvkrrer)rtr�reshaper6rr)r��tabler^rD�var�skew�kurt�computeds        r0rzTestLoggamma.test_stats[s����������
�w�r�1�~�~�	�).�	1�	1�$�A�t�S�$���~�+�+�A�v�+�>�>�H�%�h��s�D�$�0G�./�
1�
1�
1�
1�
1�	1�	1r2r^rlr
c��tj�|d���}tj|�����sJ�tj�|��}tjtj||k��t|����}|�
d���}|jdcxkr|jksnJ�dS)N順r�r)�confidence_levelrx)
r6rr�rt�isfiniter�median�	binomtest�
count_nonzeror��
proportion_cir�r�)r�r^rG�med�btest�cis      r0rzTestLoggamma.test_rvsjs���
�N���q�v��.�.���{�1�~�~�!�!�#�#�#�#�#��n�#�#�A�&�&����� 0��S�� 9� 9�3�q�6�6�B�B��
�
 �
 �%�
 �
8�
8���v��%�%�%�%�b�g�%�%�%�%�%�%�%�%r2N)
r#r$r%r�r�r
r	r�rrr8r2r0rrDs��������[���Z�*J�*H�*J�K�K�*�*�K�K�*�1�1�1�
1�
1�
1��[���S�3��,�/�/�&�&�0�/�&�&�&r2rc���eZdZd�Zd�Zd�Zd�Zd�Zej	�
dddg��d	���Zd
�Zej	�
ddd
g��d���Z
dS)�TestLogisticc���tjdd��}tj�|��}tj�|��}t
||��dS�NrYr�)rtrr6�logisticrKr]r�r�rGrr`s    r0razTestLogistic.test_cdf_ppf{sS���K��R� � ���N���q�!�!��
�^�
�
��
"�
"����2�����r2c���tjdd��}tj�|��}tj�|��}t
||��dSr$)rtrr6r%rgrlrr&s    r0r	zTestLogistic.test_sf_isf�sS���K��R� � ���N���a� � ��
�^�
�
��
"�
"����2�����r2c���d}d}ttj�d|z
��|��ttj�|��|��dS)Ng�<g�g|�EA@r)rr6r%r]rl)r�r?�desireds   r0�test_extreme_valuesz TestLogistic.test_extreme_values�sV��!��#�����*�*�1�q�5�1�1�7�;�;�;����*�*�1�-�-�w�7�7�7�7�7r2c�t�tj�gd���}gd�}t||d���dS)N)i�rrn)gm\�.�g�9��B.��gO&��$�r�r�)r6r%r�r)r�r�ris   r0�test_logpdf_basiczTestLogistic.test_logpdf_basic�sI���~�$�$�\�\�\�2�2��)�)�)��	��h�U�3�3�3�3�3�3r2c�l�tj�ddg��}t|ddg��dS)Nrw���)r6r%r�r�r�r�s  r0�test_logpdf_extreme_valuesz'TestLogistic.test_logpdf_extreme_values�s7���~�$�$�c�4�[�1�1��	�T�D�$�<�(�(�(�(�(r2zloc_rvs,scale_rvs)g��9x&��?gi� �'�?)g��B"�?g�s�^&��?c��tj�d||���}d�}t|tj�|��|f���j}tj�|��}t||d���dS)Nro�rr<r=c��|\}}t|��}tjtj||z
|z��dtj||z
|z��zz��|dzz
}tj||z
|ztj||z
|z��dz
tj||z
|z��dzzz��|z
}||fS�Nrr>)r�rtr6r�)�inputr�r.r/r>�x1�x2s       r0r�z#TestLogistic.test_fit.<locals>.func�s����D�A�q��D�	�	�A������q��A�~�.�.��R�V�T�A�X��N�3�3�3�5�6�6�89�A��>�B���$��(�a���&�$��(�a��0�0�1�4��&�$��(�a��0�0�1�4�6�7�8�8�:;�<�B��r�6�Mr2r�g���KH�9r�)r6r%r�r#�	_fitstartrGr�r)r��loc_rvs�	scale_rvsr�r��expected_solution�
fit_methods       r0�test_fitzTestLogistic.test_fit�s����~�!�!�s��y�!�I�I��	�	�	�!��u�~�'?�'?��'E�'E��M�����	��^�'�'��-�-�
�	�
�$5�E�B�B�B�B�B�Br2c��tj�ddd���}|tj�|��fg}tj�|i��d}ttj||��ttj||d���ttj||d���dS)Nrorxr>r2r�r��r�)r6r%r�r8�_reduce_funcr�)r�r�r�r�s    r0�test_fit_comp_optimizerz$TestLogistic.test_fit_comp_optimizer�s����~�!�!�s��1�!�=�=���u�~�/�/��5�5�7�8���~�*�*�4��4�4�Q�7��%�e�n�d�D�A�A�A�%�e�n�d�D�q�I�I�I�I�%�e�n�d�D��K�K�K�K�K�Kr2�
testlogcdfTFc���tjgd���}|r tj�|��}n tj�|��}gd�}t
||d���dS)N)r�r.�rr�)���g��g%h��9f�g?~T}%m�g�C�x^&ْgV瞯�<r�)rtrr6r%r~r%r)r�rCrGrris     r0�test_logcdfsf_tailsz TestLogistic.test_logcdfsf_tails�s��
�H�0�0�0�1�1���	)���%�%�a�(�(�A�A���$�$�a�R�(�(�A�F�F�F����8�%�0�0�0�0�0�0r2N)r#r$r%rar	r*r,r0r�r�r
r=rBrGr8r2r0r"r"ys�������������8�8�8�4�4�4�)�)�)��[���0�3J�3K�3M�N�N�C�C�N�N�C�,	L�	L�	L��[���\�D�%�=�9�9�1�1�:�9�1�1�1r2r"c�&�eZdZd�Zd�Zd�Zd�ZdS)�
TestLogserc�D�tj�d��dSr�r�r�s r0r�zTestLogser.setup_method�r�r2c��tj�dd���}tt	j|dk����tt	j|��dk��t|jjtdv��tj�d��}tt|t����tjd���d��}tt|tj����t|jjtdv��dS)Nr+rr�rrr�)
r6�logserr�r	rArrr�r	rr
r-r.r
s   r0rzTestLogser.test_rvs�s����|����7��3�3����	�$�!�)�$�$�%�%�%����D�!�!�W�,�-�-�-���
��9�\�#:�:�;�;�;��l���t�$�$���
�3��$�$�%�%�%��l�4� � �$�$�Q�'�'���
�3��
�.�.�/�/�/���	��)�L�"9�9�:�:�:�:�:r2c�f�tj�dd��}t|d��dS)Nrer+g&����Y3)r6rLrr�r�r�s  r0�test_pmf_small_pzTestLogser.test_pmf_small_p�s1���L���Q��&�&��	��7�#�#�#�#�#r2c�d�tj�d��}t|d��dS)N�:�0�yE>g�W�?)r6rLrDrrNs  r0�test_mean_small_pzTestLogser.test_mean_small_p�s/���L���d�#�#��	��;�'�'�'�'�'r2N)r#r$r%r�rrOrRr8r2r0rIrI�sP���������	;�	;�	;�$�$�$�(�(�(�(�(r2rIc	���eZdZejd���d���Zej�dej	ej
g��ej�dgd���ej�dgd���ej�d	d
dgdd
gf��d���������Zej�d
ej	dfej
dfg��d���ZdS)�TestGumbel_r_l�function��scopec�@�tj�d��Sr��rtr�r�r�s r0r�zTestGumbel_r_l.rng�����y�$�$�T�*�*�*r2rRr9�r�rrr:�rlrr�zfix_loc, fix_scaleTFc��|�d|||���}||�|��fg}|�|i��d}	t��}
|r|dz|
d<|r|dz|
d<t	|||	fi|
��dS)Nro)rr<r=r�rr>r�r�)r�r8rAr�r�)r�rRr9r:�fix_loc�	fix_scaler�r�r�r�r�s           r0rBz&TestGumbel_r_l.test_fit_comp_optimizer�s����x�x�S�g�Y�%(��*�*���t�~�~�d�+�+�-�.��� � ��r�*�*�1�-���v�v���	'�"�Q�;�D��L��	+�&��]�D��N�	&�d�D�$�?�?�$�?�?�?�?�?r2z	dist, sgnrr�c��|tjgd���z}|�|��\}}t||dz��t|dd���dS)N)r�r�r�r�r�r�r�g�W@g�@g3q�tw>r�r�)rtrr�r)r�rR�sgn�zr<r=s      r0r=zTestGumbel_r_l.test_fitsf��
���:�:�:�;�;�;���X�X�a�[�[�
��U�	��S�!3�3�4�4�4���5�D�A�A�A�A�A�Ar2N)
r#r$r%r��fixturer�r�r
r6�gumbel_r�gumbel_lrBr=r8r2r0rTrT�s@�������V�^�*�%�%�%�+�+�&�%�+��[���V�e�n�e�n�%E�F�F��[���Y�
�
�
�3�3��[���[�*�*�*�5�5��[���1�#�U�m�e�T�]�;�=�=�@�@�=�=�6�5�4�3�G�F�
@�(�[���[�E�N�A�+>�,1�N�B�+?�+A�B�B�B�B�B�B�B�B�Br2rTc��eZdZd�Zd�Zejd���d���Zej�	d��ej�
ddd	g��ej�
d
dd	g��ej�
ddd
g��d���������Zej�
ddd	g��ej�
d
dd	g��ej�
ddd
g��ej�
dd�eddgd���D����e
jd���d�����������Ze
jd���d���Zd�Zd�ZdS)�
TestParetoc��	�tj��5tjdt��tj�dd���\}}}}t
|tj��t
|tj��t
|tj	��t
|tj	��tj�dd���\}}}}t
|tj��t
|tj��t
|tj	��t
|tj	��tj�dd���\}}}}t
|d��t
|tj��t
|tj	��t
|tj	��tj�dd���\}}}}t
|d��t
|tj��t
|tj	��t
|tj	��tj�d	d���\}}}}t|d
��t|d��t
|tj	��t
|tj	��tj�dd���\}}}}t|d��t|d��t
|tj	��t
|tj	��tj�d
d���\}}}}t|d��t|d��t|dtjd��z��t
|tj	��tj�dd���\}}}}t|d��t|d��t|dtjd��z��t
|tj	��tj�dd���\}}}}t|d��t|d��t|dtjd��z��t|d��ddd��dS#1swxYwYdS)Nr<rxrrrr�rr|�@g�������?gr�q�@r+�@gffffff�?g�K~���?g2@g۶m۶m�?r
�UUUUUU�?g�q�q�?�$@�@g%I�$I��?g�Q�����?gUUUUUU@gr�q��?g�8��8Nb@)r@rArBrCr6�paretorrtr�r�rr��r�r�r�rFrDs     r0rzTestPareto.test_stats sk���
$�
&�
&�7	N�7	N��!�'�>�:�:�:���+�+�C��+�@�@�J�A�q�!�Q���B�F�#�#�#���B�F�#�#�#���B�F�#�#�#���B�F�#�#�#���+�+�C��+�@�@�J�A�q�!�Q���B�F�#�#�#���B�F�#�#�#���B�F�#�#�#���B�F�#�#�#���+�+�C��+�@�@�J�A�q�!�Q���C� � � ���B�F�#�#�#���B�F�#�#�#���B�F�#�#�#���+�+�C��+�@�@�J�A�q�!�Q���C� � � ���B�F�#�#�#���B�F�#�#�#���B�F�#�#�#���+�+�C��+�@�@�J�A�q�!�Q��A�y�)�)�)��A�2�3�3�3���B�F�#�#�#���B�F�#�#�#���+�+�C��+�@�@�J�A�q�!�Q��A�s�#�#�#��A�t�$�$�$���B�F�#�#�#���B�F�#�#�#���+�+�C��+�@�@�J�A�q�!�Q��A�y�)�)�)��A�2�3�3�3��A�	�2�7�7�+;�+;�;�<�<�<���B�F�#�#�#���+�+�C��+�@�@�J�A�q�!�Q��A�y�)�)�)��A�z�*�*�*��A�0�2�7�;�3G�3G�G�H�H�H���B�F�#�#�#���+�+�C��+�@�@�J�A�q�!�Q��A�y�)�)�)��A�2�3�3�3��A�	�R�W�W�-=�-=�=�>�>�>��A�L�M�M�M�o7	N�7	N�7	N�7	N�7	N�7	N�7	N�7	N�7	N�7	N�7	N�7	N����7	N�7	N�7	N�7	N�7	N�7	Ns�R8S�S� Sc��d}d}d}tj�||d|���}||z|z}t||��dS)Nge��Ar>r�rr;)r6rnrgr)r�rGr/r=r?ris      r0rjzTestPareto.test_sf\sO����
�����L�O�O�A�q�a�u�O�5�5���!�G�a�<����8�$�$�$�$�$r2rUrVc�@�tj�d��Sr�rYr�s r0r�zTestPareto.rngdrZr2z2ignore:invalid value encountered in double_scalars�	rvs_shaperr>�rvs_locr�	rvs_scaler�c�,�tj�d||||���}tj�|dd���d}tj�|dd���d}tj�|dd���d}||cxkr
|cxkrdksnJ�tj�d|||dz|���}tj�|d�	��\}	}
}t	|dz|�����|dz
}|jd}
t	|	|
tjtj	||���z����z��t	|
d��dS)
Nro�rr/r=r<r�r�p=
ף�?)r�r�)r��fix_b)r��fbr>r?)
r6rnr�r�rr~rrtr6rU)r�rrrsrtr�r��shape_mle_analytical1�shape_mle_analytical2�shape_mle_analytical3�shape_mle_a�	loc_mle_a�scale_mle_a�
data_shift�ndatas              r0r=zTestPareto.test_fiths����|���S�I�Y�$+�#� �?�?��!&�� 0� 0��A�$� 0� G� G�� J�� %�� 0� 0��A�T� 0� J� J�1� M�� %�� 0� 0��A�$� 0� G� G�� J��%�)>�.�.�.�.�%�.�.�.�.�)-�.�.�.�.�.�.��|���S�I�Y�%,�q�[�� �E�E��.3�l�.>�.>�t�!�.>�.L�.L�+��Y���[�1�_�d�h�h�j�j�1�1�1��A�X�
�� ��#���[��R�V�B�F�:�j�n�n�6F�6F�+F�$G�$G�H�H�H�	J�	J�	J��Y��"�"�"�"�"r2rl�fix_shape, fix_loc, fix_scalec��g|]}d|v�|��	S�Fr8�rQr?s  r0rSzTestPareto.<listcomp>��(��-�-�-�A�!&�!��� �!+��r2TFr���repeat�ignore��invalidc�>�tj�d||||���}|tj�|��fg}	tj�|	i��d}
i}|r||d<|r||d<|r||d<ttj||
fi|��dS)Nrorvrr�r�r�)r6rnr�r8rAr��r�rrrsrt�	fix_shaper^r_r�r�r�r�r�s            r0�test_fit_MLE_comp_optimizerz&TestPareto.test_fit_MLE_comp_optimizer�s����|���S�I�Y�$+�#� �?�?���u�|�-�-�d�3�3�6�7���|�(�(��r�2�2�1�5�����	#�"�D��J��	#�"�D��L��	'�&�D��N�%�e�l�D�$�G�G�$�G�G�G�G�Gr2c	�Z�d\}}}tj�|||dtj�d�����}|tj�|��fg}tj�|i��d}ttj||��dS)N)rrrroið&r�r)	r6rnr�rtr�r�r8rAr�)r�r�locationr=r�r�r�s       r0�test_fit_known_bad_seedz"TestPareto.test_fit_known_bad_seed�s���
")���x���|����x��S�-/�Y�-B�-B�7�-K�-K� �M�M���u�|�-�-�d�3�3�6�7���|�(�(��r�2�2�1�5��%�e�l�D�$�?�?�?�?�?r2c���ttj��tttjjgd�d���tttjjgd�dd���dS)Nr�r>r?)r�r>r�rr��r�r�)r�r6rn�
assert_raisesr!r�r�s r0�test_fit_warningszTestPareto.test_fit_warnings�sg���E�L�)�)�)��l�E�L�$4�i�i�i�a�H�H�H�H��l�E�L�$4�i�i�i�a��	 �	 �	 �	 �	 �	 r2c��tj�ddd|���}t|d��tj�|��}dS)Ni~���rro)r<r/rr�r)r6rnr�rr�)r�r�r�r�s    r0�test_negative_datazTestPareto.test_negative_data�sK���|���D�A�C�c��J�J���$��"�"�"�

�L���T�"�"���r2N)r#r$r%rrjr�rcr�r��filterwarningsr
r=r%rt�errstater�r�r�r�r8r2r0rgrgs(������:N�:N�:N�x%�%�%��V�^�*�%�%�%�+�+�&�%�+��[���!1�2�2��[���[�1�a�&�1�1��[���Y��A��/�/��[���[�1�a�&�1�1�#�#�2�1�0�/�2�1�2�2�
#�.�[���[�2�q�'�2�2��[���Y��A��/�/��[���[�1�a�&�1�1��[���<�-�-���$���q�)I�)I�)I�-�-�-�.�.��R�[��"�"�"�H�H�#�"�.�.�2�1�0�/�3�2�H�"�R�[��"�"�"�	@�	@�#�"�	@� � � �#�#�#�#�#r2rgc��eZdZd�Zd�Zd�Zd�Zd�Zd�Zd�Z	d�Z
d	�Zej
�d
dgd�fd
dddejd��zejgfddddejd��zdgfdgd�fg��d���Zd�ZdS)�
TestGenparetoc�r�dD]i}tj|��}tj�|��\}}t|d��t
tj|�����jtjd��}tj�|��\}}t||gddg��dS)N�rr}r}��rx)	rt�asarrayr6�	genpareto�_get_supportrr	�isposinfr)r�r^r.r/s    r0�test_abzTestGenpareto.test_ab�s����	$�	$�A��
�1�
�
�A��?�/�/��2�2�D�A�q���B�����B�K��N�N�#�#�#�#�
�J�s�O�O����+�+�A�.�.���1���A���S�	�*�*�*�*�*r2c��tjd���}tjddd��}t	|�|��tj�|����t	|�|��tj�|����t	|�|��tj�|����tjddd��}t	|�	|��tj�	|����dS)Nr}�r^rrlr�rrn)
r6r�rtrrr@rRrKrgr]�r��rvrGr_s    r0�test_c0zTestGenpareto.test_c0�s����_�r�
"�
"�
"���K��3��#�#������q�	�	�5�;�?�?�1�#5�#5�6�6�6�����q�	�	�5�;�?�?�1�#5�#5�6�6�6�����a���%�+�.�.��"3�"3�4�4�4��K��B��#�#������q�	�	�5�;�?�?�1�#5�#5�6�6�6�6�6r2c���tjd���}tjddd��}t	|�|��tj�|����t	|�|��tj�|����t	|�|��tj�|����tjddd��}t	|�	|��tj�	|����t	|�
d	��d��dS)
Nrrr�rrlr�r}rrnr)r6r�rtrrr@�uniformrKrgr]r�r�s    r0�test_cm1zTestGenpareto.test_cm1�s��
�_�s�
#�
#�
#���K��3��#�#������q�	�	�5�=�#4�#4�Q�#7�#7�8�8�8�����q�	�	�5�=�#4�#4�Q�#7�#7�8�8�8�����a���%�-�"2�"2�1�"5�"5�6�6�6��K��B��#�#������q�	�	�5�=�#4�#4�Q�#7�#7�8�8�8�	��	�	�!���a�(�(�(�(�(r2c���tjd���}t|�tj��|�tj��gddg��tt	j|�	tj������tjd���}t|�tj��|�tj��gddg��tt	j|�	tj������tjd���}t|�tj��|�tj��gddg��tt	j|�	tj������dS)Nrlr�r}rrr)
r6r�rr@rtr�rKr	�isneginfr��r�r�s  r0�
test_x_infzTestGenpareto.test_x_inf�sC��
�_�s�
#�
#�
#�����������������8�2�r�(�C�C�C����B�I�I�b�f�-�-�.�.�/�/�/�
�_�r�
"�
"�
"�����������������8�2�r�(�C�C�C����B�I�I�b�f�-�-�.�.�/�/�/�
�_�s�
#�
#�
#�����������������8�2�r�(�C�C�C����B�I�I�b�f�-�-�.�.�/�/�/�/�/r2c��tjddd��}dD]�}tj�||��}dD]7}tj�|||z��}t||d����8tj�||��}dD]7}tj�|||z��}t||d����8��dS)	Nrrnr��rr��r,g�+�����rqr�)r,r,)rtrr6r�r@rrK)r�rGr^�pdf0�dc�pdfc�cdf0�cdfcs        r0�test_c_continuityzTestGenpareto.test_c_continuity�s����K��2�r�"�"���		8�		8�A��?�&�&�q�!�,�,�D�%�
8�
8����*�*�1�a�"�f�5�5����d��7�7�7�7�7��?�&�&�q�!�,�,�D�$�
8�
8����*�*�1�a�"�f�5�5����d��7�7�7�7�7�
8�		8�		8r2c	�t�tjtjddd���tjdddd���d	tjddd���z
f}d
D]\}tj�||��}dD]7}tj�|||z��}t||d����8�]dS�
Nrqrzrl��baserr�F��endpointr)r}rrr�r�)rtrr[rr6r�r]r)r�r_r^�ppf0r��ppfcs      r0�test_c_continuity_ppfz#TestGenpareto.test_c_continuity_ppf������E�"�+�e�T��4�4�4��+�d�A�r�E�:�:�:��r�{�5�$�S�9�9�9�9�:�
;���	8�	8�A��?�&�&�q�!�,�,�D�%�
8�
8����*�*�1�a�"�f�5�5����d��7�7�7�7�7�
8�	8�	8r2c	�t�tjtjddd���tjdddd���d	tjddd���z
f}d
D]\}tj�||��}dD]7}tj�|||z��}t||d����8�]dSr�)rtrr[rr6r�rlr)r�r_r^�isf0r��isfcs      r0�test_c_continuity_isfz#TestGenpareto.test_c_continuity_isfr�r2c	�\�tjtjddd���tjdddd���d	tjddd���z
f}d
D]P}t	t
j�t
j�||��|��|d����QdS)
Nrqrzrlr�rr�Fr�r)rQg�C��]r2�r�gV瞯Ҽr�r�)	rtrr[rrr6r�rKr])r�r_r^s   r0�test_cdf_ppf_roundtripz$TestGenpareto.test_cdf_ppf_roundtrips����E�"�+�e�T��4�4�4��+�d�A�r�E�:�:�:��r�{�5�$�S�9�9�9�9�:�
;��/�	+�	+�A��E�O�/�/���0C�0C�A�q�0I�0I�1�M�M��E�
+�
+�
+�
+�
+�	+�	+r2c�j�tj�dddd��}t|d��dS)Ng _�Brzrrg�p��EȜ�)r6r�r%rr/s  r0r7zTestGenpareto.test_logsfs4����$�$�T�3��1�5�5����1�2�2�2�2�2r2zc, expected_statsr)rrr>rr5rkg�q�q@rnr>g�q�q�?g�?g�$I�$	�?�r�q��?rsgl�l�0@r�)rx�UUUUUU�?r�333333�c�n�tj�|d���}t||dd���dS)Nrrr�r�r0)r6r�r)r�r^�expected_statsr�s    r0rzTestGenpareto.test_stats!s;����&�&�q�&�&�9�9�����U��G�G�G�G�G�Gr2c�h�tj�d��}t|dd���dS)NrQgvǼ
�?r�r�)r6r�rr)r�r�s  r0�test_varzTestGenpareto.test_var+s3���O����%�%����,�5�9�9�9�9�9�9r2N)r#r$r%r�r�r�r�r�r�r�r�r7r�r�r
rtr�r�rr�r8r2r0r�r��sA������+�+�+�
7�
7�
7�)�)�)�0�0�0�8�8�8�8�8�8�8�8�8�+�+�+�3�3�3��[����
�\�\�\�	�

��T�2�g�b�g�a�j�j�=�"�&�1�	2�

��]�T�7�2�7�1�:�:�$5�v�>�	?�
�"�"�"�	#�	%�&�&�H�H�
&�&�H�:�:�:�:�:r2r�c�8�eZdZd�Zd�Zd�Zd�Zd�Zd�Zd�Z	dS)	�TestPearson3c�D�tj�d��dSr�r�r�s r0r�zTestPearson3.setup_method2r�r2c��tj�dd���}tt	j|��dk��t|jjtdv��tj�d��}tt|t����tjd���d��}tt|tj����t|jjtdv��tt|��dk��dS)Nrlrr��AllFloatrxr�)
r6r)r�r	rArr�r	rr
�floatr.r�r
s   r0rzTestPearson3.test_rvs5s����~�!�!�#�G�!�4�4�����D�!�!�W�,�-�-�-���
��9�Z�#8�8�9�9�9��n� � ��%�%���
�3��&�&�'�'�'��n�S�!�!�%�%�a�(�(���
�3��
�.�.�/�/�/���	��)�J�"7�7�8�8�8���C���A�
�����r2c��tj�dgd���}t|t	jgd���d���tj�dd��}t|t	jdg��d���tj�gd	�d��}t|t	jgd
���d���dS)Nr>�r}rlr�)gt��T���?g��+�q�?g����?r�r�r�rl�ND}���i?�r�r�r�rr)r�g��j���?gd��C��?g��s���?g�'�M��?)r6r)r@rrtrrs  r0rzTestPearson3.test_pdf@s����~�!�!�!�_�_�_�5�5����b�h�'K�'K�'K�L�L�!�	#�	#�	#�	#��~�!�!�"�c�*�*����b�h�
�|�4�4�4�@�@�@�@��~�!�!�"4�"4�"4�c�:�:����b�h�(@�(@�(@�A�A�GK�	M�	M�	M�	M�	M�	Mr2c�h�tj�dgd���}t|t	jgd���d���tj�dd��}t|dgd���tj�gd	�d��}t|gd
�d���dS)Nr>r�)g�r���E�?gTp�]0�?g��~���?r�r�r�rlg�E&"�J?r�)gi��Z*�J?g�h+�8w�?g���cK�?g�|<x6�?gG���?)r6r)rKrrtrrs  r0r"zTestPearson3.test_cdfJs����~�!�!�!�_�_�_�5�5����b�h�'K�'K�'K�L�L�!�	#�	#�	#�	#��~�!�!�"�c�*�*����z�l��6�6�6�6��~�!�!�"4�"4�"4�c�:�:����?�?�?�EI�	K�	K�	K�	K�	K�	Kr2c����gd�}d�d�tj��|��}��fd�|D��}t||��dS)N�r�r�rrxrxr�c�l��g|]0}ttj|��j����d��1SrO)rr6r)r@)rQr�neg_inf�x_evals  ��r0rSz<TestPearson3.test_negative_cdf_bug_11186.<locals>.<listcomp>[sG���'�'�'������-�-�1�7�F�C�C�A�F�'�'�'r2)r6r)rKr)r��skewsr��int_pdfsr�r�s    @@r0�test_negative_cdf_bug_11186z(TestPearson3.test_negative_cdf_bug_11186Tsu����!� � �������~�!�!�&�%�0�0��'�'�'�'�'� %�'�'�'����h�'�'�'�'�'r2c�6�tj�dd��}t|d��t	|t
j��sJ�tj�dd��}t|d��t	|t
j��sJ�dS)Nrr>rr�)r6r)r�rr
rt�number)r�r�s  r0�test_return_array_bug_11746z(TestPearson3.test_return_array_bug_11746_s�����&�&�q�!�,�,���V�Q�����&�"�)�,�,�,�,�,���&�&�q�(�3�3���V�Q�����&�"�)�,�,�,�,�,�,�,r2c���gd�}d}tj�tj�||��|��}t	||��tjdgdgg��}tjdd��}t	tj�||��tj�||����t	tj�||��tj�	||����t	tj�||��tj�
||����dS)Nr�rxr�r�r�r>)r6r)r]rKrrtrrr@rgrl)r�r�r�r�rrGs      r0�test_ppf_bug_17050zTestPearson3.test_ppf_bug_17050js?��!� � �����n� � ���!3�!3�F�E�!B�!B�E�J�J����V�$�$�$��x�$��#���(�(���K��A�������*�*�1�d�3�3���*�*�A�2��u�5�5�	7�	7�	7����*�*�1�d�3�3���)�)�1�"�t�e�4�4�	6�	6�	6����*�*�1�d�3�3���+�+�A��u�5�5�5�	7�	7�	7�	7�	7r2N)
r#r$r%r�rrr"r�r�r�r8r2r0r�r�1s����������	�	�	�M�M�M�K�K�K�	(�	(�	(�	-�	-�	-�7�7�7�7�7r2r�c�8�eZdZd�Zd�Zd�Zd�Zd�Zd�Zd�Z	dS)	�
TestKappa4c��gd�}d}dD]T}tj�|||��}tj�||��}t	||���UdS)N)r}rlr�rxr)
gffffff��rrr�皙����ɿ皙������rlr�rxr�ffffff�?)r6�kappa4rKr�r�r�rGr9rDr�	vals_comps      r0�test_cdf_genparetozTestKappa4.test_cdf_genparetosr�� � � �����	-�	-�A��<�#�#�A�q�!�,�,�D���+�+�A��r�2�2�I��D�)�,�,�,�,�	-�	-r2c��tjddd��}d}tjddd��}tj�|||��}tj�||��}t
||��dS)Nr�r�rnr}r�r�)rtrr6r�rK�
genextremerr�s      r0�test_cdf_genextremezTestKappa4.test_cdf_genextreme�st���K��A�r�"�"�����K��A�r�"�"���|����1�a�(�(���$�(�(��A�.�.�	���i�(�(�(�(�(r2c���tjddd��}d}d}tj�|||��}tj�|��}t
||��dS)Nrrnrr})rtrr6r�rKrRrr�s      r0�test_cdf_exponzTestKappa4.test_cdf_expon�sa���K��2�r�"�"�������|����1�a�(�(���K�O�O�A�&�&�	���i�(�(�(�(�(r2c���tjddd��}d}d}tj�|||��}tj�|��}t
||��dS)Nr�r�rnr})rtrr6r�rKrdrr�s      r0�test_cdf_gumbel_rzTestKappa4.test_cdf_gumbel_r�sc���K��A�r�"�"�������|����1�a�(�(���N�&�&�q�)�)�	���i�(�(�(�(�(r2c���tjddd��}d}d}tj�|||��}tj�|��}t
||��dS)Nr�r�rnrrr})rtrr6r�rKr%rr�s      r0�test_cdf_logisticzTestKappa4.test_cdf_logistic�sc���K��A�r�"�"�������|����1�a�(�(���N�&�&�q�)�)�	���i�(�(�(�(�(r2c���tjddd��}d}d}tj�|||��}tj�|��}t
||��dS)Nr�r�rnr)rtrr6r�rKr�rr�s      r0�test_cdf_uniformzTestKappa4.test_cdf_uniform�sc���K��A�r�"�"�������|����1�a�(�(���M�%�%�a�(�(�	���i�(�(�(�(�(r2c�0�tjdd��dSr4)r6r�r�s r0�test_integers_ctorzTestKappa4.test_integers_ctor�s��	��Q������r2N)
r#r$r%r�r�r�r�r�r�rr8r2r0r�r�~s}������	-�	-�	-�)�)�)�)�)�)�)�)�)�)�)�)�)�)�)�����r2r�c�,�eZdZd�Zd�Zd�Zd�Zd�ZdS)�TestPoissonc�D�tj�d��dSr�r�r�s r0r�zTestPoisson.setup_method�r�r2c��tjd��}tj�gd�|��}d|dz|dzdzg}t||��dS)Nr>�rrr>rxre)rtrUr6�poissonrr)r��ln2rris    r0�test_pmf_basiczTestPoisson.test_pmf_basic�sX���f�Q�i�i���}� � ����C�0�0����Q���Q��q��)����h�'�'�'�'�'r2c���tj�gd�d��}gd�}t||��tj�dd��}t|d��dS)Nrr)rrr�ffffff�?�rr)r6rrr�intervalr)r�rrirs    r0�test_mu0zTestPoisson.test_mu0�sc���}� � ����A�.�.���9�9���4��*�*�*��=�)�)�$��2�2���X�v�&�&�&�&�&r2c��tj�dd���}tt	j|dk����tt	j|��dk��t|jjtdv��tj�d��}tt|t����tjd���d��}tt|tj����t|jjtdv��dS)Nrxrr�rrr�)
r6rr�r	rArrr�r	rr
r-r.r
s   r0rzTestPoisson.test_rvs�s����}� � ��7� �3�3����	�$�!�)�$�$�%�%�%����D�!�!�W�,�-�-�-���
��9�\�#:�:�;�;�;��m����$�$���
�3��$�$�%�%�%��m�C� � �$�$�Q�'�'���
�3��
�.�.�/�/�/���	��)�L�"9�9�:�:�:�:�:r2c	��d}tj�|d���}t|||tjd|z��d|zg��tjgd���}tj�|d���}||tjddtjd��zgtjddgf}t||��dS)	Ng0@rrr)r}rr|rr>rx)r6rrrtr�rr�)r�r�r�ris    r0rzTestPoisson.test_stats�s���
����$�$�R��$�8�8�����R����R����#�b�&� A�B�B�B�
�X�o�o�o�
&�
&����$�$�R��$�8�8����R�V�Q��"�'�!�*�*��5����3�7G�H�����)�)�)�)�)r2N)r#r$r%r�rr
rrr8r2r0rr�s_���������(�(�(�'�'�'�	;�	;�	;�*�*�*�*�*r2rc�V�eZdZd�Zd�Zd�Zd�Zd�Zd�Zd�Z	d�Z
d	�Zd
�Zd�Z
d�Zd
S)�	TestKSTwoc�D�tj�d��dSr�r�r�s r0r�zTestKSTwo.setup_method�r�r2c���dD]�}tjdd|zd|zddd|zz
dg��}d|z|z}tj�|dz��}|dkrtj|��nd}tjdd||zddtj�d|��zz
tdd|zz
d��dg��}tj
�||��}t||����dS)N�rr>r�rnror�rrxrrr>r})
rtrr5r�gammalnr�r6�ksonergr�kstworKr)r�r>rG�v1�lg�elgrir�s        r0r"zTestKSTwo.test_cdf�s��)�	0�	0�A���!�S��U�A�a�C��a��A��g�q�9�:�:�A��a�%�!��B���&�&�q��s�+�+�B�!#�q���2�6�"�:�:�:�a�C��x��A�r�C�x�!"�Q�u�{�~�~�c�1�'=�'=�%=�!=�!$�Q��2��X�s�!3�!3�!$�!&�'�'�H��{���q�!�,�,�H��H�h�/�/�/�/�#	0�	0r2c��tjddd��}dD]�}tjdd|zd|zddd|zz
dg��}d|z|z}tj�|dz��}|dkrtj|��nd}tjddd||zz
dtj�	d|��ztd|zd��dg��}tj�	||��}t||����dS)Nrrr|rrxrr>)
rtrrr5rrr�r6rrgr~rr)r�rGr>rrrrir"s        r0rjzTestKSTwo.test_sf�s���K��1�b�!�!��)�	/�	/�A���!�S��U�A�a�C��a��A��g�q�9�:�:�A��a�%�!��B���&�&�q��s�+�+�B�!#�q���2�6�"�:�:�:�a�C��x��c�!"�R�#�X��!"�5�;�>�>�#�q�#9�#9�!9�!$�Q�r�T�3����!4�5�5�H��k�n�n�Q��*�*�G��G�X�.�.�.�.�	/�	/r2c�
�tjddd��dd�}gd�}|D]]}|tj|��z}tj�||��}tj|��}t|d���^dS)Nrr>r|r)rro��i�r�r�rQ)rtrr�r6rrK�diffr)r�rG�ns�_x�xn�probs�diffss       r0�test_cdf_sqrtnzTestKSTwo.test_cdf_sqrtns���

�K��1�b�!�!�!�"�"�%��
,�
,�
,���	+�	+�B��b�g�b�k�k�!�B��K�O�O�B��+�+�E��G�E�N�N�E��e�T�*�*�*�*�		+�	+r2c���tjddd��}dD]U}tj�||��}tj�||��}t
|d|z
���VdS�Nrrr|r)rtrr6rrKrgr)r�rGr>r�r"s     r0r#zTestKSTwo.test_cdf_sfso���K��1�b�!�!��)�	=�	=�A��{���q�!�,�,�H��k�n�n�Q��*�*�G�%�h��G��<�<�<�<�	=�	=r2c��tjddd��}dD]l}|tj|��z}tj�||��}tj�||��}t|d|z
���mdSr&)rtrr�r6rrKrgr)r�rGr>r!r�r"s      r0�test_cdf_sf_sqrtnzTestKSTwo.test_cdf_sf_sqrtn"s����K��1�b�!�!��)�	=�	=�A��R�W�Q�Z�Z��B��{���r�1�-�-�H��k�n�n�R��+�+�G�%�h��G��<�<�<�<�		=�	=r2c�0�tjddd��}dD]|}||d|zk}tj�||��}d|k|dkz}tj�||��}t
||||d����}dS)	Nrrr|rrxr�r�r�)rtrr6rrKr]r�r�rGr>r!r��condrs       r0�test_ppf_of_cdfzTestKSTwo.test_ppf_of_cdf*s����K��1�b�!�!��)�	=�	=�A��1�s�1�u�9��B��{���r�1�-�-�H���L�X��_�5�D��;�?�?�8�Q�/�/�D��D��J��4��t�<�<�<�<�<�
	=�	=r2c�0�tjddd��}dD]|}||d|zk}tj�||��}d|k|dkz}tj�||��}t
||||d����}dS)	Nrrr|rrxrr�r�)rtrr6rrlrgr)r�rGr>r!�vals_isfr+rs       r0�test_isf_of_sfzTestKSTwo.test_isf_of_sf4s����K��1�b�!�!��)�	=�	=�A��1�s�1�u�9��B��{���r�1�-�-�H���L�X��^�4�D��;�>�>�(�A�.�.�D��D��J��4��t�<�<�<�<�<�	=�	=r2c�V�tjddd��}dD]�}|tj|��z|d|zk}tj�||��}d|k|dkz}tj�||��}t||||����dS)Nrrr|rrxr)rtrr�r6rrKr]rr*s       r0�test_ppf_of_cdf_sqrtnzTestKSTwo.test_ppf_of_cdf_sqrtn=s����K��1�b�!�!��)�	2�	2�A��b�g�a�j�j�.�!�c�!�e�)�,�B��{���r�1�-�-�H���L�X��^�4�D��;�?�?�8�Q�/�/�D��D��J��4��1�1�1�1�	2�	2r2c�V�tjddd��}dD]�}|tj|��z|d|zk}tj�||��}d|k|dkz}tj�||��}t||||����dS)Nrrr|rrxr
)rtrr�r6rrgrlr)r�rGr>r!r"r+rs       r0�test_isf_of_sf_sqrtnzTestKSTwo.test_isf_of_sf_sqrtnFs����K��1�b�!�!��)�	2�	2�A��b�g�a�j�j�.�!�c�!�e�)�,�B��k�n�n�R��+�+�G���K�G�d�N�3�D��;�?�?�7�A�.�.�D��D��J��4��1�1�1�1�
	2�	2r2c���tjddd��dd�}dD]R}tj�||��}tj�||��}t
||���SdSr&)rtrr6rr]rKr)r�r"r>r!r�s     r0rzTestKSTwo.test_ppfPst����A�q�"�%�%�a�b�b�)��)�	-�	-�A�������*�*�B��{���r�1�-�-�H��H�e�,�,�,�,�	-�	-r2c	��gd�}tjgd���}tjgd�gd�gd�gd�gd�gd�g��}t|��D]y\}}|tjd	��ztjtjd	z|z��z}tj�||��}t|||d
����zdS)N)rnrrorr�r�)r5gUUUUUU�?rxrr>r�)g�R���T>g�r��	?g�K��5
�?gn
�"�5�?g���U
��?gk����?)g�&6��#>gb
�E���>g���I5�?g\�
�?g;�����?g%��1���?)g����6>g�L�����>g�<��?g䓀^��?gO|6$��?g qO���?)g}�+:�>gwQ�9���>g���H{��?g]qv
(��?g)�~/v��?gmJ���?)gV2�7J�=g��Gk(��>g�W�X���?g��Y�?g{�0`��?g�Z
H���?)g_�R���=g���4���>g�̪���?gT��_@�?g��<e��?g2����?r>��h㈵��>r�)
rtr�	enumeraterUr�rBr6rrKr)r�r�ratiosri�idxr>rGr�s        r0�test_simard_lecuyer_table1z$TestKSTwo.test_simard_lecuyer_table1Ws��+�
*�
*����8�8�8�9�9���8�
-�
-�
-�
-�
-�
-�
-�
-�
-�
-�
-�
-�
-�
-�
-�
-�
-�
-�

�
�
�� ��m�m�	@�	@�F�C��������"�R�W�R�U�1�W�Q�Y�%7�%7�7�A��{���q�!�,�,�H��H�h�s�m�$�?�?�?�?�?�	@�	@r2N)r#r$r%r�r"rjr$r#r(r,r/r1r3rr:r8r2r0rr�s����������0�0�0�(
/�
/�
/�+�+�+�=�=�=�=�=�=�=�=�=�=�=�=�2�2�2�2�2�2�-�-�-�@�@�@�@�@r2rc� �eZdZd�Zd�Zd�ZdS)�TestZipfc�D�tj�d��dSr�r�r�s r0r�zTestZipf.setup_methodzr�r2c��tj�dd���}tt	j|dk����tt	j|��dk��t|jjtdv��tj�d��}tt|t����tjd���d��}tt|tj����t|jjtdv��dS)Nr�rr�rrr�)
r6�zipfr�r	rArrr�r	rr
r-r.r
s   r0rzTestZipf.test_rvs}s����z�~�~�c��~�0�0����	�$�!�)�$�$�%�%�%����D�!�!�W�,�-�-�-���
��9�\�#:�:�;�;�;��j�n�n�S�!�!���
�3��$�$�%�%�%��j��o�o�!�!�!�$�$���
�3��
�.�.�/�/�/���	��)�L�"9�9�:�:�:�:�:r2c�v�tj�d���\}}ttj|����t|tj��tj�dd���\}}ttj||g�������dS)N�ffffff@�r.r|�sk�r.r)r6r?r	rtrrr�rros     r0r�zTestZipf.test_moments�s����z���#��&�&���1����A�������Q�������z���#�t��4�4���1��B�K��A��'�'�+�+�-�-�-�.�.�.�.�.r2N)r#r$r%r�rr�r8r2r0r<r<ysA���������	;�	;�	;�/�/�/�/�/r2r<c�&�eZdZd�Zd�Zd�Zd�ZdS)�TestDLaplacec�D�tj�d��dSr�r�r�s r0r�zTestDLaplace.setup_method�r�r2c��tj�dd���}tt	j|��dk��t|jjtdv��tj�d��}tt|t����tjd���d��}tt|tj����t|jjtdv��ttj�d��du��dS)Nr�rr�rr�rb)r6�dlaplacer�r	rArr�r	rr
r-r.r
s   r0rzTestDLaplace.test_rvs�s����~�!�!�#�G�!�4�4�����D�!�!�W�,�-�-�-���
��9�\�#:�:�;�;�;��n� � ��%�%���
�3��$�$�%�%�%��n�S�!�!�%�%�a�(�(���
�3��
�.�.�/�/�/���	��)�L�"9�9�:�:�:����"�"�3�'�'�t�3�4�4�4�4�4r2c��d}tj|��}|�d��\}}}}d}tj||dz��}|�|��}	tj|	|dzz��tj|	|dzz��}}
t
||fd��t||f|
||
dzzdz
fd	d
���dS)Nrr�%rr>rerrr,rQr�)r6rIrtr�rr6rr)r�r.�dlr�r�rFrDr�r`�pp�m2�m4s            r0rzTestDLaplace.test_stats�s�����
�^�A�
�
���X�X�f�%�%�
��1�a����
�Y��r�1�Q�3�
�
��
�V�V�B�Z�Z�����2�q�5��!�!�2�6�"�R��U�(�#3�#3�B���a��V�V�$�$�$���A���R��A��X��]� 3�%�d�K�K�K�K�K�Kr2c���tjd��}tj|��}|�d��\}}}}t	||fd��t||fd��dS)Nr|r)r}r})r
g
@)rtrUr6rIrr)r�r.rLr�r�rFrDs       r0�test_stats2zTestDLaplace.test_stats2�se���F�2�J�J��
�^�A�
�
���X�X�f�%�%�
��1�a���a��V�X�&�&�&���A��
�+�+�+�+�+r2N)r#r$r%r�rrrQr8r2r0rFrF�sS���������	5�	5�	5�L�L�L�,�,�,�,�,r2rFc����eZdZd�Zej�dddg��d���Zej�dddg���fd���Zd�Z	d	�Z
d
�Z�xZS)�TestInvgaussc�D�tj�d��dSr�r�r�s r0r�zTestInvgauss.setup_method�r�r2zrvs_mu,rvs_loc,rvs_scale)r>rr)g
ףp=�@g��� �r@g��ʡE6@c��tj�d|||���}tj�||���\}}}||z
}t	j|��}t
|��t	j|dz|dzz
��z}	||	z}
t|
|dd���t|	|dd���t||��tj�d|||���}tj�||dz
|dz���\}}}t|dz|��t|dz
|��tj�|d	�
��d}tj�|d	���d}tj�|d	�
��d}
||cxkr
|
cxkrd	ksnJ�dS)Nro�rr�r<r=r?r�r�r�rr�rw)�fmur)�fix_mu�r�)
r6r�r�r�rtrDr�r6rr)r��rvs_mursrtr�r�r<r=�mu_temp�	scale_mle�mu_mle�
shape_mle1�
shape_mle2�
shape_mle3s              r0r=zTestInvgauss.test_fit�s����~�!�!�s�v�&-�Y�"�@�@����+�+�D�w�+�?�?���C���g�~���'�$�-�-����I�I����r�
�W�r�]�(B�!C�!C�D�	���"��	����U�;�;�;�;��	�5�u�5�A�A�A�A��S�'�"�"�"��~�!�!�s�v�&-�Y�"�@�@����+�+�D�w��{�3<�q�=�,�B�B���C���Y��]�E�*�*�*��W�q�[�#�&�&�&��^�'�'��$�'�7�7��:�
��^�'�'��T�'�:�:�1�=�
��^�'�'���'�6�6�q�9�
��Z�=�=�=�=�:�=�=�=�=��=�=�=�=�=�=�=�=r2)gX9��v>@g������	@g�G�z@c	���tj�d|||���}tt	tj��tj��j}||��}tj�|��}t
||��||dd���}tj�|dd���}t
||��|tj�|��fg}tj�|i��d}	ttj||	|���tj||dz
z
dk��sJ�ttj||	|dz
���ttj||	d���ttj||	|tj�
d��d���dS)	NrorVrr>)r�rWrr?r�)r6r�r�r�r�r�rr8rAr�rtrr��rand)r�rZrsrtr��	super_fit�super_fitted�invgauss_fitr�r��	__class__s          �r0r�z(TestInvgauss.test_fit_MLE_comp_optimizer�s�����~�!�!�s�v�&-�Y�"�@�@���$�u�~�.�.���?�?�C�	� �y������~�)�)�$�/�/���\�<�0�0�0�!�y��A�1�5�5�5���~�)�)�$�Q�A�)�>�>���\�<�0�0�0��u�~�/�/��5�5�8�9���~�*�*�4��4�4�Q�7��	&�e�n�d�D�w�O�O�O�O��v�t�w��{�+�q�0�1�1�1�1�1�%�e�n�d�D�+2�Q�;�	8�	8�	8�	8�
	&�e�n�d�D�q�I�I�I�I�	&�e�n�d�D�w�-/�Y�^�^�A�->�->�q�-A�	C�	C�	C�	C�	C�	Cr2c���ttj��tjt
��5tj�gd�d���ddd��dS#1swxYwYdS)Nr�r>r?)r�r6r�r�rr!r�r�s r0�test_fit_raise_errorsz"TestInvgauss.test_fit_raise_errors	s����E�N�+�+�+�
�]�<�
(�
(�	2�	2��N���y�y�y�q��1�1�1�	2�	2�	2�	2�	2�	2�	2�	2�	2�	2�	2�	2����	2�	2�	2�	2�	2�	2s�$A$�$A(�+A(c��gd�}gd�}tj�d|���}t||��tj�dd���}t	|d��tj�dd���}t	|d	��tj�d
d��}t	|d��tj�d
d��}t	|d��dS)N)g�4��wT;?gjdV&�}?g�{�i\0�>gDV�_�h?gqacX?)rrrrr皙�����?�r�r
��������?g�f��F��ng��r^��:g�9̗?r�g��x_�;�:g��!��?g��ٷ
?�?)r6r�rKrrrg)r�r�rire�
cdf_actual�	sf_actuals      r0r#zTestInvgauss.test_cdf_sf	s���.�.�.��"�?�?����#�#�C�B�#�/�/���X�v�&�&�&��^�'�'��$�'�7�7�
��
�$9�:�:�:��N�%�%�c�d�%�3�3�	��	�#7�8�8�8���#�#�G�V�4�4���� 3�4�4�4���#�#�H�f�5�5���� 1�2�2�2�2�2r2c��tj�dd���}t|d��tj�dd��}t|d��tj�dd���}t|d��tj�dd��}t|d	��dS)
Nr�rlrkgiJ.���rmg��r^�ߺr
gpg��F��gpQ^�L�)r6r�r~rr%)r�r~r%s   r0r'zTestInvgauss.test_logcdf_logsf*	s�����&�&�v�$�&�7�7���� 1�2�2�2���&�&�s�D�1�1���� 5�6�6�6���$�$�U�t�$�4�4����5�6�6�6���$�$�S�$�/�/����0�1�1�1�1�1r2)
r#r$r%r�r�r�r
r=r�rhr#r'�
__classcell__�rfs@r0rSrS�s������������[���7�'�)>�?�A�A�>�>�A�A�>�:�[���7�'�)>�?�A�A�$C�$C�$C�$C�A�A�$C�L2�2�2�!3�!3�!3�F2�2�2�2�2�2�2r2rSc���eZdZej�dgd���ej�dgd���d�����Zej�dgd����fd���Zd	�Zd
�Z	d�Z
d�Z�xZS)
�TestLaplacers)r�rrr>rt)rr>r�rnc��tj�d||���}tj|��}tjtj||z
����t|��z}tj�|��\}}t||dd���t||dd���tj�||���\}}t||dd���tj�||���\}}t||��|dz}tjtj||z
����t|��z}tj�||���\}}t||��tj�||���\}}t||��tttjj|||���tttjjtjg��tttjjtjg��dS)	Nror2r�r�r?r@r>r�)r6r�r�rtrr6r�r�r�rrr�r�r�r�r�)r�rsrtr��loc_mler\r<r=s        r0r=zTestLaplace.test_fit>	s���
�}� � �c�w�i� �H�H���)�D�/�/���F�2�6�$��.�1�1�2�2�S��Y�Y�>�	��]�&�&�t�,�,�
��U���W�5�u�=�=�=�=���y�u�5�A�A�A�A��]�&�&�t�'�&�:�:�
��U���y�u�5�A�A�A�A��]�&�&�t�I�&�>�>�
��U���W�%�%�%���k���F�2�6�$��*�-�-�.�.��T���:�	��]�&�&�t�#�&�6�6�
��U��Y��&�&�&��]�&�&�t�I�&�>�>�
��U��W�c�"�"�"�	�l�E�M�$5�t�'�&�	(�	(�	(�	(�	�j�%�-�"3�b�f�X�>�>�>��j�%�-�"3�b�f�X�>�>�>�>�>r2zrvs_scale,rvs_loc))rnr�)r�rn)r�rxc���tj�d||���}d�}tj�|��\}}t	ttj��tj���|��\}}||||��}	||||��}
|	|
kst
j|	|
dd���sJ�dSdS)Nr�r2c	��dt|��tjd|z��zd|ztjtj||z
����zz
zS)Nr�r>r)r�rtrUr6r�)r<r=r�s   r0�llz3TestLaplace.test_fit_MLE_comp_optimizer.<locals>.llr	sS���C��I�I�����%����8��E�'�2�6�"�&����*<�*<�#=�#=�=�>�?�
?r2r�r�)r6r�r�r�r�r�rtr�)r�rsrtr�ryr<r=�loc_opt�	scale_opt�ll_mle�ll_optrfs           �r0r�z'TestLaplace.test_fit_MLE_comp_optimizerk	s�����}� � �d��y� �I�I��	?�	?�	?��]�&�&�t�,�,�
��U�"�4��
�#6�#6�#(�=�2�2�25�#�d�)�)�	�����C���%�%����G�Y��-�-������"�+�f�f�38�u�#F�#F�#F�������r2c��tjgd���}tj�|d���\}}t|ddd���tj�|d���\}}t|ddd���dS)N)rrrr� @r�rr?rer�r�r@)rtrr6r�r�r)r�r�r<r=s    r0�test_fit_simple_non_random_dataz+TestLaplace.test_fit_simple_non_random_data�	s����x�7�7�7�8�8���]�&�&�t�!�&�4�4�
��U���q�u�5�9�9�9�9��]�&�&�t�A�&�6�6�
��U���Q�U��7�7�7�7�7�7r2c�F�d}tj�|��}|dksJ�tj�|��}|dksJ�tj�|��}|dksJ�tj�|��}|dksJ�dS)Nr�r}r)r6r�rKrg)r�rG�p0�p1s    r0�test_sf_cdf_extremesz TestLaplace.test_sf_cdf_extremes�	s�����
�]�
�
��r�
"�
"���S�y�y�y�y��]�
�
�q�
!�
!���S�y�y�y�y�
�]�
�
�a�
 �
 ���S�y�y�y�y��]�
�
�q�b�
!�
!���S�y�y�y�y�y�yr2c��d}tj�|��}t|t	j|��dzd���dS)Nrr>r�r�)r6r�rgrrtr�)r�rGr?s   r0rjzTestLaplace.test_sf�	sE�����M���Q������2�6�1�"�:�:�a�<�e�4�4�4�4�4�4r2c��d}tj�|��}t|t	jd|z��d���dS)Ng�}���:r>r�r�)r6r�rlrrtrU)r�r?rGs   r0rnzTestLaplace.test_isf�	sE�����M���a� � ����B�F�1�Q�3�K�K�<�e�4�4�4�4�4�4r2)
r#r$r%r�r�r
r=r�r�r�rjrnrqrrs@r0rtrt=	s���������[���Y�
�
�
�6�6��[���[�-�-�-�8�8�)?�)?�9�8�7�6�)?�V�[���0�3=�3=�3=�>�>�F�F�F�F�>�>�F�$8�8�8����*5�5�5�
5�5�5�5�5�5�5r2rtc��eZdZejd���d���Zej�dgd���ej�dgd���ej�dgd	���ej�d
d�edd
gd���D����d���������Z	d�Z
d�Zd�ZdS)�TestPowerlawrUrVc�@�tj�d��Sr�rYr�s r0r�zTestPowerlaw.rng�	rZr2rr)rlrxr+rr>rsr[rtr\r�c��g|]}d|v�|��	Sr�r8r�s  r0rSzTestPowerlaw.<listcomp>�	r�r2TFr�r�c��tj�d||||���}|tj�|��fg}	tj�|	i��d}
t��}|r||d<|r5t
j|���tj	��|d<|r||d<ttj||
fi|��dS)Nr)rr.r<r=r�rr�r�r�)r6�powerlawr�r8rAr�rt�	nextafterr~r�r�r�s            r0r�z(TestPowerlaw.test_fit_MLE_comp_optimizer�	s����~�!�!�s�i�W�(1��"�E�E���u�~�/�/��5�5�8�9���~�*�*�4��4�4�Q�7���v�v���	#�"�D��J��	=��<����
�
�R�V�G�<�<�D��L��	'�&�D��N�%�e�n�d�D�I�I�D�I�I�I�I�Ir2c	��d}d}d}tj�|||dtj�d�����}d|���dzi}|tj�|��fg}tj�|i��d	}ttj||fi|��dS)
Ng�`-@r}g�r�ߟA@ror�)r.r<r=rr�r�r>r)
r6r�r�rtr�r��ptpr8rAr�)r�r.r�r=r�r�r�r�s        r0�test_problem_casezTestPowerlaw.test_problem_case�	s���
#����"���~�!�!�A�8�5�s�/1�y�/D�/D�Q�/G�/G�"�I�I���$�(�(�*�*�q�.�)���u�~�/�/��5�5�8�9���~�*�*�4��4�4�Q�7��%�e�n�d�D�I�I�D�I�I�I�I�Ir2c�z�ttj��d}tt|���5tj�gd�dd���ddd��n#1swxYwYd}tt|���5tj�gd�d���ddd��n#1swxYwYd}tt|���5tj�gd�d	���ddd��n#1swxYwYd
}tt|���5tj�gd�d���ddd��n#1swxYwYd
}tt|���5tj�gd�d���ddd��dS#1swxYwYdS)Nz7 Maximum likelihood estimation with 'powerlaw' requiresr��rr>rerr�r�r>r?rz$Negative or zero `fscale` is outsider�r@z0`fscale` must be greater than the range of data.)r�r6r�r�r!r�r�)r�r,s  r0r�zTestPowerlaw.test_fit_warnings�	s����E�N�+�+�+�H��
�<�s�
3�
3�
3�	<�	<��N���y�y�y�q���;�;�;�	<�	<�	<�	<�	<�	<�	<�	<�	<�	<�	<����	<�	<�	<�	<�I��
�<�s�
3�
3�
3�	2�	2��N���y�y�y�q��1�1�1�	2�	2�	2�	2�	2�	2�	2�	2�	2�	2�	2����	2�	2�	2�	2�I��
�<�s�
3�
3�
3�	2�	2��N���y�y�y�q��1�1�1�	2�	2�	2�	2�	2�	2�	2�	2�	2�	2�	2����	2�	2�	2�	2�6��
�:�S�
1�
1�
1�	5�	5��N���y�y�y���4�4�4�	5�	5�	5�	5�	5�	5�	5�	5�	5�	5�	5����	5�	5�	5�	5�B��
�:�S�
1�
1�
1�	4�	4��N���y�y�y���3�3�3�	4�	4�	4�	4�	4�	4�	4�	4�	4�	4�	4�	4����	4�	4�	4�	4�	4�	4sY�%A#�#A'�*A'�$B6�6B:�=B:�$D	�	D
�D
�,$E�E �#E �?$F0�0F4�7F4c��gd�}tj}tjd���5t	|||j��ddd��dS#1swxYwYdS)N)rrr>r>r�r�r�r�rerer�rr���over)r6r�rtr�r��nnlf)r�r�rRs   r0�test_minimum_data_zero_gh17801z+TestPowerlaw.test_minimum_data_zero_gh17801�	s���4�3�3���~��
�[�h�
'�
'�
'�	A�	A�)�$��d�i�@�@�@�	A�	A�	A�	A�	A�	A�	A�	A�	A�	A�	A�	A����	A�	A�	A�	A�	A�	As�A
�
A�AN)
r#r$r%r�rcr�r�r
r%r�r�r�r�r8r2r0r�r��	s<�������V�^�*�%�%�%�+�+�&�%�+��[���[�*=�*=�*=�>�>��[���Y�
�
�
�3�3��[���[�*�*�*�5�5��[���<�-�-���$���q�)I�)I�)I�-�-�-�.�.�J�J�.�.�6�5�4�3�?�>�J�"J�J�J� 4�4�4�6A�A�A�A�Ar2r�c� �eZdZd�Zd�Zd�ZdS)�TestInvGammac��tj��5tjdt��tj�dd���}gd�}t
||��gd�}tj�|d���}gd�tjdd	gtj	d
dgtj	tj	dgf}t||��D]\}}t||���	ddd��dS#1swxYwYdS)
Nr<g��(\O3@rrD)g���~��?g�C�X��&?g*�7�gS�?gBqF�q@)皙�����?�������@gffffff@)rlg�E��y�?gў3oz��?g��^Y�b�?g,��r��?gO���D@g2�*Z@g�F�N�8@)r@rArBrCr6r�rrtr�r�r�r)r�rrir.rGrs      r0�test_invgamma_inf_gh_1866z&TestInvGamma.test_invgamma_inf_gh_1866�	sU���
$�
&�
&�	*�	*��!�'�>�:�:�:��>�'�'�%��'�@�@�D�%�%�%�H��D�(�+�+�+����A��>�'�'�!�V�'�<�<�D�8�8�8����}�=����k�:������5�7�H��D�(�+�+�
*�
*���1�#�A�q�)�)�)�)�
*�	*�	*�	*�	*�	*�	*�	*�	*�	*�	*�	*�	*����	*�	*�	*�	*�	*�	*s�CC6�6C:�=C:c���tjdd��}tj�|d��}tj�|d��}t
||��dS)Ng�������rr)rtr[r6r�rKr]rr&s    r0razTestInvGamma.test_cdf_ppf
sW���K��a� � ���N���q�!�$�$��
�^�
�
��1�
%�
%����2�����r2c� �tjdkrtjdd��}ntjdd��}tj�|d��}tj�|d��}t||d���dS)N�r>ro�rrr�)	�sys�maxsizertr[r6r�rgrlrr&s    r0r	zTestInvGamma.test_sf_isf
s~���;������A�s�#�#�A�A���A�r�"�"�A��N���a��#�#��
�^�
�
��1�
%�
%����2�C�(�(�(�(�(�(r2N)r#r$r%r�rar	r8r2r0r�r��	sA������*�*�*�&���)�)�)�)�)r2r�c��eZdZd�Zd�Zej�ej	���
d��d���d���Zd�ZdS)	�TestFc��tjddgg}|D]\}}}|j|jg|�R�}�d�|D��}d�|D��}t	||��dS)N)r>rrc�<�g|]\}}}|j|jg|�R���Sr8�r@r.�rQ�_f�_argsr�s    r0rSz(TestF.test_endpoints.<locals>.<listcomp>)
�4��=�=�=���E�1�v�r�v�b�d�#�U�#�#�#�=�=�=r2c��g|]\}}}|��	Sr8r8�rQr�r��	_correct_s    r0rSz(TestF.test_endpoints.<locals>.<listcomp>*
���>�>�>�!5��U�I�9�>�>�>r2)r6�fr@r.r)r�r�r�r��_correct�ans�corrects       r0rMzTestF.test_endpoints#
s�����&�#�&�'��#'�	'�	'��B��x��"�&���&��&�&�&�C�C�=�=��=�=�=��>�>��>�>�>��!�#�w�/�/�/�/�/r2c�^�tj�ddd���\}}}}ttj|����ttj|����ttj|����ttj|����dS)Nr|�@rr)r6r�r	rtrros     r0�test_f_momentszTestF.test_f_moments-
s����W�]�]�2�s�F�]�;�;�
��1�a�����A���������A���������A�������B�K��N�N�"�#�#�#�#�#r2�mips64ztest known to fail on mips64el��reasonc���tj��5tjdt��tj�dgdzgd�d���ddd��dS#1swxYwYdS)Nr<r|re)r>rerr�r��dfn�dfdr)r@rArBrCr6r�r�s r0�test_moments_warningszTestF.test_moments_warnings5
s����
$�
&�
&�	H�	H��!�'�>�:�:�:��G�M�M�r�d�1�f�,�,�,��M�G�G�G�	H�	H�	H�	H�	H�	H�	H�	H�	H�	H�	H�	H����	H�	H�	H�	H�	H�	Hs�AA$�$A(�+A(c��tjdgdgg��}tjddg��}tj�||d���\}}}}||dz
zgdz}t	||��d|dzz||zdz
z|z|dz
dzz|dz
z}t	||��d|z|zdz
tjd|dz
z��z|d	z
tj|||zdz
z��zz}	t	||	��d|d
|zdz
z||zdz
z|dz
|dz
dzzzz}
||d	z
z|dz
z||zdz
z}|
|z}t	||��dS)Nr�r|rrr�r>rer�rr��)rtrr6r�rr�)
r�r�r�r�r�rFrDrN�v2�s2�k2num�k2den�k2s
             r0�test_stats_broadcastzTestF.test_stats_broadcast<
s����h���b�T�{�#�#���h��B�x� � ���W�]�]�s��V�]�D�D�
��1�a���S�1�W�o�
�q�
 ����2����
��a��Z�3��9�q�=�
)�C�
/�3��7�Q�,�
>�#��'�
J����2������u�s�{�Q��"�'�!�S�1�W�+�"6�"6�6��a��2�7�3��c�	�A�
�#6�7�7�7�9����2�����c�Q�s�U�R�Z�(�C�#�I��M�:��Q�w�3��7�Q�,�.�/�0���s�Q�w��3��7�+�s�S�y�1�}�=��
�U�]����2�����r2N)
r#r$r%rMr�r�r��skipif�platform�machiner]r�r�r8r2r0r�r�"
s�������0�0�0�$�$�$��[���(��(�*�*�5�5�h�?�?�Hh��i�i�H�H�j�i�H�����r2r�c�h�ttj�ddg��ddg��dS)Nr�rg�����?g�����?)rr6r�rEr8r2r0�test_rvgeneric_stdr�N
s/���e�g�k�k�1�a�&�1�1�J�
�3K�L�L�L�L�Lr2c�b�ttj�dd���tjtjtjtjf��ttj�dd���dtjtjtjf��ttj�dd���dtjtjtjf��ttj�dd���ddtjtjf��ttj�d	d
���tjtjf��ttj�dd
���dtjf��ttj�dd
���dtjf��ttj�d
d
���d��dS)Nrr)rpr�)\��(�?r}r>g�G�z@g� i@r�rCg�G�z@reg
ףp=
@)r}gq��@)rr6r�rtr�r�r8r2r0�test_moments_tr�S
s��������!�V��4�4��&�"�&�"�&�"�&�1�3�3�3������$���7�7��r�v�r�v�r�v�.�0�0�0������!�V��4�4��r�v�r�v�r�v�.�0�0�0������$���7�7�������7�9�9�9������!�T��2�2�R�V�R�V�4D�E�E�E������$���5�5��R�V�}�E�E�E������!�T��2�2�S�"�&�M�B�B�B������$���5�5�7N�O�O�O�O�Or2c�t�gd�}gd�}ttj�|��|d���dS)N)rr>r�ro)g���?@g�T�M]�?g���gY�?gU����?r�r�)rr6r�rX)rpris  r0�test_t_entropyr�c
sE��	���B�7�7�7�H��E�G�O�O�B�'�'���>�>�>�>�>�>r2�methname)r@r�rKr]rgrl�
df_infmask)rrr)rrrc��tj�d��tj|t���}tj�dd|j���}tjj|j�}tj||<tj
|dd���}tj
||dd���}tjdd���}t||��}t||��}t||��}	||��}
t|
||	||����t|
||||����dS)	Nr�r�rnr�r�r�rpr<r=r;)rtr�r�r��boolr�r�randnr�r6r�rEr�r)r�r�rprG�t_dist�
t_dist_ref�	norm_dist�t_meth�
t_meth_ref�	norm_methr�s           r0�
test_t_inf_dfr�k
s6���I�N�N�1������J�d�3�3�3�J�	��	�	�1�b�z�'7�	�	8�	8�B�
�	���)�*�A��V�B�z�N�
�W����
+�
+�
+�F���B�
�{�O��!�<�<�<�J��
�q��*�*�*�I�
�V�X�
&�
&�F���X�.�.�J��	�8�,�,�I�
�&��)�)�C���Z��)�)�A�j�M�":�":�;�;�;���j�[�!�:�:�a���n�#=�#=�>�>�>�>�>r2c��tj�d��tj|t���}tj�dd|j���}tj||<tj	�|ddd���}tj
�ddd�	��}tj	�||ddd���}td
��D]G}t|||||��t|||||���Htj	�
|dd���}tj
�
dd���}tj	�
||dd���}t|||��t|||��dS)
Nrr�rnr�r�rr)rpr<r=r�r<r=rrer�r;)rtr�r�r�r�r�rr�r6r�rEr�rrX)r�rpr��
res_ex_inf�res_ex_noinfr�s      r0�test_t_inf_df_stats_entropyr��
s���
�I�N�N�1������J�d�3�3�3�J�	��	�	�1�b�z�'7�	�	8�	8�B��V�B�z�N�
�'�-�-�2�1�A�v�-�
>�
>�C���!�!�a�q�&�!�A�A�J��7�=�=�B�
�{�O��!�)/�!�1�1�L�
�1�X�X�;�;���S��V�J�'��A��7�7�7��S��V�Z�K�(�,�q�/�:�:�:�:�
�'�/�/�R�Q�a�/�
0�
0�C���#�#���#�3�3�J��7�?�?�b�*��o�1�A�?�F�F�L���Z��*�-�-�-���j�[�!�<�0�0�0�0�0r2c�V�eZdZd�Zd�Zd�Zd�Zd�Zd�Zd�Z	d�Z
d	�Zd
�Zd�Z
d�Zd
S)�TestRvDiscretec�D�tj�d��dSr�r�r�s r0r�zTestRvDiscrete.setup_method�
r�r2c	��gd�}gd�}d}tjd||f���}|�|���}tt	|t
j����t||��D]I\}}ttt||k��t|��z|z
��dk���J|���}tjt|��tj��sJ�dS)N)r�rrr>r�re)r}�333333�?rjr}r�r}r��sample)�name�valuesr�r�)r6rUr�r	r
rAr.r�r�r6r�rtr�r��integer)r��states�probability�samples�rrGrFr?s        r0rzTestRvDiscrete.test_rvs�
s���$�$�$��4�4�4������8�V�[�4I�J�J�J��
�E�E�w�E�����
�1�e�m�,�,�-�-�-����,�,�	@�	@�D�A�q��C��A��F���E�'�N�N�2�Q�6�7�7�$�>�?�?�?�?�
�E�E�G�G���}�T�!�W�W�b�j�1�1�1�1�1�1�1r2c�f�tjgd���}tjgd�|f���}t	t||����}|���}t||��tjgd�gd�f���}|���}t|d��dS)N)r5r}r�r�r�)rrrr})	rtrr6rUr6rrXrr)r�r�r?r8r9s     r0r:zTestRvDiscrete.test_entropy�
s�����*�*�*�+�+����i�i�i��%7�8�8�8���%��u�-�-�.�.�.�
�
�I�I�K�K����:�&�&�&���i�i�i����%=�>�>�>��
�I�I�K�K���Q������r2c��gd�}gd�}tj||f���}ddgddgg}t|�|��dd	gd
dggd�
��dS)Nr��rxr�r�r�rr
rr>rxr�r}r�r,r�)r6rUrr)r��xk�pkr�rGs     r0r3zTestRvDiscrete.test_pmf�
s���
�Y�Y��
�_�_��
�
�r�2�h�
/�
/�
/���"�X��!�W�
������q�	�	��s���c��$�*/�	1�	1�	1�	1�	1�	1r2c����gd�}gd�}tj||f����gd�}gd�}t��|��|d���t�fd�|D��|d���dS)	Nr�r�r�)r�rr�r�r|rrer�)rrxrxrxrbrbrrr,r�c�:��g|]}��|����Sr8)rK)rQr`r�s  �r0rSz+TestRvDiscrete.test_cdf.<locals>.<listcomp>�
s#���7�7�7��������7�7�7r2)r6rUrrK)r�r�r��x_valuesrir�s     @r0r"zTestRvDiscrete.test_cdf�
s����
�Y�Y��
�_�_��
�
�r�2�h�
/�
/�
/��5�5�5��5�5�5������x�(�(�(��?�?�?�?�	�7�7�7�7�h�7�7�7� �u�	.�	.�	.�	.�	.�	.r2c����gd�}gd�}tj||f����gd�}gd�}t��|��|d���t�fd�|D��|d���dS)	Nr�r�r�)rlrx�333333�?rbr�r)rrr>r>rerer,r�c�:��g|]}��|����Sr8)r])rQr_r�s  �r0rSz+TestRvDiscrete.test_ppf.<locals>.<listcomp>�
s#���5�5�5�q�������5�5�5r2)r6rUrr])r�r�r��q_valuesrir�s     @r0rzTestRvDiscrete.test_ppf�
s����
�Y�Y��
�_�_��
�
�r�2�h�
/�
/�
/��0�0�0��%�%�%������x�(�(�(��?�?�?�?�	�5�5�5�5�H�5�5�5� �u�	.�	.�	.�	.�	.�	.r2c	���gd�gd�f}tj|���}t|�|�|jdd���dz��|jdd���dS)N)rr>rersr�)rlr�r�r�rlr�r�rQr)r6rUrr]rKr�)r�rr�s   r0�test_cdf_ppf_nextz TestRvDiscrete.test_cdf_ppf_next�
s|�����!:�!:�!:�;��
�
�d�
+�
+�
+���2�6�6�"�&�&���s��s��"4�"4�t�";�<�<��5����9�	&�	&�	&�	&�	&r2c�B�tjd���d��}tjgd�gd�gd�g��}t	j||f���}t
|���tj|j	|j
z��d���dS)Nr)r�re)rlrlr�r�)rlrlr�r�r�r,r�)rtr�rrr6rUrr�r6r�r��r�r�r�r�s    r0�test_multidimensionz"TestRvDiscrete.test_multidimension�
s���
�Y�r�]�]�
"�
"�6�
*�
*��
�X�-�-�-�-�-�-�-�-�-�/�0�0���
�r�2�h�
/�
/�
/����	�	���R�V�B�E�B�E�M�%:�%:��G�G�G�G�G�Gr2c��gd�}ddg}tttjfit	||f�����gd�}tttjfit	||f�����gd�}gd�}tttjfit	||f�����gd�}gd�}tttjfit	||f�����dS)Nr�rxr�)rx�333333�?gffffff濩rr>r�rer�)r�r�r�r�r�)r�r�r6rUr��r�r�r�s   r0�test_bad_inputzTestRvDiscrete.test_bad_input�
s���
�Y�Y���3�Z���j�%�"3�M�M�t�B��8�7L�7L�7L�M�M�M�
�Y�Y���j�%�"3�M�M�t�B��8�7L�7L�7L�M�M�M�
�Y�Y��
�
�
���j�%�"3�M�M�t�B��8�7L�7L�7L�M�M�M�
�_�_��
'�
'�
'���j�%�"3�M�M�t�B��8�7L�7L�7L�M�M�M�M�Mr2c��tjd���d��tjdd��}}t	t
tjfit||f�����tjd���d��tjdd��}}t	t
tjfit||f�����tjd���d��tjdd��}}tt
j||f����
d��d��dS)	Nre�r>r>)r>r�gUUUUUU�?r�r�r�r>r)rtr�rrr�r�r6rUr�rrrs   r0�test_shape_rv_samplez#TestRvDiscrete.test_shape_rv_sample�
s����1���%�%�f�-�-�r�w�v�s�/C�/C�B���j�%�"3�M�M�t�B��8�7L�7L�7L�M�M�M���1���%�%�f�-�-�r�w�v�s�/C�/C�B���j�%�"3�M�M�t�B��8�7L�7L�7L�M�M�M���1���%�%�f�-�-�r�w�v�s�/C�/C�B���U�&�r�2�h�7�7�7�;�;�A�>�>��D�D�D�D�Dr2c���gd�}gd�}tj||f���}t|���t	j|j|jz��d���dS)N)rr>rerrsr|)rlr�r�r�r�rlr�r,r�)r6rUrr�rtr6r�r�rs    r0�test_expect1zTestRvDiscrete.test_expect1
sd��
 �
 �
 ��
+�
+�
+��
�
�r�2�h�
/�
/�
/����	�	���R�V�B�E�B�E�M�%:�%:��G�G�G�G�G�Gr2c
���gd�}gd�}tj||f���}t|���|���d���t|���td�t
||��D����d���t|�d���td�t
||��D����d���dS)	N)/gi@g�r@gy@g@@g��@g�@��@g �@r4g0�@g��@gP�@g�@gp�@g�@g��@g �@g��@r5gh�@g0�@g��@g��@g��@gP�@g�@g�@g��@r6g8�@g�@gȩ@g��@gX�@g �@g�@g��@gx�@g@�@g�@gh�@g̰@g0�@g��@g��@g\�@g��@)/g-C��6:?r}gF%u�k?g:��v��z?r}r}g.n��r?g�|гY��?r}r}r}g@��6*?�4U0*�C?g@+��?g�C��6z?gn��r?g ����?g�OjM�?g n���?g��1w-!�?g��:pΈ�?gףp=
�?g�_�Le?r}g��^)�p?g�~j�t�x?g��J�4q?g��G�zt?g�J�4q?g���Mb@?g@P�sׂ?g@�^)ˀ?r}r}g���N@s?g�1w-!_?rgൄ|г�?r}g��H�}}?r}g�
F%u�?r}r}g�^)ˀ?g����v?r}r�r,r�c3�&K�|]\}}||zV��
dSrrr8�rQr��ws   r0�	<genexpr>z.TestRvDiscrete.test_expect2.<locals>.<genexpr>3s*����9�9�d�a��A��E�9�9�9�9�9�9r2c��|dzS�Nr>r8r�s r0r_z-TestRvDiscrete.test_expect2.<locals>.<lambda>6�
��A�q�D�r2c3�,K�|]\}}|dz|zV��dS)r>Nr8rs   r0rz.TestRvDiscrete.test_expect2.<locals>.<genexpr>7s.����<�<���A�A�q�D�1�H�<�<�<�<�<�<r2)r6rUrr�rDr6r�)r�r�pyr�s    r0�test_expect2zTestRvDiscrete.test_expect2s
��
=�
=�
=��?�?�?�� �
�q�"�g�
.�
.�
.��	��	�	���R�W�W�Y�Y�U�;�;�;�;���	�	����9�9�c�!�R�j�j�9�9�9�9�9��	G�	G�	G�	G�	��	�	�.�.�1�1��<�<��Q����<�<�<�<�<�5�	J�	J�	J�	J�	J�	Jr2N)r#r$r%r�rr:r3r"rrrrrrrr8r2r0r�r��
s����������2�2�2�
�
�
�	1�	1�	1�.�.�.�.�.�.�&�&�&�H�H�H�N�N�N� 
E�
E�
E�H�H�H�#J�#J�#J�#J�#Jr2r�c��eZdZd�Zd�ZdS)�TestSkewCauchyc��tjddd��}ttj�|d���tj�|����ttj�|d���tj�|����ttj�|d���tj�|����dS�Nr�r�rorrB)	rtrrr6�
skewcauchyr@r�rKr]rms  r0�test_cauchyzTestSkewCauchy.test_cauchy;s����K��A�s�#�#��!�%�"2�"6�"6�q�A�"6�">�">�"'�,�"2�"2�1�"5�"5�	7�	7�	7�!�%�"2�"6�"6�q�A�"6�">�">�"'�,�"2�"2�1�"5�"5�	7�	7�	7�!�%�"2�"6�"6�q�A�"6�">�">�"'�,�"2�"2�1�"5�"5�	7�	7�	7�	7�	7r2c��tj�d��tj�d��dzdz
}tj�d��dzdz
}gd�}gd�}t	t
j�||��|��t	t
j�||��|��t	t
j�	||��|��dS)Nrrnr>rr�)
g&4��5�?g�Q�ʶ��?g-�4<?��?g���J�?gw�@�v�?gJ%����?g>�[u��?g
�a��2�?g	�R�n�?gѦ}c�ي?)
gN@�Q���?gPPz@	�?g����?g�;
��7�?g�lg��?g!?%����?g�Qҙ�?gC��"��?g���o��?g��@�Sg�?)
rtr�r�rbrr6r r@rKr])r�r.rGr@rKs     r0�test_skewcauchy_Rz TestSkewCauchy.test_skewcauchy_RDs���&	�	���q�����I�N�N�2����"�Q�&���I�N�N�2����#�a�'��%�%�%��$�$�$��	��(�,�,�Q��2�2�C�8�8�8���(�,�,�Q��2�2�C�8�8�8���(�,�,�S�!�4�4�a�8�8�8�8�8r2N)r#r$r%r!r#r8r2r0rr:s2������7�7�7� 9� 9� 9� 9� 9r2rr>r�g������@r�gףp=
�1@rgR����Z@�ig�S㥛Ԋ@rlrmg&����@i�Qi�'i0ybi��l�HO1Zi��3r�i����rni���roik���ri1��r�i�i�4��i Ni�(l����EI.�@
i�c��eZdZd�Zd�Zd�Zd�Zd�Zd�Ze	j
�de��d���Z
d	�Zd
S)�TestSkewNormc�.�td��|_dSr�)rr�r�s r0r�zTestSkewNorm.setup_method�s��%�d�+�+����r2c���tjddd��}ttj�|d���tj�|����dSr)rtrrr6�skewnormr@rErms  r0�test_normalzTestSkewNorm.test_normal�sY���K��A�s�#�#��!�%�.�"4�"4�Q�!�"4�"<�"<�"'�*�.�.��"3�"3�	5�	5�	5�	5�	5r2c��d}tj�d||j���}t	||j��tj�d||j���}t	||j��dS)N)r�rer�r+)r.rr�r�)r6r*r�r�rr)r�rrGs   r0rzTestSkewNorm.test_rvs�sm�����N����E����I�I���U�A�G�$�$�$��N����%�d�h��G�G���U�A�G�$�$�$�$�$r2c���tj�dtd��dd|j���}tj|��tj|��tj|��tj	|��g}tj�dddd���}t||d���tj�d	td��dd|j���}tj|��tj|��tj|��tj	|��g}tj�d	ddd���}t||d���dS)
Nrer�r�r>)r.rr<r=r�r)r.r<r=rrrZ)r6r*r�r-r�rtrDrr�kurtosisr)r��Xrirs    r0r�zTestSkewNorm.test_moments�s,���N�����S���q��,0�H�
�
6�
6���G�A�J�J���q�	�	�5�:�a�=�=�%�.��:K�:K�L���>�'�'�!��!�V�'�L�L��!�(�H�a�@�@�@�@��N����#�c�(�(���,0�H�
�
6�
6���G�A�J�J���q�	�	�5�:�a�=�=�%�.��:K�:K�L���>�'�'�"�!�1�f�'�M�M��!�(�H�a�@�@�@�@�@�@r2c��tj�gd�d��}t|t	jd��d���tj�dd��}t|dd���dS)	N)rnr�r�r�r�r,r�r�rir)r6r*rKrrt�onesrLs  r0�test_cdf_large_xzTestSkewNorm.test_cdf_large_x�sn��
�N���|�|�|�R�0�0����2�7�1�:�:�E�2�2�2�2��N���r�3�'�'����3�U�+�+�+�+�+�+r2c��gd�gd�gd�gd�gd�g}|D]l\}}}tj�||��}t||d���tj�||��}t||d����mdS)N)r�rgzbL��e�9)rZr>g�n'�/2�;)r�r�g���:�:)r�r�g|����<)r�rZ�
�O�ul2;rQr�)r6r*rKrrg)r��cdfvalsrGr.�cdfvalr?s      r0�test_cdf_sf_small_valuesz%TestSkewNorm.test_cdf_sf_small_values�s���
.�-�-�+�+�+�,�,�,�.�.�.�/�/�/�
��$�	2�	2�L�A�q�&���"�"�1�a�(�(�A��A�v�D�1�1�1�1���!�!�1�"�q�b�)�)�A��A�v�D�1�1�1�1�1�	2�	2r2z
a, momentsc��t|d���D]7\}}tj�||��}t	||d����8dS)Nr)�startr,r�)r7r6r*r�r)r�r.r�orderri�moms      r0�test_noncentral_momentsz$TestSkewNorm.test_noncentral_moments�s[��(���:�:�:�	7�	7�O�E�8��.�'�'��q�1�1�C��C���6�6�6�6�6�	7�	7r2c	���tj�d��}d\}}}tj|||��}|�d|���}tj�|dd���\}}}	tj�|dd���\}
}}||cxkrdksnJ�||
ksJ�tj�|ddd	�
��\}
}}|dksJ�tj|
||��}|�d���}tj|��tj|��f}t||��tj
�d
d|���}tj�|��}tjtj|����sJ�tj�|d	���\}}}tj
|��sJ�tj|��tj|��}}t|||tjdtjz��zz��t||dzd
dtjzz
z��dS)Nl#
�k�e�)r�rjrxror�r�r�r?g���������mm�r��method�msrr�r@r>)rtr�r�r6r*r�r�rDrrrnrr�isinfrr�rB)r�r�r.r<r=rRr��a2�loc2�scale2�a3�loc3�scale3�a4�loc4�scale4�dist4r�r�a5�loc5�scale5r�r�s                        r0r=zTestSkewNorm.test_fit�sC���i�#�#�$7�8�8��$�
��3���~�a��e�,�,���h�h�C�c�h�2�2��!�>�-�-�c�4�a�-�@�@���D�&� �>�-�-�c�4�a�-�@�@���D�&��t� � � � �q� � � � � � ��R�x�x�x�x�!�>�-�-�c�1�Q�t�-�L�L���D�&���{�{�{�{���r�4��0�0���k�k�$�k�'�'���g�c�l�l�E�J�s�O�O�+����S�!�!�!��l���q�s���=�=���n� � ��%�%���v�b�k�#�&�&�'�'�'�'�'�!�>�-�-�c�$�-�?�?���D�&��x��|�|�����w�s�|�|�R�V�C�[�[�1����4�&�2�7�1�R�U�7�+;�+;�";�;�<�<�<���6�1�9��A���I�
�6�7�7�7�7�7r2N)r#r$r%r�r+rr�r2r7r�r�r
�_skewnorm_noncentral_momentsr<r=r8r2r0r'r'�s�������,�,�,�5�5�5�%�%�%�A�A�A�,�,�,�2�2�2�"�[���\�+G�H�H�7�7�I�H�7�
&8�&8�&8�&8�&8r2r'c�&�eZdZd�Zd�Zd�Zd�ZdS)�	TestExponc�`�ttj�d��d��dSrJ)rr6rRr@r�s r0�	test_zerozTestExpon.test_zero�s&���U�[�_�_�Q�'�'��+�+�+�+�+r2c���ttj�d��d��ttj�tj�d����d��dS)Ng�C��]r2<r�)rr6rRrKrlrgr�s r0�	test_tailzTestExpon.test_tail�sR���U�[�_�_�U�+�+�U�3�3�3��U�[�_�_�U�[�^�^�B�%7�%7�8�8�"�=�=�=�=�=r2c��tjdddddtjg��}ttt
jj|��dS�N����o_�?��46�@�yX�5ͻ@��j+����?�U���X@)rtrr�r�r�r6rRr�rms  r0�test_nan_raises_errorzTestExpon.test_nan_raises_error�<���H�f�f�f�f�f�b�f�E�F�F���j�%�+�/�1�5�5�5�5�5r2c��tjdddddtjg��}ttt
jj|��dSrY)rtrr�r�r�r6rRr�rms  r0�test_inf_raises_errorzTestExpon.test_inf_raises_errorr`r2N)r#r$r%rUrWr_rbr8r2r0rSrS�sP������,�,�,�>�>�>�6�6�6�
6�6�6�6�6r2rSc� �eZdZd�Zd�Zd�ZdS)�TestNormc��tjdddddtjg��}ttt
jj|��dSrY)rtrr�r�r�r6rEr�rms  r0r_zTestNorm.test_nan_raises_error�<���H�f�f�f�f�f�b�f�E�F�F���j�%�*�.�!�4�4�4�4�4r2c��tjdddddtjg��}ttt
jj|��dSrY)rtrr�r�r�r6rEr�rms  r0rbzTestNorm.test_inf_raises_errorrfr2c�\�gd�}tttjj|d���dS)Nr��shrimp)�plate)r�r�r6rEr�rms  r0�test_bad_keyword_argzTestNorm.test_bad_keyword_args,���I�I���i�����(�C�C�C�C�C�Cr2N)r#r$r%r_rbrkr8r2r0rdrd
sF������5�5�5�
5�5�5�
D�D�D�D�Dr2rdc��eZdZdZd�Zd�ZdS)�TestUniform�gh-10300c��tjdddddtjg��}ttt
jj|��dSrY)rtrr�r�r�r6r�r�rms  r0r_z!TestUniform.test_nan_raises_error�=���H�f�f�f�f�f�b�f�E�F�F���j�%�-�"3�Q�7�7�7�7�7r2c��tjdddddtjg��}ttt
jj|��dSrY)rtrr�r�r�r6r�r�rms  r0rbz!TestUniform.test_inf_raises_error$rpr2N)r#r$r%�__doc__r_rbr8r2r0rmrms7�������N�8�8�8�
8�8�8�8�8r2rmc
�D�eZdZd�Zd�Zd�Zd�Zej�	dgd���d���Z
ej�	dgd	�gd
�gd�gd�gd
�g��d���Zej�	dgd�gd�gd�gd�gd�gd�g��d���ZdS)�
TestExponNormc�l�d�}d\}}}d||zz}tj�|||d���}t|||||����d\}}}d||zz}tj�|||d���}t|||||����d\}}}d||zz}tj�|||d���}t|||||����d\}}}d||zz}tj�|||d���}t|||||����dS)	Nc��dd||zdzzz}d||zdzz|dzz}dd||zdzzdzz}|d|zz||zd||zzz||gS)Nrrr>r�r��@r�r8)�lam�sigr��opK2�exp_skew�exp_kurts      r0�get_momsz,TestExponNorm.test_moments.<locals>.get_moms-sz����c�#�g��\�)�)�D��C�#�I��>�)�D�4�L�8�H��a�3��9�q�.�0�B�7�7�H���3��J��C��#�s�3�w�-� 7��8�L�Lr2)rrrrrr�)r�r>rl)rr�r)r�r|rj)r6�	exponnormr)r�r}r�ryrx�K�stss       r0r�zTestExponNorm.test_moments+sp��	M�	M�	M����C���3��9����o�#�#�A�2�S�&�#�I�I���C���#�s�B�!7�!7�8�8�8�!���C���3��9����o�#�#�A�2�S�&�#�I�I���C���#�s�B�!7�!7�8�8�8����C���3��9����o�#�#�A�2�S�&�#�I�I���C���#�s�B�!7�!7�8�8�8�"���C���3��9����o�#�#�A�2�S�&�#�I�I���C���#�s�B�!7�!7�8�8�8�8�8r2c��tjdddddtjg��}ttt
jj|dd���dS�	NrZr[r\r]r^rrr�)rtrr�r�r�r6r~r�rms  r0r_z#TestExponNorm.test_nan_raises_errorF�D���H�f�f�f�f�f�b�f�E�F�F���j�%�/�"5�q�q��K�K�K�K�K�Kr2c��tjdddddtjg��}ttt
jj|dd���dSr�)rtrr�r�r�r6r~r�rms  r0rbz#TestExponNorm.test_inf_raises_errorKr�r2c�v�ttj�dd��d��ttj�dd��d��ttj�dd��d��ttj�dd��d��dS)Ni|���rr}�rz)rr6r~r@r�s r0�test_extremes_xzTestExponNorm.test_extremes_xPs����E�O�/�/��a�8�8�#�>�>�>��E�O�/�/��a�8�8�#�>�>�>��E�O�/�/��d�;�;�S�A�A�A��E�O�/�/��d�;�;�S�A�A�A�A�Ar2zx, K, expected))r�rzg�6
�N��-)rrzg�՜n+H�?)r�rzgi�����?)rYrzg�b�JI�-)rnrg�I�8�?)rnr�gQ3|�-0?c�f�ttj�||��|d���dS)Ng�dy���=r�)rr6r~r@)r�rGrris    r0�test_std_pdfzTestExponNorm.test_std_pdfbs1��	���+�+�A�q�1�1�8�%�H�H�H�H�H�Hr2zx, K, scale, expected)rrzrgVAҤ��?)r��{�G�zt?rg�'^���>)rFrzror})rFrzr�gbr�� ;)r�r
rgV�Me���?c��tj�|||���}|dkr
|dksJ�dSt||d���dS)NrDr}r�r�)r6r~rKr�r�rGrr=rir?s      r0�test_cdf_small_KzTestExponNorm.test_cdf_small_KxsU��
�O����1�E��2�2���s�?�?���8�8�8�8�8�8��A�x�e�4�4�4�4�4�4r2)rnrzrgh�G�}$;)r>r�rg]fJ��?)r�r�rxg31"�g#;)rnr�rxgf�
��+�-)r�r�rxr})r�r
rg�a9S���?c��tj�|||���}|dkr
|dksJ�dSt||d���dS)NrDr}g�-���a=r�)r6r~rgrr�s      r0�test_sf_small_KzTestExponNorm.test_sf_small_K�sU��
�O���q�!�5��1�1���s�?�?���8�8�8�8�8�8��A�x�e�4�4�4�4�4�4r2N)
r#r$r%r�r_rbr�r�r�r
r�r�r�r8r2r0rtrt*s�������9�9�9�6L�L�L�
L�L�L�
B�B�B�$�[���-�B�B�B�C�C�I�I�C�C�I��[���4�>�>�>�D�D�D�4�4�4�G�G�G�?�?�?�	A�B�B�5�5�
B�B�5�&�[���4�B�B�B�@�@�@�D�D�D�F�F�F�3�3�3�@�@�@�B�C�C�5�5�C�C�5�5�5r2rtc� �eZdZd�Zd�Zd�ZdS)�TestGenExponc��ddlm}tj�tjddd��ddd��}t||d���dd��dS)	Nr)�simpsrnrzrxr|)�dxr)�scipy.integrater�r6�genexponr@rAr�r)r�r�r?s   r0�test_pdf_unity_areaz TestGenExpon.test_pdf_unity_area�sd��)�)�)�)�)�)��N���u�|�A�r�4�8�8�#�s�C�H�H���E�E�!��-�-�-�q�!�4�4�4�4�4r2c���tj�tjddd��ddd��}ttjd|k|dkz����dS)Nrrnrzrxr|r)r6r�rKrAr�r	r)r�rKs  r0�test_cdf_boundszTestGenExpon.test_cdf_bounds�sV���n� � ���a��T�!:�!:�C��c�J�J����	�1��8��q��1�2�2�3�3�3�3�3r2c�n�tj�dddd��}t|dd���dS)Nrrr>r�g�M�<r�r�)r6r�rgrris  r0�test_sf_tailzTestGenExpon.test_sf_tail�s;��
�N���b�!�Q��,�,����1��>�>�>�>�>�>r2N)r#r$r%r�r�r�r8r2r0r�r��sA������5�5�5�4�4�4�
?�?�?�?�?r2r�c��eZdZd�ZdS)�TestExponpowc��ttj�dd��d��ttj�tj�dd��d��d��dS)N绽���|�=r|r+r�rb)rr6�exponpowrKrlrgr�s r0rWzTestExponpow.test_tail�si���E�N�.�.�u�b�9�9�5�A�A�A��E�N�.�.�u�~�/@�/@��B�/G�/G��L�L��	�	�	�	�	r2N)r#r$r%rWr8r2r0r�r��s#����������r2r�c�Z�eZdZd�Zej�d��d���ZdS)�TestSkellamc���tjdd��}d\}}tjgd���}ttj�|||��|d���dS)Nr�r�rnr�)g�YjP'?g�$�S��?g�Q`s��2?gC/q��F?gD�<]Y?g/��X�j?g8�6�y?g��]a¨�?g�:���?g�����?gQ���>�?g�����?g�:���?g��]a¨�?g9�6��?g0��X��?gA�<]�?gB/q���?g�Q`s���?g�$�S��?g�YjP'�?gX���_��?g�+r�x�?g�fS�r��?g�J�Xx~?r)rAr�rrr6�skellamr)r�rD�mu1�mu2�skpmfRs     r0r3zTestSkellam.test_pmf�sm���L��b�!�!�����S���-�-�-�
.�
.��	�E�M�-�-�a��c�:�:�F�B�O�O�O�O�O�Or2�ignore::RuntimeWarningc���tjdd��}d\}}tjgd���}ttj�|||��|d���dS)Nr�rr�)gƃQ��?gS��`X'?g��Vy)�>?g�:�	S?ge��i�.f?gȼJ�]x?g���)�?g{^'�+i�?g�L�Ig>�?g�n�Z�?g?rv�ͪ�?g���A�?g�0����?g#��J��?gqSm
.
�?g~��B��?gd���?g�FB�D��?gQ��9�?gm��F^�?g�c�M�y�?g�G�J�q�?g��C^c�?g.�)�{�?gñ��ݸ�?r�r)rAr�rrr6r�rK)r�rDr�r��skcdfRs     r0r"zTestSkellam.test_cdf�so��
�L��b�!�!�����S���-�-�-�
.�
.��	�E�M�-�-�a��c�:�:�F�A�N�N�N�N�N�Nr2N)r#r$r%r3r�r�r�r"r8r2r0r�r��sX������P�P�P�*�[��� 8�9�9�O�O�:�9�O�O�Or2r�c��eZdZd�Zd�ZdS)�TestLognormc��tj��5tjdt��tj�gd�d��}t|gd���ddd��dS#1swxYwYdS)Nr<�rrxrr)r}g~���r�?g�e3E��?)r@rArBrCr6r�r@r�r�r@s  r0rzTestLognorm.test_pdf�s����
$�
&�
&�	J�	J��!�'�>�:�:�:��-�#�#�K�K�K��3�3�C�%�c�+H�+H�+H�I�I�I�	J�	J�	J�	J�	J�	J�	J�	J�	J�	J�	J�	J����	J�	J�	J�	J�	J�	Js�AA0�0A4�7A4c	��d\}}}ttj�||z
|���tj�tj||z
��|z����ttj�||z
|���tj�tj||z
��|z����dS)N)g�(\��5i@��g�O��n�?r^)rr6r�rgrErtrUr%)r�r7r��sigmas    r0rzTestLognorm.test_logcdf�s���*�
��B����
�(�(��B��%�(�8�8��
�
�
�b�f�R��U�m�m�E�&9�:�:�	<�	<�	<���
�+�+�B�r�E�U�+�;�;��
�(�(����2����u�)<�=�=�	?�	?�	?�	?�	?r2N)r#r$r%rrr8r2r0r�r��s5������J�J�J�?�?�?�?�?r2r�c��eZdZd�Zd�Zd�Zd�Zej�	e
d���d���Zej�	e
d���d���Zej�	e
d���d	���Z
d
�Zej�ed���d���Zej�d
ejjejjg��ej�dddg��d�����ZdS)�TestBetac���tj�ddd��}t|d��tj�ddd��}t|tj��dS)Nrrrxg�;��B.�)r6r�r�rrtr��r�r�s  r0r�zTestBeta.test_logpdf
s\����"�"�1�a��-�-���F�N�3�3�3���"�"�1�c�1�-�-���F�B�F�+�+�+�+�+r2c�`�d\}}tjgd���}tj||��}t	|�|�����d��t	|�|��tj|�|������dS)N�ii��r�rxr�g�����ƒ�)	rtrr6r�rr�r6r@r��r�r�r�rGr/s     r0�test_logpdf_ticket_1866z TestBeta.test_logpdf_ticket_1866	
s������t��H�_�_�_�%�%���J�u�d�#�#�����������)�)�+=�>�>�>�����a���"�&����!���"5�"5�6�6�6�6�6r2c�`�gd�}tttjj|ddd���dS)N�rlrxr�rrri)r�r�rj)r�r�r6r�r�rms  r0�test_fit_bad_keyword_argsz"TestBeta.test_fit_bad_keyword_args
s=���O�O���i������1�$�	&�	&�	&�	&�	&�	&r2c�^�gd�}tttjj|dd���dS)Nr�rx)�fa�fix_a)r�r�r6r�r�rms  r0�#test_fit_duplicated_fixed_parameterz,TestBeta.test_fit_duplicated_fixed_parameter
s0��
�O�O���j�%�*�.�!��3�G�G�G�G�G�Gr2zOverflow, see gh-14901r�c�p�d\}}}ttj�|||��d��dS)N)g�������?g�R@g0x��Ag�x)��>)rr6r�r])r�r?r.r/s    r0�test_issue_12635zTestBeta.test_issue_12635
s8��7���1�a���
���q�!�Q�/�/�1F�G�G�G�G�Gr2c�>�tjgd���}tjgd���}d}tj�||dzd|z
��}t||��tj�||dzd|z
��}t||��dS)N)g@����3@?g��^?g�N�C�?)rnror�r�rr)rtrr6r�rlrrg)r��inv_R�
count_listr?�invr�s      r0�test_issue_12794zTestBeta.test_issue_12794'
s�����1�1�1�2�2���X�o�o�o�.�.�
����j�n�n�Q�
�Q����0C�D�D����U�#�#�#��j�m�m�C��a���*�1D�E�E����Q�����r2c��d}tjdd��}d}d|z
|dz||z
}}}tj�|||��}tj�|||��}t
|d|z
��dS)N��h㈵��>rr�r)rtr�r6r�r]rKr)	r��alpha_2�count_�nobsr_r.r/r�r�s	         r0�test_issue_12796zTestBeta.test_issue_127969
s�������1�b�!�!�����g�+�v��z�4�&�=�a�1���j�n�n�Q��1�%�%���j�n�n�S�!�Q�'�'����Q��[�)�)�)�)�)r2c��d\}}ttj�d||��tj��d\}}ttj�d||��tj��dS)Nrr)r�r�r)rr6r�r@rtr�)r�r.r/s   r0rMzTestBeta.test_endpointsE
sf�����1��U�Z�^�^�A�q�!�,�,�b�f�5�5�5����1��U�Z�^�^�A�q�!�,�,�b�f�5�5�5�5�5r2zDoes not convert boost warningc��d\}}}tjt��5tj�|||��ddd��dS#1swxYwYdS)N)gףp=
��?g�vH7Bg@�0�B)r��warnsrCr6r�r])r�r_r.r/s    r0�test_boost_eval_issue_14606z$TestBeta.test_boost_eval_issue_14606P
s���'���1�a�
�\�.�
)�
)�	$�	$��J�N�N�1�a��#�#�#�	$�	$�	$�	$�	$�	$�	$�	$�	$�	$�	$�	$����	$�	$�	$�	$�	$�	$s�"A�A�Ar@�a, b)�+�p�h�)@)r�r�c�H�d}	||||��dS#t$rYdSwxYw)Nr�)�
OverflowError)r�r@r.r/r?s     r0� test_beta_ppf_with_subnormal_a_bz)TestBeta.test_beta_ppf_with_subnormal_a_bV
sG��

��
	��F�1�a��O�O�O�O�O���	�	�	�
�D�D�	���s�
�
!�!N)r#r$r%r�r�r�r�r�r�r��MACOS_INTELr�r�r�rM�xfailr
r�r
r6r�r]rlr�r8r2r0r�r�
s�������,�,�,�7�7�7�&�&�&�
H�H�H��[����,D��E�E�	H�	H�F�E�	H��[����,D��E�E� � �F�E� �"�[����,D��E�E�	*�	*�F�E�	*�	6�	6�	6��[���w�'G��H�H�$�$�I�H�$�
�[���X��
���
��'G�H�H��[���V�n�n�%E�F�F���G�F�I�H���r2r�c��eZdZd�Zd�ZdS)�
TestBetaPrimec��d\}}tjgd���}tj||��}t	tj|�|���������t|�	|��tj
|�|������dS)Nr�r�)rtrr6r�r	rr�rrr@r�r�s     r0r�zTestBetaPrime.test_logpdfp
s������t��H�_�_�_�%�%���O�E�4�(�(�����A�H�H�Q�K�K�(�(�,�,�.�.�/�/�/�����a���"�&����!���"5�"5�6�6�6�6�6r2c�����tj�ddd��}t|d��d\��t	jgd���}tj�|����}t
t	j|�������tj	j
����fd�|D��}t||dd�	��dS)
Nrr�r�r}r�r�c�@��g|]}�tj|������Sr8)r6r�)rQrr�r��gen_cdfs  ���r0rSz*TestBetaPrime.test_cdf.<locals>.<listcomp>�
s+���J�J�J��'�'�%�/�3��t�<�<�J�J�Jr2��-����=r�)r6r�rKrrtrr	rrrV�_cdf_singler)r�rGr��cdfs_gr�r�r�s    @@@r0r"zTestBetaPrime.test_cdfw
s������
�O����3��,�,���Q��������t��H�_�_�_�%�%����"�"�1�e�T�2�2�����D�!�!�%�%�'�'�(�(�(��%�1��J�J�J�J�J�J��J�J�J����f�1�5�9�9�9�9�9�9r2N)r#r$r%r�r"r8r2r0r�r�o
s2������7�7�7�:�:�:�:�:r2r�c�&�eZdZd�Zd�Zd�Zd�ZdS)�	TestGammac���tj�ddd���}t|d��tj�ddd���}t|d��dS)	N�Zi�r�rDg
C�T�b?r�rngB����?)r6rr@rr�s  r0rzTestGamma.test_pdf�
sZ���k�o�o�b�#�T�o�2�2���C��-�-�-��k�o�o�a��4�o�0�0���C��+�+�+�+�+r2c�f�tj�dd��}t|d��dSrJ)r6rr�rr�s  r0r�zTestGamma.test_logpdf�
s1����#�#�A�q�)�)���F�A�&�&�&�&�&r2c�^�gd�}tttjj|dd���dS)Nr�rri)r�rj)r�r�r6rr�rms  r0r�z#TestGamma.test_fit_bad_keyword_args�
s.���O�O���i����!�!�8�L�L�L�L�L�Lr2c���tjtj�dd��dd���sJ�tjtj�dd��dd	���sJ�dS)
Nr�r�g�p�l�C@r,r�g��Jz�5rog��1�}�t@r�)rt�iscloser6rrlr�s r0rnzTestGamma.test_isf�
s���"�z�%�+�/�/�%��3�3�+�%�9�9�9�	9�	9�	9��z�%�+�/�/�%��5�5�+�%�9�9�9�	9�	9�	9�	9�	9r2N)r#r$r%rr�r�rnr8r2r0r�r��
sS������,�,�,�'�'�'�M�M�M�9�9�9�9�9r2r�c��eZdZd�Zd�ZdS)�TestChi2c���ttj�dd��dd���ttj�dd��dd���dS)Nr�gޞw�1D�?�rrog�6:��֜?�rr6�chi2r@r�s r0r�zTestChi2.test_precision�
sj���E�J�N�N�4��6�6�8N�$&�	(�	(�	(�	(��E�J�N�N�3��4�4�6J�$&�	(�	(�	(�	(�	(�	(r2c��d}tj�d|��}t|dd���tj�d|��}t|dd���d}tj�d	|��}t|d
d���tj�d|��}t|dd���dS)
Nr|gλm�:=6g'b�d�5<r�r�rxg�Q�8��@�
gܐ�؆0g�uu�[9�=rlg��8-�*@)r6r�r]r)r�rprGs   r0rzTestChi2.test_ppf�
s���
���J�N�N�5�"�%�%����<�5�I�I�I�I��J�N�N�3��#�#����7�e�D�D�D�D�
���J�N�N�5�"�%�%����:��G�G�G�G��J�N�N�3��#�#����8�u�E�E�E�E�E�Er2N)r#r$r%r�rr8r2r0r�r��
s7������(�(�(�F�F�F�F�Fr2r�c�l�eZdZd�Zd�Zd�Zej�dddg��d���Z	dS)	�TestGumbelLc���tjdd��}tj�|��}tj�|��}t
||��dS�Nr�rZ)rtrr6rerKr]rr&s    r0razTestGumbelL.test_cdf_ppf�
sS���K��b�!�!���N���q�!�!��
�^�
�
��
"�
"����2�����r2c��tjdd��}tj�|��}tj�|��}tj|��}tj|��}t||��dSr�)
rtrr6rer~r%r�r�expm1r)r�rGrrb�ur�s      r0r'zTestGumbelL.test_logcdf_logsf�
sr���K��b�!�!���N�!�!�!�$�$���N� � ��#�#���F�1�I�I��
�]�1�
�
�����1�����r2c���tjdd��}tj�|��}tj�|��}t
||��dS)NrYr�)rtrr6rergrlrr&s    r0r	zTestGumbelL.test_sf_isf�
sS���K��Q�����N���a� � ��
�^�
�
��
"�
"����2�����r2r<r�rc��tj�d|���}tj�||���\}}t	||��dS)Nro)rr<r?)r6rer�r�r)r�r<r��
fitted_locr�s     r0�test_fit_fixed_paramz TestGumbelL.test_fit_fixed_param�
sR���~�!�!�s��!�4�4����*�*�4�c�*�:�:�
�
�A��Z��%�%�%�%�%r2N)
r#r$r%rar'r	r�r�r
rr8r2r0r�r��
ss����������������[���U�R��G�,�,�&�&�-�,�&�&�&r2r�c��eZdZd�Zd�ZdS)�TestGumbelRc�d�ttj�d��dd���dS)Nrg?~T}%m;r,r�)rr6rdrgr�s r0rjzTestGumbelR.test_sf�
s=��	���)�)�"�-�-�/E�"�	$�	$�	$�	$�	$�	$r2c�d�ttj�d��dd���dS)Nr�r�r,r�)rr6rdrlr�s r0rnzTestGumbelR.test_isf�
s=��	���*�*�5�1�1�3D�"�	$�	$�	$�	$�	$�	$r2N)r#r$r%rjrnr8r2r0rr�
s2������$�$�$�$�$�$�$�$r2rc��eZdZejd���Zejd���Zejd���Zej�	dej
d��ej
dejj���g��ej�	dd	d
g��ej�	dgd���ej�	d
ddg��d���������Zejjej�	dddg��d�����Z
d�Zej�d���ej�	dddg��ej�	dd	d
g��d�������Zd�Zd�Zd�Zej�	d ej
gd!�gd"�gd#���ej
gd$�gd%�gd&�ejj���ej
gd'�ejd(d)d*��ejd+dd,��ejj���g��d-���Zej�	d ej
gd!�gd"�gd#���ej
gd$�gd%�gd&�ejj���ej
gd'�ejd(d)d*��ejd+dd,��ejj���g��d.���Zej�	d/d0dg��ej�	d1d2d3g��d4�����Zej�	d5d6d7gd8d7gg��d9���Zej�	d:d;d0ejejejfgd<d=gg��d>���Zej�	d?gd@���ej�	dAejj dBejdCd0dD��dEfejj!dBejdCd0dD��dEfejj dFejd0dGdD��dBfejj!dFejd0dGdD��dEfg��dH�����Z"dIS)J�TestLevyStablec��tjtt��jdz��}tjj�|jd���}|S)aSample data points for pdf computed with Nolan's stablec

        See - http://fs2.american.edu/jpnolan/www/stable/stable.html

        There's a known limitation of Nolan's executable for alpha < 0.2.

        The data table loaded below is generated from Nolan's stablec
        with the following parameter space:

            alpha = 0.1, 0.2, ..., 2.0
            beta = -1.0, -0.9, ..., 1.0
            p = 0.01, 0.05, 0.1, 0.25, 0.35, 0.5,
        and the equivalent for the right tail

        Typically inputs for stablec:

            stablec.exe <<
            1 # pdf
            1 # Nolan S equivalent to S0 in scipy
            .25,2,.25 # alpha
            -1,-1,0 # beta
            -10,10,1 # x
            1,0 # gamma, delta
            2 # output file
        z.data/levy_stable/stable-Z1-pdf-sample-data.npy�x,p,alpha,beta,pct��names�	rt�loadr�__file__�parent�core�records�
fromarrays�T�r�r�s  r0�nolan_pdf_sample_dataz$TestLevyStable.nolan_pdf_sample_data�
sO��6�w���N�N�!�<�
=�
�
���w��)�)�$�&�8L�)�M�M���r2c��tjtt��jdz��}tjj�|jd���}|S)a#Sample data points for cdf computed with Nolan's stablec

        See - http://fs2.american.edu/jpnolan/www/stable/stable.html

        There's a known limitation of Nolan's executable for alpha < 0.2.

        The data table loaded below is generated from Nolan's stablec
        with the following parameter space:

            alpha = 0.1, 0.2, ..., 2.0
            beta = -1.0, -0.9, ..., 1.0
            p = 0.01, 0.05, 0.1, 0.25, 0.35, 0.5,

        and the equivalent for the right tail

        Ideally, Nolan's output for CDF values should match the percentile
        from where they have been sampled from. Even more so as we extract
        percentile x positions from stablec too. However, we note at places
        Nolan's stablec will produce absolute errors in order of 1e-5. We
        compare against his calculations here. In future, once we less
        reliant on Nolan's paper we might switch to comparing directly at
        percentiles (those x values being produced from some alternative
        means).

        Typically inputs for stablec:

            stablec.exe <<
            2 # cdf
            1 # Nolan S equivalent to S0 in scipy
            .25,2,.25 # alpha
            -1,-1,0 # beta
            -10,10,1 # x
            1,0 # gamma, delta
            2 # output file
        z.data/levy_stable/stable-Z1-cdf-sample-data.npyr	r
rrs  r0�nolan_cdf_sample_dataz$TestLevyStable.nolan_cdf_sample_data sP��J�w���N�N�!�<�
=�
�
���w��)�)�$�&�8L�)�M�M���r2c�b�tjtt��jdz��}|S)a&Sample data where loc, scale are different from 0, 1

        Data extracted in similar way to pdf/cdf above using
        Nolan's stablec but set to an arbitrary location scale of
        (2, 3) for various important parameters alpha, beta and for
        parameterisations S0 and S1.
        z1data/levy_stable/stable-loc-scale-sample-data.npy)rtr
rrrrs  r0�nolan_loc_scale_sample_dataz*TestLevyStable.nolan_loc_scale_sample_dataLs1���w���N�N�!�?�
@�
�
���r2�sample_sizerr�)�marks�parameterization�S0�S1z
alpha,beta))rr)rr�)r�r)r�rxzgamma,delta�rrrc���|tj_tj||||���}tj|�|d���|j��\}}	|	dksJ�dS)N)r�r�r=r<r�r�r�)r6�levy_stablerr�r�rK)
r�rr�r�rr�r�lsr�r?s
          r0rzTestLevyStable.test_rvs[sp��&.>���*�
�
��d�%�U�
�
�
���|��F�F��$�F�7�7���
�
���1��4�x�x�x�x�x�xr2r�rxrc���tj�d��d}d}d}tj�||||d���}tj|d||||f���\}}|d	ksJ�d
S)z3Additional test cases for rvs for alpha equal to 1.�h�:rrxr�i��r<r=rr!r�rzN)rtr�r�r6r!r�r�)r�r�r�r<r=rG�statr?s        r0�test_rvs_alpha1zTestLevyStable.test_rvs_alpha1ws���	�	���y�!�!�!���������!�!�%��3�e�'+�
"�
-�
-���,�q�-�%*�D�#�u�$=�?�?�?���a��4�x�x�x�x�x�xr2c��gd�}tj�|��\}}}}t|ddd���t	|dd��t	|dd	��t	|d
d��|gd�z}tj�|��\}}}	}
t|d��t|d��t	|
d
d	��t	|	dd	��dS)N)�wg��r)r}r}r}r}羼���u?r*r*r*r*��67�',�?r+r+r+r+�tA}˜.�?r,r,r,r,�G�z��?rrzr0�)\��(̿r>gۧ�1��?reg���9]c?)r,r,r,r,r,r�g���v��?r+)r6r!r8rrr�r�rG�alpha1�beta1�loc1�scale1r7�alpha2�beta2rErFs           r0r=zTestLevyStable.test_fit�s��
�
�
��
',�&7�&A�&A�!�&D�&D�#���t�V����1�4�8�8�8�8��E�4��+�+�+��F�G�Q�/�/�/���'�1�	
�	
�	
�
�9�9�9�
9��&+�&7�&A�&A�"�&E�&E�#���t�V��V�Q�����U�B�����F�F�A�.�.�.��D�&�!�,�,�,�,�,r2zUnknown problem with fitstart.r�zalpha,beta,delta,gamma)r�rjr>r�)rrjr>r��parametrizationc���|tj_tj�||||dd���}tj�|��}|\}}	}
}t||||g||	|
|gd���dS)z7Test that fit agrees with rvs for each parametrization.r�r��r<r=rr�rzr�N)r6r!r6r�r8r)r�r�r�r�rr6r�r��	alpha_obs�beta_obs�	delta_obs�	gamma_obss            r0�test_fit_rvszTestLevyStable.test_fit_rvs�s���-<���)�� �$�$��4�U�%�e�$�%�
�
����)�)�$�/�/��47�1�	�8�Y�	��
�D�%��'�
��)�Y�7��	
�	
�	
�	
�	
�	
r2c�x�tjgd���}tj�|��\}}}}tj�|��\}}}}	t|d��|dksJ�t
||��t
||��t
||��t
|	|��dS)N�rrr�r�rnrnrnr�r�rororr)rtrr6r!r8rr)
r�rGr0r1r2r3r4r5rErFs
          r0�test_fit_beta_flipz!TestLevyStable.test_fit_beta_flip�s����H�?�?�?�@�@��&+�&7�&A�&A�!�&D�&D�#���t�V�&+�&7�&A�&A�1�"�&E�&E�#���t�V��U�A�����q�y�y�y�y��F�F�+�+�+��E�E�6�*�*�*��D�4�%�(�(�(��F�F�+�+�+�+�+r2c�V�d}tjgd���}tj�|��\}}}}tj�||z��\}}}	}
t||��t||��t|	||z��t|
|��dS)Nrr?)rtrr6r!r8r)r��SHIFTrGr0r1r2r3r4r5rErFs           r0�test_fit_delta_shiftz#TestLevyStable.test_fit_delta_shift�s������H�?�?�?�@�@��&+�&7�&A�&A�1�"�&E�&E�#���t�V�&+�&7�&A�&A�1�"�u�*�&M�&M�#���t�V��F�F�+�+�+��E�5�)�)�)��D�$��,�/�/�/��F�F�+�+�+�+�+r2c��gd�}tj�|��\}}}}|dks
Jd|�����|t|��ksJdt|���d|�����gd�}tj�|��\}}}	}
|dks
Jd|�����|	t	|��ksJdt	|���d|	�����dS)	N)rrr�r�rnrnrnr�r��rErzExpected alpha < 1, got zExpected loc < z, got )rrr�r�rnrnrnr�r��rFzExpected alpha > 1, got zExpected loc > )r6r!r8r~rr/s           r0�test_fit_loc_extrapz"TestLevyStable.test_fit_loc_extrap�s���6�6�6��&+�&7�&A�&A�!�&D�&D�#���t�V���z�z�z�>�f�>�>�z�z�z��c�!�f�f�}�}�}�D��A���D�D�d�D�D�}�}�}�
7�
7�
7��&+�&7�&A�&A�"�&E�&E�#���t�V���z�z�z�>�f�>�>�z�z�z��c�"�g�g�~�~�~�F��R���F�F��F�F�~�~�~�~�~r2z pct_range,alpha_range,beta_range)rzrxr�)rlrr>)r�rrb)rzr�rxr
r�)rlrxrr�r>)g�������r�rr�r�r)rzr�rlr5�ffffff�?rx��������?r+r�r
r�rlr>r�r��c�p�����|}tj��}|jdko
|jdk�d�|j|j|jg��}dd���fd�gdd���fd	�gdd
����fd�gdd���fd
�gdd���fd�gdd���fd�gg}t
|��D�]�\}	\}
}}|
tj_	|�|||��n|}
t��5}|�td��tj�
|
d|
d|
ddd���}tjd���5t!|
gd�|tj||
dz
��tj||
dz
��tj|
d��zg��}ddd��n#1swxYwY||d|ktj|��z}t'||
d|d |	�d!|
�d"|�d#|jj�d$|��
d%�&��ddd��n#1swxYwY���dS)'z2Test pdf values against Nolan's stablec.exe output�Linux�i686�/�dnir�c	�r��tj|d���tj|d���ztj|d���z|ddk|ddkz|ddk|ddkz|ddkzz|ddktj|dd	d
g��zz|ddktj|ddd
g��zz|ddktj|dddg��zz|ddktj|dddg��ztjtj|d��gd���zz|ddktj|ddg��ztjtj|d��dg��zz|ddktj|dddg��ztjtj|d��gd���zz|ddk|ddkz|ddkzz|ddk|ddkz|ddkzz|ddk|ddkz|ddkzz|ddktj|ddg��ztjtj|d��gd���zz|ddktj|dddg��ztjtj|d��gd���zz|ddktj|dd	d
g��ztjtj|d��ddg��zz|ddkzzS)N�pctr�r�rrxr�皙�����?rjrzr�r�r�r
r�rlr5r+)rxr��ffffff�?rHrI)g������ٿ�333333ӿr�rjrxrTr�rlr�r�rj)rbr�rr�r�)rt�isinr��r��alpha_range�
beta_range�	pct_ranges ���r0r_z7TestLevyStable.test_pdf_nolan_samples.<locals>.<lambda>s�����G�A�e�H�i�0�0��G�A�g�J��4�4�5��G�A�f�I�z�2�2�3�
�v�Y�!�^��u�X��_�.��v�Y�#�-��w�Z�3�.�0��u�X��_�.���w�Z�3�.��G�A�e�H�s�C�j�9�9�:���w�Z�3�.��G�A�e�H�s�C�j�9�9�:��$�w�Z�3�.��G�A�e�H�r�2�h�7�7�8�%�,�w�Z�3�.��G�A�e�H�s�C�j�9�9�:��G�B�F�1�V�9�$5�$5�|�|�|�D�D�E�-�6�w�Z�3�.��G�A�e�H�r�d�3�3�4��G�B�F�1�V�9�$5�$5��t�<�<�=�7�@�w�Z�3�.��G�A�e�H�s�C�j�9�9�:��G�B�F�1�V�9�$5�$5�7M�7M�7M�N�N�O�A#�J�w�Z�3�.��v�Y�#�-�/��u�X��-�/�K(�T�w�Z�3�.��v�Y�$�.�0��u�X��-�/�U-�^�w�Z�3�.��v�Y�#�-�/��u�X��-�/�_2�h�w�Z�2�-��G�A�e�H�r�d�3�3�4��G�B�F�1�V�9�$5�$5�7G�7G�7G�H�H�I�i7�r�w�Z�2�-��G�A�e�H�s�C�j�9�9�:��G�B�F�1�V�9�$5�$5�|�|�|�D�D�E�s<�|�w�Z�2�-��G�A�e�H�s�C�j�9�9�:��G�B�F�1�V�9�$5�$5��R�y�A�A�B�}A�F�7��s�*�GC,�E�H�r2�	piecewiser�c����tj|d���tj|d���ztj|d���z|ddkz|ddkzS)NrQr�r�r�r�rtrVrWs ���r0r_z7TestLevyStable.test_pdf_nolan_samples.<locals>.<lambda>Osq����G�A�e�H�i�0�0��G�A�g�J��4�4�5��G�A�f�I�z�2�2�3��w�Z�#�%�'��w�Z�2�%�	'�r2g�����|�=c���|ddk�ztj|d���zd�vztj|d���zS)Nr�rrQr�r])r�rXrY�is_linux_32rZs ����r0r_z7TestLevyStable.test_pdf_nolan_samples.<locals>.<lambda>Zs_����w�Z�2�%�$�_�&��G�A�e�H�i�0�0�1��;�&�(��G�A�f�I�z�2�2�	3�r2g��&�.�=c����tj|d���tj|d���ztj|d���z|ddkzS)NrQr�r�r�r]rWs ���r0r_z7TestLevyStable.test_pdf_nolan_samples.<locals>.<lambda>ds^����G�A�e�H�i�0�0��G�A�g�J��4�4�5��G�A�f�I�z�2�2�3��w�Z�3�&�(�r2�fft-simpsonr6c����|ddktj|d���ztj|d���ztj|d���zS)Nr�r�rQr�r]rWs ���r0r_z7TestLevyStable.test_pdf_nolan_samples.<locals>.<lambda>ms^����w�Z�3�&��G�A�e�H�i�0�0�1��G�A�g�J��4�4�5��G�A�f�I�z�2�2�3�r2r�c����tj|d���tj|d���ztj|d���z|ddkz|ddkzS)NrQr�r�rr�r]rWs ���r0r_z7TestLevyStable.test_pdf_nolan_samples.<locals>.<lambda>usp����G�A�e�H�i�0�0��G�A�g�J��4�4�5��G�A�f�I�z�2�2�3��w�Z�!�^�%��w�Z�#�%�	'�r2Nz2Density calculations experimental for FFT method.*rGr�r�rr�r=r<r�r���calc�abserr�relerrr?rhz	pdf test � failed with method 'z
' [platform: z]
�
F��err_msg�verbose)r��uname�systemr��join�	processorr7r6r!�pdf_default_methodr�recordrCr@rtr�rr��isnanrr�r)r�rrZrXrYr�rn�
platform_desc�tests�ix�default_methodr��filter_func�subdata�supr?�subdata2�failuresr_s  ```             @r0�test_pdf_nolan_samplesz%TestLevyStable.test_pdf_nolan_samples�s�������2%���� � ���l�g�-�I�%�-�6�2I�����
�\�5�=�%�/�:�<�<�
��t�J�J�J�J�J�J�L
�^�U�%�%�%�%�%�%�
��U�%�%�%�%�%�%�%�
��W�'�'�'�'�'�'�
��t�&�&�&�&�&�&�
��t�&�&�&�&�&�&�
�kC
��L"+�5�!1�!1�)	�)	�
�B�����3A�E��0� +� 7��;�;�t�,�,���=A�
�"�$�$�$
���
�
�"�H�����%�)�)��C�L��G�$��F�O���*�����[�h�/�/�/�	�	�0��4�4�4���F�1�w�s�|�#3�4�4��F�1�w�s�|�#3�4�4�r�v�g�c�l�7K�7K�K�� � �H�	�	�	�	�	�	�	�	�	�	�	����	�	�	�	�$��H�%��-��(�1�+�+���� ���C�L��D��R�R���������8L�8L�8L� ��"�"�	�	�	�	�7$
�$
�$
�$
�$
�$
�$
�$
�$
�$
�$
����$
�$
�$
�$
��)	�)	s9�A'H)�A(F8�,H)�8F<�<H)�?F<�AH)�)H-	�0H-	c������|}dd���fd�gdd���fd�gdd���fd�gdd	���fd
�gdd���fd�gdd
���fd�gg}t|��D�]�\}\}}	}
|tj_|
�||
|��n|}t	��5}|�td��tj�|d|d|ddd���}
tj	d���5t|gd�|
tj|
|dz
��tj|
|dz
��tj|d��zg��}ddd��n#1swxYwY||d|	ktj|
��z}t|
|d|	d|�d|�d|jj�d|��d �!��ddd��n#1swxYwY���dS)"z4 Test cdf values against Nolan's stablec.exe output.r[r�c���tj|d���tj|d���ztj|d���z|ddktj|dgd���z|ddkz|ddktj|dgd���z|ddkzzzS�	NrQr�r�r)rTr�r�rz�rlr�r�r�r]rWs ���r0r_z7TestLevyStable.test_cdf_nolan_samples.<locals>.<lambda>�s�����G�A�e�H�i�0�0��G�A�g�J��4�4�5��G�A�f�I�z�2�2�3�
�w�Z�2�-��G�A�f�I�/A�/A�/A�B�B�C��u�X��-�/�
�w�Z�2�-��G�A�f�I����?�?�@��u�X��-�/�
	���r2r�c���tj|d���tj|d���ztj|d���z|ddktj|dgd���z|ddkzz|ddktj|dgd���z|ddkzzSr�r]rWs ���r0r_z7TestLevyStable.test_cdf_nolan_samples.<locals>.<lambda>�s�����G�A�e�H�i�0�0��G�A�g�J��4�4�5��G�A�f�I�z�2�2�3��7��r�)����&�	�+=�+=�+=�>�>�?��5��T�)�+�	��7��r�)����&�	�?�?�?�;�;�<��5��T�)�+��r2rar6c����tj|d���tj|d���ztj|d���z|ddkzS)NrQr�r��333333�?r]rWs ���r0r_z7TestLevyStable.test_cdf_nolan_samples.<locals>.<lambda>�s^����G�A�e�H�i�0�0��G�A�g�J��4�4�5��G�A�f�I�z�2�2�3��w�Z�#�%�'�r2r�c����tj|d���tj|d���ztj|d���z|ddkz|ddkzS)NrQr�r�r�r�r]rWs ���r0r_z7TestLevyStable.test_cdf_nolan_samples.<locals>.<lambda>��q����G�A�e�H�i�0�0��G�A�g�J��4�4�5��G�A�f�I�z�2�2�3��w�Z�#�%�'��w�Z�3�&�	(�r2r
c����tj|d���tj|d���ztj|d���z|ddkz|ddkzS)NrQr�r���������?r�r]rWs ���r0r_z7TestLevyStable.test_cdf_nolan_samples.<locals>.<lambda>r�r2rzc����tj|d���tj|d���ztj|d���z|ddkz|ddkzS)NrQr�r�rr�r]rWs ���r0r_z7TestLevyStable.test_cdf_nolan_samples.<locals>.<lambda>
r�r2Nz[Cumulative density calculations experimental for FFT method. Use piecewise method instead.*rGr�r�rrrdr�r�rer?rhz	cdf test riz'
rjFrk)r7r6r!�cdf_default_methodrrsrCrKrtr�rr�rtrr�r)r�rrZrXrYr�rvrwrxr�ryrzr{r?r|r}s  ```           r0�test_cdf_nolan_samplesz%TestLevyStable.test_cdf_nolan_samples�sw�����2%���U�%�%�%�%�%�%�
�,�T�$�$�$�$�$�$�
�&�t�&�&�&�&�&�&�
��t�&�&�&�&�&�&�
��t�&�&�&�&�&�&�
��t�&�&�&�&�&�&�
�GL
��\"+�5�!1�!1�'	�'	�
�B�����3A�E��0� +� 7��;�;�t�,�,���=A�
�"�$�$�"
���
�
�"�@����
�%�)�)��C�L��G�$��F�O���*�����[�h�/�/�/�	�	�0��4�4�4���F�1�w�s�|�#3�4�4��F�1�w�s�|�#3�4�4�r�v�g�c�l�7K�7K�K�� � �H�	�	�	�	�	�	�	�	�	�	�	����	�	�	�	�$��H�%��-��(�1�+�+���� ���C�L��D��R�R������)=�)=�)=�x�x�I�!�
����7"
�"
�"
�"
�"
�"
�"
�"
�"
�"
�"
����"
�"
�"
�"
��'	�'	s9�A'G�2A(E&�G�&E*�*G�-E*�.AG�G	�G	r�rr�r@rKc��tj��}|jdkodtj��dv}|r|dkrt	jd��|}dtj_dtj_	||d|k}d|��tj_
|d	vsJ�|dkrtjjntjj}||d
|d|dd
d���}	t|	||d��dS)zGTests for pdf and cdf where loc, scale are different from 0, 1
        rL�32bitrr@z4Test unstable on some platforms; see gh-17839, 17859r[r��S)r@rKrGr�r�r>r�rdr6N)r�rnro�architecturer��skipr6r!r�rrrr@rKr)
r�rr�r�rnr_r�rzrUrs
          r0�test_location_scalez"TestLevyStable.test_location_scale?s���� � ���l�g�-�W�'�X�=R�=T�=T�UV�=W�2W���	P�4�5�=�=��K�N�O�O�O�*��0;���,�/:���,��t�G�}��-�.��-8��[�[���*��~�%�%�%�%�%)�U�]�]�E��!�!��8I�8M�	��X��C�L�'�'�*�G�F�O�1�!�
�
�
��	��G�D�M�4�0�0�0�0�0r2zmethod,decimal_placesrOrer[c	��tjgd���}tjgd���}tjgd���}tjd���5t��5}|�t
d���|tj_tj�	|d|dd	�
��}t||||��ddd��n#1swxYwYddd��dS#1swxYwYdS)z� sample points extracted from Tables and Graphs of Stable
        Probability Density Functions - Donald R Holt - 1973 - p 187.
        )rrrrrrrrr>r>r>r>r�r�r�r�rererere)gtF��_�?g�ܵ�|��?g���Q��?g��@����?g��e�c]�?g��HP�?g�Pk�w�?g!�rh���?g�T���N�?g������?gDio�ɴ?g��� �r�?g���H�?g+����?gHP�s�?g�X���?g���&�?g�
�0�*�?g�v��/�?ga2U0*��?)rr5rxrrr5rxrrr5rxrrr5rxrrr5rxrr��rzDensity calculation unstable.*)�category�messagerrrdN)rtrr�rrcrCr6r!rrr@r)r�r@�decimal_places�xs�density�betasr{r@s        r0�'test_pdf_alpha_equals_one_beta_non_zeroz6TestLevyStable.test_pdf_alpha_equals_one_beta_non_zeroas����X�H�H�H�
�
���(�
�
�
�
�
����
�
�
�
�
���[�X�
&�
&�
&�
	�
	�(9�(;�(;�
	�s��J�J�'�8�
�
�
�
�4:�E��0��#�'�'��A�u�A�1�'�E�E�C���W�n�f�
�
�
�
	�
	�
	�
	�
	�
	�
	�
	�
	�
	�
	����
	�
	�
	�
	�
	�
	�
	�
	�
	�
	�
	�
	�
	�
	�
	�
	����
	�
	�
	�
	�
	�
	s7�C/�'A$C�C/�C	�C/�C	�C/�/C3�6C3zparams,expected)r-r.rr)r>r�rnr�)rnrmrrc��tj�|d|d|d|dd���}t||��dS)Nrrr>r�rr�)r6r!r)r��paramsri�observeds    r0rzTestLevyStable.test_stats�sU���$�*�*��1�I�v�a�y�f�Q�i�v�a�y��+�
�
��	�H�h�/�/�/�/�/r2r�)r5rxr+zfunction,beta,points,expectedri��rnr}rrr�c��d|cxkrdksnJ�t||||���tjt|��|����dS)a\Ensure the pdf/cdf routines do not return nan outside support.

        This distribution's support becomes truncated in a few special cases:
            support is [mu, infty) if alpha < 1 and beta = 1
            support is (-infty, mu] if alpha < 1 and beta = -1
        Otherwise, the support is all reals. Here, mu is zero by default.
        rr)r�r�N)rrtrr�)r�r�rUr�r>ris      r0�!test_distribution_outside_supportz0TestLevyStable.test_distribution_outside_support�se��P�5�}�}�}�}�1�}�}�}�}�}�}���H�V�5�t�4�4�4��G�C��K�K��*�*�	
�	
�	
�	
�	
r2N)#r#r$r%r�rcrrrr�r
r�r�rr'r=r�r=r@rCrGrtr�xslowr~r�r�r�r��NaNrr6r!rKr@r�r8r2r0rr�
s�������^����^��B�^�)�)��^�)�V�^����^���[�����F�L����l�f�l�4�v�{�7G�H�H�H�
���
�[���/�$���>�>��[����C�C�C����[���]�V�V�,<�=�=���>�=���?�>����$�[���[���V�c�1�X�.�.�
�
�/�.���
�-�-�-�0�[���>��?�?��[��� ���	
����[����D�$�<���
�
�����@�?�
�
,�
,�
,�	,�	,�	,�
G�
G�
G��[���*��F�L�����
�
����
�
�

�F�L�(�(�(�#�#�#�(�(�(��k�&�	
�
�
�
�F�L�D�D�D����C��B�'�'����B��2�&�&��k�'�	
�
�
�-
���*��+��*�B�[���*��F�L�����
�
����
�
�

�F�L�(�(�(�#�#�#�(�(�(��k�&�	
�
�
�
�F�L�D�D�D����C��B�'�'����B��2�&�&��k�'�	
�
�
�-
���*y�y�+��*y�v�[���W�q�!�f�-�-��[���V�e�U�^�4�4�1�1�5�4�.�-�1�@�[����
�A�J�
�!��	
��������@�[����
�!�R�V�R�V�R�V�!<�=�
��/�	
���0�0���0��[���W�&7�&7�&7�8�8��[���'��!�%�����C��B�'�'��	
��!�%�����C��B�'�'��	
��!�%�����A�r�2�&�&��	
��!�%�����A�r�2�&�&��	
�'	
���:
�
�;��9�8�<
�
�
r2rc��eZdZd�Zd�ZdS)�TestArrayArgumentc�D�tj�d��dSr�r�r�s r0r�zTestArrayArgument.setup_method�r�r2c��tj�tjd��tjd��d���}t
|jd��dS)Nr�r�r%)r6rEr�rtr�r1rr�r�r�s  r0�test_noexceptionz"TestArrayArgument.test_noexception�sK���j�n�n�"�)�A�,�,�r�w�q�z�z�")��+�+���S�Y��(�(�(�(�(r2N)r#r$r%r�r�r8r2r0r�r��s2���������)�)�)�)�)r2r�c��eZdZd�Zd�ZdS)�
TestDocstringc��tjj�2tdtjj���v��tjj�4tdtjj���v��dSdS)Nr�rR)r6r�rrr	�lowerrRr�s r0�test_docstringszTestDocstring.test_docstrings�sn���>�!�-��J�%�.�"8�">�">�"@�"@�@�A�A�A��?�"�.��K�5�?�#:�#@�#@�#B�#B�B�C�C�C�C�C�/�.r2c�R�tj��tj��dSrr)r6rVrUr�s r0�test_no_name_argzTestDocstring.test_no_name_arg�s&��
�����
������r2N)r#r$r%r�r�r8r2r0r�r��s5������D�D�D�����r2r�c��tgd���}t|dk|d��\}}t|gd���t|gd���tddk|d��\}}t||d��t|dg��t|dk|d��\}}t||��t|dgtj|��z��dS)N)rr�r>rr>r�r�rr>)r�r>r>r�r�)r>r>r>r>r>r)rrrrAr�r.r/r^s   r0�TestArgsreducer��s���
�#�#�#�$�$�A��a�!�e�Q��"�"�D�A�q��q�/�/�/�*�*�*��q�/�/�/�*�*�*��a�!�e�Q��"�"�D�A�q��q�!�A�$�����q�1�#�����a�!�e�Q��"�"�D�A�q��q�!�����q�1�#��
�1�
�
�-�.�.�.�.�.r2c�R�eZdZgd�Zd�Zgd�Zej�de	��d���Z
d�Zd�Zd�Z
d	�Zd
�Zd�Zd�Zejjej�d
ddg��d�����Zej�d
ddg��d���ZdS)�
TestFitMethod)�ncfrrc�D�tj�d��dSr�r�r�s r0r�zTestFitMethod.setup_method�r�r2)rRrEr�z	dist,argsc�t�||jvrtjd|z��tjdddddtjg��}tjdddddtjg��}tt|��}tt|j|d���tt|j|d���d	S)
rnz"%s fit known to fail or deprecatedrZr[r\r]r^rr@N)�fitSkipNonFiniter�r�rtrr�r�r�r6r�r�r�)r�rRr�rGr�distfuncs      r0�!test_fit_w_non_finite_data_valuesz/TestFitMethod.test_fit_w_non_finite_data_values�s����4�(�(�(��K�<�t�C�D�D�D��H�f�f�f�f�f�b�f�E�F�F���H�f�f�f�f�f�b�f�E�F�F���5�$�'�'���j�(�,��!�<�<�<�<��j�(�,��!�<�<�<�<�<�<r2c
��tj�d��tjd���5tj�dddd���}tjtj|��tjd��z
d	z�	����}ttjtj�|d
d�����|d
dgd�
��ddd��dS#1swxYwYdS)Ni90r�r�r5r}�4@r�r�r>rr�rQr�)
rtr�r�r�r6r�r�r�rUrDrrr�)r�rG�expected_shapes   r0�test_fix_fit_2args_lognormz(TestFitMethod.test_fix_fit_2args_lognorms>��
�	���u����
�[�X�
&�
&�
&�	@�	@��
�!�!�$��D�r�!�:�:�A��W�r�v�a�y�y�2�6�"�:�:�'=��&A�%G�%G�%I�%I�J�J�N��B�H�U�]�%6�%6�q�q��%6�%L�%L�M�M�+�Q��3�$�
@�
@�
@�
@�	@�	@�	@�	@�	@�	@�	@�	@�	@�	@�	@�	@����	@�	@�	@�	@�	@�	@s�B;C=�=D�Dc��tjdd��}tj�|��\}}t|d��t|tjd����tj�|d���\}}t|d��t|tjd����tj�|d���\}}t|d��t|d��dS)Nrrr�r>r?r@)rtr�r6rEr�rr�r�r�rGr<r=s    r0�test_fix_fit_normzTestFitMethod.test_fix_fit_norm
s����I�a��O�O���Z�^�^�A�&�&�
��U��C��#�#�#��E�2�7�1�:�:�.�.�.��Z�^�^�A�A�^�.�.�
��U��S�!�����U�B�G�A�J�J�'�'�'��Z�^�^�A�a�^�0�0�
��U��C��#�#�#��U�A�����r2c���tjdd��}tj|�����}d}tj�||���\}}}tj|�����|z
}ttj|��tj	|��z
|d���t||��t||���|zd���d}d}tj�|||���\}}}t||��t||��t||���|zd���d	}d}tj�|||���\}}}t||��t||��t||���|zd���d}d	}	tj�|||	�
��\}}}t||��t||	��|tj|	��z
}
ttj	|��|
��dS)Nrrrr?r�rr��r�r�r>r�)rtr�rUrDr6rr�rr�digammar)r�rG�meanlogr�r.r<r=rFr�r�r^s           r0�test_fix_fit_gammaz TestFitMethod.test_fix_fit_gammas���I�a��O�O���&��)�)�.�.�"�"�����������5�5�
��3���F�1�6�6�8�8���w�&���B�F�1�I�I����(:�(:�:�A�q�I�I�I�I��S�$�����E�1�6�6�8�8�A�:�q�9�9�9�9�
���������b�t��<�<�
��3���Q������S�$�����E�1�6�6�8�8�A�:�q�9�9�9�9�
���������b�t��<�<�
��3���Q������S�$�����E�1�6�6�8�8�A�:�q�9�9�9�9�����������V��D�D�
��3���S�$�����U�F�#�#�#��b�f�V�n�n�$���G�O�A�.�.��2�2�2�2�2r2c	��d�}tjgd���}tj�|dd���\}}}}t|d��t|d��t
||||��ddgd���tjgd���}tj�|ddd�	��\}}}}t|d��t|d��t|d��||||��\}}t
|dd
���d|z
}	tj�|	ddd���\}
}}}
t|d��t|d��t|
d��||
||	��\}}t
|dd
���t|
|��tttjj|dd���tjgd
���}tttjj|dd���tttjj|ddd���tttjj|ddd���tttjj|dddd���dS)Nc�h�t|��}tj|�����}tjd|z
�����}t	j||z��}|||t	j|��zzz
|||t	j|��zzz
g}|Sr�)r�rtrUr6r�psi)r.r/rGr>�s1r��psiabr�s        r0�mlefuncz0TestFitMethod.test_fix_fit_beta.<locals>.mlefuncBs����A���A����������B����!������"�"�B��K��A��&�&�E���u�f�w�{�1�~�~�5�6�6���u�f�w�{�1�~�~�5�6�6�8�D��Kr2)rr5rxrrr�r�r�r>)r�r�r�r6)r�r�r�rxr�)r�r�r�)r�r�r�r�)r�r�r�r�)
rtrr6r�r�rrrr�r�)r�r�rGr.r/r<r=�da�dbr7rD�b2rErFrs               r0�test_fix_fit_betazTestFitMethod.test_fix_fit_beta?s���		�		�		�
�H�'�'�'�(�(�� �:�>�>�!�!�A�>�>�>���1�c�5��S�!�����U�A��������1�a�(�(�1�a�&�t�<�<�<�<�
�H�'�'�'�(�(�� �:�>�>�!���!�>�D�D���1�c�5��Q������S�!�����U�A�������A�q�!�!���B���A�D�)�)�)�)���U��$�z�~�~�b�Q�Q�q�~�I�I���B��f��R������T�1�����V�Q�������R��$�$���B���A�D�)�)�)�)��B��"�"�"�	�j�%�*�.�!�#�a�H�H�H�H��H�Z�Z�Z� � ���j�%�*�.�!�!�A�F�F�F�F��j�%�*�.�!�!�A�!�L�L�L�L��j�%�*�.�!�!�A�!�L�L�L�L�	�j�%�*�.�!��a��Q�	(�	(�	(�	(�	(�	(r2c���tjgd���}tj�|��\}}t|d��t|d��tj�|d���\}}t|d��t|d��tj�|d���\}}t|d��t|d��dS)N)r>r>rererererer�r>r�r@rr?re)rtrr6rRr�rr�s    r0�test_expon_fitzTestFitMethod.test_expon_fitvs����H�-�-�-�.�.���[�_�_�Q�'�'�
��U��S�!�����U�A�����[�_�_�Q�q�_�1�1�
��U��S�!�����U�A�����[�_�_�Q�Q�_�/�/�
��U��S�!�����U�A�����r2c	��tjgd���}tj|dz
��}tj�|d���\}}}t
||���d���t|d��t
|tj	|�
����d���tj�|dd���\}}}t
|tj|tjd��z
dz�
����d���t|d��t|d��tj�|dd	�
��\}}}t|d	��t|d��t
|tj	|�
����d���dS)N)r�r�rnr��;rr?rqr�rr�r>r+)r��fix_s)rtrrUr6r�r�rrErr�rDr�)r�rG�lnxm1rr<r=s      r0�test_lognorm_fitzTestFitMethod.test_lognorm_fit�s����H�-�-�-�.�.����q�1�u�
�
��!�M�-�-�a�a�-�8�8���s�E���u�y�y�{�{��7�7�7�7��S�!������r�v�e�j�j�l�l�3�3�%�@�@�@�@�!�M�-�-�a�a��-�B�B���s�E���r�w�������):�Q�(>�'D�'D�'F�'F�G�G�"�	$�	$�	$�	$��S�!�����U�A����!�M�-�-�a�a�t�-�D�D���s�E��U�D�!�!�!��S�!������r�v�e�j�j�l�l�3�3�%�@�@�@�@�@�@r2c���tjgd���}tj�|��\}}t||�����t||�����tj�|d���\}}t|d��t||�����tj�|d���\}}t|d��t|d��tttjj|d���tttjj|d���dS)N)rr�r�"@rr?rnr@r|r)rtrr6r�r�rr~r�rr�r�r�s    r0�test_uniform_fitzTestFitMethod.test_uniform_fit�s#���H�)�)�)�*�*���]�&�&�q�)�)�
��U��S�!�%�%�'�'�"�"�"��U�A�E�E�G�G�$�$�$��]�&�&�q�q�&�1�1�
��U��S�!�����U�A�E�E�G�G�$�$�$��]�&�&�q��&�4�4�
��U��S�!�����U�B�����j�%�-�"3�Q�S�A�A�A�A��j�%�-�"3�Q�s�C�C�C�C�C�Cr2r@�MLE�MMc
�`�d\}}tj�||dd���}tj�|d|���}tj�|d|���}t	||dd�	��tj�|d|�
��}t	||dd�	��tj�|d|���}tj�|d|�
��}t	||dd�	��tj�|d|���}t	||dd�	��tttjj|dd|���tttjj|dddd|���tj�|ddd|���}	|	\}
}}}
t|
||
ggd���d}tj�|d���}tj�|||���\}
}}
t|
|��dS)N)rr
ror�r�r)r�r@)r�r@rqr�)r�r@r
)r�r@)ryr@)rxr@rr>)r�r�r@rr�)r�r�r�r�r@)r�r�r�r@)rrrr�)	r6r�r�r�rr�r�rr)r�r@r.r/rG�res_1�res_2�res_3�res_4�res_5�aa�bbry�ssr�s               r0�test_fshapeszTestFitMethod.test_fshapes�s��
���1��J�N�N�1�a�c��N�=�=���
���q�R���7�7���
���q�R���7�7����u�5�u�=�=�=�=��
���q��6��:�:����u�5�u�=�=�=�=��
���q�R���7�7���
���q�R���7�7����u�5�u�=�=�=�=��
���q��6��:�:����u�5�u�=�=�=�=�	�j�%�*�.�!��a��O�O�O�O�	�j�%�*�.�!��a��Q�v�	7�	7�	7�	7�
�
���q�R�a��&��I�I�����B��B��b�"�b�\�:�:�:�.�.�.�
���{���q�s��+�+���[�_�_�T�a��_�?�?�
��B���R������r2c��tj}|�dd���}td���}t	t
|j|fi|�d|i��dS)Nr>ro)rri����)�enikibenikir@)r6r~r�r�r�r�r�)r�r@rRr�r�s     r0�test_extra_paramszTestFitMethod.test_extra_params�s]������x�x�!�#�x�&�&���t�$�$�$���i���4�F�F�3�F�F�v�F�F�F�F�F�Fr2N)r#r$r%r�r�r�r�r�r
rr�r�r�r�r�r�r�r�r�r�r�r8r2r0r�r��sV������$�$�$�D����4�3�3���[���[�(�3�3�=�=�4�3�=�@�@�@�
�
�
�$3�$3�$3�L5(�5(�5(�n
�
�
�A�A�A�(D�D�D�$�[���[���X��t�}�5�5�$�$�6�5���$�L�[���X��t�}�5�5�G�G�6�5�G�G�Gr2r�c�D�eZdZd�Zd�Zd�Zd�Zd�Zd�Zd�Z	d�Z
d	�Zd
S)�
TestFrozenc�D�tj�d��dSr�r�r�s r0r�zTestFrozen.setup_method�r�r2c���tj}tjdd���}|�d��}|�ddd���}t||��|�d��}|�ddd���}t||��|�d��}|�ddd���}t||��|�d��}|�ddd���}t||��|�d��}|�ddd���}t||��|���}|�dd���}t||��|�	��}|�	dd���}t||��|�
��}|�
dd���}t||��|���}|�dd���}t||��|���}|�dd���}t||��|�
d��}|�
ddd���}t||��t|j|j��t|j|j��dS)Nrlrr;r�r5r>)r6rEr@rrKr]rlrgrrDrrErXr�r.r/)r�rR�frozen�result_fr�s     r0rFzTestFrozen.test_norm�s����z�����C�0�0�0���:�:�d�#�#�����$�D���4�4���X�v�&�&�&��:�:�d�#�#�����$�D���4�4���X�v�&�&�&��:�:�d�#�#�����$�D���4�4���X�v�&�&�&��:�:�d�#�#�����$�D���4�4���X�v�&�&�&��9�9�T�?�?������4�s��3�3���X�v�&�&�&��=�=�?�?������S��1�1���X�v�&�&�&��;�;�=�=�����t�3��/�/���X�v�&�&�&��:�:�<�<�����d�#��.�.���X�v�&�&�&��:�:�<�<�����d�#��.�.���X�v�&�&�&��>�>�#�#�����$�c��2�2���X�v�&�&�&��=�=��#�#�����Q�D���4�4���X�v�&�&�&��V�X�t�v�&�&�&��V�X�t�v�&�&�&�&�&r2c���d}tj}tj|��}|�d��}|�d|��}t||��|�d��}|�d|��}t||��|�d��}|�d|��}t||��|�d��}|�d|��}t||��|�d��}|�d|��}t||��|���}|�|��}t||��|�	��}|�	|��}t||��|�
��}|�
|��}t||��|���}|�|��}t||��|���}|�|��}t||��|�
d��}|�
d|��}t||��t|j|jj��t|j|jj��dS)Nr|r�r5rlr>)r6rr@rrKr]rlrgrrDrrErXr�r.rRr/)r�r.rRr�r�r�s      r0�
test_gammazTestFrozen.test_gammasa�����{����Q�����:�:�d�#�#�����$��"�"���X�v�&�&�&��:�:�d�#�#�����$��"�"���X�v�&�&�&��:�:�d�#�#�����$��"�"���X�v�&�&�&��:�:�d�#�#�����$��"�"���X�v�&�&�&��9�9�T�?�?������q�!�!���X�v�&�&�&��=�=�?�?�����Q�����X�v�&�&�&��;�;�=�=�����1�����X�v�&�&�&��:�:�<�<�����!�����X�v�&�&�&��:�:�<�<�����!�����X�v�&�&�&��>�>�#�#�����a�����X�v�&�&�&��=�=��#�#�����Q��"�"���X�v�&�&�&��V�X�v�{�}�-�-�-��V�X�v�{�}�-�-�-�-�-r2c���tjd��}|�d��}|�d���|�d��}t||��dS)Nrr>rr)r6r�r�r)r�r��m1rNs    r0�test_regression_ticket_1293z&TestFrozen.test_regression_ticket_1293Lsa����q�!�!��
�]�]�1�
�
�����V��$�$�$��]�]�1�
�
��	�R������r2c�l�d}tj|���}|j�|��\}}t	||gddg��d}tj�d|���t	|j�|��dtjg��d}tj|���}|j�|��\}}t	||gddg��d}tj�d|��t	|jj|jj	ftj�|����tjd���}t|j|ju��dD]�}t
j|��}tj|���}|j|j	}}t	|d��tt
j|����t
jd��}tj�|��\}}t||gdd	g����dS)
Nr�r�r}rlrlrr�r�rx)r6r�rRr�rr@rtr�r.r/r	r�r�r)r�r^r�r.r/�rv1s      r0r�zTestFrozen.test_ab[s���
��
�_�q�
!�
!�
!���w�#�#�A�&�&���1��a��V�b�#�Y�'�'�'���
����A���#�#�#��R�W�)�)�!�,�,�q�"�&�k�:�:�:���
�_�q�
!�
!�
!���w�#�#�A�&�&���1��a��V�b�#�Y�'�'�'���
����A�q�!�!�!��b�g�i����+�U�_�-I�-I�!�-L�-L�M�M�M��o��$�$�$�������'�(�(�(��
	/�
	/�A��
�1�
�
�A���1�%�%�%�B��4���q�A���B�����B�K��N�N�#�#�#��
�3���A��?�/�/��2�2�D�A�q��Q��F�R��I�.�.�.�.�
	/�
	/r2c�T�tttjd����dS)N�	rv_frozen)r	r-r6r7r�s r0�test_rv_frozen_in_namespacez&TestFrozen.test_rv_frozen_in_namespace�s#�����+�[�9�9�:�:�:�:�:r2c��tj��}tt|d����d|_t|j���tj�	d�������tj�	d��}|�
d|���dS)Nr��*r�r�r�)r6rEr	r-r�r�	get_statertr��RandomStater�)r�r��rndms   r0�test_random_statezTestFrozen.test_random_state�s������������/�/�0�0�0�!����V�(�2�2�4�4��Y�*�*�2�.�.�8�8�:�:�	<�	<�	<��y�$�$�T�*�*���
�
���
�-�-�-�-�-r2c��tjdd��}tjd��}tjgd�gd�f���}|||fD�]}d|_|�d�	��t
j|��}|�d�	��}t
j|��}|�d�	��}t||��|�
d
��|�
d
��g}	t|	d|	d��t|�|	d��|�|	d������dS)
Ng��ds�z@g@��e�?r)rrr>r�rUr�r�r�r�rxrr)r6r�rrUr�r��pickle�dumps�loadsrr]rK)
r�r��poissr��distfnrF�r0�	unpickled�r1�medianss
          r0�
test_picklingzTestFrozen.test_pickling�s[���z�,�.A�B�B���
�b�!�!���"�<�<�<�+?�+?�+?�+A�B�B�B���U�F�+�	4�	4�F�"&�F���J�J�A�J������V�$�$�A������#�#�B���Q���I����A��&�&�B���R� � � ��z�z�#���	�
�
�c�(:�(:�;�G�����W�Q�Z�0�0�0�����G�A�J�/�/�"���w�q�z�2�2�
4�
4�
4�
4�	4�	4r2c
���d�}tjddd���}tjdd���5|�|ddd	�
��}tj�|dddddd	���}ddd��n#1swxYwYt||��tjdd�
��}|�|��}tj�|dd���}t||��dS)Nc��|Srrr8r�s r0r�z$TestFrozen.test_expect.<locals>.func�s���Hr2r>r�re)r.r<r=r�)r��dividerT)r��ub�conditional�r>�r�r<r=r�rr�r<)r��r�r<)r6rrtr�r�rr)r�r��gm�gm_val�	gamma_valr?�p_val�poisson_vals        r0�test_expectzTestFrozen.test_expect�sQ��	�	�	��[�1�!�1�
-�
-�
-��
�[��(�
;�
;�
;�	I�	I��Y�Y�t��a�T�Y�B�B�F���*�*�4�d���./�A�4�+�I�I�I�	I�	I�	I�	I�	I�	I�	I�	I�	I�	I�	I����	I�	I�	I�	I�	��	�*�*�*��M�!��#�#�#����������m�*�*�4�d��*�B�B����{�+�+�+�+�+s�AA=�=B�BN)r#r$r%r�rFr�r�r�r�rrrr8r2r0r�r��s����������1'�1'�1'�f2.�2.�2.�h
�
�
�*/�*/�*/�X;�;�;�.�.�.�4�4�4�0,�,�,�,�,r2r�c��eZdZd�Zd�Zd�Zd�Zd�Zd�Zd�Z	e
j�d��d	���Z
d
�Zd�Zd�Zd
�Zd�Zd�ZdS)�
TestExpectc�@�tj�d�dd���}t|dd���tj�d�dd���}t|dd���tj�d	dd���}tj�d
dd���}tj�d�dd||���}t|d
d���tj�d�dd||d���}t|dd���dS)Nc��|dz
|dz
zS)Nr�r8r�s r0r_z&TestExpect.test_norm.<locals>.<lambda>�s���1��q��s��r2r�r>r;rer�rc��|Srrr8r�s r0r_z&TestExpect.test_norm.<locals>.<lambda>�s���r2r�r
c��dSr�r8r�s r0r_z&TestExpect.test_norm.<locals>.<lambda>����Q�r2�r<r=r�rr�c��dSr�r8r�s r0r_z&TestExpect.test_norm.<locals>.<lambda>����a�r2T�r<r=r�rrr)r6rEr�rr])r�r�r�r�r�prob90�prob90cs       r0rFzTestExpect.test_norm�s���J���3�3��!��D�D���A�q�"�-�-�-�-��J���m�m��!��<�<���A�q�"�-�-�-�-�
�Z�^�^�D�a�q�^�
1�
1��
�Z�^�^�D�a�q�^�
1�
1����"�"�;�;�A�Q�2�"�"�M�M���F�C��4�4�4�4��*�#�#�K�K�Q�a�B�2�04�$�6�6���G�R��4�4�4�4�4�4r2c	�R�tj�d�ddd���}t|dd���tj�d	�dd
d���}t|dd���tj�d
dddd���}tj�ddddd���}tj�d�dd
d||d���}t|dd���tj�d�ddd||d���}t|dd���dS)Nc��|dz
|dz
zS)N�UUUUUU@r8r�s r0r_z&TestExpect.test_beta.<locals>.<lambda>�s���5��1�U�7�(;�r2r�r�r>)r�r<r=��q�q�?r�rc��|Srrr8r�s r0r_z&TestExpect.test_beta.<locals>.<lambda>�s���r2rr|r*r
rnr;r�c��dS)Nrr8r�s r0r_z&TestExpect.test_beta.<locals>.<lambda>�s��R�r2)rnrnFrr�c��dSr�r8r�s r0r_z&TestExpect.test_beta.<locals>.<lambda>�r$r2Tr)r6r�r�rr])r�r�r�rr�r&r's       r0�	test_betazTestExpect.test_beta�s=���J���;�;�'�"#�1�
�
.�
.���A�v�r�2�2�2�2��J���k�k��R�r��J�J���A�u�b�1�1�1�1�
�Z�^�^�D�"�b�a�q�^�
9�
9��
�Z�^�^�D�"�b�a�q�^�
9�
9����"�"�<�<�h�B�)+��r�u�#�N�N���F�C��4�4�4�4��*�#�#�K�K�h�A�*+��r�t�$�M�M���G�R��4�4�4�4�4�4r2c�T�tj�dddd���\}}tj�d�dd���}t||d	�
��tj�d�dd���}t||d�
��tj�d
�dddd	���}t||d�
��dtj�dd	gdddd������z
}tj�d�dddd���}t||d	�
��tj�d�ddddd���}t|dd�
��tj�d�ddd���}	t|	dd	�
��dS)Nr�rnr�rrc��|Srrr8r�s r0r_z+TestExpect.test_hypergeom.<locals>.<lambda>�r!r2)r�rnr�rr�rc��|dz
dzS�Nr�r>r8r�s r0r_z+TestExpect.test_hypergeom.<locals>.<lambda>�s��a��d�Q�Y�r2r�c��|dz
dzSr3r8r�s r0r_z+TestExpect.test_hypergeom.<locals>.<lambda>�s��Q�r�T�A�I�r2r�)r�r<r�rrc��dSr�r8r�s r0r_z+TestExpect.test_hypergeom.<locals>.<lambda>s��q�r2rrc��dSr�r8r�s r0r_z+TestExpect.test_hypergeom.<locals>.<lambda>s��1�r2T)r�r<r�rrc��dSr�r8r�s r0r_z+TestExpect.test_hypergeom.<locals>.<lambda>
s��!�r2r)r�r�r)r6r�r�rrr6)
r��m_true�v_truer�r��v_bounds�	prob_true�prob_bounds�prob_bc�prob_bs
          r0�test_hypergeomzTestExpect.test_hypergeom�s�����.�.�r�2�q�b�.�A�A�����O�"�"�;�;�[�b�"�I�I���A�v�r�2�2�2�2��O�"�"�#6�#6�[�')�
#�
+�
+���A�v�r�2�2�2�2��?�)�)�*=�*=�/:�.0�Q�2�*�?�?��	�H�f�b�9�9�9�9��e�o�)�)�1�b�'�2�r�1�!�)�D�D�H�H�J�J�J�	��o�,�,�[�[�{�13��b�-�B�B���K��B�?�?�?�?��/�(�(���;�B�,-�"�$�)�H�H���G�Q��3�3�3�3���'�'���+�+,��(�4�4���F�A�r�2�2�2�2�2�2r2c�$�tj�d�ddd���}dtj�dd��z
}t	||d�	��tj�d
�ddd���}t	|dd�	��dS)Nc��dSr�r8r�s r0r_z)TestExpect.test_poisson.<locals>.<lambda>s��Q�r2rr�F)r�r�rrr>r�rc��dSr�r8r�s r0r_z)TestExpect.test_poisson.<locals>.<lambda>s���r2T)r6rr�rKr)r�r<�prob_b_true�prob_lbs    r0�test_poissonzTestExpect.test_poissons����m�*�*�;�;�T�a�7<�+�>�>����
�)�)�!�Q�/�/�/���K��b�A�A�A�A��-�&�&�{�{��!�37�'�9�9���G�Q��3�3�3�3�3�3r2c���tj}|�d���}|�d���|�d���}t||d���dS)N)r�r�)rxr�r)r6�genhalflogisticr�r)r��halflog�res1r�s    r0�test_genhalflogisticzTestExpect.test_genhalflogisticsa���'���~�~�6�~�*�*�����F��#�#�#��~�~�6�~�*�*���D�$��3�3�3�3�3�3r2c�
�ttjtj�dd������ttjtj�d�d�������ttjtj�d�d�������ttjtj�d�d�������dS)Ni��G�z��?c��dSr�r8r�s r0r_z/TestExpect.test_rice_overflow.<locals>.<lambda>(����r2)rLr�c��dSrr8r�s r0r_z/TestExpect.test_rice_overflow.<locals>.<lambda>)rNr2c��dS�Nr�r8r�s r0r_z/TestExpect.test_rice_overflow.<locals>.<lambda>*rNr2)r	rtrr6�ricer@r�r�s r0�test_rice_overflowzTestExpect.test_rice_overflow#s���	���E�J�N�N�3��5�5�6�6�7�7�7����E�J�-�-�k�k��-�H�H�I�I�J�J�J����E�J�-�-�k�k��-�H�H�I�I�J�J�J����E�J�-�-�k�k��-�H�H�I�I�J�J�J�J�Jr2c�(�d\}}tj�d�|f���}t|||dz
zt	jd|z
��zd���tj�d�|f|���}t|||zd���dS)	N)r�r�c��|Srrr8�rDs r0r_z(TestExpect.test_logser.<locals>.<lambda>/r$r2r�rr�r�c��|Srrr8rVs r0r_z(TestExpect.test_logser.<locals>.<lambda>5r$r2r)r6rLr�rrtrU)r�r?r<�res_0�res_ls     r0�test_logserzTestExpect.test_logser,s������3���#�#�K�K�q�d�#�;�;�����Q��V��r�v�b�1�f�~�~�5�E�	C�	C�	C�	C���#�#�K�K�q�d��#�D�D����u�s�{��7�7�7�7�7�7r2r�c��d\}}tj�d�||f���}tj�d�||f���}t|||z
d���t||dzz
||zd���dS)N)r�r�c��|Srrr8r�s r0r_z)TestExpect.test_skellam.<locals>.<lambda>=���A�r2r�c��|dzSrr8r�s r0r_z)TestExpect.test_skellam.<locals>.<lambda>>rr2rqr�r>)r6r�r�r)r�r��p2r�rNs     r0�test_skellamzTestExpect.test_skellam8s������B�
�]�
!�
!�+�+�R��H�
!�
=�
=��
�]�
!�
!�.�.��B�x�
!�
@�
@����B��G�%�0�0�0�0���R��U�
�B��G�%�8�8�8�8�8�8r2c
���d\}}tj�d�||f��}t|t	d�t||��D����||z
zd���dS)N)r�qc��|Srrr8r�s r0r_z)TestExpect.test_randint.<locals>.<lambda>Fr!r2c3�K�|]}|V��dSrrr8�rQr�s  r0rz*TestExpect.test_randint.<locals>.<genexpr>Hs"����5�5�!�A�5�5�5�5�5�5r2r�r�)r6rr�rr6r�)r��lo�hir�s    r0�test_randintzTestExpect.test_randintBs������B��m�"�"�;�;��R��9�9�����5�5�u�R��}�}�5�5�5�5�5��b��A��	O�	O�	O�	O�	O�	Or2c�T�tttjjd�d��dS)Nc��|dzSrr8r�s r0r_z&TestExpect.test_zipf.<locals>.<lambda>Ms
��q�!�t�r2r)r
rCr6r?r�r�s r0�	test_zipfzTestExpect.test_zipfJs.���^�U�Z�%6�#�^�T�	+�	+�	+�	+�	+r2c��tj�d�d���}tj�d�dddd���}t||d	�
��dS)Nc��dSr�r8r�s r0r_z/TestExpect.test_discrete_kwds.<locals>.<lambda>Qr]r2rr�c��dSr�r8r�s r0r_z/TestExpect.test_discrete_kwds.<locals>.<lambda>Rr]r2r�� rQ)r��maxcount�	chunksize�	tolerancer�r)r6rr�r)r��n0�n1s   r0�test_discrete_kwdszTestExpect.test_discrete_kwdsOsd��
�]�
!�
!�+�+�D�
!�
9�
9��
�]�
!�
!�+�+�D�+/�2��"�O�O���B��B�/�/�/�/�/�/r2c��d�}dD]=}tj�d|��}t|||��d����>dS)Nc�H�|dzd|dzzzd|dzzzd|dzzz|zS)Nr�rnrer�r�rr>r8rks r0�
poiss_moment5z-TestExpect.test_moment.<locals>.poiss_moment5Ys8���q�5�2�b�!�e�8�#�b��Q��h�.��B��E��9�B�>�>r2)r�rsr�r�r�)r6rr�r)r�rxr��m5s    r0�test_momentzTestExpect.test_momentVse��	?�	?�	?��	?�	?�B���%�%�a��,�,�B��B�
�
�b� 1� 1��>�>�>�>�>�	?�	?r2c���ttj�dd���d��ttj�dd���d��ttj�dd���d��ttj�d���d	��ttj�d
���d
��dS)Nrrr;r�rnrl)�r�r|�Ur)rr6rEr�rr%r�s r0�test_challenging_cases_gh8928z(TestExpect.test_challenging_cases_gh8928`s���	��
�)�)�b��)�<�<�b�A�A�A���
�)�)�b��)�<�<�b�A�A�A���
�)�)�b��)�<�<�b�A�A�A����*�*��*�7�7��=�=�=����-�-�"�-�5�5�r�:�:�:�:�:r2c	�t�tj}|�dd���}t|�dd���|��t|�dddd���|��t|�dddd���|d	z��t|�ddddd
���|��t|�dddd���d
��t|�dddd���d��t|�ddddd
���d��dS)Nrnr�r;r�rar"r|r�r�Tr%r�g433333@g433333�r)r6r�rDrr�)r�rRrs   r0�test_lb_ub_gh15855zTestExpect.test_lb_ub_gh15855isH���}���i�i�B�a�i�(�(�������!��4�4�c�:�:�:������!��b��A�A�3�G�G�G������!��r��B�B�C��G�L�L�L������!��r�04�$�6�6�7:�	<�	<�	<�	�����!��r��B�B�F�K�K�K������!��r��B�B�G�L�L�L������!��r�04�$�6�6�79�	;�	;�	;�	;�	;r2N)r#r$r%rFr/r?rErJrSrZr�r�r�r`rhrkrurzr~r�r8r2r0rr�s������
5�5�5� 5�5�5�& 3� 3� 3�D	4�	4�	4�4�4�4�K�K�K�
8�
8�
8��[��� 8�9�9�9�9�:�9�9�O�O�O�+�+�+�
0�0�0�?�?�?�;�;�;�;�;�;�;�;r2rc�,�eZdZd�Zd�Zd�Zd�Zd�ZdS)�TestNctc���tjdd��}t|�d��d��tjdd��}t	|�d��dd���dS)Nr�rrxr�g�� �K��?rnr)r6�nctrrKrr�s  r0�test_nc_parameterzTestNct.test_nc_parameter�sc���Y�q�!�_�_���R�V�V�A�Y�Y��$�$�$�
�Y�q�"�
�
���B�F�F�1�I�I�~�r�B�B�B�B�B�Br2c	��tj�dtjdd��dd�dftjddd����}t
gd�gd�gd�g��}t||d	�
��dS)Nr�rersrlr)g���Mp^j?g6].ft�v?gȳ��'ς?g����u��?)g�!Y<��a?g�]d��1p?g�o�|?g�����?)g�Hd��Y?g2�Ԥ��g?g}o�/�u?g�.m�r�?r6r�)r6r�r@rtr�rrr)r�r�ris   r0�test_broadcastingzTestNct.test_broadcasting�s����i�m�m�A�r�y��A���q�q�q�$�w�7��K��Q��2�2�4�4���J�J�J�J�J�J�J�J�J�L�M�M��	��X�D�1�1�1�1�1�1r2c�t�tjdd��}t|���d��dS)Nrerr|)r6r�rrr�s  r0�test_variance_gh_issue_2401z#TestNct.test_variance_gh_issue_2401�s1��
�Y�q�!�_�_���R�V�V�X�X�s�#�#�#�#�#r2c��tj�ddd���\}}}}t||||gtjtjtjtjg��tj�ddd���\}}}}ttj|����t|||gtjtjtjg��tj�ddd���\}}}}ttj|||g�������t|tj��dS)Nr�r�r)rp�ncrr�r�)r6r�rrtr�r	rrros     r0�test_nct_inf_momentszTestNct.test_nct_inf_moments�s	���Y�_�_���V�_�D�D�
��1�a���a��A�q�\�B�F�B�F�B�F�B�F�#C�D�D�D��Y�_�_���V�_�D�D�
��1�a�����A�������a��A�Y������� 8�9�9�9��Y�_�_���V�_�D�D�
��1�a�����Q��1�I�&�&�*�*�,�,�-�-�-��Q�������r2c��tj�dd��}tj�dd��}ddg}t||dd���t||d���tj�dd��}tj�dd��}d	d
g}t||dd���t||d���dS)Nr�r>g�@@g�*�dn�?rr�r�rg�qI�@gL��)�?r�)r6r�rDr)r��nct_mean_df_1000�nct_stats_df_1000�expected_stats_df_1000�nct_mean�	nct_statsr�s       r0�test_nct_stats_large_df_valuesz&TestNct.test_nct_stats_large_df_values�s���!�9�>�>�$��2�2��!�I�O�O�D�!�4�4��
#5�6H�!I���(�*@��*C�"�	$�	$�	$�	$��)�+A�"�	$�	$�	$�	$��9�>�>�&�!�,�,���I�O�O�F�A�.�.�	�,�.@�A����.��"3�%�@�@�@�@��	�>��=�=�=�=�=�=r2N)r#r$r%r�r�r�r�r�r8r2r0r�r�sb������C�C�C�2�2�2�$�$�$� � � �>�>�>�>�>r2r�c�&�eZdZd�Zd�Zd�Zd�ZdS)�TestRecipInvGaussc�V�tj�dd��}|dksJ�dS)Nrr�r})r6�
recipinvgaussr@rLs  r0�test_pdf_endpointz#TestRecipInvGauss.test_pdf_endpoint�s+����#�#�A�s�+�+���C�x�x�x�x�x�xr2c�l�tj�dd��}|tjksJ�dS)Nrr�)r6r�r�rtr�r/s  r0�test_logpdf_endpointz&TestRecipInvGauss.test_logpdf_endpoint�s/���"�)�)�!�S�1�1�����w������r2c�n�tj�dd��}d}t||d���dS)Nr�rxg9V9�s�;r,r�)r6r�rKr�r�r?ris   r0�test_cdf_small_xz"TestRecipInvGauss.test_cdf_small_x�s<�� 
��#�#�D�#�.�.��(����8�%�0�0�0�0�0�0r2c�l�tj�dd��}d}t||d��dS)N�Prxgl���H<�+�����<)r6r�rgrr�s   r0�test_sf_large_xz!TestRecipInvGauss.test_sf_large_x�s7����"�"�2�s�+�+��(����8�U�+�+�+�+�+r2N)r#r$r%r�r�r�r�r8r2r0r�r��sP������������1�1�1�(,�,�,�,�,r2r�c� �eZdZd�Zd�Zd�ZdS)�TestRicec��gd�}ttjtj�|d����������ttjtj�|d����������ttjtj�|d����������ttjtj�	|d����������gd�}ttjtj�
|d����������tj�dd���}ttj|�������d}ttj�|d��tj�||��|d�	��dS)
N)r�rrr}�r/)rlrlrxr�rrrrQr�)r	rtrr6rRr@rr�rKr~r]r)r�rGr_rr/s     r0�test_rice_zero_bzTestRice.test_rice_zero_b�s����M�M�����E�J�N�N�1��N�3�3�4�4�8�8�:�:�;�;�;����E�J�-�-�a�2�-�6�6�7�7�;�;�=�=�>�>�>����E�J�N�N�1��N�3�3�4�4�8�8�:�:�;�;�;����E�J�-�-�a�2�-�6�6�7�7�;�;�=�=�>�>�>� � � �����E�J�N�N�1��N�3�3�4�4�8�8�:�:�;�;�;��z����6��2�2�����D�!�!�%�%�'�'�(�(�(�

����
���q�!�,�,�e�j�n�n�Q��.B�.B��Q�	(�	(�	(�	(�	(�	(r2c��tjj}t|d���jd��t|dd���jd��dS)Nrr�r)r�r�)r/r)r6rRr�rrrr�s  r0�
test_rice_rvszTestRice.test_rice_rvs�sO���j�n���S�S�2�Y�Y�Y�^�Q�'�'�'��S�S�2�F�+�+�+�1�6�:�:�:�:�:r2c���tj�tjddd��tjddd����}gd�}t||��tjddd��}tj�|dd���}gd	�}t||��tjj�d
tjddd����}gd�}t||��dS)
Nrn�)g/��Ƹ�?g�[jNJ\�?g� �	��?g�9�LJ��?g�#�O���?g�X�x���?g��l�O��?g٥a�%��?gVz�ӯ��?gTQ+xQ��?g?�`J��?g�o�����?gbu=���?gIo|���?g�"Z6��?rlrg@_@rerD)	g����<�~@gw�d
@gC�?�@g�b�
0@g�` �A@@g'��NxP@g3�3-�a@g��4pv@g��$F�@rx�)g�J-�$@g��+f4@gce*D>@g�$��D@gn�H�GI@g��rN@g<
�u�Q@giffT@g9�r[�V@g3bX�QY@gq�xJ�[@g`*DD^@g���@`@gd��A�a@)r6rRrKrtr�rr]r5)r�rK�cdf_exp�
probabilitiesr]�ppf_exps      r0�test_rice_gh9836zTestRice.test_rice_gh9836�s����j�n�n�R�Y�r�3��3�3�R�Y�r�3��5K�5K�L�L��O�O�O��
	��W�%�%�%��	�#�q�#�.�.�
��j�n�n�]�E��n�;�;��L�L�L��	��W�%�%�%��k��"�"�3��	�"�c�2�(>�(>�?�?��;�;�;��
	��W�%�%�%�%�%r2N)r#r$r%r�r�r�r8r2r0r�r��sA������(�(�(�*;�;�;�
'&�'&�'&�'&�'&r2r�c� �eZdZd�Zd�Zd�ZdS)�
TestErlangc�D�tj�d��dSr�r�r�s r0r�zTestErlang.setup_method'r�r2c	��tj��5tjdt��t	tt
jjdddd���gd�}t
j�|d���}t
j	�|d���}t||d	�
��ddd��dS#1swxYwYdS)Nr<r�rrrer%)rxrr|r
rYr
r�)r@rArBrCr�r6�erlangr�r�rr)r�r��
result_erlang�result_gammas    r0�test_erlang_runtimewarningz%TestErlang.test_erlang_runtimewarning*s ���
$�
&�
&�	D�	D��!�'�>�:�:�:�
�.��,�*�C�Q�a�a�
I�
I�
I�
I�(�'�'�D�!�L�,�,�T�a�,�8�8�M� �;�?�?�4�A�?�6�6�L��M�<�d�C�C�C�C�	D�	D�	D�	D�	D�	D�	D�	D�	D�	D�	D�	D����	D�	D�	D�	D�	D�	Ds�BB=�=C�Cc��ttj�dddg���tj�dddg�����dS)Nrxrr�rB)rr6r�r@rr�s r0�test_gh_pr_10949_argcheckz$TestErlang.test_gh_pr_10949_argcheck=sR���U�\�%�%�c�a��W�%�5�5��[�_�_�S�Q��G�_�4�4�	6�	6�	6�	6�	6r2N)r#r$r%r�r�r�r8r2r0r�r�&sD���������D�D�D�&6�6�6�6�6r2r�c���eZdZd�Zd�Zd�Zej�dddg��d���Z	ej�ddd	gd
dgg��d���Z
d
�Zd�ZdS)�TestRayleighc�D�tj�d��dS)Nr$r�r�s r0r�zTestRayleigh.setup_methodCs��
�	���y�!�!�!�!�!r2c�d�tj�d��}t|d��dS)Nrg�)�Zx��)r6r�r�r�r�rs  r0r�zTestRayleigh.test_logpdfGs.���N�!�!�"�%�%����.�/�/�/�/�/r2c�d�tj�d��}t|d��dS)Nri���)r6r�r%rr�s  r0r7zTestRayleigh.test_logsfKs-���N� � ��$�$����5�!�!�!�!�!r2zrvs_loc,rvs_scale)g�B�)�Q�?g�ot|��?)g�S2�P�?g��3���?c���tj�d||���}d�}|||��}tj�||���\}}t	||��t	||��tj�|d���\}}t	|d��tj�|��\}}t	||||����dS)Nrr2c�b�tj||z
dz��dt|��zzdzS)Nr>rx)rtr6r�)r�r�s  r0r\z(TestRayleigh.test_fit.<locals>.scale_mleTs/���F�D�4�K�A�-�.�.�!�c�$�i�i�-�@�R�G�Gr2r?r�r@)r6r�r�r�r)r�rsrtr�r\�scale_expectr<r=s        r0r=zTestRayleigh.test_fitOs����~�!�!�s��y�!�I�I��	H�	H�	H�!�y��w�/�/���^�'�'��7�'�;�;�
��U��S�'�"�"�"��U�L�)�)�)��^�'�'��R�'�8�8�
��U��U�B����
�^�'�'��-�-�
��U��U�I�I�d�C�0�0�1�1�1�1�1r2rLrzg̡=E��?g@1�d��?c��tj�d||���}|tj�|��fg}tj�|i��d}ttj||��dS)Nrr2r)r6r�r�r8rAr�)r�rsrtr�r�r�s      r0� test_fit_comparison_super_methodz-TestRayleigh.test_fit_comparison_super_methodhsu��
�~�!�!�s��y�!�I�I���u�~�/�/��5�5�8�9���~�*�*�4��4�4�Q�7��%�e�n�d�D�A�A�A�A�Ar2c�8�ttj��dSrr)r�r6r�r�s r0r�zTestRayleigh.test_fit_warningsus���E�N�+�+�+�+�+r2c��tj�d��}d\}}}tj�||||���}tj�|��\}}|tj|��ksJ�tj�||���\}}|tj|��ksJ�||ksJ�dS)Ni�)rr�r�r�r@)rtr�r�r6r�r�r�r~)	r�r�r<r=rr��loc_fitr��	scale_fits	         r0�test_fit_gh17088zTestRayleigh.test_fit_gh17088xs����i�#�#�C�(�(��'���U�D��n� � ��e�$�S� �I�I���^�'�'��,�,�
���������$�$�$�$�"�^�/�/��E�/�B�B����������$�$�$�$��E�!�!�!�!�!�!r2N)
r#r$r%r�r�r7r�r�r
r=r�r�r�r8r2r0r�r�Bs�������"�"�"�0�0�0�"�"�"��[���0�3K�3K�3M�N�N�2�2�N�N�2�.�[���0�D�$�<�4>�
�3K�3M�N�N�	B�	B�N�N�	B�,�,�,�
"�
"�
"�
"�
"r2r�c� �eZdZd�Zd�Zd�ZdS)�
TestExponWeibc��d}d}d}tj�|||��}tj�|||��}t	||gddg��dS)NrlrgY@g:�2�3W�+g��~`9�k�)r6�	exponweibr@r�r)r�rGr.r^r?r�s      r0�test_pdf_logpdfzTestExponWeib.test_pdf_logpdf�ss���������O����1�a�(�(����%�%�a��A�.�.����D�	�/�1D�E�	G�	G�	G�	G�	Gr2c�~�tjddd��}d}d}tj�|||��}tj�||��}t
||��tj�|||��}tj�||��}t
||��dS)NrZr�rerro)rtr[r6r�r@�weibull_minrr��r�rGr.r^r?rir�s       r0�test_a_is_1zTestExponWeib.test_a_is_1�s���
�K��B��"�"��
�����O����1�a�(�(���$�(�(��A�.�.����8�$�$�$���%�%�a��A�.�.���$�+�+�A�q�1�1����h�'�'�'�'�'r2c�z�tjddd��}d}d}tj�|||��}tj�|��}t
||��tj�|||��}tj�|��}t
||��dS)Nr�rrn)rtr[r6r�r@rRrr�r�s       r0�test_a_is_1_c_is_1z TestExponWeib.test_a_is_1_c_is_1�s����K��A�r�"�"��
��
���O����1�a�(�(���;�?�?�1�%�%����8�$�$�$���%�%�a��A�.�.���;�%�%�a�(�(����h�'�'�'�'�'r2N)r#r$r%r�r�r�r8r2r0r�r��sD������	G�	G�	G�(�(�(� (�(�(�(�(r2r�c��eZdZd�Zd�ZdS)�TestFatigueLifec�j�tj�dd��}t|dd���dS)Nrri���X��9r�r�)r6r�rgrris  r0r�zTestFatigueLife.test_sf_tail�s8��
�� � ���,�,����0�u�=�=�=�=�=�=r2c�n�d}tj�|d��}t|dd���dS)Nr�rirr�r�)r6r�rlr)r�r?r_s   r0�
test_isf_tailzTestFatigueLife.test_isf_tail�s:��!����!�!�!�S�)�)����5�u�-�-�-�-�-�-r2N)r#r$r%r�r�r8r2r0r�r��s2������>�>�>�.�.�.�.�.r2r�c� �eZdZd�Zd�Zd�ZdS)�TestWeibullc�f�tj�dd��}t|d��dSrJ)r6r�r�rr�s  r0r�zTestWeibull.test_logpdf�s0����$�$�Q��*�*���Q������r2c��d}d}d}tj�|||���}t|t	jd��dz��tj�|||���}t|dt	jd��z
��tj�|||���}t|tj
d����tj�|||���}t|t	jtj
d������tj�|||���}t|t	jd����tj�
|||���}	t|	d��tj�ddd���}t|t	jd	����tj�
ddd���}	t|	d	��d
}tj�|||���}t|t	jd��dz��tj�|||���}t|dt	jd��z
��tj�|||���}t|t	jd����tj�|||���}t|d��tj�|||���}t|tj
d����tj�
|||���}	t|	t	jtj
d������tj�ddd���}t|tj
d����tj�
ddd���}	t|	t	jtj
d������dS)
Nr�r|rrDrTr�r�r>r�r�g��&�.�gC���e�)r6r�r@rrtr�r�rUrKrr�r~rgr%�weibull_max)
r�rGr.r/r?rr^�lcrFr"s
          r0�test_with_maxima_distribz$TestWeibull.test_with_maxima_distrib�s���.
������
��!�!�!�Q�a�!�0�0����2�6�%�=�=��?�+�+�+�
�
�
%�
%�a��!�
%�
4�
4����E�B�F�1�I�I�-�.�.�.���!�!�!�Q�a�!�0�0����G�M�%�0�0�0�1�1�1�
�
�
%�
%�a��!�
%�
4�
4����B�F�G�M�%�$8�$8�#8�9�9�:�:�:��� � ��A�Q� �/�/����2�6�%�=�=�)�)�)�
�
�
$�
$�Q���
$�
3�
3����E�"�"�"�
�� � ��Q�a� �0�0����2�6�$�<�<�(�(�(�
�
�
$�
$�R��!�
$�
4�
4����D�!�!�!�
����!�!�!�Q�a�!�0�0����2�6�%�=�=��?�+�+�+�
�
�
%�
%�a��!�
%�
4�
4����E�B�F�1�I�I�-�.�.�.���!�!�!�Q�a�!�0�0����2�6�%�=�=�)�)�)�
�
�
%�
%�a��!�
%�
4�
4����E�"�"�"��� � ��A�Q� �/�/����G�M�%�0�0�0�1�1�1�
�
�
$�
$�Q���
$�
3�
3����B�F�G�M�%�$8�$8�#8�9�9�:�:�:�
�� � ���� �3�3����G�M�*@�A�A�A�B�B�B�
�
�
$�
$�U�A�Q�
$�
7�
7����B�F�G�M�2H�$I�$I�#I�J�J�K�K�K�K�Kr2c��tj�d��}d\}}}tj|||��}|�d|���}tj�|dd���\}}}	tj�|dd���\}
}}||cxkrdksnJ�||
ksJ�tj�|ddd	�
��\}
}}|dksJ�tj|
||��}|�d���}tj|��tj|��f}t||��dS)
Nl��>�)r>rjrxror�r�r�r?rRr>r?rAr)
rtr�r�r6r�r�r�rDrr)r�r�r^r<r=rRr��c2rErF�c3rHrI�c4rKrLrMr�rs                   r0�test_fit_minzTestWeibull.test_fit_min#sT���i�#�#�$7�8�8��#�
��3��� ��C��/�/���h�h�C�c�h�2�2��!�,�0�0��c��0�B�B���D�&� �,�0�0��c��0�B�B���D�&��t� � � � �q� � � � � � ��R�x�x�x�x�!�,�0�0��a��$�0�O�O���D�&���{�{�{�{��!�"�d�F�3�3���k�k�$�k�'�'���g�c�l�l�E�J�s�O�O�+����S�!�!�!�!�!r2N)r#r$r%r�r�r�r8r2r0r�r��sH���������
RL�RL�RL�h"�"�"�"�"r2r�c�D�eZdZd�Zd�Zd�Zd�Zd�Zd�Zd�Z	d�Z
d	�Zd
S)�TestTruncWeibullc�r�tj�ddgddd��}t|ddg��dS)Nrlr|g)\��(�?gףp=
�?r})r6�truncweibull_minr@rr�s  r0�test_pdf_boundsz TestTruncWeibull.test_pdf_bounds=s<���"�&�&��S�z�3��d�C�C���Q��c�
�#�#�#�#�#r2c���tj�dddtj��}t|d��tj�dddd��}t
|d��dS)Nr|rr}r
g���~��?)r6r�r�rtr�rrr�s  r0r�zTestTruncWeibull.test_logpdfBse���"�)�)�#�s�C���@�@���Q�����
�"�)�)�#�s�C��=�=����.�/�/�/�/�/r2c�r�tj�ddgddd��}t|ddg��dS)Nr}rr|rl)r6r�r]rr�s  r0�test_ppf_boundsz TestTruncWeibull.test_ppf_boundsJs<���"�&�&��S�z�3��S�A�A���Q��c�
�#�#�#�#�#r2c��gd�}tj�|ddd��}tj�|ddd��}t	||��dS�N)r}rlr5rxr+r�rr|r}r)r6r�r]rKr�r�r_rG�q_outs    r0�test_cdf_to_ppfz TestTruncWeibull.test_cdf_to_ppfOs[��0�0�0���"�&�&�q�"�b�"�5�5���&�*�*�1�b�"�b�9�9����5�!�!�!�!�!r2c��gd�}tj�|ddd��}tj�|ddd��}t	||��dSr�)r6r�rlrgrr�s    r0�test_sf_to_isfzTestTruncWeibull.test_sf_to_isfUs[��0�0�0���"�&�&�q�"�b�"�5�5���&�)�)�!�R��R�8�8����5�!�!�!�!�!r2c�����
��d�
d�d����
fd��tj�d�
����}t|d��tj�d�
����}t	�fd�����\}}t||��tj�d�
����}t	�fd	�����\}}t||��tj�d
�
����}t	�fd�����\}}t||��tj�d�
����}	t	�fd
�����\}
}t|	|
��dS)Nr|rrc�T��||ztj�|�����zSrr)r6r�r@)rGr>r.r/r^s  ���r0�xnpdfz)TestTruncWeibull.test_munp.<locals>.xnpdf`s)����a�4��.�2�2�1�a��A�>�>�>�>r2rrc����|d��Sr�r8�rGr�s �r0r_z,TestTruncWeibull.test_munp.<locals>.<lambda>g������a����r2r>c����|d��Srr8r�s �r0r_z,TestTruncWeibull.test_munp.<locals>.<lambda>kr�r2r�c����|d��SrQr8r�s �r0r_z,TestTruncWeibull.test_munp.<locals>.<lambda>or�r2rec����|d��S)Nrer8r�s �r0r_z,TestTruncWeibull.test_munp.<locals>.<lambda>sr�r2)r6r�r�rrr)r�r�r��m1_expectedr�rN�m2_expected�m3�m3_expectedrO�m4_expectedr.r/r^r�s           @@@@r0�	test_munpzTestTruncWeibull.test_munp[s�������������	?�	?�	?�	?�	?�	?�	?��
#�
*�
*�1�a��A�
6�
6���R�����
�
#�
*�
*�1�a��A�
6�
6���3�3�3�3�Q��:�:���Q���K�(�(�(�
�
#�
*�
*�1�a��A�
6�
6���3�3�3�3�Q��:�:���Q���K�(�(�(�
�
#�
*�
*�1�a��A�
6�
6���3�3�3�3�Q��:�:���Q���K�(�(�(�
�
#�
*�
*�1�a��A�
6�
6���3�3�3�3�Q��:�:���Q���K�(�(�(�(�(r2c�^�d}d}d}tjdtjdtjdtjd��z��z��z
��}tj�||||��}t|d��tj�||||��}t|tjd����tj�	d|||��}t||��tj�
||||��}t|d��tj�||||��}	t|	tjd����tj�d|||��}
t|
|��dS)Nrrr|rrxr)
rtr�rUr�r6r�rKrr~r]rgr%rl)r�r.r/r^�x_medrKr�r]rgr"rls           r0�test_reference_valuesz&TestTruncWeibull.test_reference_valuesvso�����������B�F�3���"�r�v�b�z�z�/�0B�)C�)C�#C�D�D�D�E�E���$�(�(���1�a�8�8����S�!�!�!�
�
#�
*�
*�5�!�Q��
:�
:����R�V�B�Z�Z�K�(�(�(��$�(�(��a��A�6�6����U�#�#�#�
�
#�
&�
&�u�a��A�
6�
6����C� � � �
�
#�
)�
)�%��A�q�
9�
9����R�V�B�Z�Z�K�(�(�(��$�(�(��a��A�6�6����U�#�#�#�#�#r2c�*�d}d}d}tj}d}tj�|||���}tj�|||||���}t
||��tj�|||���}tj�|||||���}	t
||	��tj�|||���}
tj�|||||���}t
|
|��tj�	|||���}tj�	|||||���}
t
||
��tj�
|||���}tj�
|||||���}t
||��tj�|||���}tj�|||||���}t
||��tj�
dd||d���}t
|tjd	����tj�dd||d���}t
|d	��dS)
Nr�r|r}rrDr�r>r�r�)
rtr�r6r�r@r�rr�rKr~rgr%r�)r�rGr^r.r/r=r?�p_truncr�lp_truncrK�	cdf_truncr��lc_truncrF�s_truncr"�ls_truncs                  r0�test_compare_weibull_minz)TestTruncWeibull.test_compare_weibull_min�se��
�������F������!�!�!�Q�e�!�4�4���(�,�,�Q��1�a�u�,�E�E����7�#�#�#�
�
�
%�
%�a��%�
%�
8�
8���)�0�0��A�q�!�5�0�I�I����H�%�%�%���#�#�A�q��#�6�6���*�.�.�q�!�Q���.�G�G�	���Y�'�'�'�
�
�
%�
%�a��%�
%�
8�
8���)�0�0��A�q�!�5�0�I�I����H�%�%�%��� � ��A�U� �3�3���(�+�+�A�q�!�Q�e�+�D�D����7�#�#�#�
�
�
$�
$�Q���
$�
7�
7���)�/�/��1�a��%�/�H�H����H�%�%�%�
�"�%�%�b�!�Q���%�;�;����2�6�$�<�<�(�(�(�
�
#�
)�
)�"�a��A�Q�
)�
?�
?����D�!�!�!�!�!r2c��d\}}}tj||d��}tj�||||��}tj�||||��}tj�||��tj�||��z
}tj�||��|z}tj�||��tj�||��z
|z}	tj�||��tj�||	��dS)N)rir5��?ro)	rtrr6r�r@rKr��testingr)
r�r^r.r/rGr?rdrEr@rfs
          r0�test_compare_weibull_min2z*TestTruncWeibull.test_compare_weibull_min2�s��"���1�a��K��1�c�"�"���%�)�)�!�Q��1�5�5���%�)�)�!�Q��1�5�5��� �$�$�Q��*�*�U�->�-B�-B�1�a�-H�-H�H��� �$�$�Q��*�*�T�1���!�%�%�a��+�+�e�.?�.C�.C�A�q�.I�.I�I�4�O��
�
�"�"�4��.�.�.�
�
�"�"�4��.�.�.�.�.r2N)r#r$r%r�r�r�r�r�rrrrr8r2r0r�r�;s�������$�$�$�
0�0�0�$�$�$�
"�"�"�"�"�"�)�)�)�6$�$�$�0'"�'"�'"�R/�/�/�/�/r2r�c��eZdZd�Zd�ZdS)�	TestRdistc��tj}gd�}t|�|�|d��d��|d���dS)Nr	g�@r�r)r6�rdistrrKr])r�rr�s   r0�test_rdist_cdf_gh1285zTestRdist.test_rdist_cdf_gh1285�sX�����$�$�$���F�J�J�v�z�z�&�%�'@�'@�%�H�H�"�A�	/�	/�	/�	/�	/�	/r2c�
�tjddd��}d}tdtj|dz|dz���|dzdz��ztj|���|����dS)Ng�G�z��r�rng������@rxr>r)rtrrr6r�r@r)r�rGr^s   r0�test_rdist_betazTestRdist.test_rdist_beta�s����K��t�R�(�(�����C��
�1�Q�3��!�� 4� 4� 8� 8�!�a�%��� C� C�C�!�K��N�N�.�.�q�1�1�	3�	3�	3�	3�	3r2N)r#r$r%rrr8r2r0rr�s2������/�/�/�3�3�3�3�3r2rc�2�eZdZd�Zd�Zd�Zd�Zd�Zd�ZdS)�
TestTrapezoidc�V�gd�}|D]�}d|dg}ttj�|||��tj�||����ttj�|||��tj�||������dS)N)rr�rxrrr)rr6rr@�triangrK)r��modes�moderGs    r0�test_reduces_to_triangz$TestTrapezoid.test_reduces_to_triang�s��� � � ���	;�	;�D��D�!��A���� 3� 3�A�t�T� B� B� %�� 0� 0��D� 9� 9�
;�
;�
;���� 3� 3�A�t�T� B� B� %�� 0� 0��D� 9� 9�
;�
;�
;�
;�		;�	;r2c�b�tjddd��}ttj�|dd��tj�|����ttj�|dd��tj�|����dS�Nrrrn)rtrrr6rr@r�rKrms  r0�test_reduces_to_uniformz%TestTrapezoid.test_reduces_to_uniform�s����K��1�b�!�!���E�O�/�/��1�a�8�8�%�-�:K�:K�A�:N�:N�O�O�O��E�O�/�/��1�a�8�8�%�-�:K�:K�A�:N�:N�O�O�O�O�Or2c�n�ttj�ddd��d��ttj�ddd��d��ttj�ddd��d��ttj�ddd��d��ttj�d	dd��d
��ttj�ddd��d��ttj�ddd��d
��ttj�d	dd��d
��ttj�ddd��d��ttj�ddd��d��ttj�ddd��d��ttj�ddd��d��dS)Nrr>rrxrbr�r�rrlg�?rr�g�?rg�?)rr6rr@rKr�s r0�
test_caseszTestTrapezoid.test_cases�s����E�O�/�/��1�a�8�8�!�<�<�<��E�O�/�/��1�a�8�8�!�<�<�<��E�O�/�/��Q��<�<�/�	1�	1�	1��E�O�/�/��S�#�>�>�/�	1�	1�	1�	�E�O�/�/��S�#�>�>��F�F�F��E�O�/�/��S�#�>�>��E�E�E��E�O�/�/��S�#�>�>��F�F�F��E�O�/�/��S�#�>�>��H�H�H��E�O�/�/��S�#�>�>��F�F�F��E�O�/�/��S�#�>�>��D�D�D��E�O�/�/��S�#�>�>��H�H�H��E�O�/�/��S�#�>�>��D�D�D�D�Dr2c��	�
���
�d\�	�
���
�	z
��	z
z��	z
��	z
z�	��	z
f\}}}}d��z�
z
�	z
z�
�	�
���
fd�}|d��}|d��|dzz
}d��z
�
z�	z
z��z�
z
�	z
ztjd��z�
z
�	z
z��z}ttj�||||��|d���ttj�||||��|d���ttj�||||��|d���ttj�ddd	d
��dd���ttj�ddd	d
��dd���ttj�ddd	d
��dd���dS)
N)r�r�r>r�r>c�����|dzz�|dzzz
��z
z�|dzz�|dzzz
��z
zz
z|dzz|dzzS)Nr>rr8)r>r.r/r^�dr9s �����r0r�z6TestTrapezoid.test_moments_and_entropy.<locals>.momentsn����!�a��c�(�Q��1��X�-�!�A�#�6���!��H�q�1�Q�3�x�/�A�a�C�8�9�:��q�S���q�S�"�
#r2rrxr�rrr�rr�r�)rtrUrr6rrDrrX)r�r�r_r<r=r�rDrrXr.r/r^r)r9s         @@@@@r0�test_moments_and_entropyz&TestTrapezoid.test_moments_and_entropy�s�������"�
��1�a����c�a��c�]�Q�q�S�Q�q�S�M�1�a��c�A���B��U�
��1��Q��q��M��	#�	#�	#�	#�	#�	#�	#�	#�	#�
�v�a�y�y���f�Q�i�i�$��'�!����1��Q��q��/�Q�q�S��U�1�W�-���s�a��c�!�e�A�g��0G�0G�G���E�O�0�0��R��e�D�D� �"�	.�	.�	.�	.��E�O�/�/��B��U�C�C���	-�	-�	-�	-��E�O�3�3�B��C��G�G�#�R�	1�	1�	1�	1�	�E�O�0�0��A�r�1�=�=�r�2�N�N�N�N��E�O�0�0��A�r�1�=�=�q�"�M�M�M�M��E�O�/�/��1�b�!�<�<�a��L�L�L�L�L�Lr2c���tjgd���}tjddg��dd�df}tjgd���}tj�|||��}tj|||��\}}}tj|j|j���}tj	|j��}	t|	|���|���|�����D]+\}
}}}
tj�|||
��||
<�,t||�
|j��d���tjtj�||d�	����}tj||��\}}tj|jd
f��}tj	|j��}	t|	|���|�����D]+\}
}}
tj�||
d�	��||
<�,t||j�
|j��d���dS)Nr�rxr�)r�r5r�r�r�r�rrre)rtrr6rr@�broadcast_arrays�emptyrr�r�r��ravelrrrr�r)r�r^r)rGr��cc�ddr`r��indr�r6�c1�d1s              r0�test_trapezoid_vectz!TestTrapezoid.test_trapezoid_vects
���H�_�_�_�%�%���H�c�3�Z� � ����D��)���H�&�&�&�'�'���O����1�a�(�(���(��A�q�1�1�
��B���h�r�w�b�h�/�/�/���i��� � �� ��b�h�h�j�j�"�(�(�*�*�b�h�h�j�j�I�I�	5�	5�M�A�r�2�r��_�(�(��R��4�4�C��F�F���3�;�;�q�w�/�/�e�<�<�<�<�
�J�u��,�,�Q��6�,�B�B�C�C���$�Q��*�*���B��h����|�$�$���i��� � ���S�"�(�(�*�*�b�h�h�j�j�9�9�	C�	C�I�A�r�2��_�*�*�2�r�6�*�B�B�C��F�F���3�5�=�=���1�1��>�>�>�>�>�>r2c���tjddd��}ttj�|dd��tj�|����dSr#)rtrrr6rYr@r�rms  r0�
test_trapzzTestTrapezoid.test_trapz0sM���K��1�b�!�!���E�K�O�O�A�q�!�4�4�e�m�6G�6G��6J�6J�K�K�K�K�Kr2N)	r#r$r%r!r$r&r*r4r6r8r2r0rr�s|������;�;�;�P�P�P�
E�E�E�(M�M�M�6?�?�?�4L�L�L�L�Lr2rc��eZdZd�ZdS)�
TestTriangc��tjd���5ttj�dd��d��ttj�dd��d��ttj�dd��d��ttj�dd��d��ttj�dd��d��ttj�dd��d	��ttj�dd��d��ttj�dd��d
��ttj�dd��d��ttj�dd��d��ttj�dd��d��ttj�dd��d��ddd��dS#1swxYwYdS)N�raiser�rr|rxrrr}r>r+r5)rtr�rr6rr@rKr�s r0�test_edge_caseszTestTriang.test_edge_cases7s��
�[�W�
%�
%�
%�	6�	6����)�)�!�Q�/�/��4�4�4����)�)�#�q�1�1�2�6�6�6����)�)�!�Q�/�/��4�4�4����)�)�!�Q�/�/��3�3�3����)�)�#�q�1�1�2�6�6�6����)�)�!�Q�/�/��3�3�3����)�)�"�b�1�1�2�6�6�6����)�)�#�r�2�2�D�9�9�9����)�)�#�r�2�2�C�8�8�8����)�)�"�b�1�1�2�6�6�6����)�)�#�r�2�2�D�9�9�9����)�)�"�b�1�1�1�5�5�5�	6�	6�	6�	6�	6�	6�	6�	6�	6�	6�	6�	6����	6�	6�	6�	6�	6�	6s�H)I�I�IN)r#r$r%r;r8r2r0r8r86s#������6�6�6�6�6r2r8c��eZdZd�Zd�ZdS)�
TestMielkec��d\}}ttj||���d��tj��ttj|d���d��tj��t
t	jtj|d���d������dS)N)g� �rh�@g��v���?rrr�)rr6�mielker�rtr�r	r)r�rDrFs   r0r�zTestMielke.test_momentsKs������1��U�\�!�Q�'�'�.�.�q�1�1�2�6�:�:�:��U�\�!�S�)�)�0�0��3�3�R�V�<�<�<����E�L��D�1�1�8�8��;�;�<�<�=�=�=�=�=r2c���tjddd��}d\}}ttj�||||z��tj�|||����dS)Nrzror)g������@gH�z�G@)rtrrr6�burrr@r?)r�rGrDrFs    r0�test_burr_equivalencez TestMielke.test_burr_equivalenceRs^���K��c�2�&�&�����1���
���q�!�Q�q�S�1�1�5�<�3C�3C�A�q�!�3L�3L�M�M�M�M�Mr2N)r#r$r%r�rBr8r2r0r=r=Js7������>�>�>�N�N�N�N�Nr2r=c� �eZdZd�Zd�Zd�ZdS)�TestBurrc	�n�tjddgtjddgtjddgtjddgtjddgtjddgtjd	d
gg}d�|D��}d�|D��}t	||��d
�|D��}d�|D��}t	||��dS)N)rr)rxr>)rr)r>rxrrxr)rr>r|c�<�g|]\}}}|j|jg|�R���Sr8r�r�s    r0rSz0TestBurr.test_endpoints_7491.<locals>.<listcomp>er�r2c��g|]\}}}|��	Sr8r8r�s    r0rSz0TestBurr.test_endpoints_7491.<locals>.<listcomp>fr�r2c�<�g|]\}}}|j|jg|�R���Sr8)r�r.r�s    r0rSz0TestBurr.test_endpoints_7491.<locals>.<listcomp>is4��@�@�@�<�2�u�a�y�r�y���&��&�&�&�@�@�@r2c�>�g|]\}}}tj|����Sr8)rtrUr�s    r0rSz0TestBurr.test_endpoints_7491.<locals>.<listcomp>js)��F�F�F�)=��U�I�2�6�)�$�$�F�F�Fr2)r6�fiskrA�burr12r)r�r�r�r�s    r0�test_endpoints_7491zTestBurr.test_endpoints_7491Ys����Z��q�!�
�Z��1�%�
�Z���#�
�Z��1�%�
�\�8�S�)�
�\�6�3�'�
�\�6�3�'�)��>�=��=�=�=��>�>��>�>�>��!�#�w�/�/�/�@�@�4�@�@�@��F�F��F�F�F��!�#�w�/�/�/�/�/r2c��d\}}tj||�����\}}d\}}t||��t||��dS)N)rr�)g��4U���?g���&MI�?)r6rAr)r�r^r)rD�variance�mean_hc�variance_hcs       r0�test_burr_stats_9544zTestBurr.test_burr_stats_9544ms_�����1���A�q�)�)�/�/�1�1���h� G������g�&�&�&���+�.�.�.�.�.r2c���d\}}tj||�����\}}ttj|����ttj|����d\}}tj||�����\}}ttj|����ttj|����d\}}tj�tjgd���||��\}}}}ttj|����ttj|����ttj|����ttj|����d\}}tj�gd�||��\}}}}ttj|����ttj|����ttj|����ttj|����d\}}tj�gd�||��\}}}}ttj|����ttj|����ttj|����ttj|����d\}}tj�gd�||��\}}}}ttj|����ttj|����ttj|����ttj|����d\}}tj�gd�||��\}}}}ttj|����ttj|����ttj|����ttj|����dS)N)rxr�)r�r�)rr>r�re)rir�)rjr�)rmr�)r6rAr	rtrtr�_munpr)	r�r^r)rDrN�e1�e2�e3�e4s	         r0�test_burr_nan_mean_var_9544z$TestBurr.test_burr_nan_mean_var_9544xsb�����1���A�q�)�)�/�/�1�1���h��������������"�"�#�#�#����1���A�q�)�)�/�/�1�1���h����D�!�!�"�"�"�����"�"�#�#�#����1���)�)�"�(�<�<�<�*@�*@�!�Q�G�G���B��B����������������������������������������1���)�)�,�,�,��1�=�=���B��B����B��� � � �������������������������������1���)�)�,�,�,��1�=�=���B��B����B��� � � ����B��� � � ����������������������1���)�)�,�,�,��1�=�=���B��B����B��� � � ����B��� � � ����B��� � � �������������1���)�)�,�,�,��1�=�=���B��B����B��� � � ����B��� � � ����B��� � � ����B��� � � � � r2N)r#r$r%rLrQrXr8r2r0rDrDXsA������0�0�0�(	/�	/�	/�(!�(!�(!�(!�(!r2rDc	�z�eZdZgd�Zgd�Zgd�Zejeeef��Zgd�Z	ddddd	ej
gZgd
�Ze
eee	ee��e����Zddddej
d
fddej
dfddej
dfgZd�Zejjd���Zej�e��ZdZeej�ee��d��5Ze j!e��Z"ddd��n#1swxYwYej�#de"d��d���Z$ej�#de"d��d���Z%ejjej�&d��ej�#de"d��d�������Z'd�Z(ejj)d���Z*ej�#d e��d!���Z+ejjej�&d��d"�����Z,ejj)d#���Z-d$�Z.d%�Z/d&�Z0dS)'�TestStudentizedRange)g���Q�1@g33333�F@g
ףp=*K@gH�z��M@g/�$@g�~j��!@g33333�$@g{�G�z&@gh��|?5	@g���Q8@g�&1�@g��|?5�@g������@g�O��n@g!�rh�m@gB`��"�@rAg�ʡE�s@g��|?5^@g/�$�@g�V-@g��/�$@gF���@g?5^�I@)gR���V@gffffffl@g�����p@g�r@g��ʡ� @gH�z�G/@g���Q82@g��Q��3@g!�rh��@g�@g+��@g�ʡE�s @gj�t�@gB`��"[@g������@g�n��J@gV-��
@gF���x@gsh��|?@g+��N@g��/�$
@g��� ��@g������@g�G�z�@)gfffff"�@i�i�
i�gH�z�G2@g��(\A@g���Q�C@gfffff�E@g��� ��@gNbX9�"@gH�z��$@g��(\&@g-����@g��n�@@gw��/�@g=
ףp� @gJ+�@g-���'@g�n���@gH�z��@gV-��@g�n��J@g�$���@g��/ݤ@)r
r�rrr�rnr��x)r>r�r�r�)rlr�i)#gupm1�f?)rrnr�g?�<-�=A?gv"���?reg�d暕�?ga�-�A?c��|jD]>\}}|\}}}tj�|||��}t	||d����?dS)Nr�r�)r�r6�studentized_rangerKr)r��pvkr_�
p_expectedr�rD�res_ps       r0�test_cdf_against_tablesz,TestStudentizedRange.test_cdf_against_tables�s_���i�	:�	:�F�C��"��J��1��+�/�/��1�a�8�8�E��E�:�D�9�9�9�9�9�	:�	:r2c��|jD]>\}}|\}}}tj�|||��}t	||d����?dS)N���Mb@?r�)r�r6r]r]r)r�r^�
q_expectedr?r�rD�res_qs       r0�test_ppf_against_tablesz,TestStudentizedRange.test_ppf_against_tables�s^��#�y�	:�	:�O�C���G�A�q�!��+�/�/��1�a�8�8�E��E�:�D�9�9�9�9�9�	:�	:r2z&data/studentized_range_mpmath_ref.jsonr�N�case_result�cdf_datac��|d}|d}|d|d|df}tjj|�}t|||d|d���dS�	N�src_case�	mp_resultr_rDr��
expected_atol�
expected_rtolr�)r6r]rKr�r�rgrkrl�qkvr�s      r0�test_cdf_against_mpz(TestStudentizedRange.test_cdf_against_mp��s���z�*����,�	��s�m�X�c�]�H�S�M�9���%�)�3�/����Y�%�o�6�%�o�6�	8�	8�	8�	8�	8�	8r2�pdf_datac��|d}|d}|d|d|df}tjj|�}t|||d|d���dSrj)r6r]r@rros      r0�test_pdf_against_mpz(TestStudentizedRange.test_pdf_against_mp�rrr2z+intermittent RuntimeWarning: invalid value.�moment_datac��|d}|d}|d|d|df}tjd���5tjj|�}ddd��n#1swxYwYt|||d|d	�
��dS)Nrkrlr�rDr�r�r�rmrnr�)rtr�r6r]r�r)r�rgrkrl�mkvr�s      r0�test_moment_against_mpz+TestStudentizedRange.test_moment_against_mp�s����z�*����,�	��s�m�X�c�]�H�S�M�9���[��
*�
*�
*�	7�	7��)�0�#�6�C�	7�	7�	7�	7�	7�	7�	7�	7�	7�	7�	7����	7�	7�	7�	7�	��Y�%�o�6�%�o�6�	8�	8�	8�	8�	8�	8s�A�A!�$A!c��d\}}ttjjdtj||f���}t
|dd��dS)N�r�rnrr�r)rr6r]r@rtr�r)r�rDr�r�s    r0�test_pdf_integrationz)TestStudentizedRange.test_pdf_integrationsG�����1��5�*�.��2�6��A��G�G�G����A���"�"�"�"�"r2c��d\}}tjddd���}tj�|||��dd�}tj�|||��}t
||��}t||d���dS)	Nr{rrnrz)�steprr�r�)rtr�r6r]rKr@rr)r�rDr�rG�y_cdf�	y_pdf_raw�y_pdf_cumulatives       r0�test_pdf_against_cdfz)TestStudentizedRange.test_pdf_against_cdf
s������1�

�I�a��$�'�'�'���'�+�+�A�q�!�4�4�Q�R�R�8���+�/�/��1�a�8�8�	�/�	�1�=�=��	�(�%�d�;�;�;�;�;�;r2�
r_case_resultc���|\}}}}tjd���5tj�|||��}ddd��n#1swxYwYt||��dS)Nr�r�)rtr�r6r]rKr)r�r�r_rDr��r_resr�s       r0�test_cdf_against_rz'TestStudentizedRange.test_cdf_against_rs���'���1�a��
�[��
*�
*�
*�	7�	7��)�-�-�a��A�6�6�C�	7�	7�	7�	7�	7�	7�	7�	7�	7�	7�	7����	7�	7�	7�	7���U�#�#�#�#�#s�"A�A�Ac��tjd���5tj�ddgddgddg��}ddd��n#1swxYwYt|jd	��tjtd
���5tj�dddggd���ddd��dS#1swxYwYdS)
Nr�r�rr>rer�rnr|rz...could not be broadcast...r�)rnr|r)
rtr�r6r]rSrrr�rr�rNs  r0�test_moment_vectorizationz.TestStudentizedRange.test_moment_vectorization%sh���[��
*�
*�
*�	H�	H��'�-�-�q�!�f�q�!�f�r�2�h�G�G�A�	H�	H�	H�	H�	H�	H�	H�	H�	H�	H�	H����	H�	H�	H�	H�	����&�&�&�
�]�:�-K�
L�
L�
L�	C�	C��#�)�)�!�a��V�\�\�\�B�B�B�	C�	C�	C�	C�	C�	C�	C�	C�	C�	C�	C�	C����	C�	C�	C�	C�	C�	Cs#�(A
�
A�A�&B8�8B<�?B<c��t��5}tjd���5|�t��t
j�gd���\}}}}ddd��n#1swxYwYddd��n#1swxYwYtt
j�	||����dS)Nr�r�r�)
rrtr�rcrr6r]r8r	�	_argcheck)r�r{rDrpr�s     r0�test_fitstart_validz(TestStudentizedRange.test_fitstart_valid5sF��
�
 �
 �	G�C���X�)F�)F�)F�	G�	G��J�J�)�*�*�*��1�;�;�I�I�I�F�F�K�A�r�1�a�	G�	G�	G�	G�	G�	G�	G�	G�	G�	G�	G����	G�	G�	G�	G�	G�	G�	G�	G�	G�	G�	G�	G�	G�	G�	G����	G�	G�	G�	G�	��'�1�1�!�R�8�8�9�9�9�9�9s5�B	�AA2�&B	�2A6	�6B	�9A6	�:B	�	B
�B
c��tj�ddtj��}tj�ddd��}t||dd���tj�ddtj��}tj�ddd��}t||dd���dS)Nr�rn韆r�r�)r6r]r@rtr�rrK)r�r��
res_finites   r0�test_infinite_dfz%TestStudentizedRange.test_infinite_df=s����%�)�)�!�R���8�8���,�0�0��B��>�>�
���Z�d��>�>�>�>��%�)�)�!�R���8�8���,�0�0��B��>�>�
���Z�d��>�>�>�>�>�>r2c�Z�tj�ddd��}tj�ddd��}tj�ddd��}ttt
||dd���t||dd���tj�ddd��}tj�ddd��}tj�ddd��}ttt
||dd���t||dd���dS)Nr�rnrr�i��r�r�)r6r]r@r��AssertionErrorrrK)r�r�r��
res_sanitys    r0�test_df_cutoffz#TestStudentizedRange.test_df_cutoffHs���%�)�)�!�R��8�8���,�0�0��B��>�>�
��,�0�0��B��>�>�
��n�o�s�J��d�	,�	,�	,�	,��
�J�T��E�E�E�E��%�)�)�!�R��8�8���,�0�0��B��>�>�
��,�0�0��B��>�>�
��n�o�s�J��d�	,�	,�	,�	,��
�J�T��E�E�E�E�E�Er2c��d\}}}tj�|||��}t|dd���|dksJ�dS)N)g��aRA@r�iSrr�r�)r6r]rgr)r�r_rDr�r?s     r0�
test_clippingz"TestStudentizedRange.test_clipping]sO��.���1�a��#�&�&�q�!�Q�/�/����1�5�)�)�)�)��A�v�v�v�v�v�vr2)1r#r$r%�q05�q01�q001rtrc�qs�psr��vs�ks�listr�r%r��r_datarar�r�r�rf�os�path�dirnamer�path_prefix�
relative_path�openrp�file�jsonr
�pregenerated_datar
rqru�xfail_on_32bitryr|r�r�r�r�r�r�r�r�r8r2r0rZrZ�sx������'�'�'�C�'�'�'�C�(�(�(�D�
����c�4�(�	)�	)�B�	�	�	�B�
�Q��B��R�V�	$�B�	���B��4���G�G�B��B�'�'��,�,�-�-�D�
	*�(�	
�A�r�v�(�)�	
�A�r�v�(�)�	
�B���)�*��F�:�:�:��[��:�:���:��'�/�/�(�+�+�K�<�M�	
��b�g�l�l�;�
�6�6��	<�	<�,��%�D�I�d�O�O��,�,�,�,�,�,�,�,�,�,�,����,�,�,�,��[���]�,=�j�,I�J�J�8�8�K�J�8��[���]�,=�j�,I�J�J�8�8�K�J�8��[���[��� M�N�N��[���]�,=�m�,L�M�M�8�8�N�M�O�N���8�#�#�#��[��
<�
<���
<��[���_�f�5�5�$�$�6�5�$��[���[��� M�N�N�C�C�O�N���C��[��:�:���:�	?�	?�	?�F�F�F�*����s�C1�1C5�8C5rZc�,�ttj�d��ddd���ttj�d��ddd���ttj�dd	d
���ddd���dS)
Ng���2��g����?rn�test_540_567)rrlg��2��gk
���?g�;��B.�?g��ͣh�?gH�Ŏ*�?r;g��.�y�?)rr6rErKr8r2r0r�r�gs�����
���}�5�5�7J� "�N�<�<�<�<���
���}�5�5�7K� "�N�<�<�<�<���
���}�.�-;�'�=�=�+� "�N�<�<�<�<�<�<r2c�F�tj�d���dS)Nr�r�)r6�_continuous_distns�	gamma_genr8r2r0�test_regression_ticket_1316r�ss$��
��&�&�G�&�4�4�4�4�4r2c�d�ttj�dd��dd��dS)Nr}r>rxr�r�r8r2r0�test_regression_ticket_1326r�ys*����
���s�A�.�.��R�8�8�8�8�8r2c	���tjddd��}tjd���5dddtjdgdgdgg��fD]{}tj�||��}t|dk�����ttj	|��������|tjdgdgd	gg��}tj�||��}ddd��n#1swxYwYttj	|�������t|d
dk�����t|ddk�����t|ddk�
����t|ddk�
����dS)
N��rrmr��rr}rrr�r|rrr>)rtrr�rr6r(r@r	rrt�any)rGrxr?s   r0�test_regression_tukey_lambdar�~s���	��D�#�s�#�#�A�	��H�	%�	%�	%�*�*���t�R�X��v��u�t�f�.E�%F�%F�G�	(�	(�C��!�%�%�a��-�-�A��Q�#�X�N�N�$�$�%�%�%��R�X�a�[�[�_�_�&�&�&�'�'�'�'��h������u�-�.�.����!�!�!�S�)�)��*�*�*�*�*�*�*�*�*�*�*����*�*�*�*��R�X�a�[�[�_�_�
�
������Q�q�T�S�[����� � � ��Q�q�T�S�[����� � � ��Q�q�T�S�[����� � � ��Q�q�T�S�[����� � � � � s�CD
�
D�Dzdocstrings strippedr�c��tdtjjv��tdtjjv��dS)Nzpdf(x, mu, loc=0, scale=1)zpmf(x,)r	r6rrrr8r2r0�test_regression_ticket_1421r��s9���(��
�0E�E�F�F�F��H��
�-�-�.�.�.�.�.r2c	�	�tjd���5ttjtj�dtj������ttjtj�dtj������ttjtj�	dtj������ttjtj�
dtj������ttjtj�dtj������ttjtj�dtj������ttjtj�
dtj������ttjtj�dtj������ttjtj�tjd������ttjtj�tjd������ttjtj�	tjd������ttjtj�
tjd������ttjtj�tjd������ttjtj�tjd������ttjtj�
tjd������ttjtj�tjd������ddd��dS#1swxYwYdS)Nr�r�rrx)rtr�r	rtr6r�r~r�rKr%rgr@r�r]rlrRrrzr8r2r0� test_nan_arguments_gh_issue_1362r��s)��	��X�	&�	&�	&�<�<���������2�6�2�2�3�3�4�4�4��������Q���/�/�0�0�1�1�1��������q�"�&�1�1�2�2�3�3�3��������A�r�v�.�.�/�/�0�0�0��������Q���/�/�0�0�1�1�1���������2�6�2�2�3�3�4�4�4��������Q���/�/�0�0�1�1�1��������Q���/�/�0�0�1�1�1������/�/����<�<�=�=�>�>�>������,�,�R�V�S�9�9�:�:�;�;�;������.�.�r�v�s�;�;�<�<�=�=�=������+�+�B�F�C�8�8�9�9�:�:�:������,�,�R�V�S�9�9�:�:�;�;�;������/�/����<�<�=�=�>�>�>������,�,�R�V�S�9�9�:�:�;�;�;������,�,�R�V�S�9�9�:�:�;�;�;�#<�<�<�<�<�<�<�<�<�<�<�<����<�<�<�<�<�<s�RR4�4R8�;R8c��tj�d��tjgd���}tj�|d|d|dd���}tjd�	��5tjtj�|d
�����}ddd��n#1swxYwYt||d���tjtj�|d
d�����}t||d���tjtj�|dd�����}t||d���tjtj�|dd�����}t||d���tj�d��d}d}tj
�|dd���}tjtj
�||�����}tj|tj||z
dz�����g��}t||d���dS)Ni.)r5r}rxrrr>ror�r�r�r}r?rrx)r�r<r5)r�r<r�r�r|re)
rtr�r�rr6r�r�r�r�rrEr�rD)�truerGr�r<r�ris      r0�test_frozen_fit_ticket_1536r��sL���I�N�N�4����
�8�O�O�O�$�$�D�
�
���$�q�'�4��7�D��G�#��>�>�A�	��H�	%�	%�	%�9�9���%�-�+�+�A�B�+�7�7�8�8��9�9�9�9�9�9�9�9�9�9�9����9�9�9�9����a�0�0�0�0�
�X�e�m�'�'��#�1�'�=�=�
>�
>�F����a�0�0�0�0�
�X�e�m�'�'��d��'�:�:�
;�
;�F����a�0�0�0�0�
�X�e�m�'�'��d��'�;�;�
<�
<�F����a�0�0�0�0��I�N�N�4����
�C��D�
�
���s�B�S��)�)�A�
�X�e�j�n�n�Q�T�n�2�2�
3�
3�F��x��r�w��4��!��'9�'9�';�';�<�<�=�>�>�H����!�4�4�4�4�4�4s�4C�C�Cc���tj�d��tj�d���}tj�|��}d}t||d���dS)Ni�	ror�)g
ףp=
�?g��ʡE�?rr)rtr�r�r6r�r�r�r)r�r�ris   r0�test_regression_ticket_1530r��sc���I�N�N�6����
�,�
�
��
�
$�
$�C�
�\�
�
�c�
"�
"�F��H����!�4�4�4�4�4�4r2c��tj�d��tj�d��}dD]K}tj�||z��\}}t||d���t|dd����LdS)Nr�r�)r�g�6�Arr�r�)rtr�r�r�r6r�r�r)rG�offsetr<r=s    r0�test_gh_pr_4806r��s����I�N�N�4����
�	������A�'�.�.���\�%�%�a�&�j�1�1�
��U���V�#�.�.�.�.���s��-�-�-�-�-�.�.r2c�t�tj�dd���}dtjdzdzddg}t	||d���tj�d	d���}gd
�}t	||d���tj�dd���}gd�}t	||d���dS)
Nrrrr>r�rrnrg
ףp=
	@)rgz5��v��?rg��RT��g�Q���?)rgN�g��@rg��JFٻ��)r6r(rtrBr)r�ris  r0�"test_tukeylambda_stats_ticket_1545r��s���
�	�	 �	 ��F�	 �	3�	3�B��2�5�!�8�A�:�q�#�&�H���H�b�1�1�1�1�	�	�	 �	 ��v�	 �	6�	6�B�C�C�C�H���H�b�1�1�1�1�	�	�	 �	 ��v�	 �	6�	6�B�C�C�C�H���H�b�1�1�1�1�1�1r2c��ttjtj�dd������dS)Nr�r)r	rtrr6rrzr8r2r0�test_poisson_logpmf_ticket_1436r��s2���B�K��
�,�,�T�3�7�7�8�8�9�9�9�9�9r2c��ddg}|D]6\}}tj�|d���}t||���7dS)a�Test the powerlaw stats function.

    This unit test is also a regression test for ticket 1548.

    The exact values are:
    mean:
        mu = a / (a + 1)
    variance:
        sigma**2 = a / ((a + 2) * (a + 1) ** 2)
    skewness:
        One formula (see https://en.wikipedia.org/wiki/Skewness) is
            gamma_1 = (E[X**3] - 3*mu*E[X**2] + 2*mu**3) / sigma**3
        A short calculation shows that E[X**k] is a / (a + k), so gamma_1
        can be implemented as
            n = a/(a+3) - 3*(a/(a+1))*a/(a+2) + 2*(a/(a+1))**3
            d = sqrt(a/((a+2)*(a+1)**2)) ** 3
            gamma_1 = n/d
        Either by simplifying, or by a direct calculation of mu_3 / sigma**3,
        one gets the more concise formula:
            gamma_1 = -2.0 * ((a - 1) / (a + 3)) * sqrt((a + 2) / a)
    kurtosis: (See https://en.wikipedia.org/wiki/Kurtosis)
        The excess kurtosis is
            gamma_2 = mu_4 / sigma**4 - 3
        A bit of calculus and algebra (sympy helps) shows that
            mu_4 = 3*a*(3*a**2 - a + 2) / ((a+1)**4 * (a+2) * (a+3) * (a+4))
        so
            gamma_2 = 3*(3*a**2 - a + 2) * (a+2) / (a*(a+3)*(a+4)) - 3
        which can be rearranged to
            gamma_2 = 6 * (a**3 - a**2 - 6*a + 2) / (a*(a+3)*(a+4))
    )r)rxr�r}r�)r|)gUUUUUU�?r+g^c�Q�g333333�rrN)r6r�r)�casesr.�
exact_mvskrs    r0�test_powerlaw_statsr��s[��>,�
=�
?�E��4�4�
��:��~�#�#�A�v�#�6�6��!�$�
�3�3�3�3�4�4r2c�f�tj�dd��}t|d��dS)Nrrr})r6r�r�r�r?s r0�test_powerlaw_edger�s/��
����a��#�#�A���C�����r2c�x�tj�dd��}t|d��tj�dgd���}t|t
jddg��tj�dgd���}t|t
jdt
jg��dS)Nrrr})r5rr�r)r6r�r�rr@rtr�r�s r0�test_exponpow_edger�!s���
����a��#�#�A���C����	����1�.�.�.�/�/�A���R�V�S�#�&�'�'�'�
����a�!1�!1�!1�2�2�A���R�V�S�2�6�'�*�+�+�+�+�+r2c�h�tj�ddd��}t|d��dS)Nrrr)r6�gengammar@rr�s r0�test_gengamma_edger�-s1��
����1�a��#�#�A���C�����r2c��tj�ddd��}|dksJ�tj�ddd��}|tjksJ�dS)Nrrr�r})r6r�r@r�rtr�)r?r�s  r0�!test_gengamma_endpoint_with_neg_cr�3sV��
����1�a��$�$�A���8�8�8�8��>� � ��A�r�*�*�D��B�F�7�?�?�?�?�?�?r2c���tj�ddd��}t|d��tj�ddd��}t|d��dS)Nr�rrg����>rng�q�q�?)r6r�rSrr�s r0�test_gengamma_munpr�:sZ��
����R��b�)�)�A���:�&�&�&�
����R��R�(�(�A���6�"�"�"�"�"r2c��tjgd���}tjd���5t��5}|�t
d��|�td��tj�	|��ddd��n#1swxYwYddd��dS#1swxYwYdS)N)0g>���X*ȿgPix�#�?g�=��?g&�^���?g�o����Ͽg�뵏�˿gO_��ZQ�?g����\�?g�~��?g)�/��?g�R�ܶ?g�핚�οgH߃[g�?gS"���gm_� пg���7�k�?g�#B�F�?gKU=
�?g�T�?gZg[Qs��?g���2���?g6Hwƭ8�?g���*-
�?g�a��]��?g���'�?g��޻?g?�J_��?g���~sw�?g0FfͿg������g"��@�ӿg���?%:˿g&�z�M��?g�?G����g��	F�?g��~��,��g�� �H��g���W�2ҿg���Tzy�?gw�m�ǿg��@��g��W�=�ֿg	�'t��?g".��v1��g��l����g�.H�ꮿg,�����?g*�����˿r�r�z:The maximum number of subdivisions .50. has been achieved.z-floating point number truncated to an integer)
rtrr�rrcrrCr6rr�)r)r{s  r0�test_ksone_fit_freezer�CsK��
��		/�		/�		/�
	0�
	0�A�
��X�	&�	&�	&���
�
 �
 �	�C��J�J�)�#�
$�
$�
$�
�J�J�~�F�
H�
H�
H��K�O�O�A����
	�	�	�	�	�	�	�	�	�	�	����	�	�	�	���������������������s5�B5�AB�B5�B!	�!B5�$B!	�%B5�5B9�<B9c	�r�tjttddd������}gd�}t	tj���|��|d���t	tj���|dz��j|d���t
j�|dz��d	zj	}tj
t
j�|��t
j�|��z
��}t	||d	���dS)
Nrr[re)g�D��B.�g�8_�$�g%`O��A�gjhwH�R�gWMB�@v`�g�7VY}i�gۦx\�Ar�g���	�x�g����#��g��/�d��g�3��$��g�e��gd���)��gN��q{3��gيGǓ��gr���
4��g�37�'
��g�f��F��g���?dJ��gض�����g�2�,�
��g�m�%����g�Ӆ��J��gz;�p���g�ɿ{��g�k�2����g�ij<�%��giVS�Ͷ�gBR4����g
Q�/�M��rQr�y�+����=y�����|�=r�)rtr�r�r�rr6rEr~�real�imagr�r�)rGri�deriv�deriv_expecteds    r0�test_norm_logcdfr�[s
��
��D��q�#�q�)�)�*�*�	+�	+�+�A�0�0�0�H��E�J�L�L�'�'��*�*�H�4�@�@�@�@��E�J�L�L�'�'��F�
�3�3�8�(��N�N�N�N��Z�
�
�q�6�z�
*�
*�5�
0�6�E��V�E�J�-�-�a�0�0�5�:�3D�3D�Q�3G�3G�G�H�H�N��E�>��6�6�6�6�6�6r2c�"�tjgd���}tjgd���}tj�|��}t||d���tj�|��}t||d���dS)N)r�rrxrlrzr
)g�~��V1�?g�|��N�?g��ba"�?g܄4�ťY?r4g$�%�+I�r�r�r�)rtrr6r�rKrr])rGrirr`s    r0�test_levy_cdf_ppfr�ts���
��3�3�3�4�4�A��x�1�1�1�2�2�H�	�
���q���A��A�x�e�,�,�,�,�
�����	!�	!�B��B���&�&�&�&�&�&r2c��tjgd���}tjgd���}tj�|��}t||d���dS)N)�4&�kCg�(,*� Eg�M�raB3Gg�d~�QJ)g�ep��[>g}XE�Q=g�ѭ�EG<g?_�%~��:r,r�)rtrr6r�rgr�rGrirs   r0�test_levy_sfr��sh��
��)�)�)�*�*�A��x�0�0�0�1�1�H�	�
�
�
�a���A��A�x�e�,�,�,�,�,�,r2zp, expected_isf))r+gj�1e�2H)rQg�����6C)g�?g�nE�@)r g��!^x1�?)rg�Z����?)g����?g:}!Nu�?c�h�tj�|��}t||d���dS)Nr�r�)r6r�rlr)r?�expected_isfrGs   r0�
test_levy_isfr��s2��	�
���q���A��A�|�%�0�0�0�0�0�0r2c��tjgd���}tjgd���}tj�|��}t||d���dS)N)g����Mb��g{�G�z��g{�G�zt�g�~j�t�X�)g#�]���<g�O�ul2;gޢ��ا6g̅7=��!r�r�)rtrr6r�rgrr�s   r0�test_levy_l_sfr��sh��
��1�1�1�2�2�A��x�1�1�1�2�2�H�	�������A��A�x�e�,�,�,�,�,�,r2c���tjgd���}tj�|��}tj�|��}t
||d���dS)N)g [n��<r5r�g�vIh�%,=r�)rtrr6r�rlrgr)r?rGr_s   r0�test_levy_l_isfr��s[��
��&�&�&�'�'�A�
�������A�
�������A��A�q�u�%�%�%�%�%�%r2c���ttj�dddd��d��ttj�dddd��d��ttj�dddd��d��ttj�d	d
d
d��d��ttj�dd
d
d��d��dS)
Nr
i��i��i�)gc@g�h@g=
ףp=�?��G�z�?)g c@g�h@�{�G�z�?ror�r)rr6r�rr]r8r2r0�test_hypergeom_interval_1802r��s������)�)�#�v�u�c�B�B��!�!�!����)�)�$���s�C�C��!�!�!����)�)�#�v�u�c�B�B��!�!�!����$�$�T�3��Q�7�7��;�;�;����$�$�Q��S�!�4�4�a�8�8�8�8�8r2c��tj�d��tjddd���}t	t
tjj|ddd�	��t	t
tjj|ddd
d�	��t	t
tjj|ddd
d��t	t
tjj|dddd���t	t
tjj	d
ddd���t	t
tjj
|d
ddd���t	t
tjj|d
ddd���t	t
tjjd
ddd���t	t
tjjd
ddd���t	t
tjj
|d
ddd���tj�|dd��tj�|ddd
��tj�d
d��tj�d
dd
��tj�d
dd
d��tj�	d
dd
d��tj�
tj�	d
d���d
��tj�|dd�	��t	t
tjj|ddd
��t	t
tjj|ddd
�	��t	t
tjj|dd�	��t	t
tjj|dd
dd�	��t	t
tjj|dd
ddd��t	t
tjj|dd
ddd�	��t	t
tjj|dd
dddd��	�	tj�|dd
ddd��dS)Nr�rlrSr���numr>r�rrrerxr;r|r�rsr�rrD)rtr�r�rr�r�r6rr@r�rKr]rXr�rrrRr�r�r�s r0�test_distribution_too_many_argsr��sX���I�N�N�4����	��C��!�$�$�$�A��)�U�[�_�a��A�3�?�?�?�?��)�U�[�_�a��A�q�c�B�B�B�B��)�U�[�_�a��A�q�!�<�<�<��)�U�[�_�a��A�3�c�J�J�J�J��)�U�[�_�b�!��C�H�H�H�H��)�U�[�_�a��Q�C�s�K�K�K�K��)�U�[�_�a��Q�C�s�K�K�K�K��)�U�[�.��A�3�c�J�J�J�J��)�U�[�0�"�a�S��L�L�L�L��)�U�[�_�a��Q�C�s�K�K�K�K�
�K�O�O�A�q�!����	�K�O�O�A�q�!�Q����	�K���b�!����	�K���b�!�Q����	�K���b�!�Q��%�%�%�	�K�O�O�B��1�a� � � �	�K�O�O�E�K�O�O�B�Q�O�/�/��4�4�4�
�J�N�N�1�a�Q�N�����)�U�Z�^�Q��1�a�8�8�8��)�U�Z�^�Q��1�!�<�<�<�<��)�U�[�_�a���<�<�<�<��)�U�_�0�!�Q��1�#�F�F�F�F��)�U�_�0�!�Q��1�c�3�G�G�G��)�U�Y�]�A�q�!�Q��s�C�C�C�C��)�U�Y�]�A�q�!�Q��3�c�J�J�J�J�	�I�M�M�!�Q��1�a��%�%�%�%�%r2r�c��tj�tjddd��dd��}tj�tjddd��dd��}t
||dd���dS)	Nr�r�r�r>g���Y�Z@r
rr0)r6�ncx2rKrtr��_cdfvecrr�s  r0�test_ncx2_tails_ticket_955r��sm��	�
���r�y��R��-�-�q�.�A�A�A�
�
���2�9�R��S�1�1�1�n�E�E�A��A�q�t�!�,�,�,�,�,�,r2c
�*�tj��5tjdt��t	t
j�dtj	dd��d��d��t
j�
dtj	dd��d��}ddd��n#1swxYwYttj|���
����tj��5tjdt��t	t
j�ddd	��d��tt
j�
ddd	��d
��ddd��dS#1swxYwYdS)Nr<riTi^r>rr�r�rgs���q6��)r@rArBrCrr6r�r@rtr�r�r	r�rr)�logvals r0�test_ncx2_tails_pdfr�s���
�	 �	"�	"�>�>���g�~�6�6�6��U�Z�^�^�A�r�y��c�':�':�A�>�>��B�B�B���"�"�1�b�i��S�&9�&9�1�=�=��>�>�>�>�>�>�>�>�>�>�>����>�>�>�>�
�B�K����#�#�%�%�&�&�&�
�	 �	"�	"�M�M���g�~�6�6�6��U�Z�^�^�E�1�b�1�1�1�5�5�5���
�)�)�%��B�7�7�9K�L�L�L�M�M�M�M�M�M�M�M�M�M�M�M����M�M�M�M�M�Ms%�BB1�1B5�8B5�A9F�F�Fzmethod, expectedrKg����
u%>g	;�Q�=r@g<��x���>gR�z�\R>r�gʎK$��/�g��ԆX�1�r]gf��H�u@ge�:�@c�z�ttj|��dddgd���}t||d���dS)Nrlrrern)r�rpr�r�)r�r6r�r)r@rir�s   r0�test_ncx2_zero_ncr�sE�� )�W�U�Z��
(�
(��!�Q��B�
?�
?�
?�F��F�H�5�1�1�1�1�1�1r2c��tj�ddd���}tj�dd���}t	||d���dS)Nrnrr)rpr�r�)rpr�r�r�)r6r�r�r�r)r�ris  r0�test_ncx2_zero_nc_rvsr�(sO���Z�^�^�r�a�a�^�
8�
8�F��z�~�~��!�~�4�4�H��F�H�5�1�1�1�1�1�1r2c��dtjdd��z}ttj�dd|���d��dS)Nrnr�r�r�rpr�r)rtr�rr6r�rK)r�s r0�test_ncx2_gh12731r	0sD��
�R�Y�q�"�
�
�	�B��������b��1�1�1�5�5�5�5�5r2c��tjgd���}d\}}tj�|||���}gd�}t||d���dS)N)
gю]�	�@gj���%@gb	�y/7@g����H@g�I��Z@g��X�l@g'/2?8@g�ݰm�А@gU���@gy]�`'��@g����@g*��\��@g�4�8�c�@)r�g�h�?8@r	)
rrrrrg�������?gy[q�D�?r}r}r}r}r}r}rqr�)rtrr6r�rgr)rG�nurxrg�sf_expecteds     r0�test_ncx2_gh8665r	7sv��	��"�"�"�	#�	#�A�
%�G�B��	����q�R�C��	(�	(�B�'�'�'�K�
�B��%�0�0�0�0�0�0r2c	��d}d}tjtj�d||��tj�d||��d���}tj�|||��}tj�|||ztjd|zd|zz����}t||d	�
��dS)Ni,i�r
rr�r�r>rer�r�)	rtrr6r�r]r@rEr�r)rpr�rG�ncx2_pdf�gauss_approxs     r0�test_ncx2_gh11777r		Ss���

�B�	
�B�
��E�J�N�N�5�"�b�1�1��J�N�N�5�"�b�1�1�u�	>�	>�	>�A��z�~�~�a��R�(�(�H��:�>�>�!�R�"�W�b�g�a�"�f�q�2�v�o�.F�.F�G�G�L��H�l��6�6�6�6�6�6r2c�x�tjdd���}t|�d��d��dS)NrrrD)r6�foldnormrrK)r�s r0�test_foldnorm_zeror	bs6��	����	#�	#�	#�B��������A�����r2c��tj�gd�dd��}tj�ddgdd��}td�|D����}t||��tj�gd���}tj�ddg��}td	�|D����}t||��dS)
N)r}rxrrrxrc3�RK�|]"}tjtj|fV��#dSrr�rtrr�res  r0rz-test_stats_shapes_argcheck.<locals>.<genexpr>ms/����8�8�q�"�%����	�*�8�8�8�8�8�8r2)r>rr�r>rc3�RK�|]"}tj|tjfV��#dSrrr	res  r0rz-test_stats_shapes_argcheck.<locals>.<genexpr>ss/����8�8�q�"�%��2�6�	�*�8�8�8�8�8�8r2)r6r��tuplerr�)�mv3�mv2�
mv2_augmenteds   r0�test_stats_shapes_argcheckr	hs����.�
�
�����3�
7�
7�C�
�.�
�
��S�z�1�c�
2�
2�C��8�8�C�8�8�8�8�8�M����$�$�$��-�
�
�l�l�l�
+�
+�C�
�-�
�
�q�#�h�
'�
'�C��8�8�C�8�8�8�8�8�M����$�$�$�$�$r2c��eZdZd�ZdS)�
_distr_genc��dS�Nr�r8�r�rGr.s   r0�_pdfz_distr_gen._pdf~s���rr2N�r#r$r%r	r8r2r0r	r	}s#����������r2r	c��eZdZd�ZdS)�_distr2_genc��d|z|zSr	r8r	s   r0�_cdfz_distr2_gen._cdf�����A�v��z�r2N)r#r$r%r 	r8r2r0r	r	�s#����������r2r	c��eZdZd�Zd�ZdS)�_distr3_genc��||zSrrr8�r�rGr.r/s    r0r	z_distr3_gen._pdf�s���1�u�r2c��d|z|zSr	r8r	s   r0r 	z_distr3_gen._cdf�s���A�v��z�r2N�r#r$r%r	r 	r8r2r0r#	r#	�s2�������������r2r#	c��eZdZd�Zd�ZdS)�_distr6_genc��||z|zSrrr8r%	s    r0r	z_distr6_gen._pdf�s����s�Q�w�r2c��d|z|zSr	r8r%	s    r0r 	z_distr6_gen._cdf�r!	r2Nr'	r8r2r0r)	r)	�s2�������������r2r)	c�t�eZdZd�Zd�Zd�Zd�Zd�Zd�Zd�Z	d�Z
d	�Zd
�Zd�Z
d�Zd
�Zd�Zd�Zd�Zd�ZdS)�TestSubclassingExplicitShapesc�r�tdd���}t|�dd���d��dS)N�dummyr.�r�r�rrBr��r	rr@�r��dummy_distrs  r0�test_correct_shapesz1TestSubclassingExplicitShapes.test_correct_shapes�s:�� �g�c�:�:�:���[�_�_�Q�!�_�,�,�b�1�1�1�1�1r2c�v�tdd���}tt|jdfit	d�����dS)Nr/	�Ar0	rrB)r	r�r�r@r�r2	s  r0�test_wrong_shapes_1z1TestSubclassingExplicitShapes.test_wrong_shapes_1�s>�� �g�c�:�:�:���i���!�A�A�t�a�y�y�y�A�A�A�A�Ar2c�~�tdd���}tddd���}tt|jdfi|��dS)Nr/	za, b, cr0	rr>r�r�)r	r�r�r�r@)r�r3	r�s   r0�test_wrong_shapes_2z1TestSubclassingExplicitShapes.test_wrong_shapes_2�sJ�� �g�i�@�@�@���Q�!�q�!�!�!���i���!�;�;�s�;�;�;�;�;r2c�X�tdd���}tttfi|��dS)Nr/	r�r0	)r�r�r�r	�r�r�s  r0�test_shapes_stringz0TestSubclassingExplicitShapes.test_shapes_string�s2�����+�+�+���i��3�3�s�3�3�3�3�3r2c�X�tdd���}tttfi|��dS)Nr/	z(!)r0	�r�r��SyntaxErrorr	r;	s  r0�test_shapes_identifiers_1z7TestSubclassingExplicitShapes.test_shapes_identifiers_1�s2�����.�.�.���k�:�5�5��5�5�5�5�5r2c�X�tdd���}tttfi|��dS)Nr/	�4chanr0	r>	r;	s  r0�test_shapes_identifiers_2z7TestSubclassingExplicitShapes.test_shapes_identifiers_2��2�����0�0�0���k�:�5�5��5�5�5�5�5r2c�X�tdd���}tttfi|��dS)Nr/	zm(fti)r0	r>	r;	s  r0�test_shapes_identifiers_3z7TestSubclassingExplicitShapes.test_shapes_identifiers_3�s2�����1�1�1���k�:�5�5��5�5�5�5�5r2c�X�tdd���}tttfi|��dS)Nr/	za=2r0	r>	r;	s  r0�"test_shapes_identifiers_nodefaultsz@TestSubclassingExplicitShapes.test_shapes_identifiers_nodefaults�s2�����.�.�.���k�:�5�5��5�5�5�5�5r2c�X�tdd���}tttfi|��dS)Nr/	z*argsr0	r>	r;	s  r0�test_shapes_argsz.TestSubclassingExplicitShapes.test_shapes_args�rD	r2c�X�tdd���}tttfi|��dS)Nr/	z**kwargsr0	r>	r;	s  r0�test_shapes_kwargsz0TestSubclassingExplicitShapes.test_shapes_kwargs�s2����
�3�3�3���k�:�5�5��5�5�5�5�5r2c�X�tdd���}tttfi|��dS)Nr/	za, b, c, lambdar0	r>	r;	s  r0�test_shapes_keywordsz2TestSubclassingExplicitShapes.test_shapes_keywords�s3����(9�:�:�:���k�:�5�5��5�5�5�5�5r2c���Gd�dtj��}|d���}t|�dd���tj�d��dz��dS)Nc��eZdZd�ZdS)�FTestSubclassingExplicitShapes.test_shapes_signature.<locals>._dist_genc�F�tj�|��|zSrr�r6rEr	r	s   r0r	zKTestSubclassingExplicitShapes.test_shapes_signature.<locals>._dist_gen._pdf�����z���q�)�)�A�-�-r2Nr	r8r2r0�	_dist_genrQ	��#������
.�
.�
.�
.�
.r2rU	r.�r�rxr>rB�r6rVrr@rE�r�rU	rRs   r0�test_shapes_signaturez3TestSubclassingExplicitShapes.test_shapes_signature�s|��	.�	.�	.�	.�	.��+�	.�	.�	.��y��$�$�$���T�X�X�c�Q�X�'�'������)<�)<�Q�)>�?�?�?�?�?r2c��Gd�dtj��}|d���}tt|jdfitdd�����dS)	Nc��eZdZd�ZdS)�STestSubclassingExplicitShapes.test_shapes_signature_inconsistent.<locals>._dist_genc�F�tj�|��|zSrrrS	r	s   r0r	zXTestSubclassingExplicitShapes.test_shapes_signature_inconsistent.<locals>._dist_gen._pdf�rT	r2Nr	r8r2r0rU	r]	�rV	r2rU	r�rW	rxrr>r�)r6rVr�r�r@r�rY	s   r0�"test_shapes_signature_inconsistentz@TestSubclassingExplicitShapes.test_shapes_signature_inconsistent�so��	.�	.�	.�	.�	.��+�	.�	.�	.��y��'�'�'���i���3�A�A�$��a�.�.�.�A�A�A�A�Ar2c��Gd�dtj��}|d���}t|�dd���tj�d��dz��t|�dd��tj�d��dz��tt|jdfitd�����dS)	Nc��eZdZd�ZdS)�?TestSubclassingExplicitShapes.test_star_args.<locals>._dist_genc�V�|d}tj�|��|zS�NrrS	)r�rGr��extra_kwargs    r0r	zDTestSubclassingExplicitShapes.test_star_args.<locals>._dist_gen._pdf�s$��"�1�g���z���q�)�)�K�7�7r2Nr	r8r2r0rU	rb	�s#������
8�
8�
8�
8�
8r2rU	re	rW	rx�!�re	)�xxx)r6rVrr@rEr�r�r�rY	s   r0�test_star_argsz,TestSubclassingExplicitShapes.test_star_args�s���	8�	8�	8�	8�	8��+�	8�	8�	8�
�y�
�.�.�.���T�X�X�c�r�X�2�2�E�J�N�N�3�4G�4G��4J�K�K�K��T�X�X�c�2�&�&��
���s�(;�(;�B�(>�?�?�?��i���3�?�?�$�2�,�,�,�?�?�?�?�?r2c�p�Gd�dtj��}|d���}t|�ddd���tj�d��dzdz��t|�ddd��tj�d��dzdz��dS)	Nc��eZdZd�ZdS)�ATestSubclassingExplicitShapes.test_star_args_2.<locals>._dist_genc�\�|d}tj�|��|z|zSrd	rS	)r�rGr�r�re	s     r0r	zFTestSubclassingExplicitShapes.test_star_args_2.<locals>._dist_gen._pdf�s)��"�1�g���z���q�)�)�K�7�&�@�@r2Nr	r8r2r0rU	rl	�s(������
A�
A�
A�
A�
Ar2rU	zoffset, extra_kwargrW	rx�orf	)r�re	rX	rY	s   r0�test_star_args_2z.TestSubclassingExplicitShapes.test_star_args_2�s���	A�	A�	A�	A�	A��+�	A�	A�	A�
�y� 5�6�6�6���T�X�X�c�#�2�X�>�>��Z�^�^�C�(�(��+�c�1�	3�	3�	3��T�X�X�c�3��+�+��Z�^�^�C�(�(��+�c�1�	3�	3�	3�	3�	3r2c���Gd�dtj��}|d���}t|�dd���tj�d����dS)Nc��eZdZd�ZdS)�BTestSubclassingExplicitShapes.test_extra_kwarg.<locals>._distr_genc�r�|�dd��}tj�|��|zS)Nre	r)�popr6rEr	)r�rGr��kwargsre	s     r0r	zGTestSubclassingExplicitShapes.test_extra_kwarg.<locals>._distr_gen._pdf�s0��%�j�j���:�:���z���q�)�)�K�7�7r2Nr	r8r2r0r	rr	�s#������
8�
8�
8�
8�
8r2r	re	rW	rr�rg	rX	)r�r	rRs   r0�test_extra_kwargz.TestSubclassingExplicitShapes.test_extra_kwarg�sw��	8�	8�	8�	8�	8��,�	8�	8�	8��z��/�/�/���T�X�X�a�Q�X�/�/������1B�1B�C�C�C�C�Cr2c���Gd�dtj��}|d���}t|�d��tj�d����dS)Nc��eZdZd�ZdS)�DTestSubclassingExplicitShapes.shapes_empty_string.<locals>._dist_genc�@�tj�|��Srr)r6rEr@rms  r0r	zITestSubclassingExplicitShapes.shapes_empty_string.<locals>._dist_gen._pdfs���z�~�~�a�(�(�(r2Nr	r8r2r0rU	ry	
s#������
)�
)�
)�
)�
)r2rU	�rW	rxrX	rY	s   r0�shapes_empty_stringz1TestSubclassingExplicitShapes.shapes_empty_stringsq��	)�	)�	)�	)�	)��+�	)�	)�	)��y��#�#�#���T�X�X�c�]�]�E�J�N�N�3�$7�$7�8�8�8�8�8r2N)r#r$r%r4	r7	r9	r<	r@	rC	rF	rH	rJ	rL	rN	rZ	r_	ri	ro	rv	r|	r8r2r0r-	r-	�s"������2�2�2�B�B�B�<�<�<�
4�4�4�
6�6�6�
6�6�6�6�6�6�6�6�6�6�6�6�6�6�6�6�6�6�
@�@�@�B�B�B�@�@�@�3�3�3�D�D�D�9�9�9�9�9r2r-	c���eZdZd�Zd�Zej�ed���d���Z	ej�ed���d���Z
d�Zd�Zd	�Z
d
�ZdS)�TestSubclassingNoShapesc�p�td���}t|�dd���d��dS)Nr/	r�rrBr�r1	r2	s  r0�test_only__pdfz&TestSubclassingNoShapes.test_only__pdfs8�� �g�.�.�.���[�_�_�Q�!�_�,�,�b�1�1�1�1�1r2c�p�td���}t|�dd���d��dS)Nr/	r�rrB)r	rr@r2	s  r0�test_only__cdfz&TestSubclassingNoShapes.test_only__cdfs8��!�w�/�/�/���K�O�O�A��O�3�3�Q�7�7�7�7�7r2�docstring strippedr�c���td���}t|jd��t|jd��t	jd|j��}tt|��dk��dS)Nr/	r�rr.zlogpdf\(x, a, loc=0, scale=1\))	r	r�numargsr��re�findallrrr	r��r�r3	r�s   r0�test_signature_inspectionz1TestSubclassingNoShapes.test_signature_inspectionsr��!�g�.�.�.���[�(�!�,�,�,��[�'��-�-�-��j�:�$�,�.�.����C���A�
�����r2c���td���}t|jd��t|jd��t	jd|j��}tt|��dk��dS)Nr/	r�r>r�z!logpdf\(x, a, b, loc=0, scale=1\)r)	r)	rr�	r�r�	r�	rrr	r�r�	s   r0�test_signature_inspection_2argsz7TestSubclassingNoShapes.test_signature_inspection_2args)sr��"�w�/�/�/���[�(�!�,�,�,��[�'��0�0�0��j�=�$�,�.�.����C���A�
�����r2c�>�tttd���dS)Nr/	r�)r�r�r#	r�s r0�0test_signature_inspection_2args_incorrect_shapeszHTestSubclassingNoShapes.test_signature_inspection_2args_incorrect_shapes3s���i��7�;�;�;�;�;�;r2c�x�Gd�dtj��}tt|fit	d�����dS)Nc��eZdZdd�ZdS)�>TestSubclassingNoShapes.test_defaults_raise.<locals>._dist_genr�c��dSr	r8r	s   r0r	zCTestSubclassingNoShapes.test_defaults_raise.<locals>._dist_gen._pdf:����rr2N)r�r	r8r2r0rU	r�	9s(������
�
�
�
�
�
r2rU	r/	r��r6rVr�r�r��r�rU	s  r0�test_defaults_raisez+TestSubclassingNoShapes.test_defaults_raise7�Z��	�	�	�	�	��+�	�	�	�	�i��A�A�d��.@�.@�.@�A�A�A�A�Ar2c�x�Gd�dtj��}tt|fit	d�����dS)Nc��eZdZd�ZdS)�>TestSubclassingNoShapes.test_starargs_raise.<locals>._dist_genc��dSr	r8)r�rGr.r�s    r0r	zCTestSubclassingNoShapes.test_starargs_raise.<locals>._dist_gen._pdfAr�	r2Nr	r8r2r0rU	r�	@�#������
�
�
�
�
r2rU	r/	r�r�	r�	s  r0�test_starargs_raisez+TestSubclassingNoShapes.test_starargs_raise>r�	r2c�x�Gd�dtj��}tt|fit	d�����dS)Nc��eZdZd�ZdS)�<TestSubclassingNoShapes.test_kwargs_raise.<locals>._dist_genc��dSr	r8)r�rGr.ru	s    r0r	zATestSubclassingNoShapes.test_kwargs_raise.<locals>._dist_gen._pdfHr�	r2Nr	r8r2r0rU	r�	Gr�	r2rU	r/	r�r�	r�	s  r0�test_kwargs_raisez)TestSubclassingNoShapes.test_kwargs_raiseEr�	r2N)r#r$r%r�	r�	r�r�r��DOCSTRINGS_STRIPPEDr�	r�	r�	r�	r�	r�	r8r2r0r~	r~	s�������2�2�2�8�8�8�
�[���+�4H��I�I���J�I���[���+�4H��I�I���J�I��<�<�<�B�B�B�B�B�B�B�B�B�B�Br2r~	r�	c��gd�}tjD]k}tt|��}t|tjtjf��r.|D]+}t
tj||j	��du���,�ldS)N)z,\s*,z\(\s*,z^\s*:)
r6rbr�r
rUrVr	r�	�searchrr)�badones�distnamerR�regexs    r0r�r�Ms���-�-�-�G��M�@�@���u�h�'�'���d�U�.��0C�D�E�E�	@� �
@�
@����	�%���6�6�$�>�?�?�?�?��	@�@r2c���ttj�tjdd��d��ttj�tjdd��d��dS)Nrnr|rr�rlr)rr6r�rgrtr�r�r 	r8r2r0�test_infinite_inputr�	WsR����
�(�(����R�8�8�!�<�<�<���
������3�7�7��;�;�;�;�;r2c��tj�tj�dd��d��}t	|d��dS�Ng0��.�++r)r6�lomaxr]rKrr�s r0�test_lomax_accuracyr�	\s=��
����������2�2�A�6�6�A��A�v�����r2c��tj�tj�dd��d��}t	|d��dSr�	)r6�gompertzr]rKrr�s r0�test_gompertz_accuracyr�	bsA��
����5�>�-�-�f�a�8�8�!�<�<�A��A�v�����r2c��tj�tj�dd��d��}t	|d��dSr�	)r6�
truncexponr]rKrr�s r0�test_truncexpon_accuracyr�	hsC��
����U�-�1�1�&�!�<�<�a�@�@�A��A�v�����r2c��tj�tj�dd��d��}t	|dd���dS)Nr�rr�rr)r6r�rlrgrr�s r0�test_rayleigh_accuracyr�	nsF��
����5�>�,�,�Q��2�2�A�6�6�A���3��+�+�+�+�+�+r2c���tjd���5}tjd��tj�dd��tj�dd��tj�dd��tj�tj
d��t|��}t|d��ddd��dS#1swxYwYdS)zregression test for gh-6219T)rs�alwaysrxrr}N)
r@rArBr6r�rKr@r]r�rtr�r�r)r�number_of_warnings_throwns  r0� test_genextreme_give_no_warningsr�	ts��
�	 ��	-�	-�	-�3����h�'�'�'�
����R��#�#�#�
����R��#�#�#�
����R��#�#�#�
��������-�-�-�$'��F�F�!��.��2�2�2�3�3�3�3�3�3�3�3�3�3�3�3����3�3�3�3�3�3s�B?C"�"C&�)C&c��d}tj�d��}t|d|zdzd���tj�d��}t||dzd���tj�d��}t	|d��tj�d	d
���}t||dztjd
��zdzd���tj�d
��}t|d
|zdzd���tj�d��}t|d|zdzd���dS)Ng�o��x�?rrr>rr,r�rrr�rnrDr�r�r�r|)r6r�rXrrrtrU)�euler_gammar9s  r0�test_genextreme_entropyr�	�sD��$�K�
�� � ��&�&�A��A�q��}�q�(�u�5�5�5�5�
�� � ��#�#�A��A�{�Q��U�3�3�3�3�
�� � ��%�%�A���A����
�� � ��R� �0�0�A��A�{�1�}�r�v�b�z�z�1�A�5�E�B�B�B�B�
�� � ��$�$�A��A�r�+�~��)��6�6�6�6�
�� � ��%�%�A��A�r�+�~��)��6�6�6�6�6�6r2c�R�d}tj�|d��}t|d��tj�|d��}t||��d}tj�|d��}t|d��tj�|d��}t||��d}tj�|d��}t|d��tj�|d��}t||��dS)	Ng�חAg��g'��b�%4g�Q���@rg��ؗ�Ҝ;rg��.��l6?)r6r�rgrrl)rGrFr7s   r0�test_genextreme_sf_isfr�	�s��2	�A�
����A�v�&�&�A��A�-�.�.�.�	�	�	�	�a��	(�	(�B��B������A�
����A�u�%�%�A��A�(�)�)�)�	�	�	�	�a��	'�	'�B��B������A�
����A�q�!�!�A��A�-�.�.�.�	�	�	�	�a��	#�	#�B��B������r2c�l�d}tj�|dd��}t|d��dS)Nr�r>r�g�H&��8>)r6rKr]r)�probr�s  r0�test_burr12_ppf_small_argr�	�s9���D��|����a��+�+�H��H�4�5�5�5�5�5r2c���tjddd��dd�}tj�|dd���}tjgd	���}t
||d
���tj�|dd���}tjgd
���}t
||d
���tj�|dddd���}tjgd���}t
||d
���tj�|dd���}tjgd���}t
||d
���tj�|dd���}tjgd���}t
||d
���tj�|dddd���}tjgd���}t
||d
���dS)z�
    All values are calculated using the independent implementation of the
    ROOT framework (see https://root.cern.ch/).
    Corresponding ROOT code is given in the comments.
    r�rrJNr�rr|)r�r�)gC�,d	Ɣ?gH��㸘?g5���?gM"M#w�?gɫs�^�?g�� @��?g����ƴ?g�@�w��?�ɫs�^�?��|?q�?g���C�?r�	r�	g�J�ó�?g,BV\�ۤ?g���;�?g�#�ek?g�%��ǓE?g(�Ȍ�x?g��>�J�>r
r�r)g#�{`?gd����f?g�J��q?g��S�S){?g'�	��w�?�uL���?� �T)�?�cAJ�?��GT�n�?��R�1�#�?g�A]�P�?r�	r�	r�	r�	gv��Ӣ�?gK��q?g����L?g�
	�N<!?g��A�w�>rx)r�r�r<r=)g=aO}�?g�I8��?g#��T��?��a�W)�?��ǓI���?�
�
J�?�E?d۳��?��GT�n�?���8���?��R�1�#�?�k���P�?g(*�T�?r�	r�	r�	r�	r�	r�	r�	r�	)g�D�
)�?g:3P��?g����9��?g?S�[��?r�	g�Oqx��?g�D�
)�?g����9��?gɫs�^�?g��K����?g����X��?g3�<֌�?g�
��M+�?g�yS�
c�?g��I`s�?g�辜��?g{i����?g^0���?g����?�#c����?)g��%�r?gc@���v?ghtj��}?g�(˔�^�?gHK�4�U�?r�	�ğ�@_�?�y��}D�?焂R�r/�?��N���?�Yj��hG�?��a0�L�?�4f��?��<��@��?�@�C�H�?g	ސF��?g5�+-#��?g������?g?q����?r�	)g}'�v��?r�	g*V
���?r�	g�r/0+�?r�	g[��8�	�?r�	gΥ����?r�	g���Y�?r�	g��#)�a�?r�	g?�̔���?r�	g��]ؚ��?r�	g|c���?r�	)rtrr6�crystalballr@rrrK)r/�
calculatedris   r0�test_crystalball_functionr�	�s<��	��D�#�r�"�"�3�B�3�'�A��"�&�&�q�s�c�&�:�:�J��x�&�&�&�'�'�H�
�H�j�u�5�5�5�5��"�&�&�q�s�c�&�:�:�J��x�3�3�3�4�4�H�
�H�j�u�5�5�5�5��"�&�&�q�s�c�s�#�&�N�N�J��x�$�$�$�%�%�H�
�H�j�u�5�5�5�5��"�&�&�q�s�c�&�:�:�J��x�K�K�K�L�L�H��H�j�u�5�5�5�5��"�&�&�q�s�c�&�:�:�J��x�#�#�#�$�$�H�
�H�j�u�5�5�5�5��"�&�&�q�s�c�s�#�&�N�N�J��x�K�K�K�L�L�H��H�j�u�5�5�5�5�5�5r2c��tjgd���}tjgd���}tjgd���}tj�d||��}t||d���tjgd���}tjdd	tjd
dg��}||z}tj�d||��}t||d���tjtjtjtjd
dg��}||z}tj�d||��}	t||	d���tjtjtjtjtjdg��}||z}
tj�d||��}t|
|d���tjtjtjtjtjdg��}||z}tj�d||��}
t||
d���tjtjtjtjtjdg��}||z}tj�d||��}t||d���dS)zg
    All values are calculated using the pdf formula and the integrate function
    of Mathematica
    )r|rrr|r)rrr|r
r�)rrrrrrr
r�)gԚ��h@g���[&@g��ͪ�@g�2d�Q@g�n6-@g�?ޫV&̿g���B�gQ��C�R��g��QF\j�rg����
@gn���(@r>g?Җ)����r�gMg'�#@reg('�UH��r�N)rtrr6r�	rSrr�)r�r��expected_0th_moment�calculated_0th_momentrEr.�expected_1th_moment�calculated_1th_moment�expected_2th_moment�calculated_2th_moment�expected_3th_moment�calculated_3th_moment�expected_4th_moment�calculated_4th_moment�expected_5th_moment�calculated_5th_moments                r0�!test_crystalball_function_momentsr�	sg���8�-�-�-�.�.�D�
��*�*�*�+�+�A��(�#<�#<�#<�=�=��!�-�3�3�A�t�Q�?�?���'�)>�U�K�K�K�K��8�D�D�D�E�E�D�
��(�H�b�f�i��C�D�D�A��d�(��!�-�3�3�A�t�Q�?�?���'�)>�U�K�K�K�K�
��"�&�"�&�"�&�&�(�;�<�<�A��d�(��!�-�3�3�A�t�Q�?�?���'�)>�U�K�K�K�K�
��"�&�"�&�"�&�"�&�*�=�>�>�A��d�(��!�-�3�3�A�t�Q�?�?���'�)>�U�K�K�K�K�
��"�&�"�&�"�&�"�&�'�:�;�;�A��d�(��!�-�3�3�A�t�Q�?�?���'�)>�U�K�K�K�K�
��"�&�"�&�"�&�"�&�(�;�<�<�A��d�(��!�-�3�3�A�t�Q�?�?���'�)>�U�K�K�K�K�K�Kr2c��tjdd��}|���}d\}}}tj|||��}tt
|�|����|��}t||d���dS)Nr>r�)i��r�r%r�r�)	r6r�	rXrtrrrr@r)�cbrIrfrgr�rGr�s       r0�test_crystalball_entropyr�	Hsy��	�	�1�a�	 �	 �B�
�:�:�<�<�D�"�I�B��A�
��B��A���A��T�"�&�&��)�)�_�_�a�(�(�D��D�$�T�*�*�*�*�*�*r2c���dd�}tjgd���}tj�|d|���\}}}t|dd���|dksJ�t|d	d���d
S)a?
    Test fitting invweibull to data.

    Here is a the same calculation in R:

    > library(evd)
    > library(fitdistrplus)
    > x = c(1, 1.25, 2, 2.5, 2.8,  3, 3.8, 4, 5, 8, 10, 12, 64, 99)
    > result = fitdist(x, 'frechet', control=list(reltol=1e-13),
    +                  fix.arg=list(loc=0), start=list(shape=2, scale=3))
    > result
    Fitting of the distribution ' frechet ' by maximum likelihood
    Parameters:
          estimate Std. Error
    shape 1.048482  0.2261815
    scale 3.099456  0.8292887
    Fixed parameters:
        value
    loc     0

    r8rc�,�t||||dd���S)Nrq)r��disp�xtol�ftol)r$)r��x0r�r�	s    r0�	optimizerz&test_invweibull_fit.<locals>.optimizerjs���D�"�4�d��U�K�K�K�Kr2)rrr>rirAr�r�rer�r�rnr�@�c)r�r�	g`����?r�r�g� [���@N)r8r)rtrr6r�r�r)r�	rGr^r<r=s     r0�test_invweibull_fitr�	Ss���.L�L�L�L�	��H�H�H�I�I�A��$�(�(���i�(�H�H�M�A�s�E��A�x�d�+�+�+�+��!�8�8�8�8��E�8�$�/�/�/�/�/�/r2zx, c, expected))r�r�g7Ը��h�?)r�r�gz���]r�>)r�g�"@goB��ڱ�9)r�r�g$��=cC;c�j�tj�||��}t||d���dSr)r6r�rgr)rGr^rirs    r0�test_invweibull_sfr�	us7����"�"�1�a�(�(�H��H�h�U�3�3�3�3�3�3r2zp, c, expected)rxrig��9����?)g�e����K<r�g�����@c�j�tj�||��}t||d���dSr)r6r�rlr)r?r^rirs    r0�test_invweibull_isfr�	�s7����#�#�A�q�)�)�H��H�h�U�3�3�3�3�3�3r2z	df1,df2,x)r�r�rr�rer|rsrErc�^�d}tj�|||��}tj�||||��}t	||d���tj�|||��}tj�||||��}t	||d���dS)Nrr,r�r�)r6r�rKr�rr@)�df1�df2rGr��expected_cdf�calculated_cdfr��calculated_pdfs        r0�test_ncf_edge_caser
�s���
�B��7�;�;�q�#�s�+�+�L��Y�]�]�1�c�3��3�3�N��L�.�u�=�=�=�=��7�;�;�q�#�s�+�+�L��Y�]�]�1�c�3��3�3�N��L�.�t�<�<�<�<�<�<r2c�l�tj�ddd��}t|dd���dS)Nr>rreg`E@r,r�)r6r�rr)r�s r0�test_ncf_variancer
�s6��	�	�
�
�a��A���A��A�u�5�)�)�)�)�)�)r2c��tj�dddd��}d}t|t	j|d�����dS)Nr�rrf	gffffff>@g&4I,)��?)�decimals)r6r�rKrrt�round)�	scipy_val�	check_vals  r0�test_ncf_cdf_spotcheckr
�sG��
�	�
�
�b�!�R��.�.�I��I��I�r�x�	�A�>�>�>�?�?�?�?�?r2r�z(On some 32-bit the warning is not raisedc���tjddd��}d|d<d�tjt��5tjj|g��R�}�fd�|D��}ddd��n#1swxYwYt||��dS)Nrrr�r�)rlr>r�rrc�>��g|]}tjj|g��R���Sr8)r6r�r])rQ�xi�pars  �r0rSz,test_ncf_ppf_issue_17026.<locals>.<listcomp>�s-���
2�
2�
2�"�e�i�m�B�%��%�%�%�
2�
2�
2r2)	rtrr�r�rCr6r�r]r)rGr_�q0r
s   @r0�test_ncf_ppf_issue_17026r
�s����	��A�q�#���A��A�a�D�
�C�	��n�	%�	%�3�3��I�M�!�"�c�"�"�"��
2�
2�
2�
2��
2�
2�
2��3�3�3�3�3�3�3�3�3�3�3����3�3�3�3��A�r�����s�&A*�*A.�1A.c�2�eZdZd�Zd�Zd�Zd�Zd�Zd�ZdS)�
TestHistogramc�J�tj�d��tjgd�d���}t	j|��|_tj�dddd�	��}tj|d
���}t	j|��|_	dS)Nr�)rr>r>r�r�r�rerererer�r�r�r�r�rrrrrsrsrsr�r�r�r���binsrrir��{r8r)
rtr�r��	histogramr6rW�templaterEr��
norm_template)r�r
r��norm_histograms    r0r�zTestHistogram.setup_method�s���
�	���t�����L�"@�"@�"@�FG�I�I�I�	��*�9�5�5��
��z�~�~�#�S�u�3�~�O�O����d��4�4�4��"�/��?�?����r2c��tjgd���}tjgd���}t|j�|��|��t
|j�d��d��t
|j�d��d��t
|j�d��d��t
|j�d��d��tjd	d
d��}t|j�|��tj
�|dd
���d���dS)N�r}rxrr�r|rirrjr
rmr�@rwr��@r�r�!@r�g#@)r}r}rr�{�G�z�?r"
���Q��?r#
�{�G�z�?r$
r�r�r$
r$
r#
r#
r#
r#
r}r}rr#
r!
r�r}rlr�r>rnrrir;rlr�)rtrr�rr
r@rrr
r6rE)r�r��
pdf_valuesrGs    r0rzTestHistogram.test_pdf�sX����M�M�M�N�N���Z�!I�!I�!I�J�J�
�
	��
�)�)�&�1�1�:�>�>�>�	�D�M�-�-�c�2�2�H�=�=�=��D�M�-�-�c�2�2�H�=�=�=�	�D�M�-�-�c�2�2�H�=�=�=��D�M�-�-�d�3�3�X�>�>�>��K��A�r�"�"����*�.�.�q�1�1��
���q�c���=�=�C�	I�	I�	I�	I�	I�	Ir2c�6�tjgd���}tjgd���}t|j�|��|��t|j�|dd���|dd���tjddd��}t|j�|j�|����|��tjddd��}t|j�|j�|����|��tjd	dd
��}t|j�|��tj
�|dd���d
���dS)Nr
)r}r}r}r�rr"
r#
�
ףp=
�?���Q��?�{�G�z�?rjrxr���(\���?�R���Q�?�=
ףp=�?�)\��(�?r�rrr>r�rr�ror}r�rnrir;rlr�)rtrr�rr
rKr]rr
r6rE)r�r��
cdf_valuesrGs    r0razTestHistogram.test_cdf_ppf�s�����M�M�M�N�N���Z�!M�!M�!M�N�N�
�
	��
�)�)�&�1�1�:�>�>�>���
�)�)�*�Q�r�T�*:�;�;�V�A�b�D�\�J�J�J�
�K��S�#�&�&����
�)�)�$�-�*;�*;�A�*>�*>�?�?��C�C�C��K��S�#�&�&����
�)�)�$�-�*;�*;�A�*>�*>�?�?��C�C�C��K��A�r�"�"����*�.�.�q�1�1��
���q�c���=�=�C�	I�	I�	I�	I�	I�	Ir2c�:�d}|j�|d���}ttj|dk��d��ttj|dk��d|zd�	��ttj|d
k��d|zd�	��ttj|dk��d
|zd�	��ttj|dk��d|zd�	��ttj|dk��d|zd�	��ttj|dk��d|zd�	��ttj|dk��d|zd�	��ttj|dk��d|zd�	��ttj|dk��d|zd�	��ttj|dk��d|zd�	��ttj|dk��d|zd�	��ttj|d k��d!|zd�	��ttj|d"k��d#|zd�	��ttj|d$k��d%|zd�	��ttj|d&k��d|zd�	��ttj|d&k��d|zd�	��ttj|d&k��d��dS)'Nr�r
r�rr}r|rr�r�rir"
rr#
rlrjr'
r
r(
rmr)
rrjr�r
rxrwr�r�r*
r 
r+
r�r,
rr-
r!
r�r�)r
r�rrtr6r)r�r�r�s   r0rzTestHistogram.test_rvss�������"�"���"�<�<���R�V�F�S�L�)�)�3�/�/�/����v��}�-�-�x�!�|�#�F�F�F�F����v��}�-�-�x�!�|�#�F�F�F�F����v��}�-�-�x�!�|�#�F�F�F�F����v��}�-�-�x�!�|�#�F�F�F�F����v��}�-�-�x�!�|�#�F�F�F�F����v��}�-�-�x�!�|�#�F�F�F�F����v��}�-�-�y�1�}�4�H�H�H�H����v��}�-�-�y�1�}�4�H�H�H�H����v��}�-�-�y�1�}�4�H�H�H�H����v��}�-�-�y�1�}�4�H�H�H�H����v��}�-�-�y�1�}�4�H�H�H�H����v��}�-�-�y�1�}�4�H�H�H�H����v��}�-�-�y�1�}�4�H�H�H�H����v��}�-�-�y�1�}�4�H�H�H�H����v��}�-�-�y�1�}�4�H�H�H�H����v��}�-�-�y�1�}�4�H�H�H�H��R�V�F�S�L�)�)�3�/�/�/�/�/r2c���td��D]R}t|j�|��t	jdd���|��d����SdS)Nrerrir�r�)r�rr
rSr6rEr�)r�r>s  r0rzTestHistogram.test_munpsx���q���	G�	G�A��D�.�4�4�Q�7�7�!�J�s�C�0�0�7�7��:�:��
G�
G�
G�
G�
G�	G�	Gr2c��t|j���tj�dd���d���dS)Nrrir;r�r�)rr
rXr6rEr�s r0r:zTestHistogram.test_entropysT����*�2�2�4�4��
�*�*�s�#�*�>�>�T�	K�	K�	K�	K�	K�	Kr2N)	r#r$r%r�rrarrr:r8r2r0r
r
�s������@�@�@� I�I�I�4I�I�I�,0�0�0�,G�G�G�
K�K�K�K�Kr2r
c���ddggd�}}tj||fd���}tj�|�ddg��ddg��|���dksJ�tj||fd���}tj�|�ddg��d	��|���d
ksJ�d}tt|���5tj||f��}|���d
ksJ�	ddd��n#1swxYwYtj|gd
�f��}|���dksJ�dS)Nr)rrr�F)r�rxrrcTg��'^P?gH@z(Bin widths are not constant. Assuming...r�r)	r6rWrtrrr@rr
rC)�countsr
rRr�s    r0�test_histogram_non_uniformr4
!s�����F�L�L�L�D�F���v�t�n�e�<�<�<�D��J���t�x�x��c�
�3�3�c�6�]�C�C�C��;�;�=�=�A�������v�t�n�d�;�;�;�D��J���t�x�x��c�
�3�3�V�<�<�<��;�;�=�=�F�"�"�"�"�9�G�	�n�G�	4�	4�	4�'�'��!�6�4�.�1�1���{�{�}�}��&�&�&�&�&�'�'�'�'�'�'�'�'�'�'�'����'�'�'�'�
��v�y�y�y�1�2�2�D��;�;�=�=�A������s�11D/�/D3�6D3c�`�eZdZd�Zej�dddg��d���ZdS)�TestLogUniformc�~�tj�d��}tjdd��}|�d|���}tj�d��}tjdd��}|�d|���}t||��tjtj	|��d���\}}d|�
��cxkr|���cxkrd	ksnJ�tjtj
|��d
z
��dksJ�dS)N�0o[r
rr�r�rnr
r�iLr�)rtr�r�r6�
loguniformr��
reciprocalrr
�log10r~rr�r)r�r�r�r��rv2r�rr�s        r0�
test_aliaszTestLogUniform.test_alias9s���i�#�#�K�0�0��
�
�h��
0�
0���f�f�%�c�f�2�2���i�#�#�K�0�0����x��1�1���w�w�E��w�4�4����c�"�"�"��,�r�x��}�}�2�6�6�6���a��d�h�h�j�j�6�6�6�6�D�H�H�J�J�6�6�6�6�$�6�6�6�6�6�6��v�b�i��o�o��,�-�-��3�3�3�3�3�3r2r@�mler>c�D�tj�d��}tj�ddd|���}tj�||���\}}}}|dksJ�tj�|d|���\}}}}|dksJ�dS)	Nr8
rlrr�r�rBr>r?)rtr�r�r6r9
r�r�)r�r@r�r�r.r/r<r=s        r0�test_fit_overridez TestLogUniform.test_fit_overrideJs����i�#�#�K�0�0����"�"�3���3�"�G�G�� �+�/�/��F�/�C�C���1�c�5���z�z�z�z� �+�/�/��A�f�/�M�M���1�c�5���z�z�z�z�z�zr2N)r#r$r%r=
r�r�r
r@
r8r2r0r6
r6
8sU������4�4�4�"�[���X��t�}�5�5�
�
�6�5�
�
�
r2r6
c���eZdZd�Zej�dddgddgddgg��d	���Zej�d
ddg��d
���Zej�dgd���d���Z	ej�dgd���d���Z
ej�dgd���d���Zej�dgd���d���Zej�dgd���d���Z
ej�dgd���d���Zej�dgd���d���Zd S)!�	TestArgusc���tj�ddd���}ttjd�����|���d���dS)Nrr��Er�rer)r6�argusr�rrDrms  r0�test_argus_rvs_large_chiz"TestArgus.test_argus_rvs_large_chiYsR���K�O�O�B�S�s�O�;�;���E�K��O�O�0�0�2�2�A�F�F�H�H�a�H�H�H�H�H�Hr2zchi, random_staterlrD
r��rj�c��tj�|d|���}tj|d|f��\}}t	|dk��dS)Nr�r�rE
r��r6rE
r�r�r	)r�rfr�rGr�r?s      r0rzTestArgus.test_rvs^sN��
�K�O�O�C�c��O�E�E���|�A�w���0�0���1���D������r2rfr�r�c��tj�|dd���}tj|d���\}}t	|dk��dS)Nr�ie�
r�c��dd|dzz
dzz
S)Nrr>r�r8r�s r0r_z.TestArgus.test_rvs_small_chi.<locals>.<lambda>ns���a�!�Q�$�h�#�->�)>�r2r�rJ
)r�rfr�r�r?s     r0�test_rvs_small_chizTestArgus.test_rvs_small_chihsN��

�K�O�O�C�c��O�?�?���|�A�>�>�?�?���1���D������r2zchi, expected_mean))rgщi��?)rng.��憃�?)r�g\�paP��?)�<g��	��?)r�	g,6����?c�l�tj�|d���}t||d���dS)NrrDr�r�)r6rE
rDr)r�rf�
expected_meanr�s    r0rszTestArgus.test_meanrs9��
�K���S���*�*����=�u�5�5�5�5�5�5r2zchi, expected_var, rtol))rg�LH'B��?r�)rng���o�$?r�)r�g
=����>rQ)rN
g2W��>rQ)r�	g��b~�P>rQc�l�tj�|d���}t|||���dS)NrrDr�)r6rE
rr)r�rf�expected_varr�r�s     r0r�zTestArgus.test_var}s6��
�K�O�O�C�q�O�)�)����<�d�3�3�3�3�3�3r2zchi, expected, rtol))r�gwM����?r,)rxg���j���?r,)rlgH�EA�`!?r,)rzgc�ف>r�)r
g����F�=r,)r�g߹Q#�B=r,)r�g�0�ޟ<r,)r�g���*W%:r�c�D�tt|��||���dSr�)rr")r�rfrir�s    r0�test_argus_phi_small_chiz"TestArgus.test_argus_phi_small_chi�s%��	�
�3�����=�=�=�=�=�=r2z
chi, expected))rx)g-��\/�?g+��p7c�?g�5Ϧ\��?)r�)g
��f��?gi{?ٸ�?g�Z@�7��?)rl)gB	��?g]�����?g��R{��?)rz)g�lͅw�?g,�����?g[Z=���?)r
)g�z,���?g]���?g�Ѫ1���?)r�)g8<��?g�E�-���?gI����?)r�)g~7���?g�7B.���?g�9����?)r�)g�;���?g�9B.���?g}5����?c��tjgd���}ttj�||��|d���dS)N�rlrxr�r�r�)rtrrr6rE
r@�r�rfrirGs    r0�test_pdf_small_chizTestArgus.test_pdf_small_chi��B��
�H�_�_�_�%�%��������3�/�/���F�F�F�F�F�Fr2))rx)gQ߹?e��?g��rJ,�?g�򺗄�?)r�)g�Ύ�d��?g�������?g����h�?)rl)g|
o���?gvb����?g�+��@�?)rz)g̈́On��?g���e���?g�C|[�3�?)r
)g�!?�m��?g{jiH���?g�u5�3�?)r�)g��~m��?g0-�.���?g���3�?)r�)g��~m��?g7;B.���?g��ŗ�3�?)r�)g�~m��?g�9B.���?g�ŗ�3�?c��tjgd���}ttj�||��|d���dS)NrV
r,r�)rtrrr6rE
rgrW
s    r0�test_sf_small_chizTestArgus.test_sf_small_chi�sB��
�H�_�_�_�%�%�������q�#�.�.��u�E�E�E�E�E�Er2))rx)g�+��&�?g���k��?gμ�m/�?)r�)g?CL\�f�?g���^N�?g]Aͭ�R�?)rl)ga�>$��?g�;�Lf�?g��a�W�?)rz)g���x��?g��43n�?g�w���Y�?)r
)go�7🤎?g
+-oGn�?gIQ9‹Y�?)r�)g�T���?g����Gn�?g@��̋Y�?)r�)g���U���?g��{�Gn�?giG͋Y�?)r�)gD�U���?g�{�Gn�?gH͋Y�?c��tjgd���}ttj�||��|d���dS)NrV
rqr�)rtrrr6rE
rKrW
s    r0�test_cdf_small_chizTestArgus.test_cdf_small_chi�rY
r2))rx)gQ)��?g�6�{�?rq)gB`��"۹?)g�����?g��8%�?r�)rl)g�388���?g��V�%�?r�)rz)g*�j����?gR�S�%�?r�)r
)g!�|��?gZ���%�?r�)r�)gy�[|��?g�G��%�?r�)r�)g�"3|��?g�W��%�?r�)r�)g�!3|��?g�W��%�?r�c�l�tj�|d���}t|||���dS)Nr�rr�)r6rE
r)r�rfrir�rs     r0�test_stats_small_chizTestArgus.test_stats_small_chi�s9���k����T��2�2����X�D�1�1�1�1�1�1r2N)r#r$r%rF
r�r�r
rrM
rsr�rT
rX
r[
r]
r_
r8r2r0rB
rB
Xs�������I�I�I�
�[���0�
�#�J�
�#�J�
�#�J�3
���
����
�
�[���U�T�4�L�1�1���2�1���[���1�8�8�8�9�9�6�6�
9�9�6�
�[���6�B�B�B�C�C�4�4�
C�C�4�
�[���2�D�D�D�E�E�>�>�E�E�>��[����	M�	M�	M�	N�	N�G�G�	N�	N�G�
�[����	N�	N�	N�	O�	O�F�F�	O�	O�F�
�[����	P�	P�	P�	Q�	Q�G�G�	Q�	Q�G�
�[����	C�	C�	C�	D�	D�2�2�	D�	D�2�2�2r2rB
c���eZdZd�Zd�Zej�d���ej�dgd���ej�dgd���ej�d	gd
���d���������Z	ej�dgd���ej�d	gd
���d�����Z
d
S)�TestNakagamic�n�d}d}tj�||��}t|d��dS)Nrir�g�M��+��)r6�nakagamir�r)r�r	rGr�s    r0r�zTestNakagami.test_logpdf�s<�������~�$�$�Q��+�+����1�2�2�2�2�2r2c���d}d}tj�||��}t|dd���tj�||��}t||d���dS)Nrirg��"S�+�:r�r�)r6rc
rgrrl)r�r	r�	rgr6s     r0r	zTestNakagami.test_sf_isf�sm����
��
�^�
�
�r�2�
&�
&����1��>�>�>�>�
�^�
�
��B�
'�
'����B�U�+�+�+�+�+�+r2z+Fit of nakagami not reliable, see gh-10908.r�r	)rRrir�r<)r{rn�#r=)r�r�r�c���
��d�
tj��
|||d����tj����\}}}t	||d���t	||d���t	||d����fd�}�
�fd�}�
�fd�}	t	||||��d	d
���t	||||��d	d
���t	|	|||��d	d
���dS)Nro�9�rr	r<r=r�r�r�c���d|zdztjd�|z
z��zd|z|dzztj�|z
��zzS)Nr�rr>�rtr6)r	r<r=r�s   �r0�	dlogl_dnuz(TestNakagami.test_fit.<locals>.dlogl_dnusR�����U�Q�Y�"�&��G�c�M�):�";�";�;���d�5�!�8�m�b�f�W�s�]�&;�&;�;�<�
=r2c����dtj|��ztd|��z
zdtjtj�|z
|z����zztj�|z
|zdz��z
S)Nrrr>)rtrUrr6�r	r<r=r�r�s   ��r0�
dlogl_dlocz)TestNakagami.test_fit.<locals>.dlogl_dloc
sv�����R�V�B�Z�Z��)�A�r�*:�*:�:�;����r�v�w��}��&=�>�>�?�?�?�@��f�w��}��5��9�:�:�;�
<r2c�h��d�z|z|zd|z|dzztj�|z
dz��zzS)Nr�r>r�rj
rm
s   ��r0�dlogl_dscalez+TestNakagami.test_fit.<locals>.dlogl_dscalesG����!�G�b�L�5�(��"�f�u��z�)�B�F�G�c�M�a�3G�,H�,H�H�I�
Jr2rr
r�)r6rc
r�r�r)r�r	r<r=�nu_est�loc_est�	scale_estrk
rn
rp
r�r�s          @@r0r=zTestNakagami.test_fit�s[����
���.�$�$�!���+0�t�%�E�E��%*�^�%7�%7��%@�%@�"��������-�-�-�-����3�/�/�/�/��	�5�s�3�3�3�3�	=�	=�	=�	=�	=�	<�	<�	<�	<�	<�	<�
	J�	J�	J�	J�	J�	J�	�	�	�&�'�9�=�=�q�t�L�L�L�L��
�
�6�7�I�>�>���M�M�M�M����V�W�i�@�@�!�$�O�O�O�O�O�Or2c��d}d}tj�||||d���}tj�||���\}}}t	j|��}	t	jt	j||z
dz����}
t||d���t||	d���t||
d���dS)	Nrxrorg
rh
rYr>r�r�)	r6rc
r�r�rtr~r�rDr)r�r<r=r	r>r�rq
rr
rs
�loc_theo�
scale_theos           r0�test_fit_nuzTestNakagami.test_fit_nus���
�����.�$�$�!���+0�t�%�E�E��%*�^�%7�%7��B�%7�%G�%G�"�����6�'�?�?���W�R�W�g��&7�A�%=�>�>�?�?�
�����.�.�.�.�����5�5�5�5��	�:�D�9�9�9�9�9�9r2N)r#r$r%r�r	r�r�r�r
r=rw
r8r2r0ra
ra
�s������3�3�3�",�,�,�&�[���K��L�L��[���T�?�?�?�3�3��[���U�N�N�N�3�3��[���W�k�k�k�2�2�P�P�3�2�4�3�4�3�M�L�P�8�[���U�N�N�N�3�3��[���W�k�k�k�2�2�:�:�3�2�4�3�:�:�:r2ra
c� �eZdZd�Zd�Zd�ZdS)�TestWrapCauchyc�L�tjddgddgg��}tjdgdgg��}tj�||��}|jdksJ�d�tj||f��D��}t|���|d	�
��dS)N���Q��?r5rxr+rr
r
c�T�g|]%\}}tj�||����&Sr8)r6�
wrapcauchyrK)rQr6r2s   r0rSz>TestWrapCauchy.test_cdf_shape_broadcasting.<locals>.<listcomp>5s?��<�<�<�%�b�"��)�-�-�b�"�5�5�<�<�<r2r�r�)	rtrr6r}
rKr�nditerrr.)r�r^rGr?�
scalar_valuess     r0�test_cdf_shape_broadcastingz*TestWrapCauchy.test_cdf_shape_broadcasting-s���
�H�t�T�l�S�$�K�0�1�1���H�s�e�c�U�^�$�$���� � ��A�&�&���w�&� � � � �<�<�)+��A�q�6�):�):�<�<�<�
�����	�	�=�u�=�=�=�=�=�=r2c�~�tj�tjd��}t|dd���dS)Nr{
rxr,r�)r6r}
rKrtrBrrLs  r0�test_cdf_centerzTestWrapCauchy.test_cdf_center9s7���� � ����-�-����3�U�+�+�+�+�+�+r2c��d}d}d}tj�||g|��}d|zd|z
z}t|dt	j|t	j|dz��z��tjz��t|ddt	j|t	jtj|dzz
��z��tjzz
��dS)Nrr
r+rrr>)r6r}
rKrrt�arctan�tanrB)r�r6r7r^r?�crs      r0r"zTestWrapCauchy.test_cdf=s���
��
������ � �"�b��1�-�-���!�e�a�!�e�_����!��b�i��2�6�"�Q�$�<�<��8�8���>�?�?�?���!��a�"�)�B�r�v�b�e�b��d�l�/C�/C�,C�"D�"D�R�U�"J�J�K�K�K�K�Kr2N)r#r$r%r�
r�
r"r8r2r0ry
ry
+sF������
>�
>�
>�,�,�,�L�L�L�L�Lr2ry
c���Gd�dtj��}|d���}ttd���5|���ddd��dS#1swxYwYdS)Nc��eZdZd�ZdS)�/test_rvs_no_size_error.<locals>.rvs_no_size_genc��dSr�r8r�s r0�_rvsz4test_rvs_no_size_error.<locals>.rvs_no_size_gen._rvsJs���1r2N)r#r$r%r�
r8r2r0�rvs_no_size_genr�
Is#������	�	�	�	�	r2r�
�rvs_no_sizer�z_rvs\(\) got (an|\d) unexpectedr�)r6rVr�r�r�)r�
r�
s  r0�test_rvs_no_size_errorr�
Gs��������%�-����"�/�}�5�5�5�K�	�y�(J�	K�	K�	K�����������������������������s�A�A!�$A!zdistname, argsc�n�|tvrtjd|�d���tt|��}t|tj��r�t|��dkrA|j|�\}}t|tj��t|tj��d\}}|jg|�|�|�R�\}}t|tj��t|tj��dS|j|�\}	}
t|	tj��t|
tj��dS)Nz6skipping test for the support method for distribution �.rr�)�$skip_test_support_gh13294_regressionr�r�r�r6r
rVr��supportrrtr�)r�	r�rR�a0�b0r2r3�a1�b1r.r/s           r0�test_support_gh13294_regressionr�
Ss8���7�7�7���0�$,�0�0�0�	1�	1�	1��5�(�#�#�D��$��+�,�,� ��t�9�9��>�>�!�T�\�4�(�F�B����R�V�$�$�$���R�V�$�$�$�
���f����2�t�2�T�2�6�2�2�2���B��R��� � � ��R��� � � � � ��t�|�T�"���1��Q�������Q�������r2c�\�tj�gd�gd���\}}tjtjtjtjtjg��}tjtjtjtjtjg��}t||��t||��|j|jksJ�|j|jksJ�tj�gg��\}}tjg��tjg��}}t||��t||��|j|jksJ�|j|jksJ�tj�gd�dg��\}}	tjdtjgz��}
tjdtjgz��}t||
��t|	|��|j|
jksJ�|	j|jksJ�dS)N)rrrr)rrrr�r�re)	r6rEr�
rtrr�r�rr)r�
r�
�ex_a0�ex_b0r�
r�
�ex_a1�ex_b1rDr��ex_a2�ex_b2s            r0�,test_support_broadcasting_gh13294_regressionr�
ns���
�Z�
�
����m�m�m�
<�
<�F�B���H�r�v�g���w������8�9�9�E��H�b�f�b�f�b�f�b�f�5�6�6�E���U������U����
�8�u�{�"�"�"�"�
�8�u�{�"�"�"�"�
�Z�
�
��B�
'�
'�F�B���8�B�<�<���"���5�E���U������U����
�8�u�{�"�"�"�"�
�8�u�{�"�"�"�"�
�Z�
�
����r�d�
3�
3�F�B���H�Q���x�Z� � �E��H�Q���x�Z� � �E���U������U����
�8�u�{�"�"�"�"�
�8�u�{�"�"�"�"�"�"r2c�2�ddg}dgdgdgg}ttj�||��ddgddgddgg��t	jd��}t	jd��}tj�||��jdksJ�dS)	Nr}rr|rr
r�rOr)rr6rErrtr-rr;s  r0�*test_stats_broadcasting_gh14953_regressionr�
�s���
�r�(�C��T�B�4�"���E�������U�+�+�r�2�h��R��2�r�(�-K�L�L�L�
�(�5�/�/�C��H�V���E��:�>�>�#�u�%�%�+�v�5�5�5�5�5�5r2r�)gn��!	�g��:I"<)g��Q�	@g�E��?c��ttj�|��|��ttj�|��|��dSrr)rr6�cosinerKrg)rGris  r0�test_cosine_cdf_sfr�
�sJ���E�L�$�$�Q�'�'��2�2�2��E�L�O�O�Q�B�'�'��2�2�2�2�2r2zp, expected))r�g�kM6O��)r�g0ӭ��!	�)g333333�?g��;'u(@c��ttj�|��|��ttj�|��|��dSrr)rr6r�
r]rl)r?ris  r0�test_cosine_ppf_isfr�
�sL��
�E�L�$�$�Q�'�'��2�2�2��E�L�$�$�Q�'�'�(��3�3�3�3�3r2c���tj�tjtjg��}t|tjtjg��dSrr)r6r�
r�rtrBrr�)r�s r0�test_cosine_logpdf_endpointsr�
�sD���<���������/�/�D������"�&��)�*�*�*�*�*r2c��d�tD��}d�tD��}||ksJ�d�tD��}d�tD��}||ksJ�dS)Nc�B�h|]\}}t|t���|��Sr8)r
r\�rQr�r�s   r0�	<setcomp>z*test_distr_params_lists.<locals>.<setcomp>�s:��4�4�4�7�4��'��c�2�2�4�$�4�4�4r2c��h|]\}}|��Sr8r8r�
s   r0r�
z*test_distr_params_lists.<locals>.<setcomp>�s��A�A�A�g�d�A�T�A�A�Ar2c��h|]\}}|��Sr8r8r�
s   r0r�
z*test_distr_params_lists.<locals>.<setcomp>�s��3�3�3�w�t�Q�d�3�3�3r2c��h|]\}}|��Sr8r8r�
s   r0r�
z*test_distr_params_lists.<locals>.<setcomp>�s��9�9�9�'�$���9�9�9r2)rr rr)�discrete_distnames�invdiscrete_distnames�cont_distnames�invcont_distnamess    r0�test_distr_params_listsr�
�s���4�4�l�4�4�4��A�A��A�A�A���!6�6�6�6�6�3�3�(�3�3�3�N�9�9�[�9�9�9���.�.�.�.�.�.�.r2c���tj�dd���tj�dd���dksJ�tj�dd��dksJ�dS)NrrDrDre)r:r.r)r6r*�_statsr�rSr8r2r0�test_moment_order_4r�
�sr��
�N���A�s��+�+�+��>� � �q�A� �.�.�#�5�5�5�5��>����1�%�%��,�,�,�,�,�,r2rr)�rrr@r�	r�r�pathlibrr�r�r��
numpy.testingrrrrrr	r
rrr
r�rr�rArtrr�numpy.lib.recfunctionsrr5r�scipy._lib._utilrr�rrrr�scipy.statsr6�!scipy.stats._distn_infrastructurer�scipy.stats.distributions�
scipy.specialrrr�scipy.stats._distr_paramsrr�test_discrete_basicrr �scipy.stats._continuous_distnsr!r"�scipy.optimizer#r$�	itertoolsr%�flags�optimizer�	r�r�r�
r1r9rIrLrjrpruryr�r
r�r�r�r�r�r�r�r�r�r'rHrOrWrcrur�r�r�rr/r9rOrZrnr�r�rr"rIrTrgr�r�r�rrr<rFrSrt�objectr�r�r�r�r�r�r�r�r�rr�rBrQr'rSrdrmrtr�r�r�r�r�r�r�r�r�rrr�r�r�r�r�rr�r�r�r�r�r�r�r�r�rrr8r=rDrZr�r�r�r�r�r�r�r�r�r�r�r�r�r�r�r�r�r�r�r�r�r�r�r�r�r�r�r�r�r�r�r�r	r	r		r	r	rVr	r	r#	r)	r-	r~	r�r�	r�	r�	r�	r�	r�	r�	r�	r�	r�	r�	r�	r�	r�	r�	r
r
r
r�r
r
r4
r6
rB
ra
ry
r�
r�
r�
r�
r�
r�
r�
r�
r�
r8r2r0�<module>r�
s��������	�	�	�	�
�
�
�
�
�
�
�
�������	�	�	�	���������J�J�J�J�J�J�J�J�J�J�J�J�J�J�J�J�J�J�J�J�J�J�J�J��
�
�
�*�*�*�*�*�*���������"�"�"�"�"�"�"�"�4�4�4�4�4�4�������/�/�/�/�/�/�3�3�3�3�3�3�3�3�3�3�3�3�������8�8�8�8�8�8� � � � �0�0�0�0�0�0�0�0�0�0�;�;�;�;�;�;�;�;�>�>�>�>�>�>�>�>�C�C�C�C�C�C�C�C�%�%�%�%�%�%�%�%��������i�(�1�,���|�x�'�M�.>�h�.>�.@�.@�H�.L��)6�z�'B�$�$�$�$�$�8�8�8�
?�?�?�
G�G�G�
���$	5�	5�	5�/�/�/�
(�(�(�����1�+�+�+�,�,�3�3�,�,�3�����2�7�7�7�8�8�;�;�
8�8�;�
,�,�,�"����4�>�>�>�?�?�9�9�?�?�9�
8�8�8�6
J�
J�
J� .�.�.�.�����>�>�>�?�?�

6�
6�?�?�

6�9�9�9�9�9�9�9�9�>+9�+9�+9�+9�+9�+9�+9�+9�\*�*�*�*�*�*�*�*���������>��������;�;�;�;�;�;�;�;�,#�#�#�#�#�#�#�#�<E;�E;�E;�E;�E;�E;�E;�E;�P`�`�`�`�`�`�`�`�FaD�aD�aD�aD�aD�aD�aD�aD�H;G�;G�;G�;G�;G�;G�;G�;G�|2�2�2�2�2�2�2�2�"'K�'K�'K�'K�'K�'K�'K�'K�T*�*�*�*�*�*�*�*�. G� G� G� G� G� G� G� G�FdE�dE�dE�dE�dE�dE�dE�dE�N4�4�4�4�4�4�4�4�J:�J:�J:�J:�J:�J:�J:�J:�Z2&�2&�2&�2&�2&�2&�2&�2&�jR1�R1�R1�R1�R1�R1�R1�R1�j%(�%(�%(�%(�%(�%(�%(�%(�P&B�&B�&B�&B�&B�&B�&B�&B�RX#�X#�X#�X#�X#�X#�X#�X#�vt:�t:�t:�t:�t:�t:�t:�t:�nJ7�J7�J7�J7�J7�J7�J7�J7�Z<�<�<�<�<�<�<�<�~'*�'*�'*�'*�'*�'*�'*�'*�TO@�O@�O@�O@�O@�O@�O@�O@�d/�/�/�/�/�/�/�/�2!,�!,�!,�!,�!,�!,�!,�!,�HD2�D2�D2�D2�D2�D2�D2�D2�Ni5�i5�i5�i5�i5�i5�i5�i5�XMA�MA�MA�MA�MA�6�MA�MA�MA�`&)�&)�&)�&)�&)�&)�&)�&)�R)�)�)�)�)�)�)�)�XM�M�M�

P�
P�
P� ?�?�?������&:�&:�&:�;�;�������A���A���A��)2���I�I�I�(>�*+�Q��!�Q��(8�*+��q�c�
�(4�5�5�?�?�	5�5�;�;�?�"������A���A���A��)2���I�I�I�(>�*+�Q��!�Q��(8�*+��q�c�
�(4�5�5�1�1�	5�5�1�(]J�]J�]J�]J�]J�]J�]J�]J�@*9�*9�*9�*9�*9�*9�*9�*9�j��7�2�7�1�a���g�;���	�	
�	
�g�b�g�a��2�5��k�"�"�	"�	
�	�����1�R�U�7��$�$�	$�	�	�����A�b�e�G��%�%�	%�	�	�G�B�G�A�q���w�K�(�(�	(�	*�+�	�7�2�7�1�c�"�%�i�=�!�!���7�2�7�1�c�"�%�i�=�)�)�)���7�2�7�1�c�"�%�i�=�)�)�)�5�0�
��g�b�g�a��R�U��m�,�,�,�g�5�������3�r�u�9�
�.�.�.�	�9�;�<��"�W�R�W�Q�r�u�W�
�
�
�
�

�r�'�"�'�!�B�E�'�"�"�"�
#�
�
��W�R�W�Q�r�u�W�%�%�%�
&�
�
�#�g�b�g�a���g�&�&�&�
'�

�
�D�����2�5��)�)�)�
*�

�
�e�G�B�G�A�b�e�G�,�,�,�
-�
�
��w�r�w�q���w�/�/�/�
0�
�
�
�'! ��He8�e8�e8�e8�e8�e8�e8�e8�P6�6�6�6�6�6�6�6�&
D�
D�
D�
D�
D�
D�
D�
D� 
8�
8�
8�
8�
8�
8�
8�
8�s5�s5�s5�s5�s5�s5�s5�s5�l?�?�?�?�?�?�?�?�6��������*O�*O�*O�*O�*O�*O�*O�*O�Z?�?�?�?�?�?�?�?�$k�k�k�k�k�k�k�k�\:�:�:�:�:�:�:�:�2'9�'9�'9�'9�'9�'9�'9�'9�TF�F�F�F�F�F�F�F�.&�&�&�&�&�&�&�&�>$�$�$�$�$�$�$�$�*E
�E
�E
�E
�E
�E
�E
�E
�P)�)�)�)�)�)�)�)���������
/�
/�
/� lG�lG�lG�lG�lG�lG�lG�lG�^e,�e,�e,�e,�e,�e,�e,�e,�Px;�x;�x;�x;�x;�x;�x;�x;�v:>�:>�:>�:>�:>�:>�:>�:>�z",�",�",�",�",�",�",�",�JB&�B&�B&�B&�B&�B&�B&�B&�J6�6�6�6�6�6�6�6�8@"�@"�@"�@"�@"�@"�@"�@"�F)(�)(�)(�)(�)(�)(�)(�)(�X.�.�.�.�.�.�.�.�.p"�p"�p"�p"�p"�p"�p"�p"�fJ/�J/�J/�J/�J/�v�J/�J/�J/�Z
3�
3�
3�
3�
3�
3�
3�
3� [L�[L�[L�[L�[L�[L�[L�[L�|6�6�6�6�6�6�6�6�(N�N�N�N�N�N�N�N�H!�H!�H!�H!�H!�H!�H!�H!�VA�A�A�A�A�A�A�A�H	<�	<�	<�5�5�5�9�9�9�
!�!�!�*����'�0E��F�F�/�/�G�F�/�
<�<�<�*5�5�5�85�5�5�.�.�.�2�2�2�(:�:�:�#4�#4�#4�L���	,�	,�	,�������#�#�#����07�7�7�2'�'�'�0	-�	-�	-�.����*�F�F�F�G�G�1�1�G�G�1�
	-�	-�	-�&�&�&�9�9�9�%&�%&�%&�P����4�5�5�-�-�6�5�-�M�M�M�"����+�
�H�B�H�o��7�8�8�9�
�H�B�H�o��7�8�8�9�
�x�r�x��|�4�5�5�6�
�H�B�H�k�;�/�0�0�1�	.���2�2�
��2�2�2�2�����4�5�5�6�6�6�5�6�����4�5�5�1�1�6�5�1�67�7�7����%�%�%�*������$����
�����%�%����
�����%�%���������%�%����u9�u9�u9�u9�u9�u9�u9�u9�p8B�8B�8B�8B�8B�8B�8B�8B�v����'�0D��E�E�@�@�F�E�@�<�<�<�
���������,�,�,�3�3�3�7�7�7�.)�)�)�X6�6�6�D6�D6�D6�N+L�+L�+L�\+�+�+�0�0�0�D����)�F�F�F�G�G�
4�4�G�G�
4�����)�<�<�>�?�?�4�4�?�?�4�
�������!�!�!�"���
"�
"�
"�#���_�_�_�����
=�
=�
��
=� *�*�*�@�@�@�����C�K�5�(�E��G�G���G�G��^K�^K�^K�^K�^K�^K�^K�^K�B���.��������@v2�v2�v2�v2�v2�v2�v2�v2�rW:�W:�W:�W:�W:�W:�W:�W:�tL�L�L�L�L�L�L�L�8	�	�	�����)�?�[�+H�I�I� � �J�I� �4#�#�#�26�6�6������;�4�6�7�7�3�3�7�7�3�
�����7�7�7�8�8�4�4�	8�8�4�
+�+�+�
/�/�/�-�-�-�-�-r2

Youez - 2016 - github.com/yon3zu
LinuXploit