o
    g                    @   s  d Z ddlZddlZddlZddlZddlZddlZddlZddlZddl	Z	ddl
Z
ddlZddlmZ ddlmZ ddlmZ ddlmZ ddlZdd Zd	d
 Zdd ZdddZG dd dZejddgdZejddgd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 d+d, d,ejZ'G d-d. d.ejZ(G d/d0 d0ejZ)G d1d2 d2ejZ*G d3d4 d4ejZ+G d5d6 d6ejZ,G d7d8 d8ejZ-G d9d: d:ejZ.G d;d< d<ejZ/G d=d> d>ejZ0G d?d@ d@ejZ1G dAdB dBZ2G dCdD dDZ3G dEdF dFe2e3Z4G dGdH dHeZ5G dIdJ dJeZ6G dKdL dLeZ7G dMdN dNe2Z8G dOdP dPee8e3Z9G dQdR dRee8e3Z:G dSdT dTee8Z;G dUdV dVee3Z<G dWdX dXe;e3Z=G dYdZ dZe;e3Z>G d[d\ d\e;Z?G d]d^ d^ee8e3Z@G d_d` d`ejZAG dadb dbejZBG dcdd dde2ZCG dedf dfeCee3ZDG dgdh dheCee3ZEG didj djeCeZFG dkdl dleCeZGG dmdn dnejZHG dodp dpejZIG dqdr drejZJG dsdt dtejZKG dudv dvejZLG dwdx dxZMG dydz dzejeMZNeOed{G d|d} d}ejeMZPd~d ZQeRdkrMeS  dS dS )z_Test suite for statistics module, including helper NumericTestCase and
approx_equal function.

    N)support)import_helper)DecimalFractionc                 C   s   t d| S )z:Return -1.0 for negatives, including -0.0, otherwise +1.0.   N)mathcopysignx r   J/Users/admin/.pyenv/versions/3.10.0/lib/python3.10/test/test_statistics.pysign   s   r   c                 C   sZ   t | t |ur
dS t| trt| ot|S |  d }| d }||ko,|dv S )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NN)type
isinstancefloatr   isnanZas_tuple)abZaexpZbexpr   r   r   
_nan_equal"   s   
r   c                 C   s:   t t| t|}t| | }|r|| ntd}||fS )zReturn 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.
    infN)maxabsr   )actualexpectedbaseabs_errrel_errr   r   r   _calc_errorsA   s   
r!   -q=Hz>c                 C   s   |dk s|dk rt dt| st|rdS | |krdS t| s(t|r*dS t| | }t||tt| t| }||kS )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-negativeFTN)
ValueErrorr   r   isinfr   r   )r   ytolrelZactual_errorZallowed_errorr   r   r   approx_equalQ   s   "r)   c                   @   s   e Zd ZdZdS )
_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)Zblocked)Zfreshc                   @   s.   e Zd ZdgZdd Zeeddd ZdS )TestModulesZ_normal_dist_inv_cdfc                 C   $   | j D ]}| tt|jd qd S Nr/   )
func_namesassertEqualgetattrpy_statisticsr,   selffnamer   r   r   test_py_functions   s   
zTestModules.test_py_functionsrequires _statisticsc                 C   r2   )Nr0   )r4   r5   r6   c_statisticsr,   r8   r   r   r   test_c_functions   s   
zTestModules.test_c_functionsN)	r+   r,   r-   r4   r;   unittest
skipUnlessr=   r>   r   r   r   r   r1      s
    
r1   c                   @   sB   e Zd ZdZd ZZ	dddZdd Zddd	Ze	d
d Z
dS )NumericTestCasezUnit test class for numeric work.

    This subclasses TestCase. In addition to the standard method
    ``TestCase.assertAlmostEqual``,  ``assertApproxEqual`` is provided.
    r   Nc                 C   sZ   |du r| j }|du r| j}t|tjjr t|tjjr | j}n| j}|||||| dS )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)r9   firstsecondr'   r(   msgcheckr   r   r   assertApproxEqual   s   z!NumericTestCase.assertApproxEqualc           
   	   C   sn   t |t |krdt |t |f }| ||}| |tt||D ]\}\}}	| ||	|||| q$d S )Nz0sequences differ in length: %d items != %d items)len_formatMessagefailureException	enumerateziprF   )
r9   rG   rH   r'   r(   rI   standardMsgir   er   r   r   rE      s   
z!NumericTestCase._check_approx_seqc                 C   s:   t ||||r	d S | |||||}| ||}| |N)r)   _make_std_err_msgrM   rN   )r9   rG   rH   r'   r(   rI   idxrQ   r   r   r   rF   	  s
   
z!NumericTestCase._check_approx_numc           	      C   sJ   | |ksJ d}|d u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!   )	rG   rH   r'   r(   rV   templateheaderr   r    r   r   r   rU     s   z!NumericTestCase._make_std_err_msg)NNNrT   )r+   r,   r-   r.   r'   r(   rK   rE   rF   staticmethodrU   r   r   r   r   rA      s    
,
	rA   c                   @   s   e Zd ZdZdd ZdS )TestSignz5Test that the helper function sign() works correctly.c                 C   s$   |  tdd |  tdd d S )N        r   g       )r5   r   r9   r   r   r   
testZeroes*  s   zTestSign.testZeroesN)r+   r,   r-   r.   r^   r   r   r   r   rZ   (  s    rZ   c                   @   ,   e Zd Zdd Zdd Zdd Zdd Zd	S )
ApproxEqualSymmetryTestc                 C   sh   dddt dtddg}ddd	t d
tddg}t|t|ks"J t||D ]
\}}| || q'd S )Ni	  gfffffB@gfffff(z2.54   6   i	  gB@gR(z2.59   )r   r   rL   rP   do_relative_symmetry)r9   Zargs1Zargs2r   r   r   r   r   test_relative_symmetry5  s   
z.ApproxEqualSymmetryTest.test_relative_symmetryc                 C   s   t ||t||}}||k sJ || }t|| t|| }}|| d }| t||d|d | t||d|d d S )Nr   r   r'   r(   )minr   r   
assertTruer)   )r9   r   r   deltaZrel_err1Zrel_err2r(   r   r   r   rd   E  s   z,ApproxEqualSymmetryTest.do_relative_symmetryc                 C   s  g d}d}|D ]v}t tttfD ]m}||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qd S )N)   k   im r   d   rf   r   r   )intr   r   r   r   r   do_symmetry_test)r9   argsri   r   type_r   r&   rr   r   r   test_symmetryQ  s"   z%ApproxEqualSymmetryTest.test_symmetryc              
   C   s@   d}t ||||}t ||||}| |||||||f d S )Nz+approx_equal comparisons don't match for %r)r)   r5   format)r9   r   r   r'   r(   rW   Zflag1Zflag2r   r   r   rp   i  s    z(ApproxEqualSymmetryTest.do_symmetry_testN)r+   r,   r-   re   rd   rt   rp   r   r   r   r   r`   2  s
    r`   c                   @   T   e Zd Zdd Zdd Zdd Zdd Zd	d
 Zdd Zdd Z	dd Z
dd ZdS )ApproxEqualExactTestc                 C   sJ   t ||||d}| |d|  t | | ||d}| |d|   d S )Nrf   zequality failure for x=%r)r)   rh   )r9   r   r'   r(   resultr   r   r   do_exactly_equal_testv  s   z*ApproxEqualExactTest.do_exactly_equal_testc                 C      dD ]	}|  |dd qd S )N)*   iM  i~:     i  iU
 i  r   ry   r9   r   r   r   r   test_exactly_equal_ints|     z,ApproxEqualExactTest.test_exactly_equal_intsc                 C   rz   )N)gzG?g/$?ge@g      7@g     pf@g!rhQ@gB`"KB@r   r}   r9   r   r   r   r   test_exactly_equal_floats  r   z.ApproxEqualExactTest.test_exactly_equal_floatsc                 C   sN   t }|dd|d|dd|dd|dd	|ddfD ]	}| |dd qd S )
Nr   r   r   rl      	      #   $   )r   ry   r9   Ffr   r   r   test_exactly_equal_fractions  s   6z1ApproxEqualExactTest.test_exactly_equal_fractionsc                 C   s*   t }t|d D ]	}| |dd q	d S )Nz8.2 31.274 912.04 16.745 1.2047r   )r   mapsplitry   )r9   Ddr   r   r   test_exactly_equal_decimals  s   z0ApproxEqualExactTest.test_exactly_equal_decimalsc                 C   sF   dD ]}|  |dd |  |d dd t|d}|  |dd qd S )N)   i  i\  i  i     {Gz?r   
   i  )ry   r   )r9   r   r   r   r   r   test_exactly_equal_absolute  s   
z0ApproxEqualExactTest.test_exactly_equal_absolutec                 C   s2   |  tdtdd |  td tdd d S )Nz3.5710.01r   z81.3971)ry   r   r]   r   r   r   $test_exactly_equal_absolute_decimals  s   z9ApproxEqualExactTest.test_exactly_equal_absolute_decimalsc                 C   s@   dddt ddfD ]	}| |dd q	| tddtd	 d S )
Ni   g33333SY@gzGrl      r   r   z11.68r   r   ry   r   r   r   r   r   test_exactly_equal_relative  s   z0ApproxEqualExactTest.test_exactly_equal_relativec                 C   sH   dddt ddfD ]	}| |dd q	t}| |d|d	|d
 d S )Ni9  gˡE0@g\(hr      皙?r   z7.2z0.1r   r   )r9   r   r   r   r   r   test_exactly_equal_both  s   z,ApproxEqualExactTest.test_exactly_equal_bothN)r+   r,   r-   ry   r   r   r   r   r   r   r   r   r   r   r   r   rw   p  s    rw   c                   @   4   e Zd Zdd Zdd Zdd Zdd Zd	d
 ZdS )ApproxEqualUnequalTestc                 C   s8   || fD ]}t ||d ddd}| |d|  qd S )Nr   r   rf   zinequality failure for x=%r)r)   assertFalse)r9   r   r   rx   r   r   r   do_exactly_unequal_test  s   z.ApproxEqualUnequalTest.do_exactly_unequal_testc                 C      dD ]}|  | qd S )N)i  i i  i  iXC  r   r~   r   r   r   test_exactly_unequal_ints     z0ApproxEqualUnequalTest.test_exactly_unequal_intsc                 C   r   )N)gQ#@g[@gfffffG@gףp=
W"@g=
ףp=1@r   r   r   r   r   test_exactly_unequal_floats  r   z2ApproxEqualUnequalTest.test_exactly_unequal_floatsc                 C   s<   t }|dd|dd|dd|ddfD ]}| | qd S )	Nr   rl   r   r         e   iς )r   r   r   r   r   r   test_exactly_unequal_fractions  s   (z5ApproxEqualUnequalTest.test_exactly_unequal_fractionsc                 C   s"   t td D ]}| | qd S )Nz!3.1415 298.12 3.47 18.996 0.00245)r   r   r   r   r9   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 Zd Zdd Zdd Zdd Zdd Zd	d
 Zdd Zdd Z	dd Z
dd Zdd Zdd Zdd Zdd Zdd Zdd Zdd  Zd!S )"ApproxEqualInexactTestc              	   C   sb   d}|| || fD ]$}| ||}| t||d| dd| | t|||d dd| q
d S )NTest failure for x={!r}, y={!r}r   r   rf   ru   rh   r)   r   r9   r   ri   rW   r&   rI   r   r   r   do_approx_equal_abs_test  s   z/ApproxEqualInexactTest.do_approx_equal_abs_testc                 C   &   dD ]}|  |d |  |d qd S )N)iiIirk   r   r   r   %   i  i&  i6jr   r   r   r~   r   r   r   test_approx_equal_absolute_ints     z6ApproxEqualInexactTest.test_approx_equal_absolute_intsc                 C   s2   dD ]}|  |d |  |d |  |d qd S )N)	gtqgfffffFXg333333g333333      ?      ?g333333@gQ@gҭ@      ?r   -C6?r   r   r   r   r   !test_approx_equal_absolute_floats  s
   z8ApproxEqualInexactTest.test_approx_equal_absolute_floatsc                 C   sF   t dd}g d}dd |D D ]}| || | |t| qd S )Nr      )irk   r\   r   r   rl   ra      "   G   c                 s   s    | ]}t |d V  qdS )r   Nr   ).0r   r   r   r   	<genexpr>  s    zNApproxEqualInexactTest.test_approx_equal_absolute_fractions.<locals>.<genexpr>)r   r   r   )r9   ri   
numeratorsr   r   r   r   $test_approx_equal_absolute_fractions  s   
z;ApproxEqualInexactTest.test_approx_equal_absolute_fractionsc                 C   s:   t d}tt d D ]}| || | | | qd S )Nr   z1.0 3.5 36.08 61.79 7912.3648)r   r   r   r   )r9   ri   r   r   r   r   #test_approx_equal_absolute_decimals  s
   z:ApproxEqualInexactTest.test_approx_equal_absolute_decimalsc                 C   s   |  tddddd d S )Ngh㈵>gh㈵r   r   rf   )rh   r)   r]   r   r   r   test_cross_zero     z&ApproxEqualInexactTest.test_cross_zeroc              	   C   sj   d}|d|  |d|  fD ]$}| ||}| t||dd| d| | t||d|d d| qd S )Nr   r   r   r   rf   r   r   r   r   r   do_approx_equal_rel_test  s   z/ApproxEqualInexactTest.do_approx_equal_rel_testc                 C   sr   |  tddddd |  tddddd |  tdddd	d |  td
ddd	d | tdddd	d d S )N@   /   r   g
ףp=
?rf   gGz?i  i         ?i  i  )rh   r)   r   r]   r   r   r   test_approx_equal_relative_ints  s
   z6ApproxEqualInexactTest.test_approx_equal_relative_intsc                 C   r   )N)g{GJf皙r   r   g\(|B@gʡE>@gx@{Gz?r   )r   r   r   r   r   !test_approx_equal_relative_floats  r   z8ApproxEqualInexactTest.test_approx_equal_relative_floatsc                 C   sh   t }t dd}|dd|dd|dd|dd	fD ]}|t|fD ]}| || | | | q!qd S )
Nr   r   T   ra      1   2   \   U   )r   r   r   )r9   r   ri   r   r   r   r   r   $test_approx_equal_relative_fractions  s   
(z;ApproxEqualInexactTest.test_approx_equal_relative_fractionsc                 C   s:   t td D ]}| |td | | td qd 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|||dd |r| j n| j}|t||d|d |s(|r+| j n| j}|t||||d d S )Nr   rf   )rh   r   r)   )r9   r   r   r'   r(   Ztol_flagZrel_flagrJ   r   r   r   do_check_both+  s   z$ApproxEqualInexactTest.do_check_bothc                 C   ,   |  dddddd |  dddd	dd d S )
NR@+@Mbp?W8?Tg?5^Ig%CMb`?g-C6*?r   r]   r   r   r   test_approx_equal_both13     z.ApproxEqualInexactTest.test_approx_equal_both1c                 C      |  dddddd d S )Nr   r   r   gVF?8?TFr   r]   r   r   r   test_approx_equal_both28     z.ApproxEqualInexactTest.test_approx_equal_both2c                 C   r   )Nr   r   MbP?r   FTr   r]   r   r   r   test_approx_equal_both3<  r   z.ApproxEqualInexactTest.test_approx_equal_both3c                 C   r   )
Ng=
ףp=@      @r   r   FgQ[@g(\[@r   giUMu>r   r]   r   r   r   test_approx_equal_both4@  r   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                   @   r_   )
ApproxEqualSpecialsTestc                 C   s   t tfD ]=}|d}| t|| | t||dd | t||dd | t| |  | t||  | t|d qd S )Nr   r   r   r     )r   r   rh   r)   r   )r9   rr   r   r   r   r   test_infI  s   z ApproxEqualSpecialsTest.test_infc                 C   s>   t tfD ]}|d}||ddfD ]
}| t|| qqd S )Nnanr   r   )r   r   r   r)   )r9   rr   r   otherr   r   r   test_nanS  s   z ApproxEqualSpecialsTest.test_nanc                 C   s&   t dd}| t|dddd d S )Nr[   r\   r   rf   )r   r	   rh   r)   r9   Znzeror   r   r   test_float_zeroesY  s   z)ApproxEqualSpecialsTest.test_float_zeroesc                 C   s&   t d}| t|t dddd d S )Nz-0.0r   r   rf   )r   rh   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   F  s
    
