Your IP : 216.73.216.52


Current Path : /opt/alt/python38/lib64/python3.8/test/__pycache__/
Upload File :
Current File : //opt/alt/python38/lib64/python3.8/test/__pycache__/test_statistics.cpython-38.opt-1.pyc

U

i�f���@s@dZddlZddlZddlZddlZddlZddlZddlZddlZddl	Z	ddl
Z
ddlZddlm
Z
ddlmZddlmZddlm
Z
ddlZdd�Zdd	�Zd
d�Zdvdd�ZGdd�d�Ze
jddgd�Ze
jddgd�ZGdd�dej�ZGdd�dej�ZGdd�dej�ZGdd�dej�ZGdd�dej�ZGd d!�d!ej�Z Gd"d#�d#ej�Z!Gd$d%�d%ej�Z"Gd&d'�d'ej�Z#Gd(d)�d)ej�Z$Gd*d+�d+ej�Z%Gd,d-�d-ej�Z&Gd.d/�d/ej�Z'Gd0d1�d1ej�Z(Gd2d3�d3ej�Z)Gd4d5�d5ej�Z*Gd6d7�d7ej�Z+Gd8d9�d9ej�Z,Gd:d;�d;ej�Z-Gd<d=�d=�Z.Gd>d?�d?�Z/Gd@dA�dAe.e/�Z0GdBdC�dCe�Z1GdDdE�dEe�Z2GdFdG�dGe�Z3GdHdI�dIe.�Z4GdJdK�dKee4e/�Z5GdLdM�dMee4e/�Z6GdNdO�dOee4�Z7GdPdQ�dQee/�Z8GdRdS�dSe7e/�Z9GdTdU�dUe7e/�Z:GdVdW�dWe7�Z;GdXdY�dYee4e/�Z<GdZd[�d[ej�Z=Gd\d]�d]ej�Z>Gd^d_�d_e.�Z?Gd`da�dae?ee/�Z@Gdbdc�dce?ee/�ZAGddde�dee?e�ZBGdfdg�dge?e�ZCGdhdi�diej�ZDGdjdk�dkej�ZEGdldm�dm�ZFGdndo�doejeF�ZGe�Hedp�Gdqdr�drejeF��ZIdsdt�ZJeKduk�r<e�L�dS)wz_Test suite for statistics module, including helper NumericTestCase and
approx_equal function.

�N)�support)�Decimal��FractioncCst�d|�S)z:Return -1.0 for negatives, including -0.0, otherwise +1.0.�)�math�copysign��x�r�9/opt/alt/python38/lib64/python3.8/test/test_statistics.py�signsr
cCsZt|�t|�k	rdSt|t�r2t�|�o0t�|�S|��d}|��d}||koX|dkS)a�Return True if a and b are both the same kind of NAN.

    >>> _nan_equal(Decimal('NAN'), Decimal('NAN'))
    True
    >>> _nan_equal(Decimal('sNAN'), Decimal('sNAN'))
    True
    >>> _nan_equal(Decimal('NAN'), Decimal('sNAN'))
    False
    >>> _nan_equal(Decimal(42), Decimal('NAN'))
    False

    >>> _nan_equal(float('NAN'), float('NAN'))
    True
    >>> _nan_equal(float('NAN'), 0.5)
    False

    >>> _nan_equal(float('NAN'), Decimal('NAN'))
    False

    NAN payloads are not compared.
    F�)�n�N)�type�
isinstance�floatr�isnanZas_tuple)�a�bZaexpZbexprrr�
_nan_equal"s
rcCs:tt|�t|��}t||�}|r*||ntd�}||fS)z�Return the absolute and relative errors between two numbers.

    >>> _calc_errors(100, 75)
    (25, 0.25)
    >>> _calc_errors(100, 100)
    (0, 0.0)

    Returns the (absolute error, relative error) between the two arguments.
    �inf)�max�absr)�actual�expected�base�abs_err�rel_errrrr�_calc_errorsAs
r ��-���q=�H�����z>cCs�|dks|dkrtd��t�|�s,t�|�r0dS||kr<dSt�|�sPt�|�rTdSt||�}t||tt|�t|���}||kS)a�approx_equal(x, y [, tol [, rel]]) => True|False

    Return True if numbers x and y are approximately equal, to within some
    margin of error, otherwise return False. Numbers which compare equal
    will also compare approximately equal.

    x is approximately equal to y if the difference between them is less than
    an absolute error tol or a relative error rel, whichever is bigger.

    If given, both tol and rel must be finite, non-negative numbers. If not
    given, default values are tol=1e-12 and rel=1e-7.

    >>> approx_equal(1.2589, 1.2587, tol=0.0003, rel=0)
    True
    >>> approx_equal(1.2589, 1.2587, tol=0.0001, rel=0)
    False

    Absolute error is defined as abs(x-y); if that is less than or equal to
    tol, x and y are considered approximately equal.

    Relative error is defined as abs((x-y)/x) or abs((x-y)/y), whichever is
    smaller, provided x or y are not zero. If that figure is less than or
    equal to rel, x and y are considered approximately equal.

    Complex numbers are not directly supported. If you wish to compare to
    complex numbers, extract their real and imaginary parts and compare them
    individually.

    NANs always compare unequal, even with themselves. Infinities compare
    approximately equal if they have the same sign (both positive or both
    negative). Infinities with different signs compare unequal; so do
    comparisons of infinities with finite numbers.
    rz%error tolerances must be non-negativeFT)�
ValueErrorrr�isinfrr)r
�y�tol�relZactual_errorZ
allowed_errorrrr�approx_equalQs"r(c@seZdZdZdS)�
_DoNothinga�
    When doing numeric work, especially with floats, exact equality is often
    not what you want. Due to round-off error, it is often a bad idea to try
    to compare floats with equality. Instead the usual procedure is to test
    them with some (hopefully small!) allowance for error.

    The ``approx_equal`` function allows you to specify either an absolute
    error tolerance, or a relative error, or both.

    Absolute error tolerances are simple, but you need to know the magnitude
    of the quantities being compared:

    >>> approx_equal(12.345, 12.346, tol=1e-3)
    True
    >>> approx_equal(12.345e6, 12.346e6, tol=1e-3)  # tol is too small.
    False

    Relative errors are more suitable when the values you are comparing can
    vary in magnitude:

    >>> approx_equal(12.345, 12.346, rel=1e-4)
    True
    >>> approx_equal(12.345e6, 12.346e6, rel=1e-4)
    True

    but a naive implementation of relative error testing can run into trouble
    around zero.

    If you supply both an absolute tolerance and a relative error, the
    comparison succeeds if either individual test succeeds:

    >>> approx_equal(12.345e6, 12.346e6, tol=1e-3, rel=1e-4)
    True

    N)�__name__�
__module__�__qualname__�__doc__rrrrr)�s#r)�
statistics�_statistics)Zblocked)Zfreshc@s.eZdZdgZdd�Ze�ed�dd��ZdS)�TestModulesZ_normal_dist_inv_cdfcCs$|jD]}|�tt|�jd�qdS�Nr.)�
func_names�assertEqual�getattr�
py_statisticsr+��selfZfnamerrr�test_py_functions�s
zTestModules.test_py_functions�requires _statisticscCs$|jD]}|�tt|�jd�qdS)Nr/)r2r3r4�c_statisticsr+r6rrr�test_c_functions�s
zTestModules.test_c_functionsN)	r*r+r,r2r8�unittest�
skipUnlessr:r;rrrrr0�s
r0c@s@eZdZdZdZZddd�Zdd�Zd
dd	�Ze	d
d��Z
dS)�NumericTestCasez�Unit test class for numeric work.

    This subclasses TestCase. In addition to the standard method
    ``TestCase.assertAlmostEqual``,  ``assertApproxEqual`` is provided.
    rNcCsZ|dkr|j}|dkr|j}t|tjj�r@t|tjj�r@|j}n|j}||||||�dS)a�Test passes if ``first`` and ``second`` are approximately equal.

        This test passes if ``first`` and ``second`` are equal to
        within ``tol``, an absolute error, or ``rel``, a relative error.

        If either ``tol`` or ``rel`` are None or not given, they default to
        test attributes of the same name (by default, 0).

        The objects may be either numbers, or sequences of numbers. Sequences
        are tested element-by-element.

        >>> class MyTest(NumericTestCase):
        ...     def test_number(self):
        ...         x = 1.0/6
        ...         y = sum([x]*6)
        ...         self.assertApproxEqual(y, 1.0, tol=1e-15)
        ...     def test_sequence(self):
        ...         a = [1.001, 1.001e-10, 1.001e10]
        ...         b = [1.0, 1e-10, 1e10]
        ...         self.assertApproxEqual(a, b, rel=1e-3)
        ...
        >>> import unittest
        >>> from io import StringIO  # Suppress test runner output.
        >>> suite = unittest.TestLoader().loadTestsFromTestCase(MyTest)
        >>> unittest.TextTestRunner(stream=StringIO()).run(suite)
        <unittest.runner.TextTestResult run=2 errors=0 failures=0>

        N)r&r'r�collections�abc�Sequence�_check_approx_seq�_check_approx_num)r7�first�secondr&r'�msg�checkrrr�assertApproxEqual�s��z!NumericTestCase.assertApproxEqualc
	Csnt|�t|�kr:dt|�t|�f}|�||�}|�|��tt||��D] \}\}}	|�||	||||�qHdS)Nz0sequences differ in length: %d items != %d items)�len�_formatMessage�failureException�	enumerate�ziprC)
r7rDrEr&r'rF�standardMsg�ir�errrrB�s��
z!NumericTestCase._check_approx_seqcCs>t||||�rdS|�|||||�}|�||�}|�|��dS�N)r(�_make_std_err_msgrJrK)r7rDrEr&r'rF�idxrNrrrrCs
z!NumericTestCase._check_approx_numc	Cs>d}|dk	rd|}||}t||�\}}|||||||fS)Nzk  %r != %r
  values differ by more than tol=%r and rel=%r
  -> absolute error = %r
  -> relative error = %rz,numeric sequences first differ at index %d.
)r )	rDrEr&r'rS�template�headerrrrrrrRs�z!NumericTestCase._make_std_err_msg)NNN)N)r*r+r,r-r&r'rHrBrC�staticmethodrRrrrrr>�s�
,
	r>c@seZdZdZdd�ZdS)�TestSignz5Test that the helper function sign() works correctly.cCs$|�td�d�|�td�d�dS)N�rg����)r3r
