|
Server : LiteSpeed System : Linux server51.dnsbootclub.com 4.18.0-553.62.1.lve.el8.x86_64 #1 SMP Mon Jul 21 17:50:35 UTC 2025 x86_64 User : nandedex ( 1060) PHP Version : 8.1.33 Disable Function : NONE Directory : /opt/cppython/lib/python3.8/test/__pycache__/ |
U
>��g�� � @ s@ d Z ddlZddlZddlZddlZddlZddlZddlZddlZddl Z ddl
Z
ddl Z ddl m
Z
ddlmZ ddlmZ ddl m
Z
ddlZdd� Zdd � Zd
d � Zdvdd�ZG dd� d�Ze
jddgd�Ze
jddgd�ZG dd� de j�ZG dd� de j�Z G dd� de j�Z G d d � d e j�Z G d d � d e j�Z G d d!� d!e j�Z G d"d#� d#e j�Z!G d$d%� d%e j�Z"G d&d'� d'e j�Z#G d(d)� d)e j�Z$G d*d+� d+e j�Z%G d,d-� d-e j�Z&G d.d/� d/e j�Z'G d0d1� d1e j�Z(G d2d3� d3e j�Z)G d4d5� d5e j�Z*G d6d7� d7e j�Z+G d8d9� d9e j�Z,G d:d;� d;e j�Z-G d<d=� d=�Z.G d>d?� d?�Z/G d@dA� dAe.e/�Z0G dBdC� dCe �Z1G dDdE� dEe �Z2G dFdG� dGe �Z3G dHdI� dIe.�Z4G dJdK� dKe e4e/�Z5G dLdM� dMe e4e/�Z6G dNdO� dOe e4�Z7G dPdQ� dQe e/�Z8G dRdS� dSe7e/�Z9G dTdU� dUe7e/�Z:G dVdW� dWe7�Z;G dXdY� dYe e4e/�Z<G dZd[� d[e j�Z=G d\d]� d]e j�Z>G d^d_� d_e.�Z?G d`da� dae?e e/�Z@G dbdc� dce?e e/�ZAG ddde� dee?e �ZBG dfdg� dge?e �ZCG dhdi� die j�ZDG djdk� dke j�ZEG dldm� dm�ZFG dndo� doe jeF�ZGe �Hedp�G dqdr� dre jeF��ZIdsdt� ZJeKduk�r<e �L� dS )wz_Test suite for statistics module, including helper NumericTestCase and
approx_equal function.
� N)�support)�Decimal��Fractionc C s t �d| �S )z:Return -1.0 for negatives, including -0.0, otherwise +1.0.� )�math�copysign��x� r �3/opt/cppython/lib/python3.8/test/test_statistics.py�sign s r
c C sZ t | �t |�k rdS t| 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�bZaexpZbexpr r r �
_nan_equal" s
r c C s: t t| �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_errr r r � _calc_errorsA s
r ��-���q=�H�����z>c C s� |dk s|dk rt d��t�| �s,t�|�r0dS | |kr<dS t�| �sPt�|�rTdS t| | �}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.
r z%error tolerances must be non-negativeFT)�
ValueErrorr r �isinfr r )r
�y�tol�relZ actual_errorZ
allowed_errorr r r � approx_equalQ s " r( c @ s e Zd ZdZdS )�
_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__r r r r r) � s #r) �
statistics� _statistics)Zblocked)Zfreshc @ s. e Zd ZdgZdd� Ze�ed�dd� �ZdS )� TestModulesZ_normal_dist_inv_cdfc C s$ | j D ]}| �tt|�jd� qd S �Nr. )�
func_names� assertEqual�getattr�
py_statisticsr+ ��selfZfnamer r r �test_py_functions� s
z TestModules.test_py_functions�requires _statisticsc C s$ | j D ]}| �tt|�jd� qd S )Nr/ )r2 r3 r4 � c_statisticsr+ r6 r r r �test_c_functions� s
z TestModules.test_c_functionsN) r* r+ r, r2 r8 �unittest�
skipUnlessr: r; r r r r r0 � s
r0 c @ s@ e Zd ZdZd ZZd 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.
r Nc C sZ |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�checkr r r �assertApproxEqual� s � �z!NumericTestCase.assertApproxEqualc
C sn t |�t |�kr:dt |�t |�f }| �||�}| �|��tt||��D ] \}\}} | �|| ||||� qHd S )Nz0sequences differ in length: %d items != %d items)�len�_formatMessage�failureException� enumerate�ziprC )
r7 rD rE r& r' rF � standardMsg�ir �er r r rB � s ��
z!NumericTestCase._check_approx_seqc C s> t ||||�rd S | �|||||�}| �||�}| �|��d S �N)r( �_make_std_err_msgrJ rK )r7 rD rE r& r' rF �idxrN r r r rC s
z!NumericTestCase._check_approx_numc C s> d}|d k r d| }|| }t | |�\}}|| |||||f S )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 ) rD rE r& r' rS �template�headerr r r r r rR s �z!NumericTestCase._make_std_err_msg)NNN)N) r* r+ r, r- r& r' rH rB rC � staticmethodrR r r r r r> � s �
,
r> c @ s e Zd ZdZdd� ZdS )�TestSignz5Test that the helper function sign() works correctly.c C s$ | � td�d� | � td�d� d S )N� r g ����)r3 r
�r7 r r r �
testZeroes( s zTestSign.testZeroesN)r* r+ r, r- r[ r r r r rW & s rW c @ s, e Zd Zdd� Zdd� Zdd� Zdd� Zd S )
�ApproxEqualSymmetryTestc C sT dddt d�tdd�g}ddd t d
�td d�g}t||�D ]\}}| �||� q:d S ) Ni� gfffff�B@gfffff�(�z2.54� �6 i� g������B@gR� ���(�z2.59� )r r rM �do_relative_symmetry)r7 Zargs1Zargs2r r r r r �test_relative_symmetry3 s
z.ApproxEqualSymmetryTest.test_relative_symmetryc C st t ||�t||� }}|| }t|| �t|| � }}|| d }| �t||d|d�� | �t||d|d�� d S )Nr r �r& r' )�minr r �
assertTruer( )r7 r r �deltaZrel_err1Zrel_err2r' r r r r` C s z,ApproxEqualSymmetryTest.do_relative_symmetryc C s dddddg}d}|D ]�}t tttfD ]�}||�d }|| }t|t||� �}| j||||d� | j|||d d| d� | j|||d |d d� | j||||d d� | j|||d |d� | j|||d d| d� | j||d
d
d� | j||d
d
d� q&qd S ) N������� �k i�m r �d rb r r )�intr r r r r �do_symmetry_test)r7 �argsre r �type_r
r% �rr r r �
test_symmetryO s z%ApproxEqualSymmetryTest.test_symmetryc
C s@ d}t ||||�}t ||||�}| �|||�||||f�� d S )Nz+approx_equal comparisons don't match for %r)r( r3 �format)r7 r r r& r' rT Zflag1Zflag2r r r rl g s z(ApproxEqualSymmetryTest.do_symmetry_testN)r* r+ r, ra r` rp rl r r r r r\ 0 s r\ c @ sT e Zd Zdd� Zdd� Zdd� Zdd� Zd d
� Zd d � Zd
d� Z dd� Z
dd� Z dS )�ApproxEqualExactTestc C sJ t ||||d�}| �|d| � t | | ||d�}| �|d| � d S )Nrb zequality failure for x=%r)r( rd )r7 r
r& r' �resultr r r �do_exactly_equal_testt s z*ApproxEqualExactTest.do_exactly_equal_testc C s dD ]}| � |dd� qd S )N)�* i M i~: �� i iU�
i� r �rt �r7 r r r r �test_exactly_equal_intsz s z,ApproxEqualExactTest.test_exactly_equal_intsc C s dD ]}| � |dd� qd S )N)g�z�G��?g/�$���?g�����e�@g 7@g pf@g!�rh��Q@gB`��"KB@r rw �r7 r
r r r �test_exactly_equal_floats s z.ApproxEqualExactTest.test_exactly_equal_floatsc C sN t }|dd�|d�|dd�|dd�|dd �|dd�fD ]}| �|dd� q6d S )
Nr r r rh � � � �# �$ )r rt �r7 �F�fr r r � test_exactly_equal_fractions� s 6z1ApproxEqualExactTest.test_exactly_equal_fractionsc C s* t }t|d�� �D ]}| �|dd� qd S )Nz 8.2 31.274 912.04 16.745 1.2047r )r �map�splitrt )r7 �D�dr r r �test_exactly_equal_decimals� s z0ApproxEqualExactTest.test_exactly_equal_decimalsc C sF dD ]<}| � |dd� | � |d dd� t|d�}| � |dd� qd S )N)� i� i\ i� i� � �{�G�z�?r �
i� )rt r )r7 r r� r r r �test_exactly_equal_absolute� s
z0ApproxEqualExactTest.test_exactly_equal_absolutec C s2 | � td�td�d� | � td� td�d� d S )Nz3.571�0.01r z81.3971)rt r rZ r r r �$test_exactly_equal_absolute_decimals� s z9ApproxEqualExactTest.test_exactly_equal_absolute_decimalsc C s@ dddt dd�fD ]}| �|dd� q| �td�dtd �� d S )
Ni� g33333SY@g�z�G�rh � r r� z11.68r� �r rt r rz r r r �test_exactly_equal_relative� s z0ApproxEqualExactTest.test_exactly_equal_relativec C sH dddt dd�fD ]}| �|dd� qt}| �|d�|d �|d
�� d S ) Ni9� gˡE��0@g\��(h��r| � 皙�����?r� z7.2z0.1r� r� )r7 r
r� r r r �test_exactly_equal_both� s z,ApproxEqualExactTest.test_exactly_equal_bothN) r* r+ r, rt ry r{ r� r� r� r� r� r� r r r r rr n s rr c @ s4 e Zd Zdd� Zdd� Zdd� Zdd� Zd d
� Zd S ) �ApproxEqualUnequalTestc C s8 || fD ](}t ||d ddd�}| �|d| � q
d S )Nr r rb zinequality failure for x=%r)r( � assertFalse)r7 r
r rs r r r �do_exactly_unequal_test� s z.ApproxEqualUnequalTest.do_exactly_unequal_testc C s dD ]}| � |� qd S )N)i� i�� i� i� iXC �r� rx r r r �test_exactly_unequal_ints� s z0ApproxEqualUnequalTest.test_exactly_unequal_intsc C s dD ]}| � |� qd S )N)g��Q� #@g���� [�@gfffff�G@gףp=
W"@g=
ףp=1@r� rz r r r �test_exactly_unequal_floats� s z2ApproxEqualUnequalTest.test_exactly_unequal_floatsc C s< t }|dd�|dd�|dd�|dd�fD ]}| �|� q(d S ) Nr rh r~ r} � � �e iς )r r� r� r r r � test_exactly_unequal_fractions� s (z5ApproxEqualUnequalTest.test_exactly_unequal_fractionsc C s" t td�� �D ]}| �|� qd S )Nz!3.1415 298.12 3.47 18.996 0.00245)r� r r� r� �r7 r� r r r � test_exactly_unequal_decimals� s z4ApproxEqualUnequalTest.test_exactly_unequal_decimalsN)r* r+ r, r� r� r� r� r� r r r r r� � s
r� c @ s� e Zd Zdd� Zdd� Zdd� Zdd� Zd d
� Zd d � Zd
d� Z dd� Z
dd� Z dd� Z dd� Z
dd� Zdd� Zdd � Zd d � Zd d � Zd!S )"�ApproxEqualInexactTestc C sb d}|| || fD ]H}|� ||�}| �t||d| dd�|� | �t|||d dd�|� qd S )N� Test failure for x={!r}, y={!r}r r rb �rq rd r( r� �r7 r
re rT r% rF r r r �do_approx_equal_abs_test� s
z/ApproxEqualInexactTest.do_approx_equal_abs_testc C s& dD ] }| � |d� | � |d� qd S )N) i��iI���i����rg r r r} �% i� i�& i6�jr� r �r� rx r r r � test_approx_equal_absolute_ints� s z6ApproxEqualInexactTest.test_approx_equal_absolute_intsc C s2 dD ](}| � |d� | � |d� | � |d� qd S )N) g�t��q�gfffffFX�g333333 �g333333�� �?� �?g333333 @g�Q� ��@g�����ҭ@� �?r� �-C ��6?r� rz r r r �!test_approx_equal_absolute_floats� s z8ApproxEqualInexactTest.test_approx_equal_absolute_floatsc C sX t dd�}dddddddd d
d d g }d
d� |D �D ] }| �||� | �|t|�� q2d S )Nr � i������rg rY r rh r] � �" �G c s s | ]}t |d �V qdS )r� Nr )�.0r r r r � <genexpr>� s zNApproxEqualInexactTest.test_approx_equal_absolute_fractions.<locals>.<genexpr>)r r� r )r7 re �
numeratorsr� r r r �$test_approx_equal_absolute_fractions� s
z;ApproxEqualInexactTest.test_approx_equal_absolute_fractionsc C s: t d�}tt d�� �D ] }| �||� | �| |� qd S )Nr� z 1.0 3.5 36.08 61.79 7912.3648)r r� r� r� )r7 re r� r r r �#test_approx_equal_absolute_decimals� s z:ApproxEqualInexactTest.test_approx_equal_absolute_decimalsc C s | � tddddd�� d S )Ng�h㈵��>g�h㈵��r� r rb )rd r( rZ r r r �test_cross_zero� s z&ApproxEqualInexactTest.test_cross_zeroc C sj d}|d| |d| fD ]H}|� ||�}| �t||dd| d�|� | �t||d|d d�|� q d S )Nr� r r r rb r� r� r r r �do_approx_equal_rel_test� s
z/ApproxEqualInexactTest.do_approx_equal_rel_testc C sr | � tddddd�� | � tddddd�� | � tdddd d�� | � td
ddd d�� | �td ddd d�� d S ) N�@ �/ r g
ףp=
�?rb g�G�z��?i� i � �?i� i� )rd r( r� rZ r r r � test_approx_equal_relative_ints s
z6ApproxEqualInexactTest.test_approx_equal_relative_intsc C s& dD ] }| � |d� | � |d� qd S )N)g{�G�Jf�皙������r� r� g\��(|B@g��ʡE>�@g��x��@�{�G�z�?r� )r� rz r r r �!test_approx_equal_relative_floats s z8ApproxEqualInexactTest.test_approx_equal_relative_floatsc C sh t }t dd�}|dd�|dd�|dd�|dd �fD ]0}|t|�fD ] }| �||� | �| |� qBq2d S )
Nr| r� �T r] � �1 �2 �\ �U )r r r� )r7 r� re r� r� r r r �$test_approx_equal_relative_fractions s
( z;ApproxEqualInexactTest.test_approx_equal_relative_fractionsc C s: t td�� �D ]&}| �|td�� | �| td�� qd S )Nz$0.02 1.0 5.7 13.67 94.138 91027.9321�0.001�0.05)r� r r� r� r� r r r �#test_approx_equal_relative_decimals s z:ApproxEqualInexactTest.test_approx_equal_relative_decimalsc C st |r
| j n| j}|t|||dd�� |r.| j n| j}|t||d|d�� |sP|rV| j n| j}|t||||d�� d S )Nr rb )rd r� r( )r7 r r r& r' Ztol_flagZrel_flagrG r r r �
do_check_both) s z$ApproxEqualInexactTest.do_check_bothc C s, | � dddddd� | � dddd dd� d S )
N�R� ��� @�+��� @���Mbp?� ��W�8?Tg?5^�I� �g%��C� ����Mb`?g-C ��6*?�r� rZ r r r �test_approx_equal_both11 s z.ApproxEqualInexactTest.test_approx_equal_both1c C s | � dddddd� d S )Nr� r� r� gV�F�?8?TFr� rZ r r r �test_approx_equal_both26 s z.ApproxEqualInexactTest.test_approx_equal_both2c C s | � dddddd� d S )Nr� r� ���MbP?r� FTr� rZ r r r �test_approx_equal_both3: s z.ApproxEqualInexactTest.test_approx_equal_both3c C s, | � dddddd� | � dddd dd� d S )
Ng=
ףp=@� @r� r� Fg�Q� �[�@g�(\��[�@r� gi UMu�>r� rZ r r r �test_approx_equal_both4> s z.ApproxEqualInexactTest.test_approx_equal_both4N)r* r+ r, r� r� r� r� r� r� r� r� r� r� r� r� r� r� r� r� r r r r r� � s r� c @ s, e Zd Zdd� Zdd� Zdd� Zdd� Zd S )
�ApproxEqualSpecialsTestc C s� t tfD ]z}|d�}| �t||�� | �t||dd�� | �t||dd�� | �t| | �� | �t|| �� | �t|d�� qd S )Nr r r r� �� )r r rd r( r� )r7 rn r r r r �test_infG s z ApproxEqualSpecialsTest.test_infc C s> t tfD ]0}|d�}||d�dfD ]}| �t||�� q"qd S )N�nanr r� )r r r� r( )r7 rn r� �otherr r r �test_nanQ s z ApproxEqualSpecialsTest.test_nanc C s& t �dd�}| �t|dddd�� d S )NrX rY r� rb )r r rd r( �r7 Znzeror r r �test_float_zeroesW s z)ApproxEqualSpecialsTest.test_float_zeroesc C s& t d�}| �t|t d�ddd�� d S )Nz-0.0r r� rb )r rd r( r� r r r �test_decimal_zeroes[ s z+ApproxEqualSpecialsTest.test_decimal_zeroesN)r* r+ r, r� r� r� r� r r r r r� D s
r� c @ s e Zd Zdd� Zdd� ZdS )�TestApproxEqualErrorsc C s | � ttdddd� d S )Nrj rY r� �� assertRaisesr# r( rZ r r r � test_bad_tolc s z"TestApproxEqualErrors.test_bad_tolc C s | � ttdddd� d S )Nrj r r� r� rZ r r r � test_bad_relg s z"TestApproxEqualErrors.test_bad_relN)r* r+ r, r� r� r r r r r� ` s r� c @ s4 e Zd Zdd� Zdd� Zdd� Zdd� Zd d
� Zd S ) �TestNumericTestCasec C s. t j|� }| j|� }|D ]}| �||� qd S rQ )r> rR �generate_substringsZassertIn)r7 rm Z
actual_msgr Z substringr r r �do_testw s
zTestNumericTestCase.do_testc C s | � tttj�� d S rQ )rd �
issubclassr> r< �TestCaserZ r r r � test_numerictestcase_is_testcase} s z4TestNumericTestCase.test_numerictestcase_is_testcasec C s d}| � |� d S )N)� @� @r� � �?N�r� �r7 rm r r r �test_error_msg_numeric� s z*TestNumericTestCase.test_error_msg_numericc C s d}| � |� d S )N)� @g � @g �?r� r~ r� r� r r r �test_error_msg_sequence� s z+TestNumericTestCase.test_error_msg_sequencec C sD t ||�\}}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) r7 rD rE r& r' rS r r Z
substringsr r r r� � s �z'TestNumericTestCase.generate_substringsN)r* r+ r, r� r� r� r� r� r r r r r� q s
r� c @ s( e Zd ZeZddgZdd� Zdd� ZdS )� GlobalsTestr- �__all__c C s( | j D ] }| �t| j|�d| � qd S )Nz%s not present)�expected_metadatard �hasattr�module)r7 �metar r r � test_meta� s
�zGlobalsTest.test_metac C sB | j }|jD ]0}| �|�d�d| � | �t||�d| � q d S )N�_z private name "%s" in __all__z missing name "%s" in __all__)r� r� r� �
startswithrd r� )r7 r� �namer r r �test_check_all� s
� �zGlobalsTest.test_check_allN)r* r+ r, r. r� r� r� r r r r r r� � s r� c @ s( e Zd Ze�ejjdkd�dd� �ZdS )�DocTestsr z)Docstrings are omitted with -OO and abovec C s0 t jtt jd�\}}| �|d� | �|d� d S )N)Z optionflagsr )�doctest�testmodr. �ELLIPSIS�
assertGreaterr3 )r7 ZfailedZtriedr r r �test_doc_tests� s zDocTests.test_doc_testsN) r* r+ r, r< ZskipIf�sys�flags�optimizer r r r r r � s �r c @ s e Zd Zdd� ZdS )�StatisticsErrorTestc C s4 d}| � ttd�� | � ttjt�|tjj � d S )NzNExpected StatisticsError to be a ValueError, but got a subclass of %r instead.�StatisticsError)rd r� r. r� r r# �__base__)r7 �errmsgr r r �test_has_exception� s �
�z&StatisticsErrorTest.test_has_exceptionN)r* r+ r, r r r r r r � s r c @ sD e Zd Zdd� Zdd� Zdd� Zdd� Zd d
� Zd d � Zd
d� Z dS )�ExactRatioTestc C s$ dD ]}| � t�|�|df� qd S )N)i�����r rh �c l F�x:^V r )r3 r. � _exact_ratio)r7 rO r r r �test_int� s zExactRatioTest.test_intc C s2 d}|D ]$}t |d�}| �t�|�|df� qd S )N)���r r� �& r� )r r3 r. r )r7 r� r r� r r r �
test_fraction� s
z ExactRatioTest.test_fractionc C sb | � t�d�d� | � t�d�d� dd� td�D �}|D ]"}t�|�\}}| � ||| � q:d S )Nr� �r r� � �?)r} r� c S s g | ]}t �d d��qS )���rj ��random�uniform�r� r� r r r �
<listcomp>� s z-ExactRatioTest.test_float.<locals>.<listcomp>rj )r3 r. r �range)r7 �datar
�num�denr r r �
test_float� s zExactRatioTest.test_floatc C sJ t }tj}| �||d��d� | �||d��d� | �||d��d� d S )Nz0.125r z12.345)i� �� z-1.98)i����r� )r r. r r3 )r7 r� r r r r � test_decimal� s
zExactRatioTest.test_decimalc C s� t d�}G dd� dt �}G dd� dt�}|| fD ]`}t |t|fD ]N}||�}t�|�}| �||d f� | �t|d �|� | �t�|d �� qBq2d S )N�INFc @ s e Zd ZdS )z(ExactRatioTest.test_inf.<locals>.MyFloatN�r* r+ r, r r r r �MyFloat� s r) c @ s e Zd ZdS )z*ExactRatioTest.test_inf.<locals>.MyDecimalNr( r r r r � MyDecimal� s r* r ) r r r. r r3 r rd r r$ )r7 r' r) r* r rn r
�ratior r r r� � s
zExactRatioTest.test_infc C st t d�}G dd� dt �}|||�fD ]J}t�|�}| �t�|d �� | �|d d � | �t|d �t|�� q$d S )N�NANc @ s e Zd ZdS )z.ExactRatioTest.test_float_nan.<locals>.MyFloatNr( r r r r r) � s r) r r ) r r. r rd r r �assertIsr3 r )r7 r, r) r� r+ r r r �test_float_nan� s
z ExactRatioTest.test_float_nanc C s� t d�}t d�}G dd� dt �}|||�|||�fD ]J}t�|�}| �t|d |�� | �|d d � | �t|d �t|�� q4d S )Nr, �sNANc @ s e Zd ZdS )z2ExactRatioTest.test_decimal_nan.<locals>.MyDecimalNr( r r r r r* s r* r r )r r. r rd r r- r3 r )r7 r, r/ r* r� r+ r r r �test_decimal_nan� s
z ExactRatioTest.test_decimal_nanN)
r* r+ r, r r r$ r& r� r. r0 r r r r r � s
r c @ s< e Zd Zdd� Zdd� Zdd� Zdd� Zd d
� Zd d � Zd
S )�DecimalToRatioTestc C s< t d�}| �t�|�|d f� | �t�| �| d f� d S )Nr' )r r3 r. r )r7 r r r r �
test_infinity s z DecimalToRatioTest.test_infinityc C sD t d�t d�fD ].}t�|�\}}| �t||�� | �|d � qd S )Nr, r/ )r r. r rd r r- )r7 r� r" r# r r r r� s zDecimalToRatioTest.test_nanc C sl t d�t d�g}|D ]R}t�|�\}}| �|d� | �|d� t�| �\}}| �|d� | �|d� qd S )Nz 9.8765e12z
9.8765e-12r )r r. r �assertGreaterEqualr ZassertLessEqual)r7 Znumbersr� r" r# r r r � test_sign s z DecimalToRatioTest.test_signc C s t �td��}| �|d� d S )Nz0.1234)ii � �r. r r r3 �r7 �tr r r �test_negative_exponent, s z)DecimalToRatioTest.test_negative_exponentc C s t �td��}| �|d� d S )Nz1.234e7)i K� r r6 r7 r r r �test_positive_exponent1 s z)DecimalToRatioTest.test_positive_exponentc C s8 t �td��}| �|d� t �td��}| �|d� d S )NZ1e2)rj r z1.47e5)i8> r r6 r7 r r r �test_regression_205366 s z(DecimalToRatioTest.test_regression_20536N) r* r+ r, r2 r� r4 r9 r: r; r r r r r1 s
r1 c @ s$ e Zd Zdd� Zdd� Zdd� ZdS )� IsFiniteTestc C s0 dt dd�dtd�fD ]}| �t�|�� qd S )Nrh r r| r� �5.5)r r rd r. � _isfiniterz r r r � test_finiteB s zIsFiniteTest.test_finitec C s* t d�td�fD ]}| �t�|�� qd S �Nr �r r r� r. r> rz r r r r2 G s zIsFiniteTest.test_infinityc C s0 t d�td�td�fD ]}| �t�|�� qd S �Nr� r, r/ rA rz r r r r� L s zIsFiniteTest.test_nanN)r* r+ r, r? r2 r� r r r r r< ? s r< c @ sd e Zd Zdd� Zdd� Zdd� Zdd� Zd d
� Zd d � Zd
d� Z dd� Z
dd� Z dd� Z dd� Z
dS )�
CoerceTestc C sN t tttfD ]<}| �t�|t�|� G dd� d|�}| �t�|t�|� q d S )Nc @ s e Zd ZdS )z%CoerceTest.test_bool.<locals>.MyClassNr( r r r r �MyClassm s rD )rk r r r r- r. �_coerce�bool)r7 �TrD r r r � test_boolg s zCoerceTest.test_boolc C s, | � t�||�|� | � t�||�|� dS )z Assert that type A coerces to B.N)r- r. rE �r7 �A�Br r r �assertCoerceTop s zCoerceTest.assertCoerceToc C sT | � ||� G dd� d|�}| � ||� G dd� d|�}| � ||� | � ||� dS )z6Checks that type A coerces to B, including subclasses.c @ s e Zd ZdS )z/CoerceTest.check_coerce_to.<locals>.SubclassOfANr( r r r r � SubclassOfAz s rM c @ s e Zd ZdS )z/CoerceTest.check_coerce_to.<locals>.SubclassOfBNr( r r r r � SubclassOfB} s rN N)rL )r7 rJ rK rM rN r r r �check_coerce_tou s zCoerceTest.check_coerce_toc C s, | � ttj||f� | � ttj||f� dS )z=Assert that coercing A to B, or vice versa, raises TypeError.N)r� � TypeErrorr. rE rI r r r �assertCoerceRaises� s z CoerceTest.assertCoerceRaisesc C s� | � t�||�|� G dd� d|�}G dd� d|�}G dd� d|�}|||fD ]}| �||� qN| �||� | �||� | �||� dS )z>Check that type T coerces correctly with subclasses of itself.c @ s e Zd ZdS )z*CoerceTest.check_type_coercions.<locals>.UNr( r r r r �U� s rR c @ s e Zd ZdS )z*CoerceTest.check_type_coercions.<locals>.VNr( r r r r �V� s rS c @ s e Zd ZdS )z*CoerceTest.check_type_coercions.<locals>.WNr( r r r r �W� s rT N)r- r. rE rL rQ )r7 rG rR rS rT �typr r r �check_type_coercions� s z CoerceTest.check_type_coercionsc C s* | � t� tttfD ]}| �t|� qd S rQ )rV rk r r r rO )r7 rU r r r r � s
zCoerceTest.test_intc C s | � t� | �tt� d S rQ )rV r rO r rZ r r r r � s
zCoerceTest.test_fractionc C s | � t� d S rQ )rV r rZ r r r r& � s zCoerceTest.test_decimalc C s | � t� d S rQ )rV r rZ r r r r$ � s zCoerceTest.test_floatc C s: t ttd �ttfD ]"}ttttfD ]}| � ||� q"qd S rQ )
�str�listr �tuple�dictrk r r r rQ )r7 Zbad_typeZ good_typer r r �test_non_numeric_types� s z!CoerceTest.test_non_numeric_typesc C s: t tfD ],}G dd� d|�}| �|t� | �|t� qd S )Nc @ s e Zd ZdS )z6CoerceTest.test_incompatible_types.<locals>.MySubclassNr( r r r r �
MySubclass� s r\ )r r rQ r )r7 rG r\ r r r �test_incompatible_types� s z"CoerceTest.test_incompatible_typesN)r* r+ r, rH rL rO rQ rV r r r& r$ r[ r] r r r r rC R s rC c @ sD e Zd Zdd� Zdd� Zdd� Zdd� Zd d
� Zd d � Zd
d� Z dS )� ConvertTestc C s$ | � ||� | �t|�t|�� dS )z5Check that x equals y, and has the same type as well.N)r3 r- r )r7 r
r% r r r �check_exact_equal� s z ConvertTest.check_exact_equalc C sP t �td�t�}| �|d� G dd� dt�}t �td�|�}| �||d�� d S )Nr� c @ s e Zd ZdS )z#ConvertTest.test_int.<locals>.MyIntNr( r r r r �MyInt� s r` r] )r. �_convertr rk r_ )r7 r
r` r r r r � s
zConvertTest.test_intc C s\ t �tdd�t�}| �|tdd�� G dd� dt�}t �tdd�|�}| �||dd�� d S )N�_ r c s e Zd Z� fdd�Z� ZS )z-ConvertTest.test_fraction.<locals>.MyFractionc s | � t� �|��S rQ �� __class__�super� __truediv__�r7 r� �rd r r rf � s z9ConvertTest.test_fraction.<locals>.MyFraction.__truediv__�r* r+ r, rf �
__classcell__r r rh r �
MyFraction� s rk r� �
)r. ra r r_ )r7 r
rk r r r r � s
zConvertTest.test_fractionc C sT t �tdd�t�}| �|d� G dd� dt�}t �tdd�|�}| �||d�� d S ) NrY r g �c s e Zd Z� fdd�Z� ZS )z'ConvertTest.test_float.<locals>.MyFloatc s | � t� �|��S rQ rc rg rh r r rf � s z3ConvertTest.test_float.<locals>.MyFloat.__truediv__ri r r rh r r) � s r) r} r� r )r. ra r r r_ )r7 r
r) r r r r$ � s
zConvertTest.test_floatc C sX t �tdd�t�}| �|td�� G dd� dt�}t �tdd�|�}| �||d�� d S ) Nr �( z0.025c s e Zd Z� fdd�Z� ZS )z+ConvertTest.test_decimal.<locals>.MyDecimalc s | � t� �|��S rQ rc rg rh r r rf � s z7ConvertTest.test_decimal.<locals>.MyDecimal.__truediv__ri r r rh r r* � s r* r� r� z-0.9375)r. ra r r r_ )r7 r
r* r r r r& � s
zConvertTest.test_decimalc C sF t d�td�fD ]0}|| fD ] }t�|t|��}| �||� q qd S r@ )r r r. ra r r_ )r7 r' r r
r r r r� � s zConvertTest.test_infc C s@ t d�td�td�fD ]$}t�|t|��}| �t||�� qd S rB )r r r. ra r rd r )r7 r� r
r r r r� � s zConvertTest.test_nanN)
r* r+ r, r_ r r r$ r& r� r� r r r r r^ � s
r^ c @ s( e Zd ZdZdd� Zdd� Zdd� ZdS ) � FailNegTestz Test _fail_neg private function.c C s2 ddt d�td�g}tt�|��}| �||� d S �Nr � @r| r� )r r rX r. � _fail_negr3 )r7 �values�newr r r �test_pass_through� s z FailNegTest.test_pass_throughc C s@ ddt d�td�fD ]&}| g}t�|�}| �tjt|� qd S ro )r r r. rq r� r �next)r7 r
�seq�itr r r �test_negatives_raise� s
z FailNegTest.test_negatives_raisec
C sp dt �dd� }ztt�dg|�� W n. tjk
rT } z|jd }W 5 d }~X Y n X | �d� | �||� d S )Nz badness #%d�' i�� rY r z(expected exception, but it didn't happen) r �randintru r. rq r rm Zfailr3 )r7 rF rP r r r r �test_error_msg s
zFailNegTest.test_error_msgN)r* r+ r, r- rt rx r{ r r r r rn � s rn c @ s\ e Zd Zdd� Zdd� Zdd� Zdd� Zd d
� Zd d � Zd
d� Z dd� Z
dd� Z dd� Z dS )�UnivariateCommonMixinc C s | � t| j� d S rQ �r� rP �funcrZ r r r � test_no_args s z"UnivariateCommonMixin.test_no_argsc C s* g dt g �fD ]}| �tj| j|� qd S )Nr )�iterr� r. r r~ )r7 �emptyr r r �test_empty_data s z%UnivariateCommonMixin.test_empty_datac C s( t td��}|t|�kr$t�|� q |S )z"Return int data for various tests.r� �rX r �sortedr �shuffle�r7 r! r r r � prepare_data s z"UnivariateCommonMixin.prepare_datac C s0 | � � }|d d � }| �|�}| �||d� d S )Nzdata has been modified)r� r~ ZassertListEqual)r7 r! Zsavedr� r r r � test_no_inplace_modifications! s
z3UnivariateCommonMixin.test_no_inplace_modificationsc C sF ddddddddgd }| � |�}t�|� | � |�}| �||� d S )Nr r r| r� rh � rj )r~ r r� r3 �r7 r! r r r r r �test_order_doesnt_matter+ s
z.UnivariateCommonMixin.test_order_doesnt_matterc C sn G dd� dt �}G dd� dt�}dd� }| �� }| �|�}t tt|||fD ] }| �||��}| �||� qJd S )Nc @ s e Zd ZdS )zBUnivariateCommonMixin.test_type_of_data_collection.<locals>.MyListNr( r r r r �MyList: s r� c @ s e Zd ZdS )zCUnivariateCommonMixin.test_type_of_data_collection.<locals>.MyTupleNr( r r r r �MyTuple< s r� c S s dd� | D �S )Nc s s | ]
}|V qd S rQ r )r� �objr r r r� ? s zXUnivariateCommonMixin.test_type_of_data_collection.<locals>.generator.<locals>.<genexpr>r �r! r r r � generator> s zEUnivariateCommonMixin.test_type_of_data_collection.<locals>.generator)rX rY r� r~ r� r3 )r7 r� r� r� r! r �kindrs r r r � test_type_of_data_collection8 s
z2UnivariateCommonMixin.test_type_of_data_collectionc C s0 t ddd�}| �t|��}| �| �|�|� d S �N� r� r| )r r~ rX r3 �r7 r! r r r r �test_range_dataF s z%UnivariateCommonMixin.test_range_datac C s. | � d � | � d� | � d� | � t� � d S )Nr� g E@)�check_for_type_error�objectrZ r r r �test_bad_arg_typesL s
z(UnivariateCommonMixin.test_bad_arg_typesc G s | j t| jf|�� d S rQ r} r� r r r r� [ s z*UnivariateCommonMixin.check_for_type_errorc sh G dd� dt �}| �� }| �|�}t |ttfD ]4� � fdd�|D �}t|�| �|��}| �||� q.d S )Nc s, e Zd Z� fdd�Z� fdd�ZeZ� ZS )z@UnivariateCommonMixin.test_type_of_data_element.<locals>.MyFloatc s t | �t� �|��S rQ �r re rf rg rh r r rf c s zLUnivariateCommonMixin.test_type_of_data_element.<locals>.MyFloat.__truediv__c s t | �t� �|��S rQ �r re �__add__rg rh r r r� e s zHUnivariateCommonMixin.test_type_of_data_element.<locals>.MyFloat.__add__)r* r+ r, rf r� �__radd__rj r r rh r r) b s r) c s g | ] }� |��qS r r �r� r
�r� r r r l s zCUnivariateCommonMixin.test_type_of_data_element.<locals>.<listcomp>)r r� r~ r r r r3 )r7 r) �rawr r! rs r r� r �test_type_of_data_element^ s
z/UnivariateCommonMixin.test_type_of_data_elementN)
r* r+ r, r r� r� r� r� r� r� r� r� r� r r r r r| s
r| c @ s e Zd ZdZdd� Zdd� ZdS )�UnivariateTypeMixinam Mixin 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.
c C s G dd� dt �}t tt|fS )z4Return the types which are expected to be conserved.c s\ e Zd Z� fdd�Z� fdd�Z� fdd�Z� fdd�Z� fd d
�Z� fd d �ZeZ � Z
S )
zHUnivariateTypeMixin.prepare_types_for_conservation_test.<locals>.MyFloatc s t | �t� �|��S rQ r� rg rh r r rf ~ s zTUnivariateTypeMixin.prepare_types_for_conservation_test.<locals>.MyFloat.__truediv__c s t | �t� �|��S rQ )r re � __rtruediv__rg rh r r r� � s zUUnivariateTypeMixin.prepare_types_for_conservation_test.<locals>.MyFloat.__rtruediv__c s t | �t� �|��S rQ )r re �__sub__rg rh r r r� � s zPUnivariateTypeMixin.prepare_types_for_conservation_test.<locals>.MyFloat.__sub__c s t | �t� �|��S rQ )r re �__rsub__rg rh r r r� � s zQUnivariateTypeMixin.prepare_types_for_conservation_test.<locals>.MyFloat.__rsub__c s t | �t� �|��S rQ )r re �__pow__rg rh r r r� � s zPUnivariateTypeMixin.prepare_types_for_conservation_test.<locals>.MyFloat.__pow__c s t | �t� �|��S rQ r� rg rh r r r� � s zPUnivariateTypeMixin.prepare_types_for_conservation_test.<locals>.MyFloat.__add__) r* r+ r, rf r� r� r� r� r� r� rj r r rh r r) } s r) )r r r )r7 r) r r r �#prepare_types_for_conservation_test{ s z7UnivariateTypeMixin.prepare_types_for_conservation_testc sF | � � }| �� D ]0� � fdd�|D �}| �|�}| �t|�� � qd S )Nc s g | ] }� |��qS r r r� r� r r r � s z<UnivariateTypeMixin.test_types_conserved.<locals>.<listcomp>)r� r� r~ r- r )r7 r! r� rs r r� r �test_types_conserved� s
z(UnivariateTypeMixin.test_types_conservedN)r* r+ r, r- r� r� r r r r r� q s r� c @ s e Zd Zdd� ZdS )�
TestSumCommonc C s dd� }|| _ d S )Nc W s t j| � \}}}t �||�S rQ )r. �_sumrE )rm rG �valuer r r r �simplified_sum� s z+TestSumCommon.setUp.<locals>.simplified_sum)r~ )r7 r� r r r �setUp� s zTestSumCommon.setUpN)r* r+ r, r� r r r r r� � s r� c @ sd e Zd Zdd� Zdd� Zdd� Zdd� Zd d
� Zd d � Zd
d� Z dd� Z
dd� Z dd� Z dd� Z
dS )�TestSumc C s t j| _d S rQ )r. r� r~ rZ r r r r� � s z
TestSum.setUpc C sp g dt g �fD ]\}| �| �|�ttd�df� | �| �|d�ttd�df� | �| �|d�ttd�df� qd S )Nr r r� gffffff@)r� r3 r~ rk r r r� r r r r� � s zTestSum.test_empty_datac C sX | � | �ddddddddg�ttd�d f� | � | �d
d ddd gd
�ttd�df� d S )Nr rh r| ���i����r� ru �< r� r� r r~ r� i� )r3 r~ rk r rZ r r r � test_ints� s � �zTestSum.test_intsc C sL | � | �dgd �ttd�df� | � | �ddddgd�ttd�d f� d S )
Nr� r� � @r� r� g �?r� g @r� )r3 r~ r r rZ r r r � test_floats� s � �zTestSum.test_floatsc C s. | � | �tdd�gd �ttdd�df� d S )Nr r� �� r )r3 r~ r rZ r r r �test_fractions� s �zTestSum.test_fractionsc C sX t }|d�|d�|d�|d�|d�|d�|d�|d�g}| �| �|�t t d �d
f� d S ) Nr� z5.246z1.702z-0.025z3.974z2.328z4.617z2.843z20.686r� )r r3 r~ �r7 r� r! r r r �
test_decimals� s � �zTestSum.test_decimalsc C s: dd� t d�D �}| jt| �|�d �t�|�dd� d S )Nc S s g | ]}t �d d��qS )r r� r r r r r r � s z7TestSum.test_compare_with_math_fsum.<locals>.<listcomp>r� r g��ؗ�Ҭ<�r' )r rH r r~ r �fsumr� r r r �test_compare_with_math_fsum� s z#TestSum.test_compare_with_math_fsumc C s| dd� t d�D �}| �|�d }| �|d | �|d�d � | �|d | �|d�d � | �|td� | �|d�d � d S ) Nc S s g | ]}t �d d��qS )r r� r r r r r r � s z/TestSum.test_start_argument.<locals>.<listcomp>rj r ru r� rf g@��x �D)r r~ r3 r )r7 r! r8 r r r �test_start_argument� s
zTestSum.test_start_argumentc C s4 | � t| jdddgd� | � t| jddddg� d S )Nr r r| Z999r} rZ r r r �test_strings_fail� s zTestSum.test_strings_failc C s4 | � t| jdddgd� | � t| jddddg� d S )Nr r r| s 999r} rZ r r r �test_bytes_fail� s zTestSum.test_bytes_failc C s8 | � t| jddtd�g� | � t| jddgtd�� d S )Nr rp )r� rP r~ r rZ r r r �test_mixed_sum� s zTestSum.test_mixed_sumN)r* r+ r, r� r� r� r� r� r� r� r� r� r� r� r r r r r� � s r� c @ s e Zd Zdd� ZdS )�SumTortureTestc C s� | � t�ddddgd �ttd�df� | � t�ddddgd �ttd�df� t�ddddgd �\}}}| �|t� | � |d� | jt|�d d
d � d S ) Nr �}Ô%�I�T�}Ô%�I��ry g ��@i@� g0��.�++rY g���^�,gV瞯�<r� )r3 r. r� r r r- rH )r7 rG r" �countr r r � test_torture� s � � zSumTortureTest.test_tortureN)r* r+ r, r� r r r r r� � s r� c @ sT e Zd Zdd� Zdd� Zdd� Zdd� Zd d
� Zd d � Zd
d� Z dd� Z
dd� Z dS )�SumSpecialValuesc C sN t tfD ]@}|d�}t�d|dg�d }| �t|�|� | �t�|�� qd S )Nr� r r ) r r r. r� r- r rd r r )r7 rn r� rs r r r r� � s
zSumSpecialValues.test_nanc C s< | � t�|�� | �t|�t|�� | �|dk|dk� dS )z8Check x is an infinity of the same type and sign as inf.r N)rd r r$ r- r r3 )r7 r
r r r r �check_infinity s z SumSpecialValues.check_infinityc C sL t �dd|dg�d }| �||� t �dd|d|dg�d }| �||� d S )Nr r r| r� )r. r� r� �r7 r rs r r r � do_test_inf
s z SumSpecialValues.do_test_infc C s$ t d�}dD ]}| �|| � q d S �Nr �r rY )r r� �r7 r r
r r r �test_float_inf s z SumSpecialValues.test_float_infc C s$ t d�}dD ]}| �|| � q d S r� )r r� r� r r r �test_decimal_inf s z!SumSpecialValues.test_decimal_infc C s8 t d�}t�dd|d| dg�d }| �t�|�� d S �Nr r r r| r� )r r. r� rd r r r� r r r �test_float_mismatched_infs s z+SumSpecialValues.test_float_mismatched_infsc C sP t d�}dd|d| dg}t�tj�� | �t�t�|�d �� W 5 Q R X d S r� ) r �decimal� localcontextZExtendedContextrd r r r. r� �r7 r r! r r r �3test_decimal_extendedcontext_mismatched_infs_to_nan" s zDSumSpecialValues.test_decimal_extendedcontext_mismatched_infs_to_nanc C sH t d�}dd|d| dg}t�tj�� | �tjtj|� W 5 Q R X d S r� )r r� r� Z BasicContextr� �InvalidOperationr. r� r� r r r �0test_decimal_basiccontext_mismatched_infs_to_nan) s zASumSpecialValues.test_decimal_basiccontext_mismatched_infs_to_nanc C s( t d�}d|dg}| �tjtj|� d S )Nr/ r r )r r� r� r� r. r� )r7 r/ r! r r r �test_decimal_snan_raises0 s
z)SumSpecialValues.test_decimal_snan_raisesN) r* r+ r, r� r� r� r� r� r� r� r� r� r r r r r� � s r� c @ s$ e Zd Zdd� Zdd� Zdd� ZdS )� AverageMixinc C s6 dddt dd�td�fD ]}| �| �|g�|� qd S )Nr� g @E@g �X_yCr_ � z0.28�r r r3 r~ rz r r r �test_single_value< s z AverageMixin.test_single_valuec C s dddt dd�td�fS )N� @r] � ��7y�!C�= �C z4.9712�r r rZ r r r �'prepare_values_for_repeated_single_testA s z4AverageMixin.prepare_values_for_repeated_single_testc C sR | � � D ]D}dD ]:}| j||d��" |g| }| �| �|�|� W 5 Q R X qqd S )N�r rh r� r� )r
r� )r� �subTestr3 r~ �r7 r
r� r! r r r �test_repeated_single_valueD s
z'AverageMixin.test_repeated_single_valueN)r* r+ r, r� r� r� r r r r r� 9 s r� c @ st e Zd Zdd� Zdd� Zdd� Zdd� Zd d
� Zd d � Zd
d� Z dd� Z
dd� Z dd� Z dd� Z
dd� Zdd� ZdS ) �TestMeanc C s t j| _d S rQ )r. �meanr~ rZ r r r r� N s zTestMean.setUpc C s | � | �ddddg�d� d S )Nr� r r| r� �r3 r~ rZ r r r �test_torture_pepQ s zTestMean.test_torture_pepc C sD ddddddddddddddd d
g}t �|� | �| �|�d � d S ) Nr r r r| r� rh r� r~ r� r} g @@�r r� r3 r~ r� r r r r� U s $
zTestMean.test_intsc C s4 ddddddddg}t �|� | �| �|�d � d S )
Ng @1@g �3@� 4@g �5@g �5@g @7@g 9@g �;@g 6@r� r� r r r r� [ s
zTestMean.test_floatsc C sJ t }|d�|d�|d�|d�|d�g}t�|� | �| �|�|d�� d S )Nz1.634z2.517z3.912z4.072z5.813z3.5896�r r r� r3 r~ r� r r r r� a s "
zTestMean.test_decimalsc C sf t }|dd�|dd�|dd�|dd�|dd�|dd�|dd�g}t�|� | �| �|�|d d
�� d S ) Nr r r| r� rh r� r~ r� i� i� �r r r� r3 r~ �r7 r� r! r r r r� h s <
zTestMean.test_fractionsc C sf dddddg}t tfD ]J}dD ]@}|d�| }||g }| �|�}| �t�|�� | �||� q qd S )Nr r| rh r~ r} r� r )r r r~ rd r r$ r3 )r7 r� r� r
r r! rs r r r r� o s
zTestMean.test_infc C s: dddt d�dddt d�g}| �|�}| �t�|�� d S ) Nr r� r� r r r| rh z-inf)r r~ rd r r �r7 r! rs r r r �test_mismatched_infsz s
z TestMean.test_mismatched_infsc C sL dddddg}t tfD ]0}|d�}||g }| �|�}| �t�|�� qd S )Nr r| rh r~ r} r� )r r r~ rd r r )r7 r� r� r r! rs r r r r� � s
zTestMean.test_nanc sP d� dddddddd d
g }| � |�� }| � � fd d �|D ��}| �||� d S )
Ng e��A�333333 @� @皙����@�������@�333333@������� @� @�333333 @�������"@c s g | ] }|� �qS r r r� ��cr r r � s z*TestMean.test_big_data.<locals>.<listcomp>�r~ r3 �r7 r! r rs r r� r �
test_big_data� s
zTestMean.test_big_datac C s: dd� t d�D �}| �|�}| �|d �}| �||� d S )Nc S s g | ]}t �d d��qS )r rh r r r r r r � s z.TestMean.test_doubled_data.<locals>.<listcomp>r� r �r r~ rH r� r r r �test_doubled_data� s
zTestMean.test_doubled_datac C s t d�}| �t�|g�|� d S )NZ1e4)r r3 r. r� r� r r r �test_regression_20561� s z TestMean.test_regression_20561c C s\ | � t�ddg�d� d}d}dD ]4}| � t�|g| �|� | � t�|g| �|� q"d S )Ng�������g �g )r r| rh r% )r3 r. r� )r7 �bigZtinyr r r r �test_regression_25177� s ��z TestMean.test_regression_25177N)r* r+ r, r� r� r� r� r� r� r� r� r� r� r� r� r r r r r r� M s r� c s� e Zd Zdd� Z� fdd�Zdd� Zdd� Zd d
� Zd d � Zd
d� Z dd� Z
dd� Z dd� Z dd� Z
dd� Zdd� Zdd � Z� ZS ) �TestHarmonicMeanc C s t j| _d S rQ )r. Z
harmonic_meanr~ rZ r r r r� � s zTestHarmonicMean.setUpc s t � �� }|�d� |S )Nr )re r� �remove�r7 rr rh r r r� � s
z TestHarmonicMean.prepare_datac C s dddt dd�td�fS )Nr� r] r� r� r� z4.125r� rZ r r r r� � s z8TestHarmonicMean.prepare_values_for_repeated_single_testc C s dddg}| � | �|�d� d S )Nr r r r� r r r r � test_zero� s
zTestHarmonicMean.test_zeroc
C sH t j}dgdddgfD ],}| j|d�� | �|| j|� W 5 Q R X qd S )NrY r rg r| )rr )r. r r� r� r~ )r7 �excrr r r r �test_negative_error� s z$TestHarmonicMean.test_negative_errorc C s0 ddddddg}t �|� | �| �|�d� d S )Nr r� r� r� g333333@r� r� r r r r� � s
zTestHarmonicMean.test_intsc C sH dddddg}t �|� | �| �|�d� | �| �ddddg�d� d S )Nr� r� r� r� r� r� r r r �test_floats_exact� s
z"TestHarmonicMean.test_floats_exactc C s( t dd�D ]}| �| �|g�|� q
d S )Nr r� �r r3 r~ rz r r r �test_singleton_lists� s z%TestHarmonicMean.test_singleton_listsc C s� t }| �| �|d�|d�|d�|d�g�|d�� |d�|d�|d�|d�g}t�|� | �| �|�|d�� |d�|d�|d �|d
�g}t�|� | �| �|�|d �d � d S )
Nr_ r� r� r� z0.10z0.20z1.68z0.32z5.94z2.75i� iC )r r3 r~ r r� r� r r r �test_decimals_exact� s .
z$TestHarmonicMean.test_decimals_exactc C sf t }|dd�|dd�|dd�|dd�|dd�|dd�|dd�g}t�|� | �| �|�|d d
�� d S ) Nr r r| r� rh r� r~ r� i| i� r� r� r r r r� � s <
z TestHarmonicMean.test_fractionsc C s$ dt d�dg}| �| �|�d� d S )Nrp r r� )r r3 r~ r r r r r� � s zTestHarmonicMean.test_infc C s( dt d�dg}| �t�| �|��� d S )Nrp r� r� )r rd r r r~ r r r r r� � s zTestHarmonicMean.test_nanc sP d� dddddddd d
g }| � |�� }| � � fd d �|D ��}| �||� d S )
N�o r� r� r� r� r� r� r� r� r� c s g | ] }|� �qS r r r� r� r r r � s z>TestHarmonicMean.test_multiply_data_points.<locals>.<listcomp>r� r� r r� r �test_multiply_data_points� s
z*TestHarmonicMean.test_multiply_data_pointsc C s: dd� t d�D �}| �|�}| �|d �}| �||� d S )Nc S s g | ]}t �d d��qS )r rh r r r r r r � s z6TestHarmonicMean.test_doubled_data.<locals>.<listcomp>r� r r� r� r r r r� � s
z"TestHarmonicMean.test_doubled_data)r* r+ r, r� r� r� r r r� r r
r r� r� r� r
r� rj r r rh r r � s r c sT e Zd Zdd� Z� fdd�Zdd� Zdd� Zd d
� Zd d � Zd
d� Z dd� Z
� Z S )�
TestMedianc C s t j| _d S rQ �r. �medianr~ rZ r r r r� s zTestMedian.setUpc s( t � �� }t|�d dkr$|�d� |S )�+Overload method from UnivariateCommonMixin.r r )re r� rI r� r� rh r r r�
s
zTestMedian.prepare_datac C s& ddddddg}| � | �|�d� d S )Nr r r| r� rh r� r� r� r� r r r �test_even_ints s zTestMedian.test_even_intsc C s( dddddddg}| � | �|�d� d S )Nr r r| r� rh r� r} r� r� r r r �
test_odd_ints s zTestMedian.test_odd_intsc C sV t }|dd�|dd�|dd�|dd�|dd�g}t�|� | �| �|�|dd�� d S )Nr r~ r r| r� rh r� r� r r r �test_odd_fractions s ,
z TestMedian.test_odd_fractionsc C s^ t }|dd�|dd�|dd�|dd�|dd�|dd�g}t�|� | �| �|�|dd�� d S �Nr r~ r r| r� rh r� r� r� r r r �test_even_fractions% s 4
z TestMedian.test_even_fractionsc C sJ t }|d�|d�|d�|d�|d�g}t�|� | �| �|�|d�� d S )N�2.5�3.1�4.2�5.7�5.8r� r� r r r �test_odd_decimals- s "
z TestMedian.test_odd_decimalsc C sP t }|d�|d�|d�|d�|d�|d�g}t�|� | �| �|�|d�� d S )Nz1.2r r r r r z3.65r� r� r r r �test_even_decimals5 s (
z TestMedian.test_even_decimals) r* r+ r, r� r� r r r r r r rj r r rh r r s r c @ s e Zd Zdd� Zdd� ZdS )�TestMedianDataTypec C s t j| _d S rQ r rZ r r r r� @ s zTestMedianDataType.setUpc C s( t td��}|t|�kr$t�|� q |S )Nr_ r� r� r r r r� C s z TestMedianDataType.prepare_dataN)r* r+ r, r� r� r r r r r > s r c @ s, e Zd Zdd� Zdd� Zdd� Zdd� Zd S )
�
TestMedianLowc C s t j| _d S rQ )r. Z
median_lowr~ rZ r r r r� L s zTestMedianLow.setUpc C s& ddddddg}| � | �|�d� d S �Nr r r| r� rh r� r� r� r r r r O s z TestMedianLow.test_even_intsc C s^ t }|dd�|dd�|dd�|dd�|dd�|dd�g}t�|� | �| �|�|dd�� d S r r� r� r r r r U s 4
z!TestMedianLow.test_even_fractionsc C sP t }|d�|d�|d�|d�|d�|d�g}t�|� | �| �|�|d�� d S �Nz1.1z2.2z3.3z4.4r= z6.6r� r� r r r r ] s (
z TestMedianLow.test_even_decimalsN�r* r+ r, r� r r r r r r r r K s r c @ s, e Zd Zdd� Zdd� Zdd� Zdd� Zd S )
�TestMedianHighc C s t j| _d S rQ )r. Z median_highr~ rZ r r r r� g s zTestMedianHigh.setUpc C s& ddddddg}| � | �|�d� d S r r� r� r r r r j s z TestMedianHigh.test_even_intsc C s^ t }|dd�|dd�|dd�|dd�|dd�|dd�g}t�|� | �| �|�|dd�� d S r r� r� r r r r p s 4
z"TestMedianHigh.test_even_fractionsc C sP t }|d�|d�|d�|d�|d�|d�g}t�|� | �| �|�|d�� d S r! r� r� r r r r x s (
z!TestMedianHigh.test_even_decimalsNr" r r r r r# f s r# c @ s\ e Zd Zdd� Zdd� Zdd� Zdd� Zd d
� Zd d � Zd
d� Z dd� Z
dd� Z dd� Z dS )�TestMedianGroupedc C s t j| _d S rQ )r. Zmedian_groupedr~ rZ r r r r� � s zTestMedianGrouped.setUpc C s� dddddddg}| � | �|�d� dddddddg}| � | �|�d� ddddddddd d d
g }| � | �|d�d � d d
d
d
d
ddddddddddg}| j| �|d�ddd� d S )Nr� rl � r_ g �+@rh r� r� � r� g `3@r� � � � � � r g������4@�:�0�yE>�r& �r3 r~ rH r� r r r �test_odd_number_repeated� s "z*TestMedianGrouped.test_odd_number_repeatedc C s� ddddddddddg
}| j | �|d�ddd � d
d d d d dg}| j | �|�d
dd � d
d d d d d ddddddg }| �| �|�d� d d d d ddddddg
}| �| �|�d� d S )Nrh r� r_ r� r& r� g�����*3@r, r- r r| r� g["8���@r� r� � @)rH r~ r3 r� r r r �test_even_number_repeated� s z+TestMedianGrouped.test_even_number_repeatedc C sL dddt dd�td�fD ].}dD ]$}|g| }| �| �|�t|�� q qd S )Ng333333@�D g ��ޗCr� r� z32.9714r� )r r r3 r~ r r� r r r r� � s
z,TestMedianGrouped.test_repeated_single_valuec C sP t }|dd�|dd�|dd�|dd�|dd�g}t�|� | �| �|�d� d S )Nrh r� r} rl r] � @r� r� r r r r � s ,
z$TestMedianGrouped.test_odd_fractionsc C sX t }|dd�|dd�|dd�|dd�|dd�|dd�g}t�|� | �| �|�d� d S )Nrh r� r} rl r] �
@r� r� r r r r � s 4
z%TestMedianGrouped.test_even_fractionsc C sF t }|d�|d�|d�|d�|d�g}t�|� | �| �|�d� d S )Nr= �6.5�7.5�8.5g @r� r� r r r r � s "
z#TestMedianGrouped.test_odd_decimalsc C s� t }|d�|d�|d�|d�|d�|d�g}t�|� | �| �|�d� |d�|d�|d�|d�|d�|d�g}t�|� | �| �|�d� d S )Nr= r5 r6 r7 � @g @r� r� r r r r � s (
(
z$TestMedianGrouped.test_even_decimalsc C s� ddddddddddg
}| � | �|d�d � dddddddddddg }| j| �|d�d
d d � d
d
ddddddddddg }| � | �|d�d� d S )Ng @r� r� r3 r4 r� r� r� g @g["8���@r, r- �� �� i i �, �@ iT r� g �p@r. r� r r r �
test_interval� s z TestMedianGrouped.test_intervalc C sx dddg}| � t| j|� dddg}| � t| j|� dddg}d}| � t| j||� dddg}d}| � t| j||� d S )N� � r r r| r} )r7 r! �intervalr r r �test_data_type_error� s
z&TestMedianGrouped.test_data_type_errorN)
r* r+ r, r� r/ r1 r� r r r r r= rA r r r r r$ � s
r$ c @ sT e Zd Zdd� Zdd� Zdd� Zdd� Zd d
� Zd d � Zd
d� Z dd� Z
dd� Z dS )�TestModec C s t j| _d S rQ )r. �moder~ rZ r r r r� � s zTestMode.setUpc C s dddddddddddg S ) r r r| r� r~ r} r r� r r rZ r r r r� � s zTestMode.prepare_datac C s" t ddd�}| �| �|�d� d S r� r r� r r r r� � s zTestMode.test_range_datac C s4 d}| � | �|�d� d�� }| � | �|�d� d S )NZabcbdbr zfe fi fo fum fi fi�fi)r3 r~ r� r� r r r �test_nominal_data s zTestMode.test_nominal_datac C sD t td��}td�D ]*}||g }t�|� | �| �|�|� qd S �Nr� �rX r r r� r3 r~ )r7 r! rO r� r r r �test_discrete_data s
zTestMode.test_discrete_datac C s< dddddddddddddddd d g}| � | �|�d� d S )
Nr r r| r� rh r� r~ r� r} r� r� r r r �test_bimodal_data s &zTestMode.test_bimodal_datac C s" t td��}| �| �|�d� d S )Nr� r )rX r r3 r~ r� r r r �test_unique_data s zTestMode.test_unique_datac C s | � t| jd � d S rQ r} rZ r r r �test_none_data s zTestMode.test_none_datac C s( t �ddddg�}| �| �|�d� d S )Nr r )r? �Counterr3 r~ r� r r r �test_counter_data% s zTestMode.test_counter_dataN) r* r+ r, r� r� r� rE rH rI rJ rK rM r r r r rB � s rB c @ s e Zd Zdd� ZdS )�
TestMultiModec C sB t j}| �|d�dg� | �|d�dddg� | �|d�g � d S )NZ aabbbbbbbbccr Zaabbbbccddddeeffffggr� r� r> )r. � multimoder3 )r7 rO r r r � test_basics/ s zTestMultiMode.test_basicsN)r* r+ r, rP r r r r rN - s rN c @ s$ e Zd Zdd� Zdd� Zdd� ZdS )� TestFMeanc
C s� t j}t}t}dddgddf|d�|d�|d�gdd f|d
d �|d d
�|dd �gddfdddddgddfdd |dd �gddfdtdddg�ddffD ]2\}}}||�}| �t|�t|� | �|||� q�d S )Nr� r� � @� @�floats�3.5�4.0�5.25�decimalsr~ r r� r r� � fractionsTF�333333�?Zbooleans� mixed types)�r� r� rR rS rY �iterator) r. �fmeanr r r� r- r r r3 )r7 r^ r� r� r! �
expected_meanr� � actual_meanr r r rP 8 s � zTestFMean.test_basicsc C s� t j}t j}| �|�� |g � W 5 Q R X | �|�� |tg �� W 5 Q R X | �t�� |d � W 5 Q R X | �t�� |dd dg� W 5 Q R X | �t�� |� W 5 Q R X | �t�� |dddgd� W 5 Q R X d S )Nr� r� r� �F )r. r^ r r� r� rP )r7 r^ r r r r �test_error_casesI s zTestFMean.test_error_casesc C s� t j}td�}td�}| �t�|d|g��d� | �t�|||g��d� | �t�|d|g��d� | �t�� ||| g� W 5 Q R X d S �NZNan�Infr� r� znan and infinityZinfinity) r. r^ r rd r r r$ r� r# )r7 r^ �NaNrd r r r �test_special_valuesY s z TestFMean.test_special_valuesN)r* r+ r, rP rb rf r r r r rQ 6 s rQ c @ s@ e Zd ZdZdd� Zdd� Zdd� Zdd � Zd
d � Zd d
� Z dS )�VarianceStdevMixinr! c C s6 dddt dd�td�fD ]}| �| �|g�d� qd S )Nr� g������3@g ���%�Br� r� z8.392r r� rz r r r r� o s z$VarianceStdevMixin.test_single_valuec C sH dddt dd�td�fD ]*}dD ] }|g| }| �| �|�d� q qd S ) Nr� r� g @�6��<Cr| r~ z62.4802)r r| rh r_ r r� r� r r r r� t s
z-VarianceStdevMixin.test_repeated_single_valuec C s4 dgd }| � |�}| j|ddd� | �|d� d S )Ng.�F7ݚ�?ry rX g��ؗ�Ҍ<r- r )r~ rH r3 r� r r r � test_domain_error_regression{ s
z/VarianceStdevMixin.test_domain_error_regressionc
sN ddddddddd d
g
}| � |�}d � � fd d
�|D �}| �| � |�|� d S )Ng{�G�z�?gR� ��Q�?g
ףp=
�?gR� ��Q @g�p=
ף@g ��Q� @r0 g�Q� ��@g�G�z�@g ��Q� @g j�@c s g | ] }|� �qS r r r� ��shiftr r r � s z6VarianceStdevMixin.test_shift_data.<locals>.<listcomp>)r~ rH �r7 r� r r! r ri r �test_shift_data� s
z"VarianceStdevMixin.test_shift_datac
sN dddddddddd g
}| � |�}d
� � fd d �|D �}| �| � |�|� d S )
Nr r| r� rh r~ r} r� r� r� i ʚ;c s g | ] }|� �qS r r r� ri r r r � s z<VarianceStdevMixin.test_shift_data_exact.<locals>.<listcomp>r� rk r ri r �test_shift_data_exact� s
z(VarianceStdevMixin.test_shift_data_exactc C s6 dd� t d�D �}| �|�}| �| �t|��|� d S )Nc S s g | ]}t �d d��qS )r r� r r r r r r � s z:VarianceStdevMixin.test_iter_list_same.<locals>.<listcomp>r� )r r~ r3 r� r� r r r �test_iter_list_same� s
z&VarianceStdevMixin.test_iter_list_sameN)
r* r+ r, r' r� r� rh rl rm rn r r r r rg g s
rg c @ s4 e Zd Zdd� Zdd� Zdd� Zdd� Zd d
� Zd S ) �
TestPVariancec C s t j| _d S rQ )r. � pvariancer~ rZ r r r r� � s zTestPVariance.setUpc C s0 t td��}t�|� d}| �| �|�|� d S )Nry g P�_ArG r� r r r �test_exact_uniform� s
z TestPVariance.test_exact_uniformc C s& ddddg}d}| � | �|�|� d S )Nr� r~ rl r� g �6@r� �r7 r! �exactr r r r� � s zTestPVariance.test_intsc C sX t }|dd�|dd�|dd�|dd�g}|dd�}| �|�}| �||� | �|t � d S )Nr r� r| r~ r� �r r~ r3 �assertIsInstance�r7 r� r! rs rs r r r r� � s $
z TestPVariance.test_fractionsc C sN t }|d�|d�|d�|d�g}|d�}| �|�}| �||� | �|t � d S )Nz12.1z12.2z12.5z12.9z0.096875�r r~ r3 ru �r7 r� r! rs rs r r r r� � s
zTestPVariance.test_decimalsN)r* r+ r, r� rq r� r� r� r r r r ro � s
ro c @ s< e Zd Zdd� Zdd� Zdd� Zdd� Zd d
� Zd d � Zd
S )� TestVariancec C s t j| _d S rQ )r. �variancer~ rZ r r r r� � s zTestVariance.setUpc C s6 dddt dd�td�fD ]}| �tj| j|g� qd S )Nr g33333�8@g �(G�!=Cr� r� z4.2084�r r r� r. r r~ rz r r r r� � s z TestVariance.test_single_valuec C s& ddddg}d}| � | �|�|� d S )Nr� r~ rl r� r� r� rr r r r r� � s zTestVariance.test_intsc C sX t }|dd�|dd�|dd�|dd�g}|dd�}| �|�}| �||� | �|t � d S )Nr r� r| r~ r rt rv r r r r� � s $
zTestVariance.test_fractionsc C sZ t }|d�|d�|d�|d�g}d|d� |d� }| �|�}| �||� | �|t � d S )Nr r~ r} r� z9.5r| rw rx r r r r� � s
zTestVariance.test_decimalsc C s0 d}| � | �|�d� | � | j|dd�d� d S )N�r� rp r� rp ��xbarr� r� r� r r r �test_center_not_at_mean� s z$TestVariance.test_center_not_at_meanN) r* r+ r, r� r� r� r� r� r r r r r ry � s ry c @ s$ e Zd Zdd� Zdd� Zdd� ZdS )�
TestPStdevc C s t j| _d S rQ )r. Zpstdevr~ rZ r r r r� � s zTestPStdev.setUpc C s8 dd� t d�D �}t�t�|��}| �| �|�|� d S )Nc S s g | ]}t �d d��qS )i��r) r r r r r r � s z7TestPStdev.test_compare_to_variance.<locals>.<listcomp>r� )r r �sqrtr. rp r3 r~ r� r r r �test_compare_to_variance� s z#TestPStdev.test_compare_to_variancec C s0 d}| � | �|�d� | � | j|dd�d� d S )N)r| r� r~ r� r� r� )�mur8 r� r� r r r r s z"TestPStdev.test_center_not_at_meanN)r* r+ r, r� r� r r r r r r� � s r� c @ s, e Zd Zdd� Zdd� Zdd� Zdd� Zd S )
� TestStdevc C s t j| _d S rQ )r. �stdevr~ rZ r r r r� s zTestStdev.setUpc C s6 dddt dd�td�fD ]}| �tj| j|g� qd S )N�Q gH�z�wi@g f7?+�Brh r� z35.719r{ rz r r r r� s zTestStdev.test_single_valuec C s8 dd� t d�D �}t�t�|��}| �| �|�|� d S )Nc S s g | ]}t �d d��qS )rg r} r r r r r r s z6TestStdev.test_compare_to_variance.<locals>.<listcomp>r� )r r r� r. rz r3 r~ r� r r r r� s z"TestStdev.test_compare_to_variancec C s d}| � | j|dd�d� d S )Nr| rp r} r� r� r� r r r r s z!TestStdev.test_center_not_at_meanN)r* r+ r, r� r� r� r r r r r r� s r� c @ s4 e Zd Zdd� Zdd� Zdd� Zdd� Zd d
� Zd S ) �TestGeometricMeanc C s t j}| �|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�d S ) Nr^ r) r� � B@r� g "@g @g �1@l ���; r rj r� ry r� r| r r� r] rl rh �x r~ c S s g | ]}t �d ��qS )� I@�r � expovariate�r� rO r r r r * s z1TestGeometricMean.test_basics.<locals>.<listcomp>c S s g | ]}t �d d��qS )r� r3 )r �lognormvariater� r r r r + s �� c S s g | ]}t �d dd��qS )r� � i� )r �
triangularr� r r r r , s r� )r. �geometric_mean�assertAlmostEqualr �seedr r �prodr� r rI rd �iscloser )r7 r� �rngZ
gm_decimalZgm_floatr r r rP s$
� z TestGeometricMean.test_basicsc C s� t j}t}t}d}dddgdf|d�|d�|d�gd f|d
d �|d d
�|dd �gdfdd |dd �gdfdtdddg�dffD ]2\}}||�}| �t|�t|� | j||dd� q|d S )Ng�d�@r� r� rR rT rU rV rW rX r~ r r� r r� rY r[ )r\ rY r] rh �Zplaces) r. r� r r r� r- r r r� )r7 r� r� r� r_ r! r� r` r r r �test_various_input_types2 s � z*TestGeometricMean.test_various_input_typesc C s� t j}d}|d| d| d| g�}| �t�|d| �� | �t�|�� d}|d| d| d| g�}| �t�|d| �� | �|d� d S )Ng p~g K@g 8@r� g prX )r. r� rd r r� r� r$ �assertNotEqual)r7 r� ZlargeZbig_gmZsmallZsmall_gmr r r �test_big_and_smallD s z$TestGeometricMean.test_big_and_smallc C s t j}t j}| �|�� |g � W 5 Q R X | �|�� |dddg� W 5 Q R X | �|�� |dddg� W 5 Q R X | �|�� |tg �� W 5 Q R X | �t�� |d � W 5 Q R X | �t�� |dd dg� W 5 Q R X | �t�� |� W 5 Q R X | �t�� |dddgd� W 5 Q R X d S ) Nr� rX rR g �r� r� r� ra )r. r� r r� r� rP )r7 r� r r r r rb S s$ z"TestGeometricMean.test_error_casesc C s� t j}td�}td�}| �t�|d|g��d� | �t�|||g��d� | �t�|d|g��d� | �t�� ||| g� W 5 Q R X d S rc ) r. r� r rd r r r$ r� r# )r7 r� re rd r r r rf g s z%TestGeometricMean.test_special_valuesN)r* r+ r, rP r� r� rb rf r r r r r� s
r� c @ s4 e Zd Zdd� Zdd� Zdd� Zdd� Zd d
� Zd S ) �
TestQuantilesc sl t j}dddddg}t�|� dg fddgfd d
d gfd d
ddgfdddddgfddd
dd dgfddd
dddddgfd d dd dddd dd g fd!d"dd
d
d#dd$d ddd%g fd&d'd(dd)d
dd*d+dd d,dd-d.gff
D �]R\}}| �||||d/�� | �t|||d/��|d � tttfD ]F� |t � |�|d/�}| �
� fd0d1�|D �� | �|t t � |��� �qt|�dk�r|| �|||d/�|� t |�}d|d2 |d }d|d3 |d4 }|||g } | �|||d/�|| |d5d6�||f� d7d8� }
t t |
|��} |t |
|�|d/�} | �
t
d9d1� t| | �D ��� q�tdd:�D ]8}
tjtd;�|
d<�}||�\}}}| �|t �|�� �q.d S )=Nr� r% � r< i^ r r � @o@r| � i@� t@r� � d@g �t@rh g a@g �k@g @r@g �u@r� � ^@� �u@r� � Y@g �j@g �r@g Xv@r� g V@g g@g `t@g �v@r� � T@� l@g �q@g �v@r_ g R@g Z@g e@� n@g �p@g �t@g @v@g w@�r c 3 s | ]}t t|�� k�V qd S rQ ��allr r� �Zdatatyper r r� � s z4TestQuantiles.test_specific_cases.<locals>.<genexpr>r rY rg � inclusive�r �methodc S s d| d S �Nr� g3333�J�@r r r r r r� � s z,TestQuantiles.test_specific_cases.<locals>.fc s s | ]\}}t �||�V qd S rQ �r r� �r� rP r r r r r� � s r� rj ��k)r. � quantilesr r� r3 rI r r r r� rd rX r� r� rM r �choicesr )r7 r� r! r r rs Zsdata�lo�hiZ padded_datar� �exp�actr� �q1�q2�q3r r� r �test_specific_casesu s`
� ��
� z!TestQuantiles.test_specific_casesc s� t j}ddddg}t�|� dg fddgfdd d
gfd d dd
gfdddddgfddd dd
dgfddddddd
dgfd d dd dddd dd g fd!d"dd d d#dd$d
d
dd%g fd&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 �� | �|t t � |��� �qd3d4� }t t ||��}|t ||�|d/d0�}| �
t d5d2� t
||�D ��� q�| �|d6dgd d/d0�d7d8d9d:d;d<d=d>d?g � | �|td6d@�d d/d0�d7d8d9d:d;d<d=d>d?g � dAdB� tdC�D �}||dDd/d0�} |�t|�� |�t|�� ||dDdE�}| �|| � tddF�D ]<}
tjtd�|
dG�}||d/dH�\} } }
| �| t �|�� �q^d S )INrj r% i� i r r g �r@r| r� g y@r� g �e@g @@rh r� r� g �v@g ��@r� g �b@g ��@r� g 0a@� r� g pw@g P�@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� c 3 s | ]}t t|�� k�V qd S rQ r� r� r� r r r� � s z>TestQuantiles.test_specific_cases_inclusive.<locals>.<genexpr>c S s d| d S r� r r r r r r� � s z6TestQuantiles.test_specific_cases_inclusive.<locals>.fc s s | ]\}}t �||�V qd S rQ r� r� r r r r� � s r � $@r� g >@g D@r� g N@g �Q@r� g �V@r� c S s g | ]}t �d ��qS )ry )r � randranger� r r r r � s z?TestQuantiles.test_specific_cases_inclusive.<locals>.<listcomp>i� � r� r� r� �r� )r. r� r r� r3 rI r r r r� rd rX r� rM r r rc r r� r )r7 r� r! r r rs r� r� r� r r� r� r� r� r r� r � test_specific_cases_inclusive� sd
� �� �� z+TestQuantiles.test_specific_cases_inclusivec C sT t j}tdd�D ]>}dg| }| �||�dddg� | �||dd�dddg� qd S )Nr r� r� r� r� )r. r� r r3 )r7 r� r r! r r r �test_equal_inputs� s
�z TestQuantiles.test_equal_inputsc s� t j}d}dd� t|�D �� tt� ��|kr>� �t�d�� q � �� dD ]:}|| }| � � fdd�|� |d�D �t
t|||��� qJdD ]\}|| || d h}� fd
d�|� |d�D �}d d � t ||d d � �D �}| � ||k� q�d S )
Nry c S s g | ]}t �d ��qS )皙�����?r� r� r r r r � s z9TestQuantiles.test_equal_sized_groups.<locals>.<listcomp>r� )
r r rh r� r� r� rj r% r� r� r� r5 ry c s g | ]}t � � |��qS r ��bisect�r� �qr� r r r � s r� )
rl r� �; �m �� i; i� is i� i)& r c s g | ]}t � � |��qS r r� r� r� r r r � s c S s h | ]\}}|| �qS r r )r� �pr� r r r � <setcomp>� s z8TestQuantiles.test_equal_sized_groups.<locals>.<setcomp>)
r. r� r rI �setr� r r� �sortr3 rX rM rd )r7 r� �totalr Z
group_sizeZ group_sizes�posZsizesr r� r �test_equal_sized_groups� s" � z%TestQuantiles.test_equal_sized_groupsc C sh t j}t j}| �t�� |� W 5 Q R X | �t�� |dddgddd� W 5 Q R X | �t�� |dddgd� W 5 Q R X | �|�� |dddgdd� W 5 Q R X | �|�� |dddgdd� W 5 Q R X | �t�� |dddgd d� W 5 Q R X | �t�� |dddgd
d � W 5 Q R X | �|�� |dgdd� W 5 Q R X | �t�� |dd dgdd� W 5 Q R X d S ) Nr� r� r� rl r� r� r rY r� �Xr� )r. r� r r� rP r# )r7 r� r r r r rb s( z TestQuantiles.test_error_casesN)r* r+ r, r� r� r� r� rb r r r r r� s s
66r� c @ s� e Zd Zdd� Zdd� Zdd� Zdd� Zd d
� Zd d � Ze j
d
d� �Z dd� Z dd� Z
dd� Zdd� Zdd� Zdd� Zdd � Zd d � Zd d � Zd!d"� Zd#S )$�TestNormalDistc C sB | j �dd�}| �t�� t|� W 5 Q R X | �t|j�d� d S )Nr; r� )Z_muZ_sigma)r� �
NormalDistr� rP �varsr3 rY � __slots__�r7 �ndr r r �
test_slots s zTestNormalDist.test_slotsc C s� | j �dd�}| �|jd� | �|jd� | �|jd� | j �� }| �|jd� | �|jd� | �|jd� | �| j j�� | j �dd� W 5 Q R X G dd� d| j j�}|d d
�}| �t|�|� d S ) Nr� r] i! r r i����c @ s e Zd ZdS )zGTestNormalDist.test_instantiation_and_attributes.<locals>.NewNormalDistNr( r r r r �
NewNormalDist7 s r� r% rh ) r� r� r3 r� r� rz r� r r )r7 r� r� �nndr r r �!test_instantiation_and_attributes& s
z0TestNormalDist.test_instantiation_and_attributesc C s� | j j}dddddg}| �|�|�|dd�� | �|�t|��|dd�� | �|�t|��|dd�� | �| j j�� |�g � W 5 Q R X | �| j j�� |�dg� W 5 Q R X G d d
� d
|�}|�|�}| �t|�|� d S ) N�` ri �Z r� �n r r} r� c @ s e Zd ZdS )zBTestNormalDist.test_alternative_constructor.<locals>.NewNormalDistNr( r r r r r� L s r� ) r� r� r3 Z from_samplesrY r� r� r r )r7 r� r! r� r� r r r � test_alternative_constructor< s
z+TestNormalDist.test_alternative_constructorc C s� | j j}d\}}|||�}d}|�|�}| �t|�|� | �ttt|��th� | j � |�}| �
||d | ko�||d kn � d}|j|dd�}|j|dd�} |j|dd�}
|j|dd�} | �||
� | �| | � | � || � d S )N)ry r3 r� r� rj zhappiness and joy)r� ztrouble and despair) r� r� Zsamplesr3 rI r� r� r r r� rd r� ) r7 r� r� �sigmar� r r! r~ Zdata1Zdata2Zdata3Zdata4r r r �test_sample_generationQ s"
. z%TestNormalDist.test_sample_generationc
3 C s� | j j}|dd�}| �|�d�|�d�� | �|�d�|�d�� td�D ]$}| �|�d| �|�d| �� qJd}tdd�D ]6}|�|| �|�|� | }| j|�|�|d d
� q~|� }td d d d d
ddddddddddddddd d d d d 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<�} | �| j j �� | �d� W 5 Q R X | �
|�t d=��d>� | �
|�t d?��d>� | � t
�|�t d@���� d S )ANrj r_ r r� r� g P?r� r r� 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 �-InfrX rd re )r� r� Z
assertLess�pdfr r� �cdfrL r� r r3 r rd r r )
r7 r� r� rO �dxr
Zest_pdf�ZZpx�Yr r r �test_pdfg s�
" �
zTestNormalDist.test_pdfc s" | j j}|dd�� � fdd�tdd�D �}| �ttt|��th� | �|t|�� | �� � d�d� |� }dD ]:\}}| j
|� |�|d d
� | j
|� | �d | d d
� qn|dd �}| � | j j �� |� d
� W 5 Q R X | �� � td��d� | �� � td��d � | �
t�� � td���� d S )Nrj r_ c s g | ]}� � |��qS r )r� r� �r� r r r � s z+TestNormalDist.test_cdf.<locals>.<listcomp>r r% r� ))rX r� )r� gqZ� �?)r� g��E_A�?)g�Q� ��?gGɫs ��?)g��(\��?g؞Y���?)g ��Q� �?g���9#�?)gH�z�G�?g&S���?)rZ g�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���?rh r� r� r r� r� rX rd re )r� r� r r3 r� r� r r r� r� r� r� r rd r r )r7 r� Zcdfsr� �zZcum_probr� r r� r �test_cdf� s
zTestNormalDist.test_cdfc C sJ | j j}|dd�}| �|�d�|j� |� }dddd�}|�� D ]b\}}t|dd �D ]L\}}|d
| } | j|�| � |d d � d
| } | j|�| �|d d � qTq@| �|dd��d�d� d}
td|
�D ]$} | |
} | �|� |�| ��| � q�tdd�D ]F} d| } | �|� |�| ��| � d
| } | �|� |�| ��| � q�td�D ]"}| j|�|� |��|dd � �qJ| �
| j j �� |�d� W 5 Q R X | �
| j j �� |�d� W 5 Q R X | �
| j j �� |�d
� W 5 Q R X | �
| j j �� |�d� W 5 Q R X | �
| j j �� |dd�}|�d� W 5 Q R X | � t
�|�td���� d S ) Nrj r_ r� )
rX gR� ��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� rm r� g��>���?g��� E@i �3 rp r% rh rX r� g�������?r re )r� r� r3 Zinv_cdfr� �itemsrL r� r r� r� r rd r r r ) r7 r� Ziqr� �ppr �rowr� r
r� r rP r r r � test_inv_cdf� sL
�
zTestNormalDist.test_inv_cdfc C sj | j �� }dg fddgfdddgfdddd gffD ]2\}}|j|d
�}| �td d � t||�D ��� q2d S )
Nr r rX r| g�ǘ���ۿg�ǘ����?r� g/�$���g/�$���?r� c s s" | ]\}}t j||d d�V qdS )r� )Zabs_tolNr� r� r r r r� � s �z0TestNormalDist.test_quantiles.<locals>.<genexpr>)r� r� r� rd r� rM )r7 r� r r r r r r �test_quantiles� s
� �z TestNormalDist.test_quantilesc C s~ | j j}|dd�|dd�df|dd�|dd�dffD ]6\}}}| j|�|�|dd� | j|�|�|dd� q4dd d
�d d �}|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�� |�� W 5 Q R X | �t�� |�||� W 5 Q R X | �t�� |�d � W 5 Q R X | �| j j�� |�|d d �� W 5 Q R X | �| j j�� |d d ��|� W 5 Q R X d S ) NrX rp r� gɎ�@���?gM-[닄�?r� r� i rh )�stepsr� c s� t j}| j|j d }|t| j|j� }|| �d| | � � �fdd�t|�D �}tt| j|��}tt|j|��} t||�|| ��}
|tt || ��|
S )z0Numerical integration cross-check for overlap() rp c s g | ]}�|� �qS r r r� �r� r� r r r
s zHTestNormalDist.test_overlap.<locals>.overlap_numeric.<locals>.<listcomp>)
r r� r� r r� r rX r� r� rc ) r� r� r� r� r� �center�widthZx_arrZxpZypr� r r� r �overlap_numeric
s z4TestNormalDist.test_overlap.<locals>.overlap_numericra �A r� rj r_ r� r i����r� � r r� gj�t��?g�~j�t�h?gj�t��?ga2U0*�3?g��MbX�?r r )r� r� r� Zoverlapr� rP r )r7 r� ZX1ZX2Zpublished_resultr� r� r r r � test_overlap� sF �� zTestNormalDist.test_overlapc C sX | j �dd�}| �|jd� | �|jd� | �|jd� | �|jd� | �|jd� d S )Nrj r_ �� )r� r� r3 r� r rC r� rz )r7 r� r r r �test_properties9
s z TestNormalDist.test_propertiesc C sL | j j}|dd�}|dd�}| �|| |dd�� | �|| |dd�� d S )Nrj r� rm rh � rl r� )r� r� r3 �r7 r� r� r� r r r �'test_same_type_addition_and_subtractionA
s
z6TestNormalDist.test_same_type_addition_and_subtractionc C s� | j j}|dd�}d}| �|
|dd�� | �| |dd�� | �|| |dd�� | �|| |dd�� | �|| |dd�� | �|| |dd�� | �|| |dd �� | �|| |dd �� | �|| |dd
�� | �t�� || W 5 Q R X d S ) Nrj r_ r� r r� r� i����r� r� r� )r� r� r3 r� rP )r7 r� r� r% r r r � test_translation_and_scalingH
s
z+TestNormalDist.test_translation_and_scalingc C s| | j j}|dd�}|
}| �||� | �|j|j� | �|j|j� | }| �||� | �|j|j � | �|j|j� d S )Nrj r� )r� r� Z assertIsNotr3 r� r� r r r r �test_unary_operationsX
s
z$TestNormalDist.test_unary_operationsc C s" | j j}|� }|dd�}|� }|dd�}|dd�}|dd�}| �||� | �||� | �||� | �||� | �||� G dd� d�}|� } | �|�| �t� | �|| kd� | �| |kd� G dd� d|�}
|
d d
d �} |d d
�}| �| |� G d d
� d
�} | d d
�}
|d d
�}| �||
� d S )Nr r� r� c @ s e Zd Zdd� ZdS )z'TestNormalDist.test_equality.<locals>.Ac S s dS rF r rg r r r �__eq__t
s z.TestNormalDist.test_equality.<locals>.A.__eq__N)r* r+ r, r r r r r rJ s
s rJ r� c s e Zd Z� fdd�Z� ZS )z5TestNormalDist.test_equality.<locals>.SizedNormalDistc s t � �||� || _d S rQ )re �__init__r )r7 r� r� r rh r r r ~
s z>TestNormalDist.test_equality.<locals>.SizedNormalDist.__init__)r* r+ r, r rj r r rh r �SizedNormalDist}
s r rj r_ �9 c @ s e Zd Zdd� ZdS )z3TestNormalDist.test_equality.<locals>.LognormalDistc S s || _ || _d S rQ )r� r� )r7 r� r� r r r r �
s z<TestNormalDist.test_equality.<locals>.LognormalDist.__init__N)r* r+ r, r r r r r �
LognormalDist�
s r )r� r� r� r3 r �NotImplemented)r7 r� �nd1�nd2�nd3Znd4Znd5Znd6rJ r r �sr Zlndr� r r r �
test_equalityd
s2
z TestNormalDist.test_equalityc C sZ | j �dd�}t�|�}| �||� t�|�}| �||� t�t�|��}| �||� d S )N� �B@� �@)r� r� �copyr3 �deepcopy�pickle�loads�dumps)r7 r� r r r
r r r �test_pickle_and_copy�
s
z#TestNormalDist.test_pickle_and_copyc C sH | j j}|dd�|dd�|dd�|dd�|dd�h}| �t|�d� d S )Nrj r_ r� g .@r� rb r| )r� r� r3 rI )r7 ZNDr r r r �test_hashability�
s ,z TestNormalDist.test_hashabilityc C s" | j �dd�}| �t|�d� d S )Nr r z NormalDist(mu=37.5, sigma=5.625))r� r� r3 �reprr� r r r � test_repr�
s zTestNormalDist.test_reprN)r* r+ r, r� r� r� r� r� r� r Zskip_if_pgo_taskr� r� r� r� r r r r r r r r r r r r� s$ '
@
? , r� c @ s e Zd ZeZdd� Zdd� ZdS )�TestNormalDistPythonc C s | j tjd< d S r1 �r� r �modulesrZ r r r r� �
s zTestNormalDistPython.setUpc C s t tjd< d S r1 �r. r r rZ r r r �tearDown�
s z TestNormalDistPython.tearDownN)r* r+ r, r5 r� r� r r r r r r �
s r r9 c @ s e Zd ZeZdd� Zdd� ZdS )�TestNormalDistCc C s | j tjd< d S r1 r rZ r r r r� �
s zTestNormalDistC.setUpc C s t tjd< d S r1 r rZ r r r r �
s zTestNormalDistC.tearDownN)r* r+ r, r: r� r� r r r r r r �
s r c C s |� t�� � |S )z&Used for doctest/unittest integration.)ZaddTestsr Z DocTestSuite)�loaderZtests�ignorer r r �
load_tests�
s r# �__main__)r! r" )Mr- r� r? �collections.abcr r� r r r r r r< �testr r rY r r. r
r r r( r) Zimport_fresh_moduler5 r: r� r0 r> rW r\ rr r� r� r� r� r� r� r r r r1 r< rC r^ rn r| r� r� r� r� r� r� r� r r r r r# r$ rB rN rQ rg ro ry r� r� r� r� r� r r= r r# r* �mainr r r r �<module> s�
:+
_
>@ w -@4d: c' H
@`X9
r: 1B%'Y %