B )`#@sdZddlmZddlmZddlmZddlZddlZddlmZddlm Z ddlm Z dd lm Z d d l m Z dddZddZddZddZddZGdddeZdS)aUtility functions for comparing proto2 messages in Python. ProtoEq() compares two proto2 messages for equality. ClearDefaultValuedFields() recursively clears the fields that are set to their default values. This is useful for comparing protocol buffers where the semantics of unset fields and default valued fields are the same. assertProtoEqual() is useful for unit tests. It produces much more helpful output than assertEqual() for proto2 messages, e.g. this: outer { inner { - strings: "x" ? ^ + strings: "y" ? ^ } } ...compared to the default output from assertEqual() that looks like this: AssertionError: != Call it inside your unit test's googletest.TestCase subclasses like this: from tensorflow.python.util.protobuf import compare class MyTest(googletest.TestCase): ... def testXXX(self): ... compare.assertProtoEqual(self, a, b) Alternatively: from tensorflow.python.util.protobuf import compare class MyTest(compare.ProtoAssertions, googletest.TestCase): ... def testXXX(self): ... self.assertProtoEqual(a, b) )absolute_import)division)print_functionN) descriptor)descriptor_pool)message) text_format)collections_abcTFc Cst}t|tjr(tj|||d}x>||fD]2}|rX|}|rX| d||f|r2t |q2Wtj ||d} tj ||d} t | dkrt | dkr|j | | |dn4ddt| d| d} | d|| fd S) akFails with a useful error if a and b aren't equal. Comparison of repeated fields matches the semantics of unittest.TestCase.assertEqual(), ie order and extra duplicates fields matter. Args: self: googletest.TestCase a: proto2 PB instance, or text string representing one. b: proto2 PB instance -- message.Message or subclass thereof. check_initialized: boolean, whether to fail if either a or b isn't initialized. normalize_numbers: boolean, whether to normalize types and precision of numbers before comparison. msg: if specified, is used as the error message on failure. )rzInitialization errors: %s %si)msg Tz%s : %sN)rZDefault isinstancesix string_typesrZMerge __class__ZFindInitializationErrorsZfailNormalizeNumberFieldsZMessageToStringlenZassertMultiLineEqualjoindifflibZ unified_diff splitlines) selfabZcheck_initializedZnormalize_numbersr poolpberrorsZa_strZb_strZdiffrV/home/dcms/DCMS/lib/python3.7/site-packages/tensorflow/python/util/protobuf/compare.pyassertProtoEqualMs"  rcCsx|D]\}}d}|jtjjk r2d}|g}d}|jtjjtjjtjjfkrbdd|D}nl|jtjj tjj tjj tjj fkrdd|D}n:|jtjj krdd|D}n|jtjjkrdd|D}|dk r|r||jt||j|nt||j|d |jtjjks,|jtjjkr |jtjjkr|jjr|jjr|jjd jtjjkrx6t|D]}t|qxWq x|D]}t|qWq W|S) aNormalizes types and precisions of number fields in a protocol buffer. Due to subtleties in the python protocol buffer implementation, it is possible for values to have different types and precision depending on whether they were set and retrieved directly or deserialized from a protobuf. This function normalizes integer values to ints and longs based on width, 32-bit floats to five digits of precision to account for python always storing them as 64-bit, and ensures doubles are floating point for when they're set to integers. Modifies pb in place. Recurses into nested objects. Args: pb: proto2 message. Returns: the given pb, modified in place. TFNcSsg|] }t|qSr)int).0xrrr sz)NormalizeNumberFields..cSsg|] }t|qSr)r )r!r"rrrr#scSsg|]}t|dqS))round)r!r"rrrr#scSsg|]}tt|dqS))r%float)r!r"rrrr#srr ) ListFieldslabelrZFieldDescriptorZLABEL_REPEATEDtypeZ TYPE_INT64Z TYPE_UINT64Z TYPE_SINT64Z TYPE_INT32Z TYPE_UINT32Z TYPE_SINT32Z TYPE_ENUMZ TYPE_FLOATZ TYPE_DOUBLEZ ClearFieldnamegetattrextendsetattrZ TYPE_MESSAGEZ TYPE_GROUPZ message_typeZ has_optionsZ GetOptionsZ map_entryZfields_by_numberr itervaluesr)rdescvaluesZ is_repeatedZnormalized_valuesvrrrrysH          rcCs t|tjS)N)rr Mapping)valuerrr_IsMapsr5cCs6t|tjrdSy t|dStk r0dSXdS)NFT)rrriter TypeError)r4rrr_IsRepeatedContainers r8cCsdd}||||}}t|tr.t|ts6||kSxNtt|t|BD].}||ksh||krldSt||||sTdSqTWdS)a8Compares two proto2 objects for equality. Recurses into nested messages. Uses list (not set) semantics for comparing repeated fields, ie duplicates and order matter. Args: a: A proto2 message or a primitive. b: A proto2 message or a primitive. Returns: `True` if the messages are equal. cSsVt|tjr"tdd|DSt|r6t|St|rNttt |S|SdS)aVReturns a dictionary or unchanged pb bases on its type. Specifically, this function returns a dictionary that maps tag number (for messages) or element index (for repeated fields) to value, or just pb unchanged if it's neither. Args: pb: A proto2 message or a primitive. Returns: A dict or unchanged pb. css|]\}}|j|fVqdS)N)number)r!r0r4rrr sz*ProtoEq..Format..N) rrMessagedictr(r5itemsr8 enumeratelist)rrrrFormats  zProtoEq..FormatFT)rr<sortedsetkeysProtoEq)rrr@tagrrrrDs "rDc@seZdZdZddZdS)ProtoAssertionsa Mix this into a googletest.TestCase class to get proto2 assertions. Usage: class SomeTestCase(compare.ProtoAssertions, googletest.TestCase): ... def testSomething(self): ... self.assertProtoEqual(a, b) See module-level definitions for method documentation. cOst|f||S)N)r)rargskwargsrrrrsz ProtoAssertions.assertProtoEqualN)__name__ __module__ __qualname____doc__rrrrrrFs rF)TFN)rL __future__rrrrrZgoogle.protobufrrrrcompatr rrr5r8rDobjectrFrrrr;s"         +D 7