�r7rrr�
testZeroes(szTestSign.testZeroesN)r*r+r,r-r[rrrrrW&srWc@s,eZdZdd�Zdd�Zdd�Zdd�Zd	S)
�ApproxEqualSymmetryTestcCsTdddtd�tdd�g}ddd	td
�tdd�g}t||�D]\}}|�||�q:dS)Ni�	gfffff�B@gfffff�(�z2.54��6i�	g������B@gR����(�z2.59�)rrrM�do_relative_symmetry)r7Zargs1Zargs2rrrrr�test_relative_symmetry3s
z.ApproxEqualSymmetryTest.test_relative_symmetrycCstt||�t||�}}||}t||�t||�}}||d}|�t||d|d��|�t||d|d��dS)Nrr�r&r')�minrr�
assertTruer()r7rr�deltaZrel_err1Zrel_err2r'rrrr`Csz,ApproxEqualSymmetryTest.do_relative_symmetrycCsdddddg}d}|D]�}ttttfD]�}||�d}||}t|t||��}|j||||d�|j|||d	d|d�|j|||d	|dd�|j||||dd�|j|||d	|d�|j|||d	d|d�|j||d
d
d�|j||d
d
d�q&qdS)N��������ki�mr�drbrr)�intrrrrr�do_symmetry_test)r7�argsrer�type_r
r%�rrrr�
test_symmetryOsz%ApproxEqualSymmetryTest.test_symmetryc
Cs@d}t||||�}t||||�}|�|||�||||f��dS)Nz+approx_equal comparisons don't match for %r)r(r3�format)r7rrr&r'rTZflag1Zflag2rrrrlgsz(ApproxEqualSymmetryTest.do_symmetry_testN)r*r+r,rar`rprlrrrrr\0sr\c@sTeZdZdd�Zdd�Zdd�Zdd�Zd	d
�Zdd�Zd
d�Z	dd�Z
dd�ZdS)�ApproxEqualExactTestcCsJt||||d�}|�|d|�t||||d�}|�|d|�dS)Nrbzequality failure for x=%r)r(rd)r7r
r&r'�resultrrr�do_exactly_equal_testtsz*ApproxEqualExactTest.do_exactly_equal_testcCsdD]}|�|dd�qdS)N)�*iMi~:��iiU�
i�r�rt�r7rrrr�test_exactly_equal_intszsz,ApproxEqualExactTest.test_exactly_equal_intscCsdD]}|�|dd�qdS)N)g�z�G��?g/�$���?g�����e�@g7@gpf@g!�rh��Q@gB`��"KB@rrw�r7r
rrr�test_exactly_equal_floatssz.ApproxEqualExactTest.test_exactly_equal_floatscCsNt}|dd�|d�|dd�|dd�|dd	�|dd�fD]}|�|dd�q6dS)
Nrrrrh��	��#�$)rrt�r7�F�frrr�test_exactly_equal_fractions�s6z1ApproxEqualExactTest.test_exactly_equal_fractionscCs*t}t|d���D]}|�|dd�qdS)Nz8.2 31.274 912.04 16.745 1.2047r)r�map�splitrt)r7�D�drrr�test_exactly_equal_decimals�sz0ApproxEqualExactTest.test_exactly_equal_decimalscCsFdD]<}|�|dd�|�|ddd�t|d�}|�|dd�qdS)N)�i�i\i�i���{�G�z�?r�
i�)rtr)r7rr�rrr�test_exactly_equal_absolute�s

z0ApproxEqualExactTest.test_exactly_equal_absolutecCs2|�td�td�d�|�td�td�d�dS)Nz3.571�0.01rz81.3971)rtrrZrrr�$test_exactly_equal_absolute_decimals�sz9ApproxEqualExactTest.test_exactly_equal_absolute_decimalscCs@dddtdd�fD]}|�|dd�q|�td�dtd	��dS)
Ni� g33333SY@g�z�G�rh�rr�z11.68r��rrtrrzrrr�test_exactly_equal_relative�sz0ApproxEqualExactTest.test_exactly_equal_relativecCsHdddtdd�fD]}|�|dd�qt}|�|d�|d	�|d
��dS)Ni9�gˡE��0@g\��(h��r|�皙�����?r�z7.2z0.1r�r�)r7r
r�rrr�test_exactly_equal_both�sz,ApproxEqualExactTest.test_exactly_equal_bothN)r*r+r,rtryr{r�r�r�r�r�r�rrrrrrnsrrc@s4eZdZdd�Zdd�Zdd�Zdd�Zd	d
�ZdS)�ApproxEqualUnequalTestcCs8||fD](}t||dddd�}|�|d|�q
dS)Nrrrbzinequality failure for x=%r)r(�assertFalse)r7r
rrsrrr�do_exactly_unequal_test�sz.ApproxEqualUnequalTest.do_exactly_unequal_testcCsdD]}|�|�qdS)N)i�i��i�i�iXC�r�rxrrr�test_exactly_unequal_ints�sz0ApproxEqualUnequalTest.test_exactly_unequal_intscCsdD]}|�|�qdS)N)g��Q�#@g����[�@gfffff�G@gףp=
W"@g=
ףp=1@r�rzrrr�test_exactly_unequal_floats�sz2ApproxEqualUnequalTest.test_exactly_unequal_floatscCs<t}|dd�|dd�|dd�|dd�fD]}|�|�q(dS)	Nrrhr~r}���eiς)rr�r�rrr�test_exactly_unequal_fractions�s(z5ApproxEqualUnequalTest.test_exactly_unequal_fractionscCs"ttd���D]}|�|�qdS)Nz!3.1415 298.12 3.47 18.996 0.00245)r�rr�r��r7r�rrr�test_exactly_unequal_decimals�sz4ApproxEqualUnequalTest.test_exactly_unequal_decimalsN)r*r+r,r�r�r�r�r�rrrrr��s
r�c@s�eZdZdd�Zdd�Zdd�Zdd�Zd	d
�Zdd�Zd
d�Z	dd�Z
dd�Zdd�Zdd�Z
dd�Zdd�Zdd�Zdd�Zdd �Zd!S)"�ApproxEqualInexactTestc	Csbd}||||fD]H}|�||�}|�t||d|dd�|�|�t|||ddd�|�qdS)N�Test failure for x={!r}, y={!r}rrrb�rqrdr(r��r7r
rerTr%rFrrr�do_approx_equal_abs_test�s
z/ApproxEqualInexactTest.do_approx_equal_abs_testcCs&dD]}|�|d�|�|d�qdS)N)i��iI���i����rgrrr}�%i�i�&i6�jr�r�r�rxrrr�test_approx_equal_absolute_ints�sz6ApproxEqualInexactTest.test_approx_equal_absolute_intscCs2dD](}|�|d�|�|d�|�|d�qdS)N)	g�t��q�gfffffFX�g333333�g333333���?��?g333333@g�Q���@g�����ҭ@��?r��-C��6?r�rzrrr�!test_approx_equal_absolute_floats�sz8ApproxEqualInexactTest.test_approx_equal_absolute_floatscCsXtdd�}dddddddd	d
ddg}d
d�|D�D] }|�||�|�|t|��q2dS)Nr�i������rgrYrrhr]��"�Gcss|]}t|d�VqdS)r�Nr)�.0rrrr�	<genexpr>�szNApproxEqualInexactTest.test_approx_equal_absolute_fractions.<locals>.<genexpr>)rr�r)r7re�
numeratorsr�rrr�$test_approx_equal_absolute_fractions�s

z;ApproxEqualInexactTest.test_approx_equal_absolute_fractionscCs:td�}ttd���D]}|�||�|�||�qdS)Nr�z1.0 3.5 36.08 61.79 7912.3648)rr�r�r�)r7rer�rrr�#test_approx_equal_absolute_decimals�sz:ApproxEqualInexactTest.test_approx_equal_absolute_decimalscCs|�tddddd��dS)Ng�h㈵��>g�h㈵��r�rrb)rdr(rZrrr�test_cross_zero�sz&ApproxEqualInexactTest.test_cross_zeroc	Csjd}|d||d|fD]H}|�||�}|�t||dd|d�|�|�t||d|dd�|�qdS)Nr�rrrrbr�r�rrr�do_approx_equal_rel_test�s
z/ApproxEqualInexactTest.do_approx_equal_rel_testcCsr|�tddddd��|�tddddd��|�tdddd	d��|�td
ddd	d��|�tdddd	d��dS)N�@�/rg
ףp=
�?rbg�G�z��?i�i��?i�i�)rdr(r�rZrrr�test_approx_equal_relative_intss
z6ApproxEqualInexactTest.test_approx_equal_relative_intscCs&dD]}|�|d�|�|d�qdS)N)g{�G�Jf�皙������r�r�g\��(|B@g��ʡE>�@g��x��@�{�G�z�?r�)r�rzrrr�!test_approx_equal_relative_floatssz8ApproxEqualInexactTest.test_approx_equal_relative_floatscCsht}tdd�}|dd�|dd�|dd�|dd	�fD]0}|t|�fD]}|�||�|�||�qBq2dS)
Nr|r��Tr]��1�2�\�U)rrr�)r7r�rer�r�rrr�$test_approx_equal_relative_fractionss
(z;ApproxEqualInexactTest.test_approx_equal_relative_fractionscCs:ttd���D]&}|�|td��|�|td��qdS)Nz$0.02 1.0 5.7 13.67 94.138 91027.9321�0.001�0.05)r�rr�r�r�rrr�#test_approx_equal_relative_decimalssz:ApproxEqualInexactTest.test_approx_equal_relative_decimalscCst|r
|jn|j}|t|||dd��|r.|jn|j}|t||d|d��|sP|rV|jn|j}|t||||d��dS)Nrrb)rdr�r()r7rrr&r'Ztol_flagZrel_flagrGrrr�
do_check_both)sz$ApproxEqualInexactTest.do_check_bothcCs,|�dddddd�|�dddd	dd�dS)
N�R����@�+���@���Mbp?���W�8?Tg?5^�I��g%��C�����Mb`?g-C��6*?�r�rZrrr�test_approx_equal_both11sz.ApproxEqualInexactTest.test_approx_equal_both1cCs|�dddddd�dS)Nr�r�r�gV�F�?8?TFr�rZrrr�test_approx_equal_both26sz.ApproxEqualInexactTest.test_approx_equal_both2cCs|�dddddd�dS)Nr�r����MbP?r�FTr�rZrrr�test_approx_equal_both3:sz.ApproxEqualInexactTest.test_approx_equal_both3cCs,|�dddddd�|�dddd	dd�dS)
Ng=
ףp=@�@r�r�Fg�Q��[�@g�(\��[�@r�giUMu�>r�rZrrr�test_approx_equal_both4>sz.ApproxEqualInexactTest.test_approx_equal_both4N)r*r+r,r�r�r�r�r�r�r�r�r�r�r�r�r�r�r�r�rrrrr��s 		r�c@s,eZdZdd�Zdd�Zdd�Zdd�Zd	S)
�ApproxEqualSpecialsTestcCs�ttfD]z}|d�}|�t||��|�t||dd��|�t||dd��|�t||��|�t||��|�t|d��qdS)Nrrrr���)rrrdr(r�)r7rnrrrr�test_infGsz ApproxEqualSpecialsTest.test_infcCs>ttfD]0}|d�}||d�dfD]}|�t||��q"qdS)N�nanrr�)rrr�r()r7rnr��otherrrr�test_nanQsz ApproxEqualSpecialsTest.test_nancCs&t�dd�}|�t|dddd��dS)NrXrYr�rb)rrrdr(�r7Znzerorrr�test_float_zeroesWsz)ApproxEqualSpecialsTest.test_float_zeroescCs&td�}|�t|td�ddd��dS)Nz-0.0rr�rb)rrdr(r�rrr�test_decimal_zeroes[sz+ApproxEqualSpecialsTest.test_decimal_zeroesN)r*r+r,r�r�r�r�rrrrr�Ds
r�c@seZdZdd�Zdd�ZdS)�TestApproxEqualErrorscCs|�ttdddd�dS)NrjrYr���assertRaisesr#r(rZrrr�test_bad_tolcsz"TestApproxEqualErrors.test_bad_tolcCs|�ttdddd�dS)Nrjrr�r�rZrrr�test_bad_relgsz"TestApproxEqualErrors.test_bad_relN)r*r+r,r�r�rrrrr�`sr�c@s4eZdZdd�Zdd�Zdd�Zdd�Zd	d
�ZdS)�TestNumericTestCasecCs.tj|�}|j|�}|D]}|�||�qdSrQ)r>rR�generate_substringsZassertIn)r7rmZ
actual_msgrZ	substringrrr�do_testws