r   c                   @      e Zd Zdd Zdd ZdS )TestApproxEqualErrorsc                 C      |  ttdddd d S )Nrn   r\   r   assertRaisesr$   r)   r]   r   r   r   test_bad_tole  r   z"TestApproxEqualErrors.test_bad_tolc                 C   r   )Nrn   r   r   r   r]   r   r   r   test_bad_reli  r   z"TestApproxEqualErrors.test_bad_relN)r+   r,   r-   r   r   r   r   r   r   r   b  s    r   c                   @   r   )TestNumericTestCasec                 C   s.   t j| }| j| }|D ]}| || qd S rT   )rA   rU   generate_substringsZassertIn)r9   rq   Z
actual_msgr   Z	substringr   r   r   do_testy  s
   

zTestNumericTestCase.do_testc                 C   s   |  tttj d S rT   )rh   
issubclassrA   r?   TestCaser]   r   r   r    test_numerictestcase_is_testcase  s   z4TestNumericTestCase.test_numerictestcase_is_testcasec                 C      d}|  | d S )N)      @      @r         ?Nr   r9   rq   r   r   r   test_error_msg_numeric     z*TestNumericTestCase.test_error_msg_numericc                 C   r  )N)      @g      @g      ?r   r   r  r  r   r   r   test_error_msg_sequence  r  z+TestNumericTestCase.test_error_msg_sequencec           	      C   sD   t ||\}}d| d| d| d| g}|dur |d|  |S )z5Return substrings we expect to see in error messages.ztol=%rzrel=%rzabsolute error = %rzrelative error = %rNzdiffer at index %d)r!   append)	r9   rG   rH   r'   r(   rV   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   s  s    r   c                   @   s(   e Zd ZeZddgZdd Zdd ZdS )GlobalsTestr.   __all__c                 C   s(   | j D ]}| t| j|d|  qd S )Nz%s not present)expected_metadatarh   hasattrmodule)r9   metar   r   r   	test_meta  s
   
zGlobalsTest.test_metac                 C   sB   | j }|jD ]}| |dd|  | t||d|  qd S )N_zprivate name "%s" in __all__zmissing name "%s" in __all__)r  r  r   
startswithrh   r  )r9   r  namer   r   r   test_check_all  s   
zGlobalsTest.test_check_allN)r+   r,   r-   r/   r  r  r  r  r   r   r   r   r    s
    r  c                   @   s(   e Zd Zeejjdkddd ZdS )DocTestsr   z)Docstrings are omitted with -OO and abovec                 C   s0   t jtt jd\}}| |d | |d d S )N)Zoptionflagsr   )doctesttestmodr/   ELLIPSISassertGreaterr5   )r9   ZfailedZtriedr   r   r   test_doc_tests  s   zDocTests.test_doc_testsN)	r+   r,   r-   r?   ZskipIfsysflagsoptimizer  r   r   r   r   r    s
    r  c                   @      e Zd Zdd ZdS )StatisticsErrorTestc                 C   s4   d}|  ttd |  ttjt|tjj  d S )NzNExpected StatisticsError to be a ValueError, but got a subclass of %r instead.StatisticsError)rh   r  r/   r   r"  r$   __base__)r9   errmsgr   r   r   test_has_exception  s   

z&StatisticsErrorTest.test_has_exceptionN)r+   r,   r-   r%  r   r   r   r   r!        r!  c                   @   sD   e Zd Zdd Zdd Zdd Zdd Zd	d
 Zdd Zdd Z	dS )ExactRatioTestc                 C   s$   dD ]}|  t||df qd S )N)ir   rl   c   l      Fx:^V r   )r5   r/   _exact_ratio)r9   rR   r   r   r   test_int  s   zExactRatioTest.test_intc                 C   s2   d}|D ]}t |d}| t||df qd S )N)r   r   &   r   )r   r5   r/   r*  )r9   r   r   r   r   r   r   test_fraction  s
   
zExactRatioTest.test_fractionc                 C   sb   |  tdd |  tdd dd tdD }|D ]}t|\}}|  |||  qd S )Nr   r   r         ?)r   r   c                 S      g | ]}t d dqS )rn   randomuniformr   r  r   r   r   
<listcomp>      z-ExactRatioTest.test_float.<locals>.<listcomp>rn   )r5   r/   r*  range)r9   datar   numdenr   r   r   
test_float  s   zExactRatioTest.test_floatc                 C   sJ   t }tj}| ||dd | ||dd | ||dd d S )Nz0.125r/  z12.345)i	     z-1.98)ir   )r   r/   r*  r5   )r9   r   r*  r   r   r   test_decimal  s
   zExactRatioTest.test_decimalc                 C   s   t d}G dd dt }G dd dt}|| fD ]0}t |t|fD ]'}||}t|}| ||d f | t|d | | t|d  q!qd S )NINFc                   @      e Zd ZdS )z(ExactRatioTest.test_inf.<locals>.MyFloatNr+   r,   r-   r   r   r   r   MyFloat      rC  c                   @   rA  )z*ExactRatioTest.test_inf.<locals>.MyDecimalNrB  r   r   r   r   	MyDecimal  rD  rE  r   )	r   r   r/   r*  r5   r   rh   r   r%   )r9   r@  rC  rE  r   rr   r   ratior   r   r   r     s   
zExactRatioTest.test_infc                 C   st   t d}G dd dt }|||fD ]%}t|}| t|d  | |d d  | t|d t| qd S )NNANc                   @   rA  )z.ExactRatioTest.test_float_nan.<locals>.MyFloatNrB  r   r   r   r   rC    rD  rC  r   r   )	r   r/   r*  rh   r   r   assertIsr5   r   )r9   rG  rC  r   rF  r   r   r   test_float_nan  s   
zExactRatioTest.test_float_nanc                 C   s   t d}t d}G dd dt }||||||fD ]%}t|}| t|d | | |d d  | t|d t| qd S )NrG  sNANc                   @   rA  )z2ExactRatioTest.test_decimal_nan.<locals>.MyDecimalNrB  r   r   r   r   rE    rD  rE  r   r   )r   r/   r*  rh   r   rH  r5   r   )r9   rG  rJ  rE  r   rF  r   r   r   test_decimal_nan  s   
zExactRatioTest.test_decimal_nanN)
r+   r,   r-   r+  r.  r=  r?  r   rI  rK  r   r   r   r   r'    s    
r'  c                   @   <   e Zd Zdd Zdd Zdd Zdd Zd	d
 Zdd ZdS )DecimalToRatioTestc                 C   s<   t d}| t||d f | t| | d f d S )Nr@  )r   r5   r/   r*  )r9   r   r   r   r   test_infinity  s   z DecimalToRatioTest.test_infinityc                 C   sD   t dt dfD ]}t|\}}| t|| | |d  qd S )NrG  rJ  )r   r/   r*  rh   r   rH  )r9   r   r;  r<  r   r   r   r     s
   zDecimalToRatioTest.test_nanc                 C   sx   t dt dg}|D ]/}|dksJ t|\}}| |d | |d t| \}}| |d | |d q
