o

6��a!?�@s�dZd
dl
Z
d
dlZd
dlZd
dlmZ
d
dlmZ
d
dlm	Z	
d
dl
mZ
e	dd�ZGd	d�de�Z
d
d�Zdd
d�
Zdd�Zdd�Zdadaegd�
�
Zddi
Zdd�Zdd�ZGdd�d�ZdS)as
tl;dr: all code code is licensed under simplified BSD, unless stated otherwise.

Unless stated otherwise in the source files, all code is copyright 2010 David
Wolever <david@wolever.net>. All rights reserved.

Redistribution and use in source and binary forms, with or without
modification, are permitted provided that the following conditions are met:

   1. Redistributions of source code must retain the above copyright notice,
   this list of conditions and the following disclaimer.

   2. Redistributions in binary form must reproduce the above copyright notice,
   this list of conditions and the following disclaimer in the documentation
   and/or other materials provided with the distribution.

THIS SOFTWARE IS PROVIDED BY <COPYRIGHT HOLDER> ``AS IS'' AND ANY EXPRESS OR
IMPLIED WARRANTIES, INCLUDING, BUT NOT LIMITED TO, THE IMPLIED WARRANTIES OF
MERCHANTABILITY AND FITNESS FOR A PARTICULAR PURPOSE ARE DISCLAIMED. IN NO
EVENT SHALL <COPYRIGHT HOLDER> OR CONTRIBUTORS BE LIABLE FOR ANY DIRECT,
INDIRECT, INCIDENTAL, SPECIAL, EXEMPLARY, OR CONSEQUENTIAL DAMAGES (INCLUDING,
BUT NOT LIMITED TO, PROCUREMENT OF SUBSTITUTE GOODS OR SERVICES; LOSS OF USE,
DATA, OR PROFITS; OR BUSINESS INTERRUPTION) HOWEVER CAUSED AND ON ANY THEORY OF
LIABILITY, WHETHER IN CONTRACT, STRICT LIABILITY, OR TORT (INCLUDING NEGLIGENCE
OR OTHERWISE) ARISING IN ANY WAY OUT OF THE USE OF THIS SOFTWARE, EVEN IF
ADVISED OF THE POSSIBILITY OF SUCH DAMAGE.

The views and conclusions contained in the software and documentation are those
of the authors and should not be interpreted as representing official policies,
either expressed or implied, of David Wolever.

�N)
�wraps)
�
MethodType)
�
namedtuple)
�TestCase�paramzargs kwargsc@s:eZ
dZd
Zdd�Zeddd�
�
Zedd��
Zd	d
�ZdS)ra�
 Represents a single parameter to a test case.

        For example::

            >>> p = param("foo", bar=16)
            >>> p
            param("foo", bar=16)
            >>> p.args
            ('foo', )
            >>> p.kwargs
            {'bar': 16}

        Intended to be used as an argument to ``@parameterized``::

            @parameterized([
                param("foo", bar=16),
            ])
            def test_stuff(foo, bar=16):
                pass
        c
Ost�
||
|�S�
N)�_param�__new__��cls�args�kwargs�r�F/usr/lib/python3/dist-packages/numpy/testing/_private/parameterized.pyr	Bs
z
param.__new__NcCs|
pd
}
|pi}||
i|�
�
S)a
 Creates a ``param`` by explicitly specifying ``args`` and
            ``kwargs``::

                >>> param.explicit([1,2,3])
                param(*(1, 2, 3))
                >>> param.explicit(kwargs={"foo": 42})
                param(*(), **{"foo": "42"})
            rrr
rrr�explicitEs


zparam.explicitc
Csbt|
t
�r|
St|
tf
�r|
f
}
z||
�WSty0
}
zd
t|�
v
r$�td|
|
f�
�
d}~w
w)a(
 Returns an instance of ``param()`` for ``@parameterized`` argument
            ``args``::

                >>> param.from_decorator((42, ))
                param(args=(42, ), kwargs={})
                >>> param.from_decorator("foo")
                param(args=("foo", ), kwargs={})
            zafter * must bez=Parameters must be tuples, but %r is not (hint: use '(%r, )')N)�
isinstancer�str�	TypeError)rr�
errr�from_decoratorSs 













����zparam.from_decoratorc

Csd
|S)Nzparam(*%r, **%r)r)
�selfrrr�__repr__ks
zparam.__repr__�NN)	�__name__�
__module__�__qualname__�__doc__r	�classmethodrrrrrrrr,s



cs�t�
|�
}|jd
d�dg
krdn
d}|j|d
�}tt|�j��
}|jt|�
|d
�}�jt|�
d
�}|��f
dd�t||jpCg�D�
�

dd�|D�
�
tt	��
fd	d��j
D�
�
�
}|rm|�d
|jf
t
|�
f�

|rz|�d|jf
|f�

|S)a Return tuples of parameterized arguments and their values.

        This is useful if you are writing your own doc_func
        function and need to know the values for each parameter name::

            >>> def func(a, foo=None, bar=42, **kwargs): pass
            >>> p = param(1, foo=7, extra=99)
            >>> parameterized_argument_value_pairs(func, p)
            [("a", 1), ("foo", 7), ("bar", 42), ("**kwargs", {"extra": 99})]

        If the function's first argument is named ``self`` then it will be
        ignored::

            >>> def func(self, a): pass
            >>> p = param(1)
            >>> parameterized_argument_value_pairs(func, p)
            [("a", 1)]

        Additionally, empty ``*args`` or ``**kwargs`` will be ignored::

            >>> def func(foo, *args): pass
            >>> p = param(1)
            >>> parameterized_argument_value_pairs(func, p)
            [("foo", 1)]
            >>> p = param(1, 16)
            >>> parameterized_argument_value_pairs(func, p)
            [("foo", 1), ("*args", (16, ))]
    N�
rr
c
s"g|]
\}
}|
�j�
|
|�f�qSr)r
�get)�.0�name�default)
�
prr�
<listcomp>�s��z6parameterized_argument_value_pairs.<locals>.<listcomp>c
Ssh|]\}
}|
�qSrr)r �
n�
_rrr�	<setcomp>�sz5parameterized_argument_value_pairs.<locals>.<setcomp>c
s"g|]
}
|
�
v
r|
�j|
f�qSr)
r
)r r!�r#�seen_arg_namesrrr$�s

�z*%sz**%s)�inspect�
getargspecr�list�zip�len�extend�defaults�dict�sortedr
�append�varargs�tuple�keywords)�funcr#�argspec�
arg_offset�
named_args�resultr4r6rr(r�"parameterized_argument_value_pairsos$



�

�

r<�@cCszt|�
}t
|t�rzt|d
�}Wnty


t|d�}Yn
wt|�
|
kr;|d|
d�d|t|�
|
dd�}|S)z� A shortened repr of ``x`` which is guaranteed to be ``unicode``::

            >>> short_repr("foo")
            u"foo"
            >>> short_repr("123456789", n=4)
            u"12...89"
    zutf-8�latin1N�z...)�reprr�bytesr�UnicodeDecodeErrorr.)�
xr%�x_reprrrr�
short_repr�s	





�,

rEc
Cs�|jdurdSt
||�}d
d�|D�
}|j���d�
\}}}d}|�d�
r-d}|dd�}dt|�
r4dp5dd	�|�
f}	d�|��|	|||g�
S)
Nc
Ss"g|]
\}
}|
�dt|�
���qS)
�
=)
rE)r r%�
vrrrr$�s"z$default_doc_func.<locals>.<listcomp>�

��
.���z%s[with %s]�
 �, )rr<�lstrip�	partition�endswithr.�join�rstrip)
r7�numr#�all_args_with_values�descs�first�nl�rest�suffixrrrr�default_doc_func�s










rZcCsP|j}d
|
f
}t
|j�
dkr$t|jdtf
�r$|dt�|jd�
7}||S)Nz_%sr
r&)rr.rrr�
parameterized�to_safe_name)r7rSr#�	base_name�name_suffixrrr�default_name_func�s



 


r_�noseF)�unittest�	unittest2r`�nose2�pytest�_pytestrdc

Cs&|tv
rt
d
|d�t�
f�
�
|adS)Nz,Invalid test runner: %r (must be one of: %s)rM)�
_test_runnersrrQ�_test_runner_override)
r!rrr�set_test_runner�s


��rhcCsrtd
u
rtSt
dur7t��}t|�
D]"}
|
d}|j�d�
�d�
d}|tvr+t|}|t	vr4|a

t
Sqd
a
t
S)a

 Guess which test runner we're using by traversing the stack and looking
        for the first matching module. This *should* be reasonably safe, as
        it's done during test discovery where the test runner should be the
        stack frame immediately outside. NFr
rrJ)
rg�_test_runner_guessr*�stack�reversed�	f_globalsrrO�_test_runner_aliasesrf)rj�record�frame�modulerrr�
detect_runner�s 









�
rqc@sxeZ
dZd
Zddd�
Zdd�Zdd�Zd	d
�Zdd�Ze	d
d��
Z
e	dd��
Ze	ddd�
�
Ze	dd��
Z
e	dd��
ZdS)r[a= Parameterize a test case::

            class TestInt:
                @parameterized([
                    ("A", 10),
                    ("F", 15),
                    param("10", 42, base=42)
                ])
                def test_int(self, input, expected, base=16):
                    actual = int(input, base=base)
                    assert_equal(actual, expected)

            @parameterized([
                (2, 3, 5)
                (3, 5, 8),
            ])
            def test_add(a, b, expected):
                assert_equal(a + b, expected)
        NcCs|�|
�
|_
|p	t|_dSr)�input_as_callable�	get_inputrZ�doc_func)r�inputrtrrr�__init__
s

zparameterized.__init__csD���
t
�
�
d��
�fd
d�	�
�����_�
�_d�
jf
�
_�S)Nc

3s��|ot|�
}
�j
}t�j�
D]D\}}t�|�
}��|�
||�\}}z#|d
j
�_
|du
r4t|
�
j|�
|V
W|du
rBt	|
�
j�
|�_
q|du
rPt	|
�
j�
|�_
wdS)Nr
)
�typer�	enumerate�parameterized_inputrr�param_as_nose_tuple�setattrr�delattr)�	test_self�test_cls�original_docrSrr#�unbound_func�
nose_tuple�r�	test_func�wrapperrrr�
s$�









�

�z'parameterized.__call__.<locals>.wrapperz_parameterized_original_%sr)�assert_not_in_testcase_subclassrrsry�parameterized_funcr)rr�rr�r�__call__
s





zparameterized.__call__csVt��
�f
d
d��
}|�
�||�|_|}|
du
rt||
�}||f
|j|jp'if
fS)Nc
s�|dd
�i|d
�
�
S)NrKr)
r�
r7rr�<lambda>8
sz3parameterized.param_as_nose_tuple.<locals>.<lambda>)rrtrrrr
)rr}r7rSr#�	nose_funcr�rr�rrz7
s





z!parameterized.param_as_nose_tuplec
Cs&|��}
t
d
d�|
D�
�
rtd�
�
dS)Nc
ss�|]}
t|
t
�V
qdSr)�
issubclassr)r rrrr�	<genexpr>D
s�z@parameterized.assert_not_in_testcase_subclass.<locals>.<genexpr>zqWarning: '@parameterized' tests won't work inside subclasses of 'TestCase' - use '@parameterized.expand' instead.)�$_terrible_magic_get_defining_classes�any�	Exception)r�parent_classesrrrr�B
s


�z-parameterized.assert_not_in_testcase_subclassc
Cs�t�
�}
t|
�
d
k
rgS|
d
}|d
o|d
d��}|r#|�d�
s%gS|�d�
\}}}|�d�
\}}}td|d|dj|dj�S)a@
 Returns the list of parent classes of the class currently being defined.
            Will likely only work if called from the ``parameterized`` decorator.
            This function is entirely @brandon_rhodes's fault, as he suggested
            the implementation: http://stackoverflow.com/a/8793684/71522
            �r
zclass �
(�
)�
[�
])	r*rjr.�strip�
startswithrO�evalrl�f_locals)rrjro�code_contextr&�parentsrrrr�I
s







 
z2parameterized._terrible_magic_get_defining_classescs,t�
�
r��
fd
d�S��
�
�
��f
dd�S)Ncs���
��
Sr)
�check_input_valuesr�rrurrr�]
sz1parameterized.input_as_callable.<locals>.<lambda>c
s�Srrr)
�input_valuesrrr�_
s)�callabler�r�r)rrur�rrrZ
s



zparameterized.input_as_callablecCs t|
t
�s	t
|
�
}
d
d�|
D�
S)Nc
Ssg|]}
t�
|
�
�qSr)rr)r r#rrrr$j
sz4parameterized.check_input_values.<locals>.<listcomp>)rr,)rr�rrrr�a
s


z parameterized.check_input_valuescspd
|vrtj
dtdd�
�s|d
�d|vr$tj
dtdd�
�
s$|d�
�
p't�
�p+t�d
��
��fdd	�	}|S)am A "brute force" method of parameterizing test cases. Creates new
            test cases and injects them into the namespace that the wrapped
            function is being defined in. Useful for parameterizing tests in
            subclasses of 'UnitTest', where Nose test generators don't work.

            >>> @parameterized.expand([("foo", 1, 2)])
            ... def test_add1(name, input, expected):
            ...     actual = add1(input)
            ...     assert_equal(actual, expected)
            ...
            >>> locals()
            ... 'test_add1_foo_0': <function ...> ...
            >>>
            �testcase_func_namez1testcase_func_name= is deprecated; use name_func=r?)
�
stacklevel�testcase_func_docz/testcase_func_doc= is deprecated; use doc_func=Nc	srt�
�}|d
}|dj}����
�}t|�
D]\}}�|||�}��|||�||<�
|||�||_qd|_dS)Nrr
F)r*rjr�rrrx�param_as_standalone_funcr�__test__)	�
f�instancerjro�frame_locals�
parametersrSr#r!�rrtru�	name_funcrr�parameterized_expand_wrapper�
s








z:parameterized.expand.<locals>.parameterized_expand_wrapperr)�warnings�warn�DeprecationWarningrZr_)rrur�rt�legacyr�rr�r�expandl
s 

�


�

zparameterized.expandcsBt��
��
fd
d��
}||_
�|_z|`W|Sty 


Y|Sw)Nc
s�|�
ji�
j
�
�
Sr)rr
)
�
a�r7r#rr�standalone_func�
sz?parameterized.param_as_standalone_func.<locals>.standalone_func)rr�place_as�__wrapped__�AttributeError)rr#r7r!r�rr�rr��
s

�

�z&parameterized.param_as_standalone_funccCstt
�d
d|
��
S)Nz[^a-zA-Z0-9_]+r&)r�re�sub)r�
srrrr\�
szparameterized.to_safe_namerr)rrrrrvr�rzr�r�rrrr�r�r�r\rrrrr[
s"







-


r[)
r=)rr�r*r��	functoolsr�typesr�collectionsrrarrrr<rErZr_rgri�setrfrmrhrqr[rrrr�<module>
s, 





C
;	