zTestNumericTestCase.do_testcCs|�tttj��dSrQ)rd�
issubclassr>r<�TestCaserZrrr� test_numerictestcase_is_testcase}sz4TestNumericTestCase.test_numerictestcase_is_testcasecCsd}|�|�dS)N)�@�@r���?N�r��r7rmrrr�test_error_msg_numeric�sz*TestNumericTestCase.test_error_msg_numericcCsd}|�|�dS)N)�@g� @g�?r�r~r�r�rrr�test_error_msg_sequence�sz+TestNumericTestCase.test_error_msg_sequencec	CsDt||�\}}d|d|d|d|g}|dk	r@|�d|�|S)z5Return substrings we expect to see in error messages.ztol=%rzrel=%rzabsolute error = %rzrelative error = %rNzdiffer at index %d)r �append)	r7rDrEr&r'rSrrZ
substringsrrrr��s�z'TestNumericTestCase.generate_substringsN)r*r+r,r�r�r�r�r�rrrrr�qs
r�c@s(eZdZeZddgZdd�Zdd�ZdS)�GlobalsTestr-�__all__cCs(|jD]}|�t|j|�d|�qdS)Nz%s not present)�expected_metadatard�hasattr�module)r7�metarrr�	test_meta�s
�zGlobalsTest.test_metacCsB|j}|jD]0}|�|�d�d|�|�t||�d|�qdS)N�_zprivate name "%s" in __all__zmissing name "%s" in __all__)r�r�r��
startswithrdr�)r7r��namerrr�test_check_all�s
��zGlobalsTest.test_check_allN)r*r+r,r.r�r�r�rrrrrr��sr�c@s(eZdZe�ejjdkd�dd��ZdS)�DocTestsrz)Docstrings are omitted with -OO and abovecCs0tjttjd�\}}|�|d�|�|d�dS)N)Zoptionflagsr)�doctestZtestmodr.�ELLIPSIS�
assertGreaterr3)r7ZfailedZtriedrrr�test_doc_tests�szDocTests.test_doc_testsN)	r*r+r,r<ZskipIf�sys�flags�optimizerrrrrr�s�rc@seZdZdd�ZdS)�StatisticsErrorTestcCs4d}|�ttd��|�ttjt�|tjj�dS)NzNExpected StatisticsError to be a ValueError, but got a subclass of %r instead.�StatisticsError)rdr�r.r�rr#�__base__)r7�errmsgrrr�test_has_exception�s�

�z&StatisticsErrorTest.test_has_exceptionN)r*r+r,rrrrrr
�sr
c@sDeZdZdd�Zdd�Zdd�Zdd�Zd	d
�Zdd�Zd
d�Z	dS)�ExactRatioTestcCs$dD]}|�t�|�|df�qdS)N)i�����rrh�cl F�x:^Vr)r3r.�_exact_ratio)r7rOrrr�test_int�szExactRatioTest.test_intcCs2d}|D]$}t|d�}|�t�|�|df�qdS)N)���rr��&r�)rr3r.r)r7r�rr�rrr�
test_fraction�s
zExactRatioTest.test_fractioncCsb|�t�d�d�|�t�d�d�dd�td�D�}|D]"}t�|�\}}|�|||�q:dS)Nr��rr���?)r}r�cSsg|]}t�dd��qS)���rj��randomZuniform�r�r�rrr�
<listcomp>�sz-ExactRatioTest.test_float.<locals>.<listcomp>rj)r3r.r�range)r7�datar
�num�denrrr�
test_float�szExactRatioTest.test_floatcCsJt}tj}|�||d��d�|�||d��d�|�||d��d�dS)Nz0.125rz12.345)i�	��z-1.98)i����r�)rr.rr3)r7r�rrrr�test_decimal�s
zExactRatioTest.test_decimalcCs�td�}Gdd�dt�}Gdd�dt�}||fD]`}t|t|fD]N}||�}t�|�}|�||df�|�t|d�|�|�t�|d��qBq2dS)N�INFc@seZdZdS)z(ExactRatioTest.test_inf.<locals>.MyFloatN�r*r+r,rrrr�MyFloat�sr'c@seZdZdS)z*ExactRatioTest.test_inf.<locals>.MyDecimalNr&rrrr�	MyDecimal�sr(r)	rrr.rr3rrdrr$)r7r%r'r(rrnr
�ratiorrrr��s
zExactRatioTest.test_infcCsttd�}Gdd�dt�}|||�fD]J}t�|�}|�t�|d��|�|dd�|�t|d�t|��q$dS)N�NANc@seZdZdS)z.ExactRatioTest.test_float_nan.<locals>.MyFloatNr&rrrrr'�sr'rr)	rr.rrdrr�assertIsr3r)r7r*r'r�r)rrr�test_float_nan�s
zExactRatioTest.test_float_nancCs�td�}td�}Gdd�dt�}|||�|||�fD]J}t�|�}|�t|d|��|�|dd�|�t|d�t|��q4dS)Nr*�sNANc@seZdZdS)z2ExactRatioTest.test_decimal_nan.<locals>.MyDecimalNr&rrrrr(sr(rr)rr.rrdrr+r3r)r7r*r-r(r�r)rrr�test_decimal_nan�s
zExactRatioTest.test_decimal_nanN)
r*r+r,rrr"r$r�r,r.rrrrr�s
rc@s<eZdZdd�Zdd�Zdd�Zdd�Zd	d
�Zdd�Zd
S)�DecimalToRatioTestcCs<td�}|�t�|�|df�|�t�|�|df�dS)Nr%)rr3r.r)r7rrrr�
test_infinitysz DecimalToRatioTest.test_infinitycCsDtd�td�fD].}t�|�\}}|�t||��|�|d�qdS)Nr*r-)rr.rrdrr+)r7r�r r!rrrr�szDecimalToRatioTest.test_nancCsltd�td�g}|D]R}t�|�\}}|�|d�|�|d�t�|�\}}|�|d�|�|d�qdS)Nz	9.8765e12z
9.8765e-12r)rr.r�assertGreaterEqualrZassertLessEqual)r7Znumbersr�r r!rrr�	test_signszDecimalToRatioTest.test_signcCst�td��}|�|d�dS)Nz0.1234)ii��r.rrr3�r7�trrr�test_negative_exponent,sz)DecimalToRatioTest.test_negative_exponentcCst�td��}|�|d�dS)Nz1.234e7)i K�rr4r5rrr�test_positive_exponent1sz)DecimalToRatioTest.test_positive_exponentcCs8t�td��}|�|d�t�td��}|�|d�dS)NZ1e2)rjrz1.47e5)i8>rr4r5rrr�test_regression_205366sz(DecimalToRatioTest.test_regression_20536N)	r*r+r,r0r�r2r7r8r9rrrrr/s
r/c@s$eZdZdd�Zdd�Zdd�ZdS)�IsFiniteTestcCs0dtdd�dtd�fD]}|�t�|��qdS)Nrhrr|r��5.5)rrrdr.�	_isfiniterzrrr�test_finiteBszIsFiniteTest.test_finitecCs*td�td�fD]}|�t�|��qdS�Nr�rrr�r.r<rzrrrr0GszIsFiniteTest.test_infinitycCs0td�td�td�fD]}|�t�|��qdS�Nr�r*r-r?rzrrrr�LszIsFiniteTest.test_nanN)r*r+r,r=r0r�rrrrr:?sr:c@sdeZdZdd�Zdd�Zdd�Zdd�Zd	d
�Zdd�Zd
d�Z	dd�Z
dd�Zdd�Zdd�Z
dS)�
CoerceTestcCsNttttfD]<}|�t�|t�|�Gdd�d|�}|�t�|t�|�qdS)Nc@seZdZdS)z%CoerceTest.test_bool.<locals>.MyClassNr&rrrr�MyClassmsrB)rkrrrr+r.�_coerce�bool)r7�TrBrrr�	test_boolgszCoerceTest.test_boolcCs,|�t�||�|�|�t�||�|�dS)z Assert that type A coerces to B.N)r+r.rC�r7�A�Brrr�assertCoerceTopszCoerceTest.assertCoerceTocCsT|�||�Gdd�d|�}|�||�Gdd�d|�}|�||�|�||�dS)z6Checks that type A coerces to B, including subclasses.c@seZdZdS)z/CoerceTest.check_coerce_to.<locals>.SubclassOfANr&rrrr�SubclassOfAzsrKc@seZdZdS)z/CoerceTest.check_coerce_to.<locals>.SubclassOfBNr&rrrr�SubclassOfB}srLN)rJ)r7rHrIrKrLrrr�check_coerce_touszCoerceTest.check_coerce_tocCs,|�ttj||f�|�ttj||f�dS)z=Assert that coercing A to B, or vice versa, raises TypeError.N)r��	TypeErrorr.rCrGrrr�assertCoerceRaises�szCoerceTest.assertCoerceRaisescCs�|�t�||�|�Gdd�d|�}Gdd�d|�}Gdd�d|�}|||fD]}|�||�qN|�||�|�||�|�||�dS)z>Check that type T coerces correctly with subclasses of itself.c@seZdZdS)z*CoerceTest.check_type_coercions.<locals>.UNr&rrrr�U�srPc@seZdZdS)z*CoerceTest.check_type_coercions.<locals>.VNr&rrrr�V�srQc@seZdZdS)z*CoerceTest.check_type_coercions.<locals>.WNr&rrrr�W�srRN)r+r.rCrJrO)r7rErPrQrR�typrrr�check_type_coercions�szCoerceTest.check_type_coercionscCs*|�t�tttfD]}|�t|�qdSrQ)rTrkrrrrM)r7rSrrrr�s
zCoerceTest.test_intcCs|�t�|�tt�dSrQ)rTrrMrrZrrrr�s
zCoerceTest.test_fractioncCs|�t�dSrQ)rTrrZrrrr$�szCoerceTest.test_decimalcCs|�t�dSrQ)rTrrZrrrr"�szCoerceTest.test_floatcCs:tttd�ttfD]"}ttttfD]}|�	||�q"qdSrQ)
�str�listr�tuple�dictrkrrrrO)r7Zbad_typeZ	good_typerrr�test_non_numeric_types�sz!CoerceTest.test_non_numeric_typescCs:ttfD],}Gdd�d|�}|�|t�|�|t�qdS)Nc@seZdZdS)z6CoerceTest.test_incompatible_types.<locals>.MySubclassNr&rrrr�
MySubclass�srZ)rrrOr)r7rErZrrr�test_incompatible_types�sz"CoerceTest.test_incompatible_typesN)r*r+r,rFrJrMrOrTrrr$r"rYr[rrrrrARs	rAc@sDeZdZdd�Zdd�Zdd�Zdd�Zd	d
�Zdd�Zd
d�Z	dS)�ConvertTestcCs$|�||�|�t|�t|��dS)z5Check that x equals y, and has the same type as well.N)r3r+r)r7r
r%rrr�check_exact_equal�szConvertTest.check_exact_equalcCsPt�td�t�}|�|d�Gdd�dt�}t�td�|�}|�||d��dS)Nr�c@seZdZdS)z#ConvertTest.test_int.<locals>.MyIntNr&rrrr�MyInt�sr^r])r.�_convertrrkr])r7r
r^rrrr�s
zConvertTest.test_intcCs\t�tdd�t�}|�|tdd��Gdd�dt�}t�tdd�|�}|�||dd��dS)N�_rcseZdZ�fdd�Z�ZS)z-ConvertTest.test_fraction.<locals>.MyFractioncs|�t��|��SrQ��	__class__�super�__truediv__�r7r��rbrrrd�sz9ConvertTest.test_fraction.<locals>.MyFraction.__truediv__�r*r+r,rd�
__classcell__rrrfr�
MyFraction�srir��
)r.r_rr])r7r
rirrrr�s
zConvertTest.test_fractioncCsTt�tdd�t�}|�|d�Gdd�dt�}t�tdd�|�}|�||d��dS)	NrYrg�cseZdZ�fdd�Z�ZS)z'ConvertTest.test_float.<locals>.MyFloatcs|�t��|��SrQrarerfrrrd�sz3ConvertTest.test_float.<locals>.MyFloat.__truediv__rgrrrfrr'�sr'r}r�r)r.r_rrr])r7r
r'rrrr"�s
zConvertTest.test_floatcCsXt�tdd�t�}|�|td��Gdd�dt�}t�tdd�|�}|�||d��dS)	Nr�(z0.025cseZdZ�fdd�Z�ZS)z+ConvertTest.test_decimal.<locals>.MyDecimalcs|�t��|��SrQrarerfrrrd�sz7ConvertTest.test_decimal.<locals>.MyDecimal.__truediv__rgrrrfrr(�sr(r�r�z-0.9375)r.r_rrr])r7r
r(rrrr$�s
zConvertTest.test_decimalcCsFtd�td�fD]0}||fD] }t�|t|��}|�||�qqdSr>)rrr.r_rr])r7r%rr
rrrr��szConvertTest.test_infcCs@td�td�td�fD]$}t�|t|��}|�t||��qdSr@)rrr.r_rrdr)r7r�r
rrrr��szConvertTest.test_nanN)
r*r+r,r]rrr"r$r�r�rrrrr\�s


r\c@s(eZdZdZdd�Zdd�Zdd�ZdS)	�FailNegTestz Test _fail_neg private function.cCs2ddtd�td�g}tt�|��}|�||�dS�Nr�@r|r�)rrrVr.�	_fail_negr3)r7�values�newrrr�test_pass_through�szFailNegTest.test_pass_throughcCs@ddtd�td�fD]&}|g}t�|�}|�tjt|�qdSrm)rrr.ror�r�next)r7r
�seq�itrrr�test_negatives_raise�s
z FailNegTest.test_negatives_raisec
Cspdt�dd�}ztt�dg|��Wn.tjk
rT}z|jd}W5d}~XYnX|�d�|�||�dS)Nzbadness #%d�'i��rYrz(expected exception, but it didn't happen)	rZrandintrsr.rorrmZfailr3)r7rFrPr
rrr�test_error_msgs
zFailNegTest.test_error_msgN)r*r+r,r-rrrvrxrrrrrl�srlc@s\eZdZdd�Zdd�Zdd�Zdd�Zd	d
�Zdd�Zd
d�Z	dd�Z
dd�Zdd�ZdS)�UnivariateCommonMixincCs|�t|j�dSrQ�r�rN�funcrZrrr�test_no_argssz"UnivariateCommonMixin.test_no_argscCs*gdtg�fD]}|�tj|j|�qdS)Nr)�iterr�r.rr{)r7�emptyrrr�test_empty_datasz%UnivariateCommonMixin.test_empty_datacCs(ttd��}|t|�kr$t�|�q|S)z"Return int data for various tests.r��rVr�sortedr�shuffle�r7rrrr�prepare_datasz"UnivariateCommonMixin.prepare_datacCs0|��}|dd�}|�|�}|�||d�dS)Nzdata has been modified)r�r{ZassertListEqual)r7rZsavedr�rrr�test_no_inplace_modifications!s
z3UnivariateCommonMixin.test_no_inplace_modificationscCsFddddddddgd}|�|�}t�|�|�|�}|�||�dS)Nrrr|r�rh�rj)r{rr�r3�r7rrrrrr�test_order_doesnt_matter+s



z.UnivariateCommonMixin.test_order_doesnt_mattercCsnGdd�dt�}Gdd�dt�}dd�}|��}|�|�}ttt|||fD]}|�||��}|�||�qJdS)Nc@seZdZdS)zBUnivariateCommonMixin.test_type_of_data_collection.<locals>.MyListNr&rrrr�MyList:sr�c@seZdZdS)zCUnivariateCommonMixin.test_type_of_data_collection.<locals>.MyTupleNr&rrrr�MyTuple<sr�cSsdd�|D�S)Ncss|]
}|VqdSrQr)r��objrrrr�?szXUnivariateCommonMixin.test_type_of_data_collection.<locals>.generator.<locals>.<genexpr>r�rrrr�	generator>szEUnivariateCommonMixin.test_type_of_data_collection.<locals>.generator)rVrWr�r{r}r3)r7r�r�r�rr�kindrsrrr�test_type_of_data_collection8s
z2UnivariateCommonMixin.test_type_of_data_collectioncCs0tddd�}|�t|��}|�|�|�|�dS�N�r�r|)rr{rVr3�r7rrrrr�test_range_dataFsz%UnivariateCommonMixin.test_range_datacCs.|�d�|�d�|�d�|�t��dS)Nr�gE@)�check_for_type_error�objectrZrrr�test_bad_arg_typesLs



z(UnivariateCommonMixin.test_bad_arg_typescGs|jt|jf|��dSrQrzr�rrrr�[sz*UnivariateCommonMixin.check_for_type_errorcshGdd�dt�}|��}|�|�}t|ttfD]4��fdd�|D�}t|�|�|��}|�||�q.dS)Ncs,eZdZ�fdd�Z�fdd�ZeZ�ZS)z@UnivariateCommonMixin.test_type_of_data_element.<locals>.MyFloatcst|�t��|��SrQ�rrcrdrerfrrrdcszLUnivariateCommonMixin.test_type_of_data_element.<locals>.MyFloat.__truediv__cst|�t��|��SrQ�rrc�__add__rerfrrr�eszHUnivariateCommonMixin.test_type_of_data_element.<locals>.MyFloat.__add__)r*r+r,rdr��__radd__rhrrrfrr'bsr'csg|]}�|��qSrr�r�r
�r�rrrlszCUnivariateCommonMixin.test_type_of_data_element.<locals>.<listcomp>)rr�r{rrrr3)r7r'�rawrrrsrr�r�test_type_of_data_element^s
z/UnivariateCommonMixin.test_type_of_data_elementN)
r*r+r,r|rr�r�r�r�r�r�r�r�rrrrrys

ryc@s eZdZdZdd�Zdd�ZdS)�UnivariateTypeMixinamMixin class for type-conserving functions.

    This mixin class holds test(s) for functions which conserve the type of
    individual data points. E.g. the mean of a list of Fractions should itself
    be a Fraction.

    Not all tests to do with types need go in this class. Only those that
    rely on the function returning the same type as its input data.
    cCsGdd�dt�}ttt|fS)z4Return the types which are expected to be conserved.cs\eZdZ�fdd�Z�fdd�Z�fdd�Z�fdd�Z�fd	d
�Z�fdd�ZeZ	�Z
S)
zHUnivariateTypeMixin.prepare_types_for_conservation_test.<locals>.MyFloatcst|�t��|��SrQr�rerfrrrd~szTUnivariateTypeMixin.prepare_types_for_conservation_test.<locals>.MyFloat.__truediv__cst|�t��|��SrQ)rrc�__rtruediv__rerfrrr��szUUnivariateTypeMixin.prepare_types_for_conservation_test.<locals>.MyFloat.__rtruediv__cst|�t��|��SrQ)rrc�__sub__rerfrrr��szPUnivariateTypeMixin.prepare_types_for_conservation_test.<locals>.MyFloat.__sub__cst|�t��|��SrQ)rrc�__rsub__rerfrrr��szQUnivariateTypeMixin.prepare_types_for_conservation_test.<locals>.MyFloat.__rsub__cst|�t��|��SrQ)rrc�__pow__rerfrrr��szPUnivariateTypeMixin.prepare_types_for_conservation_test.<locals>.MyFloat.__pow__cst|�t��|��SrQr�rerfrrr��szPUnivariateTypeMixin.prepare_types_for_conservation_test.<locals>.MyFloat.__add__)r*r+r,rdr�r�r�r�r�r�rhrrrfrr'}sr')rrr)r7r'rrr�#prepare_types_for_conservation_test{sz7UnivariateTypeMixin.prepare_types_for_conservation_testcsF|��}|��D]0��fdd�|D�}|�|�}|�t|���qdS)Ncsg|]}�|��qSrrr�r�rrr�sz<UnivariateTypeMixin.test_types_conserved.<locals>.<listcomp>)r�r�r{r+r)r7rr�rsrr�r�test_types_conserved�s

z(UnivariateTypeMixin.test_types_conservedN)r*r+r,r-r�r�rrrrr�qs	r�c@seZdZdd�ZdS)�
TestSumCommoncCsdd�}||_dS)NcWstj|�\}}}t�||�SrQ)r.�_sumrC)rmrE�valuerrrr�simplified_sum�sz+TestSumCommon.setUp.<locals>.simplified_sum)r{)r7r�rrr�setUp�szTestSumCommon.setUpN)r*r+r,r�rrrrr��sr�c@sdeZdZdd�Zdd�Zdd�Zdd�Zd	d
�Zdd�Zd
d�Z	dd�Z
dd�Zdd�Zdd�Z
dS)�TestSumcCstj|_dSrQ)r.r�r{rZrrrr��sz
TestSum.setUpcCspgdtg�fD]\}|�|�|�ttd�df�|�|�|d�ttd�df�|�|�|d�ttd�df�qdS)Nrrr�gffffff@)r}r3r{rkrrr�rrrr�szTestSum.test_empty_datacCsX|�|�ddddddddg�ttd�d	f�|�|�d
ddddgd
�ttd�df�dS)Nrrhr|���i����r�ru�<r�r�rr~r�i�)r3r{rkrrZrrr�	test_ints�s��zTestSum.test_intscCsL|�|�dgd�ttd�df�|�|�ddddgd�ttd�d	f�dS)
Nr�r��@r�r�g�?r�g	@r�)r3r{rrrZrrr�test_floats�s��zTestSum.test_floatscCs.|�|�tdd�gd�ttdd�df�dS)Nrr���r)r3r{rrZrrr�test_fractions�s�zTestSum.test_fractionsc	CsXt}|d�|d�|d�|d�|d�|d�|d�|d�g}|�|�|�ttd	�d
f�dS)Nr�z5.246z1.702z-0.025z3.974z2.328z4.617z2.843z20.686r�)rr3r{�r7r�rrrr�
test_decimals�s��zTestSum.test_decimalscCs:dd�td�D�}|jt|�|�d�t�|�dd�dS)NcSsg|]}t�dd��qS)rr�rrrrrr�sz7TestSum.test_compare_with_math_fsum.<locals>.<listcomp>r�rg��ؗ�Ҭ<�r')rrHrr{r�fsumr�rrr�test_compare_with_math_fsum�sz#TestSum.test_compare_with_math_fsumcCs|dd�td�D�}|�|�d}|�|d|�|d�d�|�|d|�|d�d�|�|td�|�|d�d�dS)	NcSsg|]}t�dd��qS)rr�rrrrrr�sz/TestSum.test_start_argument.<locals>.<listcomp>rjrrur�rfg@��x�D)rr{r3r)r7rr6rrr�test_start_argument�s
zTestSum.test_start_argumentcCs4|�t|jdddgd�|�t|jddddg�dS)Nrrr|Z999rzrZrrr�test_strings_fail�szTestSum.test_strings_failcCs4|�t|jdddgd�|�t|jddddg�dS)Nrrr|s999rzrZrrr�test_bytes_fail�szTestSum.test_bytes_failcCs8|�t|jddtd�g�|�t|jddgtd��dS)Nrrn)r�rNr{rrZrrr�test_mixed_sum�szTestSum.test_mixed_sumN)r*r+r,r�rr�r�r�r�r�r�r�r�r�rrrrr��sr�c@seZdZdd�ZdS)�SumTortureTestcCs�|�t�ddddgd�ttd�df�|�t�ddddgd�ttd�df�t�ddddgd�\}}}|�|t�|�|d�|jt|�d	d
d�dS)Nr�}Ô%�I�T�}Ô%�I��rwg��@i@�g0��.�++rYg���^�,gV瞯�<r�)r3r.r�rrr+rH)r7rEr �countrrr�test_torture�s��zSumTortureTest.test_tortureN)r*r+r,r�rrrrr��sr�c@sTeZdZdd�Zdd�Zdd�Zdd�Zd	d
�Zdd�Zd
d�Z	dd�Z
dd�ZdS)�SumSpecialValuescCsNttfD]@}|d�}t�d|dg�d}|�t|�|�|�t�|��qdS)Nr�rr)	rrr.r�r+rrdrr)r7rnr�rsrrrr��s
zSumSpecialValues.test_nancCs<|�t�|��|�t|�t|��|�|dk|dk�dS)z8Check x is an infinity of the same type and sign as inf.rN)rdrr$r+rr3)r7r
rrrr�check_infinityszSumSpecialValues.check_infinitycCsLt�dd|dg�d}|�||�t�dd|d|dg�d}|�||�dS)Nrrr|r�)r.r�r��r7rrsrrr�do_test_inf
szSumSpecialValues.do_test_infcCs$td�}dD]}|�||�qdS�Nr�rrY)rr��r7rr
rrr�test_float_infszSumSpecialValues.test_float_infcCs$td�}dD]}|�||�qdSr�)rr�r�rrr�test_decimal_infsz!SumSpecialValues.test_decimal_infcCs8td�}t�dd|d|dg�d}|�t�|��dS�Nrrrr|r�)rr.r�rdrrr�rrr�test_float_mismatched_infssz+SumSpecialValues.test_float_mismatched_infsc	CsPtd�}dd|d|dg}t�tj�� |�t�t�|�d��W5QRXdSr�)	r�decimal�localcontextZExtendedContextrdrrr.r��r7rrrrr�3test_decimal_extendedcontext_mismatched_infs_to_nan"szDSumSpecialValues.test_decimal_extendedcontext_mismatched_infs_to_nanc	CsHtd�}dd|d|dg}t�tj��|�tjtj|�W5QRXdSr�)rr�r�ZBasicContextr��InvalidOperationr.r�r�rrr�0test_decimal_basiccontext_mismatched_infs_to_nan)szASumSpecialValues.test_decimal_basiccontext_mismatched_infs_to_nancCs(td�}d|dg}|�tjtj|�dS)Nr-rr)rr�r�r�r.r�)r7r-rrrr�test_decimal_snan_raises0s
z)SumSpecialValues.test_decimal_snan_raisesN)r*r+r,r�r�r�r�r�r�r�r�r�rrrrr��sr�c@s$eZdZdd�Zdd�Zdd�ZdS)�AverageMixincCs6dddtdd�td�fD]}|�|�|g�|�qdS)Nr�g@E@g�X_yCr_�z0.28�rrr3r{rzrrr�test_single_value<szAverageMixin.test_single_valuecCsdddtdd�td�fS)N�@r]���7y�!C�=�Cz4.9712�rrrZrrr�'prepare_values_for_repeated_single_testAsz4AverageMixin.prepare_values_for_repeated_single_testcCsR|��D]D}dD]:}|j||d��"|g|}|�|�|�|�W5QRXqqdS)N�rrhr�r�)r
r�)r��subTestr3r{�r7r
r�rrrr�test_repeated_single_valueDs

z'AverageMixin.test_repeated_single_valueN)r*r+r,r�r�r�rrrrr�9sr�c@steZdZdd�Zdd�Zdd�Zdd�Zd	d
�Zdd�Zd
d�Z	dd�Z
dd�Zdd�Zdd�Z
dd�Zdd�ZdS)�TestMeancCstj|_dSrQ)r.�meanr{rZrrrr�NszTestMean.setUpcCs|�|�ddddg�d�dS)Nr�rr|r��r3r{rZrrr�test_torture_pepQszTestMean.test_torture_pepcCsDddddddddddddddd	d
g}t�|�|�|�|�d�dS)Nrrrr|r�rhr�r~r�r}g@@�rr�r3r{r�rrrr�Us$
zTestMean.test_intscCs4ddddddddg}t�|�|�|�|�d	�dS)
Ng@1@g�3@�4@g�5@g�5@g@7@g 9@g�;@g6@r�r�rrrr�[s
zTestMean.test_floatscCsJt}|d�|d�|d�|d�|d�g}t�|�|�|�|�|d��dS)Nz1.634z2.517z3.912z4.072z5.813z3.5896�rrr�r3r{r�rrrr�as"
zTestMean.test_decimalsc	Csft}|dd�|dd�|dd�|dd�|dd�|dd�|dd�g}t�|�|�|�|�|d	d
��dS)Nrrr|r�rhr�r~r�i�i��rrr�r3r{�r7r�rrrrr�hs<
zTestMean.test_fractionscCsfdddddg}ttfD]J}dD]@}|d�|}||g}|�|�}|�t�|��|�||�qqdS)Nrr|rhr~r}r�r)rrr{rdrr$r3)r7r�r�r
rrrsrrrr�os

zTestMean.test_infc	Cs:dddtd�dddtd�g}|�|�}|�t�|��dS)	Nrr�r�rrr|rhz-inf)rr{rdrr�r7rrsrrr�test_mismatched_infszs
zTestMean.test_mismatched_infscCsLdddddg}ttfD]0}|d�}||g}|�|�}|�t�|��qdS)Nrr|rhr~r}r�)rrr{rdrr)r7r�r�rrrsrrrr��s

zTestMean.test_nanc	sPd�dddddddd	d
g	}|�|��}|��fdd�|D��}|�||�dS)
Nge��A�333333@�@皙����@�������@�333333@�������@� @�333333 @�������"@csg|]}|��qSrrr���crrr�sz*TestMean.test_big_data.<locals>.<listcomp>�r{r3�r7rrrsrr�r�
test_big_data�s
zTestMean.test_big_datacCs:dd�td�D�}|�|�}|�|d�}|�||�dS)NcSsg|]}t�dd��qS)rrhrrrrrr�sz.TestMean.test_doubled_data.<locals>.<listcomp>r�r�rr{rHr�rrr�test_doubled_data�s
zTestMean.test_doubled_datacCs td�}|�t�|g�|�dS)NZ1e4)rr3r.r�r�rrr�test_regression_20561�szTestMean.test_regression_20561cCs\|�t�ddg�d�d}d}dD]4}|�t�|g|�|�|�t�|g|�|�q"dS)Ng�������g�g)rr|rhr#)r3r.r�)r7ZbigZtinyrrrr�test_regression_25177�s��zTestMean.test_regression_25177N)r*r+r,r�r�r�r�r�r�r�r�r�r�r�r�r�rrrrr�Ms		r�cs�eZdZdd�Z�fdd�Zdd�Zdd�Zd	d
�Zdd�Zd
d�Z	dd�Z
dd�Zdd�Zdd�Z
dd�Zdd�Zdd�Z�ZS)�TestHarmonicMeancCstj|_dSrQ)r.Z
harmonic_meanr{rZrrrr��szTestHarmonicMean.setUpcst���}|�d�|S)Nr)rcr��remove�r7rprfrrr��s

zTestHarmonicMean.prepare_datacCsdddtdd�td�fS)Nr�r]r�r�r�z4.125r�rZrrrr��sz8TestHarmonicMean.prepare_values_for_repeated_single_testcCs dddg}|�|�|�d�dS)Nrrrr�rrrr�	test_zero�s
zTestHarmonicMean.test_zeroc
CsHtj}dgdddgfD],}|j|d��|�||j|�W5QRXqdS)NrYrrgr|)rp)r.rr�r�r{)r7�excrprrr�test_negative_error�sz$TestHarmonicMean.test_negative_errorcCs0ddddddg}t�|�|�|�|�d�dS)Nrr�r�r�g333333@r�r�rrrr��s
zTestHarmonicMean.test_intscCsHdddddg}t�|�|�|�|�d�|�|�ddddg�d�dS)Nr�r�r�r�r�r�rrr�test_floats_exact�s
z"TestHarmonicMean.test_floats_exactcCs(tdd�D]}|�|�|g�|�q
dS)Nrr��rr3r{rzrrr�test_singleton_lists�sz%TestHarmonicMean.test_singleton_listsc	Cs�t}|�|�|d�|d�|d�|d�g�|d��|d�|d�|d�|d�g}t�|�|�|�|�|d��|d�|d�|d	�|d
�g}t�|�|�|�|�|d�d�dS)
Nr_r�r�r�z0.10z0.20z1.68z0.32z5.94z2.75i�iC)rr3r{rr�r�rrr�test_decimals_exact�s.

z$TestHarmonicMean.test_decimals_exactc	Csft}|dd�|dd�|dd�|dd�|dd�|dd�|dd�g}t�|�|�|�|�|d	d
��dS)Nrrr|r�rhr�r~r�i|i�r�r�rrrr��s<
zTestHarmonicMean.test_fractionscCs$dtd�dg}|�|�|�d�dS)Nrnrr�)rr3r{rrrrr��szTestHarmonicMean.test_infcCs(dtd�dg}|�t�|�|���dS)Nrnr�r�)rrdrrr{rrrrr��szTestHarmonicMean.test_nanc	sPd�dddddddd	d
g	}|�|��}|��fdd�|D��}|�||�dS)
N�or�r�r�r�r�r�r�r�r�csg|]}|��qSrrr�r�rrr�sz>TestHarmonicMean.test_multiply_data_points.<locals>.<listcomp>r�r�rr�r�test_multiply_data_points�s
z*TestHarmonicMean.test_multiply_data_pointscCs:dd�td�D�}|�|�}|�|d�}|�||�dS)NcSsg|]}t�dd��qS)rrhrrrrrr�sz6TestHarmonicMean.test_doubled_data.<locals>.<listcomp>r�rr�r�rrrr��s
z"TestHarmonicMean.test_doubled_data)r*r+r,r�r�r�rrr�rrrr�r�r�r	r�rhrrrfrr��sr�csTeZdZdd�Z�fdd�Zdd�Zdd�Zd	d
�Zdd�Zd
d�Z	dd�Z
�ZS)�
TestMediancCstj|_dSrQ�r.�medianr{rZrrrr�szTestMedian.setUpcs(t���}t|�ddkr$|�d�|S)�+Overload method from UnivariateCommonMixin.rr)rcr�rIr�r�rfrrr�
s

zTestMedian.prepare_datacCs&ddddddg}|�|�|�d�dS)Nrrr|r�rhr�r�r�r�rrr�test_even_intsszTestMedian.test_even_intscCs(dddddddg}|�|�|�d�dS)Nrrr|r�rhr�r}r�r�rrr�
test_odd_intsszTestMedian.test_odd_intscCsVt}|dd�|dd�|dd�|dd�|dd�g}t�|�|�|�|�|dd��dS)Nrr~rr|r�rhr�r�rrr�test_odd_fractionss,
zTestMedian.test_odd_fractionscCs^t}|dd�|dd�|dd�|dd�|dd�|dd�g}t�|�|�|�|�|dd��dS�Nrr~rr|r�rhr�r�r�rrr�test_even_fractions%s4
zTestMedian.test_even_fractionscCsJt}|d�|d�|d�|d�|d�g}t�|�|�|�|�|d��dS)N�2.5�3.1�4.2�5.7�5.8r�r�rrr�test_odd_decimals-s"
zTestMedian.test_odd_decimalscCsPt}|d�|d�|d�|d�|d�|d�g}t�|�|�|�|�|d��dS)Nz1.2rrrrrz3.65r�r�rrr�test_even_decimals5s(
zTestMedian.test_even_decimals)r*r+r,r�r�rrrrrrrhrrrfrr
sr
c@seZdZdd�Zdd�ZdS)�TestMedianDataTypecCstj|_dSrQrrZrrrr�@szTestMedianDataType.setUpcCs(ttd��}|t|�kr$t�|�q|S)Nr_r�r�rrrr�CszTestMedianDataType.prepare_dataN)r*r+r,r�r�rrrrr>src@s,eZdZdd�Zdd�Zdd�Zdd�Zd	S)
�
TestMedianLowcCstj|_dSrQ)r.Z
median_lowr{rZrrrr�LszTestMedianLow.setUpcCs&ddddddg}|�|�|�d�dS�Nrrr|r�rhr�r�r�rrrrOszTestMedianLow.test_even_intscCs^t}|dd�|dd�|dd�|dd�|dd�|dd�g}t�|�|�|�|�|dd��dSrr�r�rrrrUs4
z!TestMedianLow.test_even_fractionscCsPt}|d�|d�|d�|d�|d�|d�g}t�|�|�|�|�|d��dS�Nz1.1z2.2z3.3z4.4r;z6.6r�r�rrrr]s(
z TestMedianLow.test_even_decimalsN�r*r+r,r�rrrrrrrrKsrc@s,eZdZdd�Zdd�Zdd�Zdd�Zd	S)
�TestMedianHighcCstj|_dSrQ)r.Zmedian_highr{rZrrrr�gszTestMedianHigh.setUpcCs&ddddddg}|�|�|�d�dSrr�r�rrrrjszTestMedianHigh.test_even_intscCs^t}|dd�|dd�|dd�|dd�|dd�|dd�g}t�|�|�|�|�|dd��dSrr�r�rrrrps4
z"TestMedianHigh.test_even_fractionscCsPt}|d�|d�|d�|d�|d�|d�g}t�|�|�|�|�|d��dSrr�r�rrrrxs(
z!TestMedianHigh.test_even_decimalsNrrrrrrfsrc@s\eZdZdd�Zdd�Zdd�Zdd�Zd	d
�Zdd�Zd
d�Z	dd�Z
dd�Zdd�ZdS)�TestMedianGroupedcCstj|_dSrQ)r.Zmedian_groupedr{rZrrrr��szTestMedianGrouped.setUpcCs�dddddddg}|�|�|�d�dddddddg}|�|�|�d�ddddddddd	d	d
g}|�|�|d�d�dd
d
d
d
ddddddddddg}|j|�|d�ddd�dS)Nr�rj�r_g�+@rhr�r��r�g`3@r������rg������4@�:�0�yE>�r&�r3r{rHr�rrr�test_odd_number_repeated�s"z*TestMedianGrouped.test_odd_number_repeatedcCs�ddddddddddg
}|j|�|d�ddd	�d
dddddg}|j|�|�d
dd	�d
dddddddddddg}|�|�|�d�ddddddddddg
}|�|�|�d�dS)Nrhr�r_r�r"r�g�����*3@r(r)rr|r�g["8���@r�r��@)rHr{r3r�rrr�test_even_number_repeated�sz+TestMedianGrouped.test_even_number_repeatedcCsLdddtdd�td�fD].}dD]$}|g|}|�|�|�t|��q qdS)Ng333333@�Dg��ޗCr�r�z32.9714r�)rrr3r{rr�rrrr��s
z,TestMedianGrouped.test_repeated_single_valuecCsPt}|dd�|dd�|dd�|dd�|dd�g}t�|�|�|�|�d�dS)Nrhr�r}rjr]�@r�r�rrrr�s,
z$TestMedianGrouped.test_odd_fractionscCsXt}|dd�|dd�|dd�|dd�|dd�|dd�g}t�|�|�|�|�d�dS)Nrhr�r}rjr]�
@r�r�rrrr�s4
z%TestMedianGrouped.test_even_fractionscCsFt}|d�|d�|d�|d�|d�g}t�|�|�|�|�d�dS)Nr;�6.5�7.5�8.5g@r�r�rrrr�s"
z#TestMedianGrouped.test_odd_decimalscCs�t}|d�|d�|d�|d�|d�|d�g}t�|�|�|�|�d�|d�|d�|d�|d�|d�|d�g}t�|�|�|�|�d�dS)Nr;r1r2r3�@g@r�r�rrrr�s(
(
z$TestMedianGrouped.test_even_decimalscCs�ddddddddddg
}|�|�|d�d	�dddddddddddg}|j|�|d�d
dd�d
d
ddddddddddg}|�|�|d�d�dS)Ng@r�r�r/r0r�r�r�g@g["8���@r(r)����ii�,�@iTr�g�p@r*r�rrr�
test_interval�szTestMedianGrouped.test_intervalcCsxdddg}|�t|j|�dddg}|�t|j|�dddg}d}|�t|j||�dddg}d}|�t|j||�dS)N��rrr|rz)r7rZintervalrrr�test_data_type_error�s



