ó Du—ac@s`dZdZddlZddlZddlZddlZyddlZWnek rkddlZnXddl Z ddl Z ej dƒZ e j ƒjZeƒZeƒZd„Zd„Zd„Zd„Zd efd „ƒYZd „Zd „Zd „Zd„Zd„Zdefd„ƒYZd„Zdej fd„ƒYZ!d„Z"dS(s(Adds support for parameterized tests to Python's unittest TestCase class. A parameterized test is a method in a test case that is invoked with different argument tuples. A simple example: class AdditionExample(parameterized.ParameterizedTestCase): @parameterized.Parameters( (1, 2, 3), (4, 5, 9), (1, 1, 3)) def testAddition(self, op1, op2, result): self.assertEqual(result, op1 + op2) Each invocation is a separate test case and properly isolated just like a normal test method, with its own setUp/tearDown cycle. In the example above, there are three separate testcases, one of which will fail due to an assertion error (1 + 1 != 3). Parameters for invididual test cases can be tuples (with positional parameters) or dictionaries (with named parameters): class AdditionExample(parameterized.ParameterizedTestCase): @parameterized.Parameters( {'op1': 1, 'op2': 2, 'result': 3}, {'op1': 4, 'op2': 5, 'result': 9}, ) def testAddition(self, op1, op2, result): self.assertEqual(result, op1 + op2) If a parameterized test fails, the error message will show the original test name (which is modified internally) and the arguments for the specific invocation, which are part of the string returned by the shortDescription() method on test cases. The id method of the test, used internally by the unittest framework, is also modified to show the arguments. To make sure that test names stay the same across several invocations, object representations like >>> class Foo(object): ... pass >>> repr(Foo()) '<__main__.Foo object at 0x23d8610>' are turned into '<__main__.Foo>'. For even more descriptive names, especially in test logs, you can use the NamedParameters decorator. In this case, only tuples are supported, and the first parameters has to be a string (or an object that returns an apt name when converted via str()): class NamedExample(parameterized.ParameterizedTestCase): @parameterized.NamedParameters( ('Normal', 'aa', 'aaa', True), ('EmptyPrefix', '', 'abc', True), ('BothEmpty', '', '', True)) def testStartsWith(self, prefix, string, result): self.assertEqual(result, strings.startswith(prefix)) Named tests also have the benefit that they can be run individually from the command line: $ testmodule.py NamedExample.testStartsWithNormal . -------------------------------------------------------------------- Ran 1 test in 0.000s OK Parameterized Classes ===================== If invocation arguments are shared across test methods in a single ParameterizedTestCase class, instead of decorating all test methods individually, the class itself can be decorated: @parameterized.Parameters( (1, 2, 3) (4, 5, 9)) class ArithmeticTest(parameterized.ParameterizedTestCase): def testAdd(self, arg1, arg2, result): self.assertEqual(arg1 + arg2, result) def testSubtract(self, arg2, arg2, result): self.assertEqual(result - arg1, arg2) Inputs from Iterables ===================== If parameters should be shared across several test cases, or are dynamically created from other sources, a single non-tuple iterable can be passed into the decorator. This iterable will be used to obtain the test cases: class AdditionExample(parameterized.ParameterizedTestCase): @parameterized.Parameters( c.op1, c.op2, c.result for c in testcases ) def testAddition(self, op1, op2, result): self.assertEqual(result, op1 + op2) Single-Argument Test Methods ============================ If a test method takes only one argument, the single argument does not need to be wrapped into a tuple: class NegativeNumberExample(parameterized.ParameterizedTestCase): @parameterized.Parameters( -1, -3, -4, -5 ) def testIsNegative(self, arg): self.assertTrue(IsNegative(arg)) s!tmarek@google.com (Torsten Marek)iÿÿÿÿNs0\<([a-zA-Z0-9_\-\.]+) object at 0x[a-fA-F0-9]+\>cCstjdt|ƒƒS(Ns<\1>(tADDR_REtsubtrepr(tobj((sK/usr/lib/python2.7/site-packages/google/protobuf/internal/_parameterized.pyt _CleanRepr¥scCsd|j|jfS(Ns%s.%s(t __module__t__name__(tcls((sK/usr/lib/python2.7/site-packages/google/protobuf/internal/_parameterized.pyt _StrClass«scCs#t|tjƒo"t|tjƒ S(N(t isinstancet collectionstIterabletsixt string_types(R((sK/usr/lib/python2.7/site-packages/google/protobuf/internal/_parameterized.pyt_NonStringIterable¯scCsbt|tjƒr/djd„|jƒDƒƒSt|ƒrQdjtt|ƒƒSt|fƒSdS(Ns, css+|]!\}}d|t|ƒfVqdS(s%s=%sN(R(t.0targnametvalue((sK/usr/lib/python2.7/site-packages/google/protobuf/internal/_parameterized.pys ¶s( R R tMappingtjointitemsRtmapRt_FormatParameterList(ttestcase_params((sK/usr/lib/python2.7/site-packages/google/protobuf/internal/_parameterized.pyR´s   t_ParameterizedTestItercBs)eZdZd„Zd„Zd„ZRS(s9Callable and iterable class for producing new test cases.cCs||_||_||_dS(s\Returns concrete test functions for a test and a list of parameters. The naming_type is used to determine the name of the concrete functions as reported by the unittest framework. If naming_type is _FIRST_ARG, the testcases must be tuples, and the first element must have a string representation that is a valid Python identifier. Args: test_method: The decorated test method. testcases: (list of tuple/dict) A list of parameter tuples/dicts for individual test invocations. naming_type: The test naming type, either _NAMED or _ARGUMENT_REPR. N(t _test_methodt testcasest _naming_type(tselft test_methodRt naming_type((sK/usr/lib/python2.7/site-packages/google/protobuf/internal/_parameterized.pyt__init__Ás  cOstdƒ‚dS(Ns¹You appear to be running a parameterized test case without having inherited from parameterized.ParameterizedTestCase. This is bad because none of your test cases are actually being run.(t RuntimeError(Rtargstkwargs((sK/usr/lib/python2.7/site-packages/google/protobuf/internal/_parameterized.pyt__call__Óscs;|j‰|j‰‡‡fd†‰‡fd†|jDƒS(NcsÙtjˆƒ‡‡fd†ƒ}ˆtkr\t|_|jtˆdƒ7_ˆd‰n8ˆtkrdtˆƒf|_ nt dˆfƒ‚d|jtˆƒf|_ ˆj rÕ|j dˆj f7_ n|S(NcsOtˆtjƒr"ˆ|ˆn)tˆƒr>ˆ|ˆŒn ˆ|ˆƒdS(N(R R RR(R(RR(sK/usr/lib/python2.7/site-packages/google/protobuf/internal/_parameterized.pytBoundParamTestÞs  iis(%s)s%s is not a valid naming type.s%s(%s)s %s( t functoolstwrapst _FIRST_ARGtTruet__x_use_name__Rtstrt_ARGUMENT_REPRRt__x_extra_id__R t__doc__(RR$(RR(RsK/usr/lib/python2.7/site-packages/google/protobuf/internal/_parameterized.pytMakeBoundParamTestÝs!     c3s|]}ˆ|ƒVqdS(N((Rtc(R.(sK/usr/lib/python2.7/site-packages/google/protobuf/internal/_parameterized.pys üs(RRR(R((R.RRsK/usr/lib/python2.7/site-packages/google/protobuf/internal/_parameterized.pyt__iter__Ùs  (RRR-RR#R0(((sK/usr/lib/python2.7/site-packages/google/protobuf/internal/_parameterized.pyR¾s  cCs$t|ƒdko#t|dtƒ S(s<True iff testcases contains only a single non-tuple element.ii(tlenR ttuple(R((sK/usr/lib/python2.7/site-packages/google/protobuf/internal/_parameterized.pyt_IsSingletonListÿsc CsÀi|_}x¬|jjƒjƒD]•\}}|jtjjƒr#t|t j ƒr#t ||ƒi}t |||t |||ƒƒx-|jƒD]\}}t|||ƒq•Wq#q#WdS(N(t _id_suffixt__dict__tcopyRt startswithtunittestt TestLoaderttestMethodPrefixR ttypest FunctionTypetdelattrt _UpdateClassDictForParamTestCaseRtsetattr(t class_objectRRt id_suffixtnameRtmethodstmeth((sK/usr/lib/python2.7/site-packages/google/protobuf/internal/_parameterized.pyt _ModifyClasss "  cs/‡‡fd†}tˆƒr+ˆd‰n|S(sÃImplementation of the parameterization decorators. Args: naming_type: The naming type. testcases: Testcase parameters. Returns: A function for modifying the decorated object. csUt|tƒrAt|tˆtjƒs3tˆƒnˆˆƒ|St|ˆˆƒSdS(N(R ttypeRER tSequencetlistR(R(RR(sK/usr/lib/python2.7/site-packages/google/protobuf/internal/_parameterized.pyt_Apply!si(R3(RRRI((RRsK/usr/lib/python2.7/site-packages/google/protobuf/internal/_parameterized.pyt_ParameterDecorators   cGs tt|ƒS(siA decorator for creating parameterized tests. See the module docstring for a usage example. Args: *testcases: Parameters for the decorated method, either a single iterable, or a list of tuples/dicts/objects (for tests with only one argument). Returns: A test generator to be handled by TestGeneratorMetaclass. (RJR+(R((sK/usr/lib/python2.7/site-packages/google/protobuf/internal/_parameterized.pyt Parameters4s cGs tt|ƒS(sŸA decorator for creating parameterized tests. See the module docstring for a usage example. The first element of each parameter tuple should be a string and will be appended to the name of the test method. Args: *testcases: Parameters for the decorated method, either a single iterable, or a list of tuples. Returns: A test generator to be handled by TestGeneratorMetaclass. (RJR'(R((sK/usr/lib/python2.7/site-packages/google/protobuf/internal/_parameterized.pytNamedParametersCstTestGeneratorMetaclasscBseZdZd„ZRS(sìMetaclass for test cases with test generators. A test generator is an iterable in a testcase that produces callables. These callables must be single-argument methods. These methods are injected into the class namespace and the original iterable is removed. If the name of the iterable conforms to the test pattern, the injected methods will be picked up as tests by the unittest framework. In general, it is supposed to be used in conjunction with the Parameters decorator. cCs‘i|d<}xj|jƒD]\\}}|jtjjƒrt|ƒrt|ƒ}|j|ƒt||||ƒqqWt j ||||ƒS(NR4( RR7R8R9R:RtitertpopR>RFt__new__(tmcst class_nametbasestdctRARBRtiterator((sK/usr/lib/python2.7/site-packages/google/protobuf/internal/_parameterized.pyRPas   (RRR-RP(((sK/usr/lib/python2.7/site-packages/google/protobuf/internal/_parameterized.pyRMTs cCsrxkt|ƒD]]\}}t|dtƒr7|j}nd|t|f}|||ms   tParameterizedTestCasecBs/eZdZeZd„Zd„Zd„ZRS(s9Base class for test cases using the Parameters decorator.cCs|jjtƒdS(Ni(t_testMethodNametsplitRZ(R((sK/usr/lib/python2.7/site-packages/google/protobuf/internal/_parameterized.pyt _OriginalName‡scCsd|jƒt|jƒfS(Ns%s (%s)(RaRt __class__(R((sK/usr/lib/python2.7/site-packages/google/protobuf/internal/_parameterized.pyt__str__ŠscCs2dt|jƒ|jƒ|jj|jdƒfS(s½Returns the descriptive ID of the test. This is used internally by the unittesting framework to get a name for the test to be used in reports. Returns: The test id. s%s.%s%sRV(RRbRaR4tgetR_(R((sK/usr/lib/python2.7/site-packages/google/protobuf/internal/_parameterized.pytids  (RRR-RMt __metaclass__RaRcRe(((sK/usr/lib/python2.7/site-packages/google/protobuf/internal/_parameterized.pyR^ƒs   cCs1td|jtfiƒ}|d|tfiƒS(s;Returns a new base class with a cooperative metaclass base. This enables the ParameterizedTestCase to be used in combination with other base classes that have custom metaclasses, such as mox.MoxTestBase. Only works with metaclasses that do not override type.__new__. Example: import google3 import mox from google3.testing.pybase import parameterized class ExampleTest(parameterized.CoopParameterizedTestCase(mox.MoxTestBase)): ... Args: other_base_class: (class) A test case base class. Returns: A new class object. t CoopMetaclasstCoopParameterizedTestCase(RFRfRMR^(tother_base_classt metaclass((sK/usr/lib/python2.7/site-packages/google/protobuf/internal/_parameterized.pyRh›s(#R-t __author__R R%treR;t unittest2R8t ImportErrortuuidR tcompileRtuuid1thexRZtobjectR'R+RRRRRR3RERJRKRLRFRMR>tTestCaseR^Rh(((sK/usr/lib/python2.7/site-packages/google/protobuf/internal/_parameterized.pyts:             A