d S )Nz	9.8765e12z
9.8765e-12r   )r   r/   r*  assertGreaterEqualr  ZassertLessEqual)r9   Znumbersr   r;  r<  r   r   r   	test_sign   s   zDecimalToRatioTest.test_signc                 C      t td}| |d d S )Nz0.1234)ii    r/   r*  r   r5   r9   tr   r   r   test_negative_exponent.     z)DecimalToRatioTest.test_negative_exponentc                 C   rQ  )Nz1.234e7)i K r   rS  rT  r   r   r   test_positive_exponent3  rW  z)DecimalToRatioTest.test_positive_exponentc                 C   s8   t td}| |d t td}| |d d S )NZ1e2)rn   r   z1.47e5)i8> r   rS  rT  r   r   r   test_regression_205368  s   z(DecimalToRatioTest.test_regression_20536N)	r+   r,   r-   rN  r   rP  rV  rX  rY  r   r   r   r   rM    s    
rM  c                   @   $   e Zd Zdd Zdd Zdd ZdS )IsFiniteTestc                 C   s0   dt dddtdfD ]
}| t| qd S )Nrl   r   r   r  5.5)r   r   rh   r/   	_isfiniter   r   r   r   test_finiteD     zIsFiniteTest.test_finitec                 C   s*   t dtdfD ]
}| t| qd S Nr   r   r   r   r/   r]  r   r   r   r   rN  I  s   zIsFiniteTest.test_infinityc                 C   s0   t dtdtdfD ]
}| t| qd S Nr   rG  rJ  ra  r   r   r   r   r   N  r_  zIsFiniteTest.test_nanN)r+   r,   r-   r^  rN  r   r   r   r   r   r[  A  s    r[  c                   @   d   e Zd Zdd Zdd Zdd Zdd Zd	d
 Zdd Zdd Z	dd Z
dd Zdd Zdd ZdS )
CoerceTestc                 C   sN   t tttfD ]}| t|t| G dd d|}| t|t| qd S )Nc                   @   rA  )z%CoerceTest.test_bool.<locals>.MyClassNrB  r   r   r   r   MyClasso      re  )ro   r   r   r   rH  r/   _coercebool)r9   Tre  r   r   r   	test_booli  s
   zCoerceTest.test_boolc                 C   s,   |  t||| |  t||| dS )z Assert that type A coerces to B.N)rH  r/   rg  r9   ABr   r   r   assertCoerceTor  r   zCoerceTest.assertCoerceToc                 C   sT   |  || G dd d|}|  || G dd d|}|  || |  || dS )z6Checks that type A coerces to B, including subclasses.c                   @   rA  )z/CoerceTest.check_coerce_to.<locals>.SubclassOfANrB  r   r   r   r   SubclassOfA|  rf  ro  c                   @   rA  )z/CoerceTest.check_coerce_to.<locals>.SubclassOfBNrB  r   r   r   r   SubclassOfB  rf  rp  N)rn  )r9   rl  rm  ro  rp  r   r   r   check_coerce_tow  s   zCoerceTest.check_coerce_toc                 C   s,   |  ttj||f |  ttj||f dS )z=Assert that coercing A to B, or vice versa, raises TypeError.N)r   	TypeErrorr/   rg  rk  r   r   r   assertCoerceRaises  r   zCoerceTest.assertCoerceRaisesc                 C   s   |t usJ | t||| G dd d|}G dd d|}G dd d|}|||fD ]}| || q-| || | || | || dS )z>Check that type T coerces correctly with subclasses of itself.c                   @   rA  )z*CoerceTest.check_type_coercions.<locals>.UNrB  r   r   r   r   U  rf  rt  c                   @   rA  )z*CoerceTest.check_type_coercions.<locals>.VNrB  r   r   r   r   V  rf  ru  c                   @   rA  )z*CoerceTest.check_type_coercions.<locals>.WNrB  r   r   r   r   W  rf  rv  N)rh  rH  r/   rg  rn  rs  )r9   ri  rt  ru  rv  typr   r   r   check_type_coercions  s   zCoerceTest.check_type_coercionsc                 C   s*   |  t tttfD ]}| t| q
d S rT   )rx  ro   r   r   r   rq  )r9   rw  r   r   r   r+    s   
zCoerceTest.test_intc                 C   s   |  t | tt d S rT   )rx  r   rq  r   r]   r   r   r   r.    s   
zCoerceTest.test_fractionc                 C      |  t d S rT   )rx  r   r]   r   r   r   r?       zCoerceTest.test_decimalc                 C   ry  rT   )rx  r   r]   r   r   r   r=    rz  zCoerceTest.test_floatc                 C   s:   t ttd ttfD ]}ttttfD ]}| 	|| qq	d S rT   )
strlistr   tupledictro   r   r   r   rs  )r9   Zbad_typeZ	good_typer   r   r   test_non_numeric_types  s
   z!CoerceTest.test_non_numeric_typesc                 C   s:   t tfD ]}G dd d|}| |t | |t qd S )Nc                   @   rA  )z6CoerceTest.test_incompatible_types.<locals>.MySubclassNrB  r   r   r   r   
MySubclass  rf  r  )r   r   rs  r   )r9   ri  r  r   r   r   test_incompatible_types  s
   z"CoerceTest.test_incompatible_typesN)r+   r,   r-   rj  rn  rq  rs  rx  r+  r.  r?  r=  r  r  r   r   r   r   rd  T  s    	rd  c                   @   sL   e Zd Zdd Zdd Zdd Zdd Zd	d
 Zdd Zdd Z	dd Z
dS )ConvertTestc                 C   s$   |  || | t|t| dS )z5Check that x equals y, and has the same type as well.N)r5   rH  r   r9   r   r&   r   r   r   check_exact_equal  s   zConvertTest.check_exact_equalc                 C   sP   t tdt}| |d G dd dt}t td|}| ||d d S )Nr   c                   @   rA  )z#ConvertTest.test_int.<locals>.MyIntNrB  r   r   r   r   MyInt  rf  r  ra   )r/   _convertr   ro   r  )r9   r   r  r   r   r   r+    s
   zConvertTest.test_intc                 C   s\   t tddt}| |tdd G dd dt}t tdd|}| ||dd d S )N_   r)  c                          e Zd Z fddZ  ZS )z-ConvertTest.test_fraction.<locals>.MyFractionc                       |  t |S rT   	__class__super__truediv__r9   r   r  r   r   r       z9ConvertTest.test_fraction.<locals>.MyFraction.__truediv__r+   r,   r-   r  __classcell__r   r   r  r   
MyFraction      r  r      )r/   r  r   r  )r9   r   r  r   r   r   r.    s
   zConvertTest.test_fractionc                 C   sT   t tddt}| |d G dd dt}t tdd|}| ||d d S )	Nr\   r   g      c                       r  )z'ConvertTest.test_float.<locals>.MyFloatc                    r  rT   r  r  r  r   r   r    r  z3ConvertTest.test_float.<locals>.MyFloat.__truediv__r  r   r   r  r   rC    r  rC  r   r   r0  )r/   r  r   r   r  )r9   r   rC  r   r   r   r=    s
   zConvertTest.test_floatc                 C   sX   t tddt}| |td G dd dt}t tdd|}| ||d d S )	Nr   (   z0.025c                       r  )z+ConvertTest.test_decimal.<locals>.MyDecimalc                    r  rT   r  r  r  r   r   r    r  z7ConvertTest.test_decimal.<locals>.MyDecimal.__truediv__r  r   r   r  r   rE    r  rE  r   r   z-0.9375)r/   r  r   r   r  )r9   r   rE  r   r   r   r?    s
   zConvertTest.test_decimalc                 C   sF   t dtdfD ]}|| fD ]}t|t|}| || qqd S r`  )r   r   r/   r  r   r  )r9   r@  r   r   r   r   r   r     s   zConvertTest.test_infc                 C   s@   t dtdtdfD ]}t|t|}| t|| qd S rb  )r   r   r/   r  r   rh   r   )r9   r   r   r   r   r   r     s   zConvertTest.test_nanc                 C   s<   |  t td t W d    d S 1 sw   Y  d S rT   )r   rr  r/   r  r   r]   r   r   r   test_invalid_input_type  s   "z#ConvertTest.test_invalid_input_typeN)r+   r,   r-   r  r+  r.  r=  r?  r   r   r  r   r   r   r   r    s    


r  c                   @   s(   e Zd ZdZdd Zdd Zdd ZdS )	FailNegTestz Test _fail_neg private function.c                 C   s2   ddt dtdg}tt|}| || d S Nr          @r   r   )r   r   r|  r/   	_fail_negr5   )r9   valuesnewr   r   r   test_pass_through  s   zFailNegTest.test_pass_throughc                 C   s@   ddt dtdfD ]}| g}t|}| tjt| q
d S r  )r   r   r/   r  r   r"  next)r9   r   seqitr   r   r   test_negatives_raise  s
   
z FailNegTest.test_negatives_raisec              
   C   sr   dt dd }zttdg| W n tjy+ } z|jd }W Y d }~n
d }~ww | d | || d S )Nzbadness #%d'  i r\   r   z(expected exception, but it didn't happen)	r4  randintr  r/   r  r"  rq   Zfailr5   )r9   rI   rS   r$  r   r   r   test_error_msg  s   
zFailNegTest.test_error_msgN)r+   r,   r-   r.   r  r  r  r   r   r   r   r    s
    r  c                   @   r   )FindLteqTestc              
   C   s   g dfddgdfddgdffD ]6\}}| j ||d# | t t|| W d    n1 s1w   Y  W d    q1 s@w   Y  qd S )Nr   r   r   r   r   )subTestr   r$   r/   
_find_lteq)r9   r   r   r   r   r   test_invalid_input_values  s   

z&FindLteqTest.test_invalid_input_valuesc              	   C   sz   g dddfg dddfg dddffD ]&\}}}| j ||d | |t|| W d    q1 s5w   Y  qd S )	Nr   r   r   r   r   r   r   r   r   r   r   r   r   r   r   r   r   r   r   r   r  )r  r5   r/   r  )r9   r   r   
expected_ir   r   r   test_locate_successfully  s   z%FindLteqTest.test_locate_successfullyNr+   r,   r-   r  r  r   r   r   r   r    s    
r  c                   @   r   )FindRteqTestc              	   C   sb   dgddfddgddffD ]!\}}}|  t t||| W d    q1 s)w   Y  qd S )Nr   r   r   r   )r   r$   r/   
_find_rteq)r9   r   lr   r   r   r   r  ,  s   
z&FindRteqTest.test_invalid_input_valuesc              
   C   s   g ddddfg ddddfg ddddffD ])\}}}}| j |||d	 | |t||| W d    q1 s;w   Y  qd S )
Nr  r   r   r   r  r   r  r   )r   r  r   )r  r5   r/   r  )r9   r   r  r   r  r   r   r   r  4  s   z%FindRteqTest.test_locate_successfullyNr  r   r   r   r   r  )  s    r  c                   @   \   e Zd Zdd Zdd Zdd Zdd Zd	d
 Zdd Zdd Z	dd Z
dd Zdd ZdS )UnivariateCommonMixinc                 C   s   |  t| j d S rT   r   rr  funcr]   r   r   r   test_no_argsC  s   z"UnivariateCommonMixin.test_no_argsc                 C   s*   g dt g fD ]}| tj| j| qd S )Nr   )iterr   r/   r"  r  )r9   emptyr   r   r   test_empty_dataG  s   z%UnivariateCommonMixin.test_empty_datac                 C   s2   t td}|t|krt| |t|ks|S )z"Return int data for various tests.r   N)r|  r9  sortedr4  shuffler9   r:  r   r   r   prepare_dataL  s
   
z"UnivariateCommonMixin.prepare_datac                 C   s\   |   }t|dksJ |t|ksJ |d d  }||us J | |}| ||d d S )Nr   zdata has been modified)r  rL   r  r  ZassertListEqual)r9   r:  Zsavedr  r   r   r   test_no_inplace_modificationsS  s   
z3UnivariateCommonMixin.test_no_inplace_modificationsc                 C   s:   g dd }|  |}t| |  |}| || d S )N)r   r   r   r   r   r   rl      rn   )r  r4  r  r5   r9   r:  r   r   r   r   r   test_order_doesnt_matter]  s
   


z.UnivariateCommonMixin.test_order_doesnt_matterc                 C   sn   G dd dt }G dd dt}dd }|  }| |}t tt|||fD ]}| ||}| || q%d S )Nc                   @   rA  )zBUnivariateCommonMixin.test_type_of_data_collection.<locals>.MyListNrB  r   r   r   r   MyListl  rD  r  c                   @   rA  )zCUnivariateCommonMixin.test_type_of_data_collection.<locals>.MyTupleNrB  r   r   r   r   MyTuplen  rD  r  c                 S   s   dd | D S )Nc                 s   s    | ]}|V  qd S rT   r   )r   objr   r   r   r   q  s    zXUnivariateCommonMixin.test_type_of_data_collection.<locals>.generator.<locals>.<genexpr>r   r:  r   r   r   	generatorp     zEUnivariateCommonMixin.test_type_of_data_collection.<locals>.generator)r|  r}  r  r  r  r5   )r9   r  r  r  r:  r   kindrx   r   r   r   test_type_of_data_collectionj  s   
z2UnivariateCommonMixin.test_type_of_data_collectionc                 C   s0   t ddd}| t|}| | || d S N   r   r   )r9  r  r|  r5   r9   r:  r   r   r   r   test_range_datax  s   z%UnivariateCommonMixin.test_range_datac                 C   s.   |  d  |  d |  d |  t  d S )Nr   g      E@)check_for_type_errorobjectr]   r   r   r   test_bad_arg_types~  s   



z(UnivariateCommonMixin.test_bad_arg_typesc                 G   s   | j t| jg|R   d S rT   r  r  r   r   r   r    s   z*UnivariateCommonMixin.check_for_type_errorc                    sh   G dd dt }|  }| |}t |ttfD ]  fdd|D }t|| |}| || qd S )Nc                       s,   e Zd Z fddZ fddZeZ  ZS )z@UnivariateCommonMixin.test_type_of_data_element.<locals>.MyFloatc                       t | t |S rT   r   r  r  r  r  r   r   r       zLUnivariateCommonMixin.test_type_of_data_element.<locals>.MyFloat.__truediv__c                    r  rT   r   r  __add__r  r  r   r   r    r  zHUnivariateCommonMixin.test_type_of_data_element.<locals>.MyFloat.__add__)r+   r,   r-   r  r  __radd__r  r   r   r  r   rC    s    rC  c                       g | ]} |qS r   r   r   r   r  r   r   r7        zCUnivariateCommonMixin.test_type_of_data_element.<locals>.<listcomp>)r   r  r  r   r   r   r5   )r9   rC  rawr   r:  rx   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 Zd ZdZdd Zdd ZdS )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 dd dt }t tt|fS )z4Return the types which are expected to be conserved.c                       s\   e Zd Z fddZ fddZ fddZ fddZ fd	d
Z fddZeZ	  Z
S )zHUnivariateTypeMixin.prepare_types_for_conservation_test.<locals>.MyFloatc                    r  rT   r  r  r  r   r   r    r  zTUnivariateTypeMixin.prepare_types_for_conservation_test.<locals>.MyFloat.__truediv__c                    r  rT   )r   r  __rtruediv__r  r  r   r   r    r  zUUnivariateTypeMixin.prepare_types_for_conservation_test.<locals>.MyFloat.__rtruediv__c                    r  rT   )r   r  __sub__r  r  r   r   r    r  zPUnivariateTypeMixin.prepare_types_for_conservation_test.<locals>.MyFloat.__sub__c                    r  rT   )r   r  __rsub__r  r  r   r   r    r  zQUnivariateTypeMixin.prepare_types_for_conservation_test.<locals>.MyFloat.__rsub__c                    r  rT   )r   r  __pow__r  r  r   r   r    r  zPUnivariateTypeMixin.prepare_types_for_conservation_test.<locals>.MyFloat.__pow__c                    r  rT   r  r  r  r   r   r    r  zPUnivariateTypeMixin.prepare_types_for_conservation_test.<locals>.MyFloat.__add__)r+   r,   r-   r  r  r  r  r  r  r  r  r   r   r  r   rC    s    rC  N)r   r   r   )r9   rC  r   r   r   #prepare_types_for_conservation_test  s   z7UnivariateTypeMixin.prepare_types_for_conservation_testc                    sF   |   }|  D ]  fdd|D }| |}| t|  qd S )Nc                    r  r   r   r  r  r   r   r7    r  z<UnivariateTypeMixin.test_types_conserved.<locals>.<listcomp>)r  r  r  rH  r   )r9   r:  r   rx   r   r  r   test_types_conserved  s   
z(UnivariateTypeMixin.test_types_conservedN)r+   r,   r-   r.   r  r  r   r   r   r   r    s    	r  c                   @   r   )TestSumCommonc                 C   s   dd }|| _ d S )Nc                  W   s   t j|  \}}}t ||S rT   )r/   _sumrg  )rq   ri  valuer   r   r   r   simplified_sum  s   z+TestSumCommon.setUp.<locals>.simplified_sum)r  )r9   r  r   r   r   setUp  s   
zTestSumCommon.setUpN)r+   r,   r-   r  r   r   r   r   r    s    r  c                   @   rc  )TestSumc                 C      t j| _d S rT   )r/   r  r  r]   r   r   r   r       zTestSum.setUpc                 C   sp   g dt g fD ].}| | |ttddf | | |dttddf | | |dttddf qd S )Nr   r   r   ffffff@)r  r5   r  ro   r   r   r  r   r   r   r    s
    zTestSum.test_empty_datac                 C   sF   |  | g dttddf |  | g ddttddf d S )N)r   rl   r   r  r{   r   <   r   )r   r   r   r  r   r   i  rl   )r5   r  ro   r   r]   r   r   r   	test_ints  s   zTestSum.test_intsc                 C   sH   |  | dgd ttddf |  | g ddttddf d S )Nr  r        @)r   r  r   g      ?r   g      	@r   )r5   r  r   r   r]   r   r   r   test_floats  s   zTestSum.test_floatsc                 C   s.   |  | tddgd ttdddf d S )Nr   r     r   )r5   r  r   r]   r   r   r   test_fractions  s   zTestSum.test_fractionsc              	   C   sX   t }|d|d|d|d|d|d|d|dg}| | |t t d	d
f d S )Nr   z5.246z1.702z-0.025z3.974z2.328z4.617z2.843z20.686r   )r   r5   r  r9   r   r:  r   r   r   test_decimals  s   zTestSum.test_decimalsc                 C   s:   dd t dD }| jt| |d t|dd d S )Nc                 S   r1  )r2  r   r3  r6  r   r   r   r7    r8  z7TestSum.test_compare_with_math_fsum.<locals>.<listcomp>r   r   gؗҬ<r(   )r9  rK   r   r  r   fsumr  r   r   r   test_compare_with_math_fsum  s   (z#TestSum.test_compare_with_math_fsumc                 C   s|   dd t dD }| |d }| |d | |dd  | |d | |dd  | |td | |dd  d S )	Nc                 S   r1  )r   r   r3  r6  r   r   r   r7    r8  z/TestSum.test_start_argument.<locals>.<listcomp>rn   r   r{   r   rj   g@xD)r9  r  r5   r   )r9   r:  rU  r   r   r   test_start_argument  s
   $zTestSum.test_start_argumentc                 C   .   |  t| jg dd |  t| jg d d S )Nr   r   r   999)r   r   r   r  r  r]   r   r   r   test_strings_fail     zTestSum.test_strings_failc                 C   r  )Nr     999)r   r   r   r  r  r]   r   r   r   test_bytes_fail  r  zTestSum.test_bytes_failc                 C   s8   |  t| jddtdg |  t| jddgtd d S )Nr   r  )r   rr  r  r   r]   r   r   r   test_mixed_sum  s   zTestSum.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                   @   r   )SumTortureTestc                 C   s   |  tg dd ttddf |  tg dd ttddf tg dd \}}}| |t |  |d | jt|ddd	 d S )
N)r   }Ô%ITr   }Ô%Ir  g     @i@  )r  r   r   r  )0.++r   r  r\   g^,gV瞯<r  )r5   r/   r  r   r   rH  rK   )r9   ri  r;  countr   r   r   test_torture  s   zSumTortureTest.test_tortureN)r+   r,   r-   r
  r   r   r   r   r    r&  r  c                   @   rv   )SumSpecialValuesc                 C   sN   t tfD ] }|d}td|dgd }| t|| | t| qd S )Nr   r   r   )	r   r   r/   r  rH  r   rh   r   r   )r9   rr   r   rx   r   r   r   r   .  s   zSumSpecialValues.test_nanc                 C   sH   |  t| | t|t| | |dk|dk ||ks"J dS )z8Check x is an infinity of the same type and sign as inf.r   N)rh   r   r%   rH  r   r5   )r9   r   r   r   r   r   check_infinity5  s   zSumSpecialValues.check_infinityc                 C   sL   t dd|dgd }| || t dd|d|dgd }| || d S )Nr   r   r   r   )r/   r  r  r9   r   rx   r   r   r   do_test_inf<  s   zSumSpecialValues.do_test_infc                 C   $   t d}dD ]	}| ||  qd S Nr   r   r\   )r   r  r9   r   r   r   r   r   test_float_infD     zSumSpecialValues.test_float_infc                 C   r  r  )r   r  r  r   r   r   test_decimal_infI  r  z!SumSpecialValues.test_decimal_infc                 C   s8   t d}tdd|d| dgd }| t| d S Nr   r   r   r   r   )r   r/   r  rh   r   r   r  r   r   r   test_float_mismatched_infsN  s   z+SumSpecialValues.test_float_mismatched_infsc                 C   sf   t d}dd|d| dg}ttj | tt|d  W d    d S 1 s,w   Y  d S r  )	r   decimallocalcontextZExtendedContextrh   r   r   r/   r  r9   r   r:  r   r   r   3test_decimal_extendedcontext_mismatched_infs_to_nanT  s
   "zDSumSpecialValues.test_decimal_extendedcontext_mismatched_infs_to_nanc                 C   s^   t d}dd|d| dg}ttj | tjtj| W d    d S 1 s(w   Y  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|dg}| tjtj| d S )NrJ  r   r   )r   r   r  r  r/   r  )r9   rJ  r:  r   r   r   test_decimal_snan_raisesb     
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                   @   rZ  )AverageMixinc                 C   s6   dddt ddtdfD ]}| | |g| qd S )Nr   g     @E@g  X_yCrc      z0.28r   r   r5   r  r   r   r   r   test_single_valuen     zAverageMixin.test_single_valuec                 C      dddt ddtdfS )N      @ra    7y!C=   C   z4.9712r   r   r]   r   r   r   'prepare_values_for_repeated_single_tests  s   z4AverageMixin.prepare_values_for_repeated_single_testc              
   C   sf   |   D ],}dD ]'}| j||d |g| }| | || W d    q1 s*w   Y  qqd S )Nr   rl   r   r  )r   r	  )r+  r  r5   r  r9   r   r	  r:  r   r   r   test_repeated_single_valuev  s   
z'AverageMixin.test_repeated_single_valueN)r+   r,   r-   r#  r+  r.  r   r   r   r   r   k  s    r   c                   @   st   e Zd Zdd Zdd Zdd Zdd Zd	d
 Zdd Zdd Z	dd Z
dd Zdd Zdd Zdd Zdd ZdS )TestMeanc                 C   r  rT   )r/   meanr  r]   r   r   r   r    r  zTestMean.setUpc                 C   s   |  | g dd d S )N)r  r   r   r  r   r5   r  r]   r   r   r   test_torture_pep  r   zTestMean.test_torture_pepc                 C   (   g d}t | | | |d d S )N)r   r   r   r   r   r   r   rl   rl   r  r   r   r   r   r   r   g     @@r4  r  r5   r  r  r   r   r   r    r  zTestMean.test_intsc                 C   r3  )N)g     @1@g     3@      4@g     5@g     5@g     @7@g      9@g     ;@g     6@r4  r  r   r   r   r    r  zTestMean.test_floatsc                 C   sJ   t }|d|d|d|d|dg}t| | | ||d d S )Nz1.634z2.517z3.912z4.072z5.813z3.5896)r   r4  r  r5   r  r  r   r   r   r    s   "
zTestMean.test_decimalsc              	   C   f   t }|dd|dd|dd|dd|dd|dd|ddg}t| | | ||d	d
 d S )Nr   r   r   r   rl   r  r   r   i  i  r   r4  r  r5   r  r9   r   r:  r   r   r   r       <
zTestMean.test_fractionsc                 C   s`   g d}t tfD ]%}dD ] }|d| }||g }| |}| t| | || qqd S )Nr   r   rl   r   r   r  r   )r   r   r  rh   r   r%   r5   )r9   r  r  r   r   r:  rx   r   r   r   r     s   

zTestMean.test_infc              	   C   s:   dddt ddddt dg}| |}| t| d S )	Nr   r   r  r   r   r   rl   z-inf)r   r  rh   r   r   r9   r:  rx   r   r   r   test_mismatched_infs  s   
zTestMean.test_mismatched_infsc                 C   sF   g d}t tfD ]}|d}||g }| |}| t| qd S )Nr:  r   )r   r   r  rh   r   r   )r9   r  r  r   r:  rx   r   r   r   r     s   

zTestMean.test_nanc                    sN   d g d}|  |  }| ksJ |   fdd|D }| || d S )Ng    eA	333333@      @g@g@g333333@@g       @g333333 @g"@c                       g | ]}|  qS r   r   r  cr   r   r7    r  z*TestMean.test_big_data.<locals>.<listcomp>r  r5   r9   r:  r   rx   r   rB  r   test_big_data  s   zTestMean.test_big_datac                 C   :   dd t dD }| |}| |d }| || d S )Nc                 S   r1  )r(  rl   r3  r6  r   r   r   r7    r8  z.TestMean.test_doubled_data.<locals>.<listcomp>r   r   r9  r  rK   r  r   r   r   test_doubled_data     
zTestMean.test_doubled_datac                 C   s    t d}| t|g| d S )NZ1e4)r   r5   r/   r0  r   r   r   r   test_regression_20561  s   zTestMean.test_regression_20561c                 C   s\   |  tddgd d}d}dD ]}|  t|g| | |  t|g| | qd S )Ngg      g       )r   r   rl   r>  )r5   r/   r0  )r9   bigZtinyr   r   r   r   test_regression_25177  s   zTestMean.test_regression_25177N)r+   r,   r-   r  r2  r  r  r  r  r   r<  r   rF  rI  rK  rM  r   r   r   r   r/    s    		r/  c                       s   e Zd Zdd Z fddZdd Zdd Zd	d
 Zdd Zdd Z	dd Z
dd Zdd Zdd Zdd Zdd Zdd Zdd Zdd  Z  ZS )!TestHarmonicMeanc                 C   r  rT   )r/   Zharmonic_meanr  r]   r   r   r   r    r  zTestHarmonicMean.setUpc                    s   t   }|d |S )Nr   )r  r  remover9   r  r  r   r   r    s   

zTestHarmonicMean.prepare_datac                 C   r%  )Nr&  ra   r'  r(  r)  z4.125r*  r]   r   r   r   r+    r   z8TestHarmonicMean.prepare_values_for_repeated_single_testc                 C   s   g d}|  | |d d S )N)r   r   r   r   r1  rP  r   r   r   	test_zero  s   zTestHarmonicMean.test_zeroc              	   C   sZ   t j}dgg dfD ] }| j|d | || j| W d    q
1 s%w   Y  q
d S )Nr\   )r   rk   r   )r  )r/   r"  r  r   r  )r9   excr  r   r   r   test_negative_error  s   z$TestHarmonicMean.test_negative_errorc              
   C   s   dgg dg dg dfD ]2}| j |d" | t | | W d    n1 s+w   Y  W d    q1 s:w   Y  qd S )Nz3.14)123)r   rU  r   4rl   )r  r>  r?  z5.6r  )r  r   rr  r  r  r   r   r   test_invalid_type_error  s   z(TestHarmonicMean.test_invalid_type_errorc                 C   r3  )N)r   r   r   r   r   r   g333333@r4  r  r   r   r   r    r  zTestHarmonicMean.test_intsc                 C   s>   g d}t | | | |d | | g dd d S )N)r   r  r  r   r   r  )r  r   r   r   r   r4  r  r   r   r   test_floats_exact  s   
z"TestHarmonicMean.test_floats_exactc                 C   s(   t ddD ]}| | |g| qd S )Nr   r   r9  r5   r  r   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 )Nrc   r   r  r   z0.10z0.20z1.68z0.32z5.94z2.75i iC )r   r5   r  r4  r  r  r   r   r   test_decimals_exact  s   .

z$TestHarmonicMean.test_decimals_exactc              	   C   r6  )Nr   r   r   r   rl   r  r   r   i|  i  r7  r8  r   r   r   r  "  r9  zTestHarmonicMean.test_fractionsc                 C   s$   dt ddg}| | |d d S )Nr  r   r   )r   r5   r  rP  r   r   r   r   )  s   zTestHarmonicMean.test_infc                 C   s(   dt ddg}| t| | d S )Nr  r   r   )r   rh   r   r   r  rP  r   r   r   r   .  s   zTestHarmonicMean.test_nanc                    sB   d g d}|  |  }|   fdd|D }| || d S )No   r=  c                    s   g | ]}|  qS r   r   r  rB  r   r   r7  8  r  z>TestHarmonicMean.test_multiply_data_points.<locals>.<listcomp>rD  rE  r   rB  r   test_multiply_data_points3  s
   z*TestHarmonicMean.test_multiply_data_pointsc                 C   rG  )Nc                 S   r1  )r   rl   r3  r6  r   r   r   r7  =  r8  z6TestHarmonicMean.test_doubled_data.<locals>.<listcomp>r   r   rH  r  r   r   r   rI  ;  rJ  z"TestHarmonicMean.test_doubled_datac              	   C   s  |  | ddgddgd |  | jddgddgdd |  | tddgtddgd |  | tddtd	dtd
dgg d| tddgd td	dgd  td
dgd   |  | dgd
gd | t | g dg d W d    n1 sw   Y  | tj | g dddg W d    n1 sw   Y  | tj | dgdg W d    n1 sw   Y  | tj | ddgddg W d    d S 1 sw   Y  d S )Nr  r  rl   r   g      L@)weightsr   r   r   r   r   )rl   r   r   r  )r   r   r   r   r   r  )r5   r  r  r   r   rr  r/   r"  r]   r   r   r   test_with_weightsB  s@   
&"z"TestHarmonicMean.test_with_weights)r+   r,   r-   r  r  r+  rQ  rS  rX  r  rY  r[  r\  r  r   r   r^  rI  r`  r  r   r   r  r   rN    s"    rN  c                       sT   e Zd Zdd Z fddZdd Zdd Zd	d
 Zdd Zdd Z	dd Z
  ZS )
TestMedianc                 C   r  rT   r/   medianr  r]   r   r   r   r  Z  r  zTestMedian.setUpc                    s(   t   }t|d dkr|d |S )+Overload method from UnivariateCommonMixin.r   r   N)r  r  rL   r  r  r  r   r   r  ]  s   

zTestMedian.prepare_datac                 C   2   g d}t |d dksJ | | |d d S )Nr   r   r   r   rl   r  r   r   r&  rL   r5   r  r  r   r   r   test_even_intsd     zTestMedian.test_even_intsc                 C   re  )N)r   r   r   r   rl   r  r   r   r   r   rg  r  r   r   r   test_odd_intsj  ri  zTestMedian.test_odd_intsc                 C   sj   t }|dd|dd|dd|dd|ddg}t|d dks"J t| | | ||dd d S )Nr   r   r   r   r   rl   r   rL   r4  r  r5   r  r8  r   r   r   test_odd_fractionsp  s
   ,
zTestMedian.test_odd_fractionsc                 C   sr   t }|dd|dd|dd|dd|dd|ddg}t|d dks&J t| | | ||dd d S 	Nr   r   r   r   r   rl   r  r   rk  r8  r   r   r   test_even_fractionsx  
   4
zTestMedian.test_even_fractionsc                 C   s^   t }|d|d|d|d|dg}t|d dksJ t| | | ||d d S )N2.53.14.25.75.8r   r   r   rL   r4  r  r5   r  r  r   r   r   test_odd_decimals  s
   "
zTestMedian.test_odd_decimalsc                 C   sd   t }|d|d|d|d|d|dg}t|d dks J t| | | ||d	 d S )
Nz1.2rp  rq  rr  rs  rt  r   r   z3.65ru  r  r   r   r   test_even_decimals  
   (
zTestMedian.test_even_decimals)r+   r,   r-   r  r  rh  rj  rl  rn  rv  rw  r  r   r   r  r   ra  X  s    ra  c                   @   r   )TestMedianDataTypec                 C   r  rT   rb  r]   r   r   r   r    r  zTestMedianDataType.setUpc                 C   sF   t td}t|d dksJ |t|kr!t| |t|ks|S )Nrc   r   r   )r|  r9  rL   r  r4  r  r  r   r   r   r    s   
zTestMedianDataType.prepare_dataN)r+   r,   r-   r  r  r   r   r   r   ry    s    ry  c                   @   r_   )
TestMedianLowc                 C   r  rT   )r/   Z
median_lowr  r]   r   r   r   r    r  zTestMedianLow.setUpc                 C   re  )Nrf  r   r   r   rg  r  r   r   r   rh    ri  zTestMedianLow.test_even_intsc                 C   sr   t }|dd|dd|dd|dd|dd|ddg}t|d dks&J t| | | ||dd d S rm  rk  r8  r   r   r   rn    ro  z!TestMedianLow.test_even_fractionsc                 C   sd   t }|d|d|d|d|d|dg}t|d dks J t| | | ||d d S 	Nz1.1z2.2z3.3z4.4r\  z6.6r   r   ru  r  r   r   r   rw    rx  z TestMedianLow.test_even_decimalsNr+   r,   r-   r  rh  rn  rw  r   r   r   r   rz    
    rz  c                   @   r_   )
TestMedianHighc                 C   r  rT   )r/   Zmedian_highr  r]   r   r   r   r    r  zTestMedianHigh.setUpc                 C   re  )Nrf  r   r   r   rg  r  r   r   r   rh    ri  zTestMedianHigh.test_even_intsc                 C   sr   t }|dd|dd|dd|dd|dd|ddg}t|d dks&J t| | | ||dd d S rm  rk  r8  r   r   r   rn    ro  z"TestMedianHigh.test_even_fractionsc                 C   sd   t }|d|d|d|d|d|dg}t|d dks J t| | | ||d d S r{  ru  r  r   r   r   rw    rx  z!TestMedianHigh.test_even_decimalsNr|  r   r   r   r   r~    r}  r~  c                   @   r  )TestMedianGroupedc                 C   r  rT   )r/   Zmedian_groupedr  r]   r   r   r   r    r  zTestMedianGrouped.setUpc                 C   s   g d}t |d dksJ | | |d g d}t |d dks%J | | |d g d}t |d dks<J | | |dd	 g d
}t |d dksTJ | j| |dddd d S )N)r   r     r  r  rc   rc   r   r   r  )r   r  r  r  r  r  rc   g     +@)rl   r   r   rc   r  r  r  r     r  r   rl   g     `3@)r      r  r  r  r  r  r     r  r     r        g4@:0yE>r'   )rL   r5   r  rK   r  r   r   r   test_odd_number_repeated  s   z*TestMedianGrouped.test_odd_number_repeatedc                 C   s   g d}t |d dksJ | j| |dddd g d}t |d dks(J | j| |d	dd g d
}t |d dksAJ | | |d g d}t |d dksXJ | | |d d S )N)
rl   r   r   rc   r  r  r  r  r  r   r   r   rl   g*3@r  r  )r   r   r   r   r   rl   g["8@)r   r   r   r   r   r   rl   rl   rl   rl   r  r  r?  )
r   r   r   r   rl   rl   rl   rl   r  r        @)rL   rK   r  r5   r  r   r   r   test_even_number_repeated  s   z+TestMedianGrouped.test_even_number_repeatedc                 C   sL   dddt ddtdfD ]}dD ]}|g| }| | |t| qqd S )N333333@D   g ޗCr   r   z32.9714r,  )r   r   r5   r  r   r-  r   r   r   r.    s   
z,TestMedianGrouped.test_repeated_single_valuec                 C   sd   t }|dd|dd|dd|dd|ddg}t|d dks"J t| | | |d d S )	Nrl   r   r   r  ra   r   r         @rk  r8  r   r   r   rl    s
   ,
z$TestMedianGrouped.test_odd_fractionsc                 C   sl   t }|dd|dd|dd|dd|dd|ddg}t|d dks&J t| | | |d d S )	Nrl   r   r   r  ra   r   r         
@rk  r8  r   r   r   rn    s
   4
z%TestMedianGrouped.test_even_fractionsc                 C   sZ   t }|d|d|d|d|dg}t|d dksJ t| | | |d d S )Nr\  6.57.58.5r   r   g      @ru  r  r   r   r   rv    s
   "
z#TestMedianGrouped.test_odd_decimalsc                 C   s   t }|d|d|d|d|d|dg}t|d dks J t| | | |d |d|d|d|d|d|dg}t|d dksLJ t| | | |d d S )	Nr\  r  r  r  r   r         @g      @ru  r  r   r   r   rw    s   (
(
z$TestMedianGrouped.test_even_decimalsc                 C   s\   g d}|  | |dd g d}| j| |dddd g d}|  | |d	d
 d S )N)
      @r  r  r   r   r  r  r  r&  r	  r  g      @)r  r  r  r   r   r   r  r  r  r&  r	  g["8@r  r  )   r       r  r  r    r  ,  @  iT  r  g     p@)r5   r  rK   r  r   r   r   test_interval,  s   zTestMedianGrouped.test_intervalc                 C   sp   g d}|  t| j| g d}|  t| j| g d}d}|  t| j|| g d}d}|  t| j|| d S )N) r  r  )    r  r  r  r  r  r  )r9   r:  intervalr   r   r   test_data_type_error5  s   z&TestMedianGrouped.test_data_type_errorN)r+   r,   r-   r  r  r  r.  rl  rn  rv  rw  r  r  r   r   r   r   r    s    		r  c                   @   rv   )TestModec                 C   r  rT   )r/   moder  r]   r   r   r   r  H  r  zTestMode.setUpc                 C   s   g dS )rd  )r   r   r   r   r   r   r   r   r   r   r   Nr   r]   r   r   r   r  K  s   zTestMode.prepare_datac                 C   s"   t ddd}| | |d d S r  rZ  r  r   r   r   r  P  s   zTestMode.test_range_datac                 C   s4   d}|  | |d d }|  | |d d S )NZabcbdbr   zfe fi fo fum fi fiZfi)r5   r  r   r  r   r   r   test_nominal_dataU  s   zTestMode.test_nominal_datac                 C   sD   t td}tdD ]}||g }t| | | || q
d S Nr   r|  r9  r4  r  r5   r  )r9   r:  rR   r   r   r   r   test_discrete_data\  s   

zTestMode.test_discrete_datac                 C   sF   g d}| d| d  krdksJ  J | | |d d S )N)r   r   r   r   r   r   r   r   rl   r  r  r  r  r   r   r   r   r   r  r   )r	  r5   r  r  r   r   r   test_bimodal_datad  s   (zTestMode.test_bimodal_datac                 C   s"   t td}| | |d d S )Nr   r   )r|  r9  r5   r  r  r   r   r   test_unique_datak  s   zTestMode.test_unique_datac                 C   s   |  t| jd  d S rT   r  r]   r   r   r   test_none_dataq  s   zTestMode.test_none_datac                 C   s$   t g d}| | |d d S )N)r   r   r   r   r   )rB   Counterr5   r  r  r   r   r   test_counter_datax  s   zTestMode.test_counter_dataN)r+   r,   r-   r  r  r  r  r  r  r  r  r  r   r   r   r   r  F  s    r  c                   @   r   )TestMultiModec                 C   s@   t j}| |ddg | |dg d | |dg  d S )NZaabbbbbbbbccr   Zaabbbbccddddeeffffgg)r   r   r   r  )r/   	multimoder5   )r9   r  r   r   r   test_basics  s   zTestMultiMode.test_basicsN)r+   r,   r-   r  r   r   r   r   r    s    r  c                   @   rZ  )	TestFMeanc              	   C   s   t j}t}t}g dddf|d|d|dgddf|dd	|d
d|dd
gddfg dddfdd
|dd
gddfdtg dddffD ]\}}}||}| t|t| | ||| qEd S )Nr&  r        @      @floats3.54.05.25decimalsr   r   r   r   r   	fractions)TFTTF333333?Zbooleansr&  mixed types)r  r  r}  iterator)	r/   fmeanr   r   r  rH  r   r   r5   )r9   r  r   r   r:  expected_meanr  actual_meanr   r   r   r    s    	zTestFMean.test_basicsc                 C   sJ  t j}t j}| | |g  W d    n1 sw   Y  | | |tg  W d    n1 s5w   Y  | t |d  W d    n1 sNw   Y  | t |g d W d    n1 siw   Y  | t |  W d    n1 sw   Y  | t |g dd W d    d S 1 sw   Y  d S )Nr   Nr  r   r  r  F   )r/   r  r"  r   r  rr  )r9   r  r"  r   r   r   test_error_cases  s(   

"zTestFMean.test_error_casesc                 C      t j}td}td}| t|d|gd | t|||gd | t|d|gd | t ||| g W d    d S 1 sJw   Y  d S NZNanInfr   r   znan and infinityZinfinity)	r/   r  r   rh   r   r   r%   r   r$   )r9   r  NaNr  r   r   r   test_special_values     "zTestFMean.test_special_valuesN)r+   r,   r-   r  r  r  r   r   r   r   r    s    r  c                   @   s@   e Zd ZdZdd Zdd Zdd Zdd	 Zd
d Zdd Z	dS )VarianceStdevMixinr"   c                 C   s6   dddt ddtdfD ]}| | |gd qd S )Nr   g3@g  %Br   r   z8.392r   r"  r   r   r   r   r#    r$  z$VarianceStdevMixin.test_single_valuec                 C   sH   dddt ddtdfD ]}dD ]}|g| }| | |d qqd S )	Nr@  r   g @6<Cr   r   z62.4802)r   r   rl   rc   r   r"  r-  r   r   r   r.    s   
z-VarianceStdevMixin.test_repeated_single_valuec                 C   s4   dgd }|  |}| j|ddd | |d d S )Ng.F7ݚ?r  r[   gؗҌ<r  r   )r  rK   rO  r;  r   r   r   test_domain_error_regression  s   

z/VarianceStdevMixin.test_domain_error_regressionc                    s>   g d}|  |}d  fdd|D }| |  || d S )N)
g{Gz?gRQ?g
ףp=
?gRQ @gp=
ף@gQ	@r  gQ@gGz@gQ@g     j@c                    rA  r   r   r  shiftr   r   r7    r  z6VarianceStdevMixin.test_shift_data.<locals>.<listcomp>)r  rK   r9   r  r   r:  r   r  r   test_shift_data  s
   
z"VarianceStdevMixin.test_shift_datac                    sT   g d}t dd |D sJ | |}d  fdd|D }| | || d S )N)
r   r   r   r   rl   r   r   r   r   r   c                 s   s    | ]	}|t |kV  qd S rT   )ro   r  r   r   r   r     s    z;VarianceStdevMixin.test_shift_data_exact.<locals>.<genexpr>i ʚ;c                    rA  r   r   r  r  r   r   r7    r  z<VarianceStdevMixin.test_shift_data_exact.<locals>.<listcomp>)allr  r5   r  r   r  r   test_shift_data_exact  s   
z(VarianceStdevMixin.test_shift_data_exactc                 C   s6   dd t dD }| |}| | t|| d S )Nc                 S   r1  )r(  r   r3  r6  r   r   r   r7    r8  z:VarianceStdevMixin.test_iter_list_same.<locals>.<listcomp>r   )r9  r  r5   r  r  r   r   r   test_iter_list_same  s   
z&VarianceStdevMixin.test_iter_list_sameN)
r+   r,   r-   r(   r#  r.  r  r  r  r  r   r   r   r   r    s    
	r  c                   @   r   )TestPVariancec                 C   r  rT   )r/   	pvariancer  r]   r   r   r   r    r  zTestPVariance.setUpc                 C   s0   t td}t| d}| | || d S )Nr  g   P_Ar  r  r   r   r   test_exact_uniform  s   
z TestPVariance.test_exact_uniformc                 C   "   g d}d}|  | || d S )Nr   r   r  r   g     6@r1  r9   r:  exactr   r   r   r       zTestPVariance.test_intsc                 C   sX   t }|dd|dd|dd|ddg}|dd}| |}| || | |t  d S )Nr   r   r   r   r   r   r  r5   assertIsInstancer9   r   r:  r  rx   r   r   r   r       $

zTestPVariance.test_fractionsc                 C   sN   t }|d|d|d|dg}|d}| |}| || | |t  d S )Nz12.1z12.2z12.5z12.9z0.096875r   r  r5   r  r9   r   r:  r  rx   r   r   r   r    s   
zTestPVariance.test_decimalsN)r+   r,   r-   r  r  r  r  r  r   r   r   r   r    s    	r  c                   @   rL  )TestVariancec                 C   r  rT   )r/   variancer  r]   r   r   r   r  #  r  zTestVariance.setUpc                 C   6   dddt ddtdfD ]}| tj| j|g qd S )Nr   g333338@g (G!=Cr!  r   z4.2084r   r   r   r/   r"  r  r   r   r   r   r#  &  r$  zTestVariance.test_single_valuec                 C   r  )Nr  r   r1  r  r   r   r   r  +  r  zTestVariance.test_intsc                 C   sX   t }|dd|dd|dd|ddg}|dd}| |}| || | |t  d S )Nr   r   r   r   r   r  r  r   r   r   r  1  r  zTestVariance.test_fractionsc                 C   sZ   t }|d|d|d|dg}d|d |d }| |}| || | |t  d S )Nr   r   r   r   z9.5r   r  r  r   r   r   r  :  s   
zTestVariance.test_decimalsc                 C   0   d}|  | |d |  | j|ddd d S )Nr   r  r   r  xbarr   r1  r  r   r   r   test_center_not_at_meanC  s   z$TestVariance.test_center_not_at_meanN)	r+   r,   r-   r  r#  r  r  r  r  r   r   r   r   r  !  s    		r  c                   @   rZ  )
TestPStdevc                 C   r  rT   )r/   Zpstdevr  r]   r   r   r   r  J  r  zTestPStdev.setUpc                 C   8   dd t dD }tt|}| | || d S )Nc                 S   r1  )ir  r3  r6  r   r   r   r7  O  r8  z7TestPStdev.test_compare_to_variance.<locals>.<listcomp>r   )r9  r   sqrtr/   r  r5   r  r  r   r   r   test_compare_to_varianceM     z#TestPStdev.test_compare_to_variancec                 C   r  )N)r   r  r   r   r  r   )mur  r1  r  r   r   r   r  S  s   z"TestPStdev.test_center_not_at_meanN)r+   r,   r-   r  r  r  r   r   r   r   r  H  s    r  c                   @   r_   )
	TestStdevc                 C   r  rT   )r/   stdevr  r]   r   r   r   r  [  r  zTestStdev.setUpc                 C   r  )NQ   gHzwi@g  f7?+Brl   r   z35.719r  r   r   r   r   r#  ^  r$  zTestStdev.test_single_valuec                 C   r  )Nc                 S   r1  )rk   r   r3  r6  r   r   r   r7  e  r8  z6TestStdev.test_compare_to_variance.<locals>.<listcomp>r   )r9  r   r  r/   r  r5   r  r  r   r   r   r  c  r  z"TestStdev.test_compare_to_variancec                 C   s   d}|  | j|ddd d S )Nr  r  r  r   r1  r  r   r   r   r  i  s   z!TestStdev.test_center_not_at_meanN)r+   r,   r-   r  r#  r  r  r   r   r   r   r  Y  s
    r  c                   @   r   )TestGeometricMeanc                 C   s   t j}| |g dd | |ddgd | |dgd td tdd	tdd
tddtdddtdddg ddd td
D dd tdD dd tdD f	D ]!}ttt	|t	dt
|  }||}| t|t| qXd S )N)rb   r  r         B@r  g      "@g      @g     1@l   ; r   rn   r   r  r  r   r(  )r   ra   r  rl   x   r   c                 S      g | ]}t d qS )      I@r4  expovariater   rR   r   r   r   r7  }      z1TestGeometricMean.test_basics.<locals>.<listcomp>c                 S   r1  )r5  r  )r4  lognormvariater  r   r   r   r7  ~  r8    c                 S   s   g | ]	}t d ddqS )r    i  )r4  
triangularr  r   r   r   r7    s    r  )r/   geometric_meanassertAlmostEqualr4  seedr9  r   prodr   r   rL   rh   iscloser   )r9   r  rngZ
gm_decimalZgm_floatr   r   r   r  o  s&   


 zTestGeometricMean.test_basicsc                 C   s   t j}t}t}d}g ddf|d|d|dgdf|dd	|d
d|dd
gdfdd
|dd
gdfdtg ddffD ]\}}||}| t|t| | j||dd q<d S )Ngd@r  r  r  r  r  r  r   r   r   r   r   r  r&  r  )r  r}  r  rl   Zplaces)	r/   r  r   r   r  rH  r   r   r  )r9   r  r   r   r  r:  r  r  r   r   r   test_various_input_types  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      pr[   )r/   r  rh   r   r  r   r%   assertNotEqual)r9   r  ZlargeZbig_gmZsmallZsmall_gmr   r   r   test_big_and_small  s   z$TestGeometricMean.test_big_and_smallc                 C   s  t j}t j}| | |g  W d    n1 sw   Y  | | |g d W d    n1 s5w   Y  | | |g d W d    n1 sPw   Y  | | |tg  W d    n1 skw   Y  | t |d  W d    n1 sw   Y  | t |g d W d    n1 sw   Y  | t |  W d    n1 sw   Y  | t |g dd W d    d S 1 sw   Y  d S )N)r&  r[   r  )r&  g      r  r  r  r  )r/   r  r"  r   r  rr  )r9   r  r"  r   r   r   r    s4   

"z"TestGeometricMean.test_error_casesc                 C   r  r  )	r/   r  r   rh   r   r   r%   r   r$   )r9   r  r  r  r   r   r   r    r  z%TestGeometricMean.test_special_valuesN)r+   r,   r-   r  r  r  r  r  r   r   r   r   r  m  s    r  c                   @   r   )TestQuantilesc                    s  t j}g d}t| dg fddgfdddgfdg d	fd
g dfdg dfdg dfdg dfdg dfdg dff
D ]\}}| ||||d | t|||d|d  tttfD ]" |t	 ||d}| 
 fdd|D  | |tt	 | q^t|dkr| |||d| t|}d|d  |d  }d|d  |d  }|||g }	| |||d||	|dd||f dd }
tt	|
|}|t	|
||d}| 
td d t||D  q=tdd!D ]}tjtd"|d#}||\}}}| |t | qd S )$N)r  r>     r  i^  r   r        @o@r         i@      t@r   )      d@r
       t@rl   )      a@     k@     @r@     u@r  )      ^@r  r
  r       u@r   )      Y@r  g     j@r
  g     r@r  g     Xv@r   )	g      V@r  g      g@r  r
  r  g     `t@r  g     v@r   )      T@r  r  r        l@r
  g     q@r  r  r  g     v@rc   )g      R@g      Z@r  g      e@r  r        n@g     p@r  r  g     t@r  g     @v@g      w@r   c                 3        | ]}t t| kV  qd S rT   r  r   r  Zdatatyper   r   r         z4TestQuantiles.test_specific_cases.<locals>.<genexpr>r   r\   rk   	inclusiver   methodc                 S      d|  d S Nr&  g3333J@r   r
   r   r   r   r     r  z,TestQuantiles.test_specific_cases.<locals>.fc                 s        | ]\}}t ||V  qd S rT   r   r  r   rS   r   r   r   r   r     r  r  rn   k)r/   	quantilesr4  r  r5   rL   r   r   r   r   rh   r|  r  r  rP   r9  choicesrc  )r9   r(  r:  r   r   rx   ZsdatalohiZpadded_datar   expactr'  q1q2q3r   r  r   test_specific_cases  sP   









 z!TestQuantiles.test_specific_casesc                    s.  t j}g d}t| dg fddgfdddgfdg d	fd
g dfdg dfdg dfdg dfdg dfdg dff
D ]k\}}| ||||dd | t|||dd|d  tttfD ]# |t	 ||dd}| 
 fdd|D  | |tt	 | q`dd }tt	||}|t	|||dd}| 
tdd t||D  q=| |ddgdddg d | |tdd dddg d d!d" td#D }||d$dd}	|t| |t| ||d$d%}| ||	 tdd&D ]}
tjtd|
d'}||dd(\}}}| |t | qd S ))N)rn   r>  i  i   r   r        r@r   r        y@r   )     e@r2       @@rl   )r  r       v@     @r  )     b@r  r2  r3       @r   )g     0a@   r  r2  g     pw@r5  g     P@r   )	g     @`@r  g     g@r  r2  r6  g     {@r7  g     @@r   )g     @_@r8  r4  r  r
  r2  r  r3  r5  r9  g     @rc   )r  g     a@r  g     f@r  r  g     q@r  r6  r3  g      ~@r7  g      @g     @r  r  c                 3   r  rT   r  r  r  r   r   r   	  r  z>TestQuantiles.test_specific_cases_inclusive.<locals>.<genexpr>c                 S   r!  r"  r   r
   r   r   r   r   	  r  z6TestQuantiles.test_specific_cases_inclusive.<locals>.fc                 s   r#  rT   r$  r%  r   r   r   r   	  r  r   rn   )	      $@r5  g      >@g      D@r  g      N@g     Q@r  g     V@r   c                 S   r  )r  )r4  	randranger  r   r   r   r7  (	  r  z?TestQuantiles.test_specific_cases_inclusive.<locals>.<listcomp>i      r  r  r&  r   )r/   r(  r4  r  r5   rL   r   r   r   r   rh   r|  r  rP   r9  rO  rg   r   r)  rc  )r9   r(  r:  r   r   rx   r   r,  r-  r   r'  r.  r/  r0  r   r  r   test_specific_cases_inclusive  sR   








 z+TestQuantiles.test_specific_cases_inclusivec                 C   sP   t j}tddD ]}dg| }| ||g d | ||ddg d qd S )Nr   r   r;  )r;  r;  r;  r  r>  )r/   r(  r9  r5   )r9   r(  r   r:  r   r   r   test_equal_inputs4	  s   
zTestQuantiles.test_equal_inputsc              	      s   t j}d}dd t|D  tt |kr& td tt |ks   dD ]}|| }| 	 fdd| |dD t
t||| q,dD ].}|| || d	 h} fd
d| |dD }dd t||d	d  D }| ||k qLd S )Nr  c                 S   r  )皙?r  r  r   r   r   r7  ?	  r  z9TestQuantiles.test_equal_sized_groups.<locals>.<listcomp>rA  )r   r   rl   r   r  r   rn   r>  r  r   r  rR  r  c                       g | ]}t   |qS r   bisectr   qr  r   r   r7  H	  r8  r  )
r  r!  ;   m      i;  i  is  i  i)&  r   c                    rB  r   rC  rE  r  r   r   r7  O	  r8  c                 S   s   h | ]\}}|| qS r   r   )r   prF  r   r   r   	<setcomp>P	  r8  z8TestQuantiles.test_equal_sized_groups.<locals>.<setcomp>)r/   r(  r9  rL   setr  r4  r  sortr5   r|  rP   rh   )r9   r(  totalr   Z
group_sizeZgroup_sizesposZsizesr   r  r   test_equal_sized_groups<	  s&   z%TestQuantiles.test_equal_sized_groupsc                 C   s  t j}t j}| t |  W d    n1 sw   Y  | t |g dddd W d    n1 s7w   Y  | t |g dd W d    n1 sSw   Y  | | |g ddd W d    n1 spw   Y  | | |g ddd W d    n1 sw   Y  | t |g ddd W d    n1 sw   Y  | t |g ddd	 W d    n1 sw   Y  | | |d
gdd W d    n1 sw   Y  | t |g ddd W d    d S 1 sw   Y  d S )N)r   r  r   r  r   r  r   r\   r   Xr>  r   )r   Nr   )r/   r(  r"  r   rr  r$   )r9   r(  r"  r   r   r   r  S	  s:   $zTestQuantiles.test_error_casesN)r+   r,   r-   r1  r?  r@  rP  r  r   r   r   r   r    s    66r  c                   @   r   )TestBivariateStatisticsc              	   C   s   g dddgfddgg dffD ]X\}}|  tj t|| W d    n1 s+w   Y  |  tj t|| W d    n1 sGw   Y  |  tj t|| W d    q1 scw   Y  qd S )Nr  r   r   r   r/   r"  
covariancecorrelationlinear_regressionr  r   r   r   test_unequal_size_errorl	  s   z/TestBivariateStatistics.test_unequal_size_errorc              	   C   s   g g fg ddgfddgg fdgdgfdgddgfddgdgffD ]X\}}|  tj t|| W d    n1 s;w   Y  |  tj t|| W d    n1 sWw   Y  |  tj t|| W d    q 1 ssw   Y  q d S )Nr   r   rS  r  r   r   r   test_small_sample_errorx	  s"   


z/TestBivariateStatistics.test_small_sample_errorN)r+   r,   r-   rW  rX  r   r   r   r   rR  j	  s    rR  c                   @   r   )TestCorrelationAndCovariancec                 C   s   g dg ddfg dg ddfg dg ddfg dg ddfg dg dd	ffD ]\}}}|  t||| |  t||| q*d S )
Nr  r   r\   rk   r(  r\   )r   r   r   )r   r   r   r   )r   r   r   r   r  r/   rU  rT  )r9   r   r&   rx   r   r   r   test_results	  s   z)TestCorrelationAndCovariance.test_resultsc                 C   sl   g d}g d}|  t||d |  t||d g d}|  t||d |  t||d d S )Nr  )r   r   r  r   rl   )r   rA  g333333?r   r   r[  r  r   r   r   test_different_scales	  s   z2TestCorrelationAndCovariance.test_different_scalesN)r+   r,   r-   r\  r]  r   r   r   r   rY  	  s    rY  c                   @   r   )TestLinearRegressionc                 C   sN   g d}g d}|  tj t|| W d    d S 1 s w   Y  d S )N)r   r   r   r  )r   r/   r"  rV  r  r   r   r   test_constant_input_error	  s
   "z.TestLinearRegression.test_constant_input_errorc              
   C   s   g dg dddfg dg dddfg dg dddfg dg ddd	fg dg d
ddfg dg dddfg dg dddffD ]\}}}}t ||\}}| || | || qAd S )Nr  )r   r   r   r   r   )rn   rn   rn   rn   )r   r  r   r   r   rZ  r\   )r   r  r   r  )gffffff@g@r  rl   r   )r/   rV  r  )r9   r   r&   Ztrue_interceptZ
true_slopeZslopeZ	interceptr   r   r   r\  	  s   	z!TestLinearRegression.test_resultsN)r+   r,   r-   r_  r\  r   r   r   r   r^  	  s    r^  c                   @   s   e Zd Zdd Zdd Zdd Zdd Zd	d
 Zdd Ze	j
dd Zdd Zdd Zdd Zdd Zdd Zdd Zdd Zdd Zdd  Zd!d" Zd#d$ Zd%S )&TestNormalDistc                 C   sV   | j dd}| t t| W d    n1 sw   Y  | t|jd d S )Nr  r   )Z_muZ_sigma)r  
NormalDistr   rr  varsr5   r}  	__slots__r9   ndr   r   r   
test_slots	  s
   
zTestNormalDist.test_slotsc                 C   s   | j dd}| |jd | |jd | |jd | j  }| |jd | |jd | |jd | | j j | j dd W d    n1 sOw   Y  G dd d| j j}|d	d
}| t|| d S )Nr  ra   i!  r   r   ic                   @   rA  )zGTestNormalDist.test_instantiation_and_attributes.<locals>.NewNormalDistNrB  r   r   r   r   NewNormalDist	  rD  rg  r>  rl   )	r  ra  r5   r0  r  r  r   r"  r   )r9   re  rg  nndr   r   r   !test_instantiation_and_attributes	  s   

z0TestNormalDist.test_instantiation_and_attributesc                 C   s   | j j}g d}| |||dd | |t||dd | |t||dd | | j j |g  W d    n1 sGw   Y  | | j j |dg W d    n1 sdw   Y  G dd d|}||}| t|| d S )N)`   rm   Z   r   n   r)  r   r   c                   @   rA  )zBTestNormalDist.test_alternative_constructor.<locals>.NewNormalDistNrB  r   r   r   r   rg  	  rD  rg  )	r  ra  r5   Zfrom_samplesr}  r  r   r"  r   )r9   ra  r:  rg  rh  r   r   r   test_alternative_constructor	  s   
z+TestNormalDist.test_alternative_constructorc                 C   s   | j j}d\}}|||}d}||}| t|| | ttt|th | j 	|}| 
||d  |  koA||d  kn   d}|j|dd}|j|dd}	|j|dd}
|j|dd}| ||
 | |	| | ||	 d S )N)r  r  r   r   rn   zhappiness and joy)r   ztrouble and despair)r  ra  Zsamplesr5   rL   rL  r   r   r   r0  rh   r  )r9   ra  r  sigmarQ  r   r:  r  Zdata1Zdata2Zdata3Zdata4r   r   r   test_sample_generation	  s"   

.z%TestNormalDist.test_sample_generationc           
      C   s  | j j}|dd}| |d|d | |d|d tdD ]}| |d| |d|  q%d}tddD ]}||| || | }| j|||d	d
 q?| }tg dD ]\}}| j||d |d	d
 | j|| d |d	d
 qd|dd}	| | j j	 |	d W d    n1 sw   Y  | 
|tdd | 
|tdd | t|td d S )Nrn   rc   r)  r   r   g      P?rk  r]  r   r  )2+ݓ?rp  rp  ggDio?g?gV}b?gQ|?gFx?ggs?g٬\m?g rh?gK7A`?g|гY?gQ?gQI?gsh|??g=yX5?g|a2U0*?gQ?gN@?g/$?g~k	?g]C?gw/?g~:p?g>W[?gMO?gW[?g鷯?g{Gz?gqh?g|a2U?gK7A?gvq-?gjt?gc]F?g\C?gףp=
?g?ga2U0*?gy):?g(y?gN@a?gfc]F?g"lxz,?gOn?g37?gec]?gn?gT?r  r   -Infr[   r  r  )r  ra  Z
assertLesspdfr9  r  cdfrO   r   r"  r5   r   rh   r   r   )
r9   ra  rQ  rR   dxr   Zest_pdfZZpxYr   r   r   test_pdf
  s*   
"
zTestNormalDist.test_pdfc                    s6  | j j}|dd  fddtddD }| ttt|th | |t| |  	dd | }dD ]\}}| j
|	||d	d
 | j
|	| d| d	d
 q7|dd}| | j j |	d W d    n1 sqw   Y  |  	tdd |  	tdd | t 	td d S )Nrn   rc   c                    s   g | ]}  |qS r   )rs  r  rQ  r   r   r7  2
  r  z+TestNormalDist.test_cdf.<locals>.<listcomp>r   r>  r   ))r[   r   )r   gqZ ?)r   gE_A?)gQ?gGɫs?)g(\?g؞Y?)gQ?g9#?)gHzG?g&S?)r  gMbX9?)gQ?gT㥛 ?)g?g??)gffffff @g_xZ?)gQ@g#0?)g)\(@gu<f2?)gףp=
@gVe?)gHzG@g9?rl   r  r   r   rk  rq  r[   r  r  )r  ra  r9  r5   rL  r   r   r   r  rs  r  r   r"  rh   r   r   )r9   ra  Zcdfsru  zZcum_probrv  r   rx  r   test_cdf/
  s"   

zTestNormalDist.test_cdfc                 C   s  | j j}|dd}| |d|j | }dddd}| D ]1\}}t|dd	D ]&\}}|d
|   }	| j||	 |dd d|	 }	| j||	|dd q*q | |dddd d}
td|
D ]}	|	|
 }	| |	||	|	 qetddD ]#}d|  }	| |	||	|	 d|	 }	| |	||	|	 q}tdD ]}| j||	||dd q| 
| j j |d W d    n1 sw   Y  | 
| j j |d W d    n1 sw   Y  | 
| j j |d W d    n	1 sw   Y  | 
| j j |d W d    n	1 s#w   Y  | 
| j j |dd}|d W d    n	1 sEw   Y  | t|td d S )Nrn   rc   r   )
r[   gRQ?gS㥛@gS
@gT㥛 @g^I@g rh@g+N@gC@gV-o@)
g rh?g\(\?g~jt@g+@gMbX9@g(\B@gv@g@gSc@gK7A@)
gPn?gS㥛@gQ@gn@g(\@gPn@g@g r@gˡE@glq@)r  r  r   r   )startr;  r   r  r   r  r   g>?g  E@i   3   r  r>  rl   r[   r   g?r   r  )r  ra  r5   Zinv_cdfr0  itemsrO   r  r9  rs  r   r"  rh   r   r   r   )r9   ra  Ziqru  Zppr   rowr,  r   rJ  r   rS   r   r   r   test_inv_cdfM
  sX   


zTestNormalDist.test_inv_cdfc                 C   sh   | j  }dg fddgfdddgfdg dffD ]\}}|j|d	}| td
d t||D  qd S )Nr   r   r[   r   gǘۿgǘ?r   )g/$r[   g/$?r  c                 s   s$    | ]\}}t j||d dV  qdS )r   )Zabs_tolNr$  r%  r   r   r   r   
  s    z0TestNormalDist.test_quantiles.<locals>.<genexpr>)r  ra  r(  rh   r  rP   )r9   ru  r   r   r   r   r   r   test_quantiles
  s   


zTestNormalDist.test_quantilesc                 C   s  | j j}|dd|dddf|dd|dddffD ]\}}}| j|||dd | j|||dd qdd	d
dd}|dd|ddf|dd|ddf|dd|ddf|dd|ddf|dd|ddf|dd|ddf|dd|ddf|dd|ddf|dd|ddf|dd|ddf|dd|ddf|dd|ddf|dd|ddf|dd|ddffD ] \}}| j|||||d	d | j|||||d	d q| }| t |  W d    n1 sw   Y  | t ||| W d    n	1 sw   Y  | t |d  W d    n	1 s-w   Y  | | j j ||dd W d    n	1 sMw   Y  | | j j |dd| W d    d S 1 snw   Y  d S ) Nr[   r  r   gɎ@?gM-[닄?r   r  i    rl   )stepsry  c                   s   t j}| j|j d }|t| j|j }|| d| |   fddt|D }tt| j|}tt|j|}	t||||	}
|tt	||	|
 S )z0Numerical integration cross-check for overlap() r  c                    s   g | ]}|   qS r   r   r  rt  r{  r   r   r7  
  r8  zHTestNormalDist.test_overlap.<locals>.overlap_numeric.<locals>.<listcomp>N)
r   r  r0  r   r  r9  r|  r   rr  rg   )rQ  rv  r  ry  r  centerwidthZx_arrZxpZyprN  r   r  r   overlap_numeric
  s   z4TestNormalDist.test_overlap.<locals>.overlap_numericr  A   r&  rn   rc   rl  r2  ir      r   r   gjt?g~jth?gjt?ga2U0*3?gMbX?r   r   )r  ra  r  Zoverlapr   rr  r"  )r9   ra  ZX1ZX2Zpublished_resultr  rQ  r   r   r   test_overlap
  sP   
$zTestNormalDist.test_overlapc                 C   s(  | j j}|dd}| |dd | |dd | |dd | t |  W d    n1 s8w   Y  | t |dd W d    n1 sSw   Y  | t |d  W d    n1 smw   Y  | | j j |dd	d W d    d S 1 sw   Y  d S )
Nrn   rc      gffffff@:   gffffffr[   r   r   )r  ra  r5   Zzscorer   rr  r"  )r9   ra  rQ  r   r   r   test_zscore
  s"   

"zTestNormalDist.test_zscorec                 C   sX   | j dd}| |jd | |jd | |jd | |jd | |jd d S )Nrn   rc      )r  ra  r5   r0  rc  r  r  r  )r9   rQ  r   r   r   test_properties
  s   zTestNormalDist.test_propertiesc                 C   sL   | j j}|dd}|dd}| || |dd | || |dd d S )Nrn   r   r  rl      r  r  )r  ra  r5   r9   ra  rQ  rv  r   r   r   'test_same_type_addition_and_subtraction
  s
   

z6TestNormalDist.test_same_type_addition_and_subtractionc                 C   s  | j j}|dd}d}| |
 |dd | | |dd | || |dd | || |dd | || |dd | || |dd | || |dd	 | || |dd	 | || |dd
 | t ||  W d    d S 1 sw   Y  d S )Nrn   rc   r   r2  rl  rk  ir   r  r   )r  ra  r5   r   rr  )r9   ra  rQ  r&   r   r   r   test_translation_and_scaling
  s   

"z+TestNormalDist.test_translation_and_scalingc                 C   s|   | j j}|dd}|
 }| || | |j|j | |j|j | }| || | |j|j  | |j|j d S )Nrn   r   )r  ra  ZassertIsNotr5   r0  r  r  r   r   r   test_unary_operations  s   
z$TestNormalDist.test_unary_operationsc                 C   s"  | j j}| }|dd}| }|dd}|dd}|dd}| || | || | || | || | || G dd d}| }	| ||	t | ||	kd | |	|kd G dd d|}
|
d	d
d}|d	d
}| || G dd d}|d	d
}|d	d
}| || d S )Nr   r   r   c                   @   r   )z'TestNormalDist.test_equality.<locals>.Ac                 S   s   dS r  r   r  r   r   r   __eq__$  s   z.TestNormalDist.test_equality.<locals>.A.__eq__N)r+   r,   r-   r  r   r   r   r   rl  #  r&  rl  r   c                       r  )z5TestNormalDist.test_equality.<locals>.SizedNormalDistc                    s   t  || || _d S rT   )r  __init__r   )r9   r  rn  r   r  r   r   r  .  s   
z>TestNormalDist.test_equality.<locals>.SizedNormalDist.__init__)r+   r,   r-   r  r  r   r   r  r   SizedNormalDist-  r  r  rn   rc   9   c                   @   r   )z3TestNormalDist.test_equality.<locals>.LognormalDistc                 S   s   || _ || _d S rT   )r  rn  )r9   r  rn  r   r   r   r  9  s   
z<TestNormalDist.test_equality.<locals>.LognormalDist.__init__N)r+   r,   r-   r  r   r   r   r   LognormalDist8  r&  r  )r  ra  r  r5   r  NotImplemented)r9   ra  nd1nd2nd3Znd4Znd5Znd6rl  r   r  sr  Zlndre  r   r   r   test_equality  s2   






zTestNormalDist.test_equalityc                 C   sZ   | j dd}t|}| || t|}| || tt|}| || d S )N     B@     @)r  ra  copyr5   deepcopypickleloadsdumps)r9   re  r  r  r  r   r   r   test_pickle_and_copy@  s   

z#TestNormalDist.test_pickle_and_copyc                 C   sH   | j j}|dd|dd|dd|dd|ddh}| t|d d S )Nrn   rc   r  g      .@r   r  r   )r  ra  r5   rL   )r9   ZNDr  r   r   r   test_hashabilityI  s   ,zTestNormalDist.test_hashabilityc                 C   s"   | j dd}| t|d d S )Nr  r  z NormalDist(mu=37.5, sigma=5.625))r  ra  r5   reprrd  r   r   r   	test_reprN  s   zTestNormalDist.test_reprN)r+   r,   r-   rf  ri  rm  ro  rw  rz  r   Zskip_if_pgo_taskr  r  r  r  r  r  r  r  r  r  r  r  r   r   r   r   r`  	  s(    	'
@?,	r`  c                   @       e Zd ZeZdd Zdd ZdS )TestNormalDistPythonc                 C      | j tjd< d S r3   r  r  modulesr]   r   r   r   r  X     zTestNormalDistPython.setUpc                 C      t tjd< d S r3   r/   r  r  r]   r   r   r   tearDown[  r  zTestNormalDistPython.tearDownN)r+   r,   r-   r7   r  r  r  r   r   r   r   r  V  s    r  r<   c                   @   r  )TestNormalDistCc                 C   r  r3   r  r]   r   r   r   r  b  r  zTestNormalDistC.setUpc                 C   r  r3   r  r]   r   r   r   r  e  r  zTestNormalDistC.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.N)ZaddTestsr  ZDocTestSuite)loaderZtestsignorer   r   r   
load_testsk  s   r  __main__)r"   r#   )Tr.   rD  rB   collections.abcr  r  r  r   r  r4  r  r?   testr   Ztest.supportr   r   r  r   r/   r   r   r!   r)   r*   Zimport_fresh_moduler7   r=   r   r1   rA   rZ   r`   rw   r   r   r   r   r   r  r  r!  r'  rM  r[  rd  r  r  r  r  r  r  r  r  r  r  r   r/  rN  ra  ry  rz  r~  r  r  r  r  r  r  r  r  r  r  r  rR  rY  r^  r`  r  r@   r  r  r+   mainr   r   r   r   <module>   s    
:+_
>@w-@4d>c'H@`y9r:	1B%'Y %   !
	