z&TestMedianGrouped.test_data_type_errorN)
r*r+r,r�r+r-r�rrrrr9r<rrrrr �s	
	r c@sTeZdZdd�Zdd�Zdd�Zdd�Zd	d
�Zdd�Zd
d�Z	dd�Z
dd�ZdS)�TestModecCstj|_dSrQ)r.�moder{rZrrrr��szTestMode.setUpcCsdddddddddddgS)	r
rr|r�r~r}rr�rrrZrrrr��szTestMode.prepare_datacCs"tddd�}|�|�|�d�dSr�rr�rrrr��szTestMode.test_range_datacCs4d}|�|�|�d�d��}|�|�|�d�dS)NZabcbdbrzfe fi fo fum fi fiZfi)r3r{r�r�rrr�test_nominal_dataszTestMode.test_nominal_datacCsDttd��}td�D]*}||g}t�|�|�|�|�|�qdS�Nr��rVrrr�r3r{)r7rrOr�rrr�test_discrete_data	s


zTestMode.test_discrete_datacCs<dddddddddddddddd	d	g}|�|�|�d�dS)
Nrrr|r�rhr�r~r�r}r�r�rrr�test_bimodal_datas&zTestMode.test_bimodal_datacCs"ttd��}|�|�|�d�dS)Nr�r)rVrr3r{r�rrr�test_unique_dataszTestMode.test_unique_datacCs|�t|jd�dSrQrzrZrrr�test_none_dataszTestMode.test_none_datacCs(t�ddddg�}|�|�|�d�dS)Nrr)r?�Counterr3r{r�rrr�test_counter_data%szTestMode.test_counter_dataN)r*r+r,r�r�r�r?rBrCrDrErGrrrrr=�sr=c@seZdZdd�ZdS)�
TestMultiModecCsBtj}|�|d�dg�|�|d�dddg�|�|d�g�dS)NZaabbbbbbbbccrZaabbbbccddddeeffffggr�r�r:)r.�	multimoder3)r7rIrrr�test_basics/szTestMultiMode.test_basicsN)r*r+r,rJrrrrrH-srHc@s$eZdZdd�Zdd�Zdd�ZdS)�	TestFMeanc
Cs�tj}t}t}dddgddf|d�|d�|d�gdd	f|d
d�|dd
�|dd�gddfdddddgddfdd|dd�gddfdtdddg�ddffD]2\}}}||�}|�t|�t|�|�|||�q�dS)Nr�r��@�@�floats�3.5�4.0�5.25�decimalsr~rr�rr��	fractionsTF�333333�?Zbooleans�mixed types)�r�r�rLrMrW�iterator)	r.�fmeanrrr}r+rrr3)r7rXr�r�r�
expected_meanr��actual_meanrrrrJ8s �	zTestFMean.test_basicsc	Cs�tj}tj}|�|��|g�W5QRX|�|��|tg��W5QRX|�t��|d�W5QRX|�t��|dddg�W5QRX|�t��|�W5QRX|�t��|dddgd�W5QRXdS)Nr�r�r��F)r.rXrr�r}rN)r7rXrrrr�test_error_casesIszTestFMean.test_error_casesc	Cs�tj}td�}td�}|�t�|d|g��d�|�t�|||g��d�|�t�|d|g��d�|�t��|||g�W5QRXdS�NZNan�Infr�r�znan and infinityZinfinity)	r.rXrrdrrr$r�r#)r7rX�NaNr^rrr�test_special_valuesYszTestFMean.test_special_valuesN)r*r+r,rJr\r`rrrrrK6srKc@s@eZdZdZdd�Zdd�Zdd�Zdd	�Zd
d�Zdd
�Z	dS)�VarianceStdevMixinr!cCs6dddtdd�td�fD]}|�|�|g�d�qdS)Nr�g������3@g���%�Br�r�z8.392rr�rzrrrr�osz$VarianceStdevMixin.test_single_valuecCsHdddtdd�td�fD]*}dD] }|g|}|�|�|�d�q qdS)	Nr�r�g@�6��<Cr|r~z62.4802)rr|rhr_rr�r�rrrr�ts
z-VarianceStdevMixin.test_repeated_single_valuecCs4dgd}|�|�}|j|ddd�|�|d�dS)Ng.�F7ݚ�?rwrXg��ؗ�Ҍ<r)r)r{rHr1r�rrr�test_domain_error_regression{s

z/VarianceStdevMixin.test_domain_error_regressionc
sNddddddddd	d
g
}|�|�}d��fdd
�|D�}|�|�|�|�dS)Ng{�G�z�?gR���Q�?g
ףp=
�?gR���Q@g�p=
ף@g��Q�	@r,g�Q���@g�G�z�@g��Q�@gj�@csg|]}|��qSrrr���shiftrrr�sz6VarianceStdevMixin.test_shift_data.<locals>.<listcomp>)r{rH�r7r�rrrrcr�test_shift_data�s

z"VarianceStdevMixin.test_shift_datac
sNdddddddddd	g
}|�|�}d
��fdd�|D�}|�|�|�|�dS)
Nrr|r�rhr~r}r�r�r�iʚ;csg|]}|��qSrrr�rcrrr�sz<VarianceStdevMixin.test_shift_data_exact.<locals>.<listcomp>r�rerrcr�test_shift_data_exact�s

z(VarianceStdevMixin.test_shift_data_exactcCs6dd�td�D�}|�|�}|�|�t|��|�dS)NcSsg|]}t�dd��qS)rr�rrrrrr�sz:VarianceStdevMixin.test_iter_list_same.<locals>.<listcomp>r�)rr{r3r}r�rrr�test_iter_list_same�s
z&VarianceStdevMixin.test_iter_list_sameN)
r*r+r,r'r�r�rbrfrgrhrrrrrags
	rac@s4eZdZdd�Zdd�Zdd�Zdd�Zd	d
�ZdS)�
TestPVariancecCstj|_dSrQ)r.�	pvariancer{rZrrrr��szTestPVariance.setUpcCs0ttd��}t�|�d}|�|�|�|�dS)NrwgP�_ArAr�rrr�test_exact_uniform�s
z TestPVariance.test_exact_uniformcCs&ddddg}d}|�|�|�|�dS)Nr�r~rjr�g�6@r��r7r�exactrrrr��szTestPVariance.test_intscCsXt}|dd�|dd�|dd�|dd�g}|dd�}|�|�}|�||�|�|t�dS)Nrr�r|r~r��rr{r3�assertIsInstance�r7r�rrmrsrrrr��s$

zTestPVariance.test_fractionscCsNt}|d�|d�|d�|d�g}|d�}|�|�}|�||�|�|t�dS)Nz12.1z12.2z12.5z12.9z0.096875�rr{r3ro�r7r�rrmrsrrrr��s
zTestPVariance.test_decimalsN)r*r+r,r�rkr�r�r�rrrrri�s
	ric@s<eZdZdd�Zdd�Zdd�Zdd�Zd	d
�Zdd�Zd
S)�TestVariancecCstj|_dSrQ)r.�variancer{rZrrrr��szTestVariance.setUpcCs6dddtdd�td�fD]}|�tj|j|g�qdS)Nrg33333�8@g�(G�!=Cr�r�z4.2084�rrr�r.rr{rzrrrr��szTestVariance.test_single_valuecCs&ddddg}d}|�|�|�|�dS)Nr�r~rjr�r�r�rlrrrr��szTestVariance.test_intscCsXt}|dd�|dd�|dd�|dd�g}|dd�}|�|�}|�||�|�|t�dS)Nrr�r|r~rrnrprrrr��s$

zTestVariance.test_fractionscCsZt}|d�|d�|d�|d�g}d|d�|d�}|�|�}|�||�|�|t�dS)Nrr~r}r�z9.5r|rqrrrrrr��s
zTestVariance.test_decimalscCs0d}|�|�|�d�|�|j|dd�d�dS)N�r�rnr�rn��xbarr�r�r�rrr�test_center_not_at_mean�sz$TestVariance.test_center_not_at_meanN)	r*r+r,r�r�r�r�r�ryrrrrrs�s		rsc@s$eZdZdd�Zdd�Zdd�ZdS)�
TestPStdevcCstj|_dSrQ)r.Zpstdevr{rZrrrr��szTestPStdev.setUpcCs8dd�td�D�}t�t�|��}|�|�|�|�dS)NcSsg|]}t�dd��qS)i��r%rrrrrr�sz7TestPStdev.test_compare_to_variance.<locals>.<listcomp>r�)rr�sqrtr.rjr3r{r�rrr�test_compare_to_variance�sz#TestPStdev.test_compare_to_variancecCs0d}|�|�|�d�|�|j|dd�d�dS)N)r|r�r~r�r�r�)�mur4r�r�rrrrysz"TestPStdev.test_center_not_at_meanN)r*r+r,r�r|ryrrrrrz�srzc@s,eZdZdd�Zdd�Zdd�Zdd�Zd	S)
�	TestStdevcCstj|_dSrQ)r.�stdevr{rZrrrr�szTestStdev.setUpcCs6dddtdd�td�fD]}|�tj|j|g�qdS)N�QgH�z�wi@gf7?+�Brhr�z35.719rurzrrrr�szTestStdev.test_single_valuecCs8dd�td�D�}t�t�|��}|�|�|�|�dS)NcSsg|]}t�dd��qS)rgr}rrrrrrsz6TestStdev.test_compare_to_variance.<locals>.<listcomp>r�)rrr{r.rtr3r{r�rrrr|sz"TestStdev.test_compare_to_variancecCsd}|�|j|dd�d�dS)Nrvrnrwr�r�r�rrrrysz!TestStdev.test_center_not_at_meanN)r*r+r,r�r�r|ryrrrrr~sr~c@s4eZdZdd�Zdd�Zdd�Zdd�Zd	d
�ZdS)�TestGeometricMeancCstj}|�|dddg�d�|�|ddg�d�|�|dg�d�t�d	�td
d�td
d�td
d
�tdd
d�td
dd�ddddddgdd�td�D�dd�td�D�dd�td�D�f	D]B}t�tt	|��t	d
�t
|�}||�}|�t�|t
|���q�dS)Nr^r%r��B@r�g"@g@g�1@l���;rrjr�rwr�r|rr�r]rjrh�xr~cSsg|]}t�d��qS)�I@�r�expovariate�r�rOrrrr*sz1TestGeometricMean.test_basics.<locals>.<listcomp>cSsg|]}t�dd��qS)r�r/)rZlognormvariater�rrrr+s��cSsg|]}t�ddd��qS)r��i�)rZ
triangularr�rrrr,sr�)r.�geometric_mean�assertAlmostEqualr�seedrrZprodr�rrIrd�iscloser)r7r��rngZ
gm_decimalZgm_floatrrrrJs$


� zTestGeometricMean.test_basicsc	Cs�tj}t}t}d}dddgdf|d�|d�|d�gd	f|d
d�|dd
�|dd�gdfdd|dd�gdfdtdddg�dffD]2\}}||�}|�t|�t|�|j||dd�q|dS)Ng�d�@r�r�rLrNrOrPrQrRr~rr�rr�rSrU)rVrWrWrh�Zplaces)	r.r�rrr}r+rrr�)r7r�r�r�rYrr�rZrrr�test_various_input_types2s�z*TestGeometricMean.test_various_input_typescCs�tj}d}|d|d|d|g�}|�t�|d|��|�t�|��d}|d|d|d|g�}|�t�|d|��|�|d�dS)Ngp~gK@g8@r�gprX)r.r�rdrr�r�r$�assertNotEqual)r7r�ZlargeZbig_gmZsmallZsmall_gmrrr�test_big_and_smallDsz$TestGeometricMean.test_big_and_smallc	Cstj}tj}|�|��|g�W5QRX|�|��|dddg�W5QRX|�|��|dddg�W5QRX|�|��|tg��W5QRX|�t��|d�W5QRX|�t��|dddg�W5QRX|�t��|�W5QRX|�t��|dddgd�W5QRXdS)	Nr�rXrLg�r�r�r�r[)r.r�rr�r}rN)r7r�rrrrr\Ss$z"TestGeometricMean.test_error_casesc	Cs�tj}td�}td�}|�t�|d|g��d�|�t�|||g��d�|�t�|d|g��d�|�t��|||g�W5QRXdSr])	r.r�rrdrrr$r�r#)r7r�r_r^rrrr`gsz%TestGeometricMean.test_special_valuesN)r*r+r,rJr�r�r\r`rrrrr�s
r�c@s4eZdZdd�Zdd�Zdd�Zdd�Zd	d
�ZdS)�
TestQuantilescsltj}dddddg}t�|�dgfddgfd	d
dgfdd
ddgfdddddgfddd
dddgfddd
dddddgfdddddddddd g	fd!d"dd
d
d#dd$dddd%gfd&d'd(dd)d
dd*d+ddd,dd-d.gff
D�]R\}}|�||||d/��|�t|||d/��|d�tttfD]F�|t	�|�|d/�}|�
�fd0d1�|D��|�|tt	�|����qt|�dk�r||�|||d/�|�t|�}d|d2|d}d|d3|d4}|||g}	|�|||d/�||	|d5d6�||f�d7d8�}
tt	|
|��}|t	|
|�|d/�}|�
t
d9d1�t||�D���q�tdd:�D]8}
tjtd;�|
d<�}||�\}}}|�|t�|���q.dS)=Nr�r#�r8i^rr�@o@r|�i@�t@r��d@g�t@rhga@g�k@g@r@g�u@r��^@��u@r��Y@g�j@g�r@gXv@r�gV@gg@g`t@g�v@r��T@� l@g�q@g�v@r_gR@gZ@ge@�n@g�p@g�t@g@v@gw@�rc3s|]}tt|��k�VqdSrQ��allrr��Zdatatyperrr��sz4TestQuantiles.test_specific_cases.<locals>.<genexpr>rrYrg�	inclusive�r�methodcSsd|dS�Nr�g3333�J�@rr	rrrr��sz,TestQuantiles.test_specific_cases.<locals>.fcss|]\}}t�||�VqdSrQ�rr��r�rPrrrrr��sr�rj��k)r.�	quantilesrr�r3rIrrrr�rdrVr�r�rMr�choicesr)r7r�rrrrsZsdata�lo�hiZpadded_datar��exp�actr��q1�q2�q3rr�r�test_specific_casesus`

���
� z!TestQuantiles.test_specific_casescs�tj}ddddg}t�|�dgfddgfdd	d
gfdddd
gfdddddgfddd	dd
dgfddddddd
dgfdddddddddd g	fd!d"ddd	d#dd$d
d
dd%gfd&d'd(dd)d	dd*d+dd
d,dd-d.gff
D]�\}}|�||||d/d0��|�t|||d/d0��|d�tttfD]H�|t	�|�|d/d0�}|�
�fd1d2�|D��|�|tt	�|����qd3d4�}tt	||��}|t	||�|d/d0�}|�
td5d2�t
||�D���q�|�|d6dgdd/d0�d7d8d9d:d;d<d=d>d?g	�|�|td6d@�dd/d0�d7d8d9d:d;d<d=d>d?g	�dAdB�tdC�D�}||dDd/d0�}	|�t|��|�t|��||dDdE�}|�||	�tddF�D]<}
tjtd�|
dG�}||d/dH�\}}}
|�|t�|���q^dS)INrjr#i�i rrg�r@r|r�gy@r�g�e@g@@rhr�r�g�v@g��@r�g�b@g��@r�g0a@�r�gpw@gP�@r�g@`@g�g@g�{@g@�@r�g@_@r�r�g�@r_r�g�a@g�f@g�q@r�g~@g�@g��@r�r�c3s|]}tt|��k�VqdSrQr�r�r�rrr��sz>TestQuantiles.test_specific_cases_inclusive.<locals>.<genexpr>cSsd|dSr�rr	rrrr��sz6TestQuantiles.test_specific_cases_inclusive.<locals>.fcss|]\}}t�||�VqdSrQr�r�rrrr��sr�$@r�g>@gD@r�gN@g�Q@r�g�V@r�cSsg|]}t�d��qS)rw)rZ	randranger�rrrr�sz?TestQuantiles.test_specific_cases_inclusive.<locals>.<listcomp>i�� r�r�r��r�)r.r�rr�r3rIrrrr�rdrVr�rMrr�rcrr�r)r7r�rrrrsr�r�r�rr�r�r�r�rr�r�test_specific_cases_inclusive�sd

��� ��z+TestQuantiles.test_specific_cases_inclusivecCsTtj}tdd�D]>}dg|}|�||�dddg�|�||dd�dddg�qdS)Nrr�r�r�r�)r.r�rr3)r7r�rrrrr�test_equal_inputs�s
�zTestQuantiles.test_equal_inputsc	s�tj}d}dd�t|�D��tt���|kr>��t�d��q���dD]:}||}|�	�fdd�|�|d�D�t
t|||���qJdD]\}||||d	h}�fd
d�|�|d�D�}dd�t||d	d��D�}|�||k�q�dS)
NrwcSsg|]}t�d��qS)皙�����?r�r�rrrr�sz9TestQuantiles.test_equal_sized_groups.<locals>.<listcomp>r�)
rrrhr�r�r�rjr#r�r�r�r3rwcsg|]}t��|��qSr��bisect�r��qr�rrr�sr�)
rjr��;�m��i;i�isi�i)&rcsg|]}t��|��qSrr�r�r�rrr�scSsh|]\}}||�qSrr)r��pr�rrr�	<setcomp>�sz8TestQuantiles.test_equal_sized_groups.<locals>.<setcomp>)
r.r�rrI�setr�rr��sortr3rVrMrd)r7r��totalrZ
group_sizeZgroup_sizes�posZsizesrr�r�test_equal_sized_groups�s"�z%TestQuantiles.test_equal_sized_groupsc	Cshtj}tj}|�t��|�W5QRX|�t��|dddgddd�W5QRX|�t��|dddgd�W5QRX|�|��|dddgdd�W5QRX|�|��|dddgdd�W5QRX|�t��|dddgd	d�W5QRX|�t��|dddgd
d�W5QRX|�|��|dgdd�W5QRX|�t��|dddgdd�W5QRXdS)Nr�r�r�rjr�r�rrYr��Xr�)r.r�rr�rNr#)r7r�rrrrr\	s(zTestQuantiles.test_error_casesN)r*r+r,r�r�r�r�r\rrrrr�ss
66r�c@s�eZdZdd�Zdd�Zdd�Zdd�Zd	d
�Zdd�Ze	j
d
d��Zdd�Zdd�Z
dd�Zdd�Zdd�Zdd�Zdd�Zdd�Zdd �Zd!d"�Zd#S)$�TestNormalDistc	CsB|j�dd�}|�t��t|�W5QRX|�t|j�d�dS)Nr7r�)Z_muZ_sigma)r��
NormalDistr�rN�varsr3rW�	__slots__�r7�ndrrr�
test_slots 	szTestNormalDist.test_slotsc	Cs�|j�dd�}|�|jd�|�|jd�|�|jd�|j��}|�|jd�|�|jd�|�|jd�|�|jj��|j�dd�W5QRXGdd�d|jj�}|d	d
�}|�t|�|�dS)Nr�r]i!rri����c@seZdZdS)zGTestNormalDist.test_instantiation_and_attributes.<locals>.NewNormalDistNr&rrrr�
NewNormalDist7	sr�r#rh)	r�r�r3r�rrtr�rr)r7r�r��nndrrr�!test_instantiation_and_attributes&	s

z0TestNormalDist.test_instantiation_and_attributesc	Cs�|jj}dddddg}|�|�|�|dd��|�|�t|��|dd��|�|�t|��|dd��|�|jj��|�g�W5QRX|�|jj��|�dg�W5QRXGd	d
�d
|�}|�|�}|�t|�|�dS)N�`ri�Zr��nrr}r�c@seZdZdS)zBTestNormalDist.test_alternative_constructor.<locals>.NewNormalDistNr&rrrrr�L	sr�)	r�r�r3Zfrom_samplesrWr}r�rr)r7r�rr�r�rrr�test_alternative_constructor<	s
z+TestNormalDist.test_alternative_constructorcCs�|jj}d\}}|||�}d}|�|�}|�t|�|�|�ttt|��th�|j�	|�}|�
||d|ko�||dkn�d}|j|dd�}|j|dd�}	|j|dd�}
|j|dd�}|�||
�|�|	|�|�||	�dS)N)rwr/r�r�rjzhappiness and joy)r�ztrouble and despair)r�r�Zsamplesr3rIr�r�rrr�rdr�)r7r�r}�sigmar�rrrxZdata1Zdata2Zdata3Zdata4rrr�test_sample_generationQ	s"

.z%TestNormalDist.test_sample_generationc
3Cs�|jj}|dd�}|�|�d�|�d��|�|�d�|�d��td�D]$}|�|�d|�|�d|��qJd}tdd�D]6}|�||�|�|�|}|j|�|�|d	d
�q~|�}tddddd
ddddddddddddddddddd d!d"d#d$d%d&d'd(d)d*d+d,d-d.d/d0d1d2d3d4d5d6d7d8d9d:g2�D]@\}}|j|�|d;�|d	d
�|j|�|d;�|d	d
��q(|dd<�}	|�|jj	��|	�d�W5QRX|�
|�td=��d>�|�
|�td?��d>�|�t
�|�td@����dS)ANrjr_rr�r�gP?r�rr�r�g+��ݓ��?ggDio��?g������?gV}��b�?g�Q�|�?gF���x�?g��g��s�?g�٬�\m�?g�� �rh�?g�K7�A`�?g��|гY�?g����Q�?g���QI�?gsh��|?�?g�=yX�5�?g|a2U0*�?g��Q��?g���N@�?g�/�$�?g~��k	��?g]�C����?gw��/��?g�~�:p��?g�>W[���?gM�
O��?gW[����?g鷯��?g{�G�z�?g
q���h�?g�|a2U�?g��K7�A�?gvq
�-�?gj�t��?g�c]�F�?g�\�C���?gףp=
��?g�����?ga2U0*��?g�y�):��?g(��y�?g��N@a�?gf�c]�F�?g"lxz�,�?g�O��n�?g�3��7��?g�e�c]��?g��n���?g��T����?r�r�-InfrXr^r_)r�r�Z
assertLess�pdfrr��cdfrLr�rr3rrdrr)
r7r�r�rO�dxr
Zest_pdf�ZZpx�Yrrr�test_pdfg	s�
"� 
zTestNormalDist.test_pdfc	s"|jj}|dd���fdd�tdd�D�}|�ttt|��th�|�|t|��|���	d�d�|�}dD]:\}}|j
|�	|�|d	d
�|j
|�	|�d|d	d
�qn|dd�}|�|jj��|�	d
�W5QRX|���	td��d�|���	td��d�|�
t���	td����dS)Nrjr_csg|]}��|��qSr)r�r��r�rrr�	sz+TestNormalDist.test_cdf.<locals>.<listcomp>rr#r�))rXr�)r�gqZ� �?)r�g��E_A�?)g�Q���?gGɫs��?)g��(\��?g؞Y���?)g��Q��?g���9#�?)gH�z�G�?g&S���?)rTg�MbX9�?)g���Q��?gT㥛� �?)g�������?g�?�?)gffffff@g_�x�Z�?)g��Q�@g��#0��?)g)\��(@gu<f�2��?)gףp=
�@gVe����?)gH�z�G@g9���?rhr�r�rr�r�rXr^r_)r�r�rr3r�r�rrr�r�r�r�rrdrr)r7r�Zcdfsr��zZcum_probr�rr�r�test_cdf�	s 

zTestNormalDist.test_cdfc	CsJ|jj}|dd�}|�|�d�|j�|�}dddd�}|��D]b\}}t|dd	�D]L\}}|d
|}	|j|�|	�|dd�d
|	}	|j|�|	�|dd�qTq@|�|dd��d�d�d}
td|
�D]$}	|	|
}	|�|�	|�|	��|	�q�tdd�D]F}d|}	|�|�	|�|	��|	�d
|	}	|�|�	|�|	��|	�q�td�D]"}|j|�|�	|��|dd��qJ|�
|jj��|�d�W5QRX|�
|jj��|�d�W5QRX|�
|jj��|�d
�W5QRX|�
|jj��|�d�W5QRX|�
|jj��|dd�}|�d�W5QRX|�t
�|�td����dS)Nrjr_r�)
rXgR���Q�?g�S㥛@g���S
@gT㥛� @g^�I�@g� �rh�@g+��N@g��C��@gV-��o@)
g� �rh��?g\��(\�?g�~j�t@g+��@g�MbX9@g�(\�B@g��v��@g����@g��Sc@g�K7�A�@)
gP��n��?g�S㥛@g���Q�@g��n��
@g��(\@gP��n@g�����@g��� �r@gˡE��@g�l�q@)r�r�r�r)�startr�r|r�r�rkr�g��>���?g���E@i�3rnr#rhrXr�g�������?rr_)r�r�r3Zinv_cdfr��itemsrLr�rr�r�rrdrrr)r7r�Ziqr�Zppr�rowr�r
r�rrPrrr�test_inv_cdf�	sL
�
 
zTestNormalDist.test_inv_cdfcCsj|j��}dgfddgfdddgfdddd	gffD]2\}}|j|d
�}|�tdd�t||�D���q2dS)
NrrrXr|g�ǘ���ۿg�ǘ����?r�g/�$���g/�$���?r�css"|]\}}tj||dd�VqdS)r�)Zabs_tolNr�r�rrrr��	s�z0TestNormalDist.test_quantiles.<locals>.<genexpr>)r�r�r�rdr�rM)r7r�rrrrrr�test_quantiles�	s

��zTestNormalDist.test_quantilescCs~|jj}|dd�|dd�df|dd�|dd�dffD]6\}}}|j|�|�|dd�|j|�|�|dd�q4dd	d
�dd�}|dd�|dd�f|dd�|dd�f|dd�|dd�f|d
d�|dd�f|dd�|dd�f|dd�|dd�f|dd�|dd�f|dd�|dd�f|dd�|dd�f|dd�|dd�f|dd�|dd�f|dd�|dd�f|dd�|dd�f|dd�|dd�ffD]B\}}|j|�|�|||�d	d�|j|�|�|||�d	d��q||�}|�t��|��W5QRX|�t��|�||�W5QRX|�t��|�d�W5QRX|�|jj��|�|dd��W5QRX|�|jj��|dd��|�W5QRXdS) NrXrnr�gɎ�@���?gM-[닄�?r�r�i rh)�stepsr�cs�tj}|j|jd}|t|j|j�}||�d||���fdd�t|�D�}tt|j|��}tt|j|��}	t||�||	��}
|tt	||	��|
S)z0Numerical integration cross-check for overlap() rncsg|]}�|��qSrrr��r�r�rrr
szHTestNormalDist.test_overlap.<locals>.overlap_numeric.<locals>.<listcomp>)
rr�r�rrrrVr�r�rc)r�r�r�r�r��center�widthZx_arrZxpZypr�rr�r�overlap_numeric
sz4TestNormalDist.test_overlap.<locals>.overlap_numericr[�Ar�rjr_r�ri����r��rr�gj�t��?g�~j�t�h?gj�t��?ga2U0*�3?g��MbX�?rr)r�r�r�Zoverlapr�rNr)r7r�ZX1ZX2Zpublished_resultr�r�rrr�test_overlap�	sF�� zTestNormalDist.test_overlapcCsX|j�dd�}|�|jd�|�|jd�|�|jd�|�|jd�|�|jd�dS)Nrjr_��)r�r�r3r�rr>rrt)r7r�rrr�test_properties9
szTestNormalDist.test_propertiescCsL|jj}|dd�}|dd�}|�|||dd��|�|||dd��dS)Nrjr�rkrh�rjr�)r�r�r3�r7r�r�r�rrr�'test_same_type_addition_and_subtractionA
s


z6TestNormalDist.test_same_type_addition_and_subtractionc	Cs�|jj}|dd�}d}|�|
|dd��|�||dd��|�|||dd��|�|||dd��|�|||dd��|�|||dd��|�|||dd	��|�|||dd	��|�|||dd
��|�t��||W5QRXdS)Nrjr_r�rr�r�i����r�r�r�)r�r�r3r�rN)r7r�r�r%rrr�test_translation_and_scalingH
s
z+TestNormalDist.test_translation_and_scalingcCs||jj}|dd�}|
}|�||�|�|j|j�|�|j|j�|}|�||�|�|j|j�|�|j|j�dS)Nrjr�)r�r�ZassertIsNotr3r�rr�rrr�test_unary_operationsX
s
z$TestNormalDist.test_unary_operationscCs"|jj}|�}|dd�}|�}|dd�}|dd�}|dd�}|�||�|�||�|�||�|�||�|�||�Gdd�d�}|�}	|�|�|	�t�|�||	kd�|�|	|kd�Gdd�d|�}
|
d	d
d�}|d	d
�}|�||�Gdd
�d
�}|d	d
�}
|d	d
�}|�||
�dS)Nrr�r�c@seZdZdd�ZdS)z'TestNormalDist.test_equality.<locals>.AcSsdSr@rrerrr�__eq__t
sz.TestNormalDist.test_equality.<locals>.A.__eq__N)r*r+r,r�rrrrrHs
srHr�cseZdZ�fdd�Z�ZS)z5TestNormalDist.test_equality.<locals>.SizedNormalDistcst��||�||_dSrQ)rc�__init__r)r7r}r�rrfrrr�~
sz>TestNormalDist.test_equality.<locals>.SizedNormalDist.__init__)r*r+r,r�rhrrrfr�SizedNormalDist}
sr�rjr_�9c@seZdZdd�ZdS)z3TestNormalDist.test_equality.<locals>.LognormalDistcSs||_||_dSrQ)r}r�)r7r}r�rrrr��
sz<TestNormalDist.test_equality.<locals>.LognormalDist.__init__N)r*r+r,r�rrrr�
LognormalDist�
sr�)r�r�r�r3r��NotImplemented)r7r��nd1�nd2�nd3Znd4Znd5Znd6rHrr��sr�Zlndr�rrr�
test_equalityd
s2






zTestNormalDist.test_equalitycCsZ|j�dd�}t�|�}|�||�t�|�}|�||�t�t�|��}|�||�dS)N��B@��@)r�r��copyr3Zdeepcopy�pickle�loads�dumps)r7r�rrrrrr�test_pickle_and_copy�
s

z#TestNormalDist.test_pickle_and_copycCsH|jj}|dd�|dd�|dd�|dd�|dd�h}|�t|�d�dS)Nrjr_r�g.@r�r`r|)r�r�r3rI)r7ZNDrrrr�test_hashability�
s,zTestNormalDist.test_hashabilitycCs"|j�dd�}|�t|�d�dS)Nrrz NormalDist(mu=37.5, sigma=5.625))r�r�r3�reprr�rrr�	test_repr�
szTestNormalDist.test_reprN)r*r+r,r�r�r�r�r�r�rZskip_if_pgo_taskr�r�r�r�r�r�r�rrrrrrrrr�	s$	'
@
?,	r�c@s eZdZeZdd�Zdd�ZdS)�TestNormalDistPythoncCs|jtjd<dSr1�r�r�modulesrZrrrr��
szTestNormalDistPython.setUpcCsttjd<dSr1�r.rrrZrrr�tearDown�
szTestNormalDistPython.tearDownN)r*r+r,r5r�r�rrrrrr�
srr9c@s eZdZeZdd�Zdd�ZdS)�TestNormalDistCcCs|jtjd<dSr1rrZrrrr��
szTestNormalDistC.setUpcCsttjd<dSr1rrZrrrr�
szTestNormalDistC.tearDownN)r*r+r,r:r�r�rrrrrr�
srcCs|�t���|S)z&Used for doctest/unittest integration.)ZaddTestsrZDocTestSuite)�loaderZtests�ignorerrr�
load_tests�
sr�__main__)r!r")Mr-r�r?Zcollections.abcrr�rrrrrr<ZtestrrrSrr.r
rr r(r)Zimport_fresh_moduler5r:r�r0r>rWr\rrr�r�r�r�r�r�rr
rr/r:rAr\rlryr�r�r�r�r�r�r�r�r
rrrr r=rHrKrarirsrzr~r�r�r�rr=rrr*�mainrrrr�<module>s�
:+
_
>@w-@4d:c'H
@`X9
r:	1B%'Y%