U C^mG@svdZddlZddlZddlZddlZddlmZddlmZddl m Z zddl m Z Wne k rpeZ YnXddlmZejddkZerdd lmZeZeZeZefZd d Zn,Gd d d ZddZeZefZeZdd Ze ddZGdddeZGddde Z ddZ!d,ddZ"ddZ#ddZ$da%d a&e'd!d"dd#d$gZ(d%d$iZ)d&d'Z*d(d)Z+Gd*d+d+e,Z-dS)-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 . 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 ``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 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) OrderedDict)TestCase) InstanceTypecCs t|||SNrfuncinstancetyperG/tmp/pip-install-6_kvzl1k/numpy/numpy/testing/_private/parameterized.py make_method9src@s eZdZdS)r N)__name__ __module__ __qualname__rrrrr =sr cGs tt|Sr )listziparrr?rcCs|dkr |St||Sr rr rrrrCsparamz args kwargsc@s:eZdZdZddZed ddZeddZd 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 cOst|||Sr )_param__new__clsargskwargsrrrr`sz param.__new__NcCs|pd}|pi}|||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"}) rrrrrrexplicitcs zparam.explicitc Csnt|tr|St|tr|f}z ||WStk rh}z"dt|krHtd||fW5d}~XYnXdS)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 string_types TypeErrorstr)rr errrfrom_decoratorqs    zparam.from_decoratorcCsd|S)Nzparam(*%r, **%r)r)selfrrr__repr__szparam.__repr__)NN) rrr__doc__r classmethodr"r(r*rrrrrJs  c@seZdZdZejZejZdS)QuietOrderedDictzu When OrderedDict is available, use it to make sure that the kwargs in doc strings are consistently ordered. N)rrrr+dict__str__r*rrrrr-sr-cst|}|jdddgkr"dnd}|j|d}t|j}|jt||d}jt|d}|fddt||jpgDdd|Dtt fd dj D}|r| d |j ft |f|r| 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, ))] Nr)rcs"g|]\}}|j||fqSr)r!get).0namedefault)prr sz6parameterized_argument_value_pairs..cSsh|] \}}|qSrr)r2n_rrr sz5parameterized_argument_value_pairs..cs"g|]}|kr|j|fqSr)r!)r2r3r5Zseen_arg_namesrrr6sz*%sz**%s)inspect getargspecr lziplenextendrdefaultsr-sortedr!appendvarargstuplekeywords)r r5ZargspecZ arg_offsetZ named_argsresultrCrErr:r"parameterized_argument_value_pairss$   rG@cCs|t|}t|tr@zt|d}Wntk r>t|d}YnXt||krx|d|dd|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-8latin1Nrz...)reprr# bytes_type text_typeUnicodeDecodeErrorr>)xr7Zx_reprrrr short_reprs   ,rOc Cs|jdkrdSt||}dd|D}|jd\}}}d}|drZd}|dd}dt|rhdpjdd |f} d|| |||gS) NcSs g|]\}}d|t|fqS)z%s=%s)rO)r2r7vrrrr6sz$default_doc_func.. .z %s[with %s] , )r+rGlstrip partitionendswithr>joinrstrip) r numr5Zall_args_with_valuesZdescsfirstnlrestsuffixr rrrdefault_doc_funcs    racCsN|j}d|f}t|jdkrFt|jdtrF|dt|jd7}||S)Nz_%srr8)rr>r r#r$ parameterized to_safe_name)r r\r5 base_nameZ name_suffixrrrdefault_name_funcs  renoseFunittestZ unittest2Znose2ZpytestZ_pytestcCs&|tkrtd|dtf|adS)Nz,Invalid test runner: %r (must be one of: %s)rV) _test_runnersr%rZ_test_runner_override)r3rrrset_test_runners rjcCstdk r tStdkrt}t|D]X}|d}|jddd}|tkrVt|}|t krf|aq|d dr$daqq$datS) 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 disocvery where the test runner should be the stack frame immediately outside. NFrrrSr0zpython2.6/unittest.pyrg) ri_test_runner_guessr;stackreversed f_globalsr1rX_test_runner_aliasesrhrY)rlrecordframemodulerrr detect_runners" rsc@sxeZdZdZdddZddZddZd d Zd d Ze d dZ e ddZ e dddZ e ddZ e ddZdS)rbaE Parameterize a test case:: class TestInt(object): @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|||_|pt|_dSr )input_as_callable get_inputradoc_func)r)inputrvrrr__init__=s zparameterized.__init__csDtdfdd __djf_S)Nc 3s|o t|}|dk r,t|tr,td|fj}tjD]p\}}t|} |||\}}z,|dj_|dk rt |j ||VW5|dk rt |j |_Xq.wrapperz_parameterized_original_%s)N)assert_not_in_testcase_subclassrrur{Zparameterized_funcr)r)rrrr__call__As zparameterized.__call__csrtfdd}||||_|}|dk rVtrBtdkrBdn|}t||t|}||f|j|jpjiffS)Ncs|dd|dS)NrTr)r r rrrirz3parameterized.param_as_nose_tuple..rf) rrvr+PY2rsrrr r!)r)rr r\r5Z nose_funcrZ func_selfrrrr|hsz!parameterized.param_as_nose_tuplecCs&|}tdd|Dr"tddS)Ncss|]}t|tVqdSr )ryr)r2rrrr }sz@parameterized.assert_not_in_testcase_subclass..zqWarning: '@parameterized' tests won't work inside subclasses of 'TestCase' - use '@parameterized.expand' instead.)$_terrible_magic_get_defining_classesany Exception)r)Zparent_classesrrrr{sz-parameterized.assert_not_in_testcase_subclasscCst}t|dkrgS|d}|do6|dd}|rF|dsJgS|d\}}}|d\}}}td|d|dj|djS)a? Returns the set 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 rzclass ()[]) r;rlr>strip startswithrXevalrnf_locals)r)rlrq code_contextr8parentsrrrrs z2parameterized._terrible_magic_get_defining_classescs,trfddSfddS)Ncs Sr )check_input_valuesrrrwrrrrz1parameterized.input_as_callable..csSr rr) input_valuesrrrr)callablerrr)rrwrrrts zparameterized.input_as_callablecCs t|tst|}dd|DS)NcSsg|]}t|qSr)rr()r2r5rrrr6sz4parameterized.check_input_values..)r#r)rrrrrrs z parameterized.check_input_valuesc spd|kr$tjdtdds$|dd|krHtjdtddsH|dpNtpVtd 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': ... >>> Ztestcase_func_namez1testcase_func_name= is deprecated; use name_func=r) stacklevelZtestcase_func_docz/testcase_func_doc= is deprecated; use doc_func=Nc srt}|d}|dj}}t|D]8\}}|||}|||||<|||||_q.d|_dS)Nr0rF)r;rlrrtrzparam_as_standalone_funcr+Z__test__) fr rlrqZ frame_locals parametersr\r5r3rrvrw name_funcrrparameterized_expand_wrappers   z:parameterized.expand..parameterized_expand_wrapper)N)warningswarnDeprecationWarningrare)rrwrrvlegacyrrrrexpands$ zparameterized.expandcsDtfdd}||_|_z|`Wntk r>YnX|S)Ncs|jjSr )r r!rr r5rrstandalone_funcsz?parameterized.param_as_standalone_func..standalone_func)rrZplace_as __wrapped__AttributeError)rr5r r3rrrrrsz¶meterized.param_as_standalone_funccCsttdd|S)Nz[^a-zA-Z0-9_]+r8)r&resub)rsrrrrcszparameterized.to_safe_name)N)NN)rrrr+rxrr|rrr,rtrrrrcrrrrrb(s  '   - rb)rH).r+rsysr;r functoolsrtypesr collectionsrrZMaybeOrderedDict ImportErrorr.rgr version_inforr rr=unicoderLr&rK basestringr$rbytesrrr-rGrOrarerirksetrhrorjrsobjectrbrrrrsT         C;