3 S╔LjAVу@s(dZdZddlZddlZddlZddlZddlZddlZddlZddl Z yddl ZWnek rpddlZYnXye Wnek rЦddДZ YnXddlmZddlmZddlmZdd lmZdd lmZddlmZddlmZdd lmZddlmZddlmZddlmZddlmZddlmZddlmZddlmZej РrXe!Z"ddДZ#ddДZ$ddДZ%ddДZ&ej'Z'ej(defdefГGdd Дd e'ГГZ)Gd!d"Дd"e'ГZ*Gd#d$Дd$e'ГZ+Gd%d&Дd&e'ГZ,Gd'd(Дd(e'ГZ-ej.ej/Гd)kРp№ej0d0kd,ГGd-d.Дd.e'ГГZ1e2d/kРr$ej3ГdS)1aЬTests python protocol buffers against the golden message. Note that the golden messages exercise every known field type, thus this test ends up exercising and verifying nearly all of the parsing and serialization code in the whole library. TODO(kenton): Merge with wire_format_test? It doesn't make a whole lot of sense to call this a test of the "message" module, which only declares an abstract interface. z!gps@google.com (Gregory P. Smith)щNcCs||k||kS)Nй)┌x┌yrr·"/usr/lib/python3.6/message_test.py┌>sr)┌map_proto2_unittest_pb2)┌map_unittest_pb2)┌unittest_pb2)┌unittest_proto3_arena_pb2)┌descriptor_pb2)┌descriptor_pool)┌message_factory)┌text_format)┌api_implementation)┌encoder)┌packed_field_test_pb2)┌ test_util)┌testing_refleaks)┌message)┌_parameterizedcCs||kS)Nr)┌valrrr┌isnanVsrcCst|Гot|dГS)Nr)r)rrrr┌isinfYsrcCst|Гo|dkS)Nr)r)rrrr┌IsPosInf\srcCst|Гo|dkS)Nr)r)rrrr┌IsNegInf^srZ_proto2Z_proto3c @s8eZdZddДZddДZddДZddДZd d ДZddДZd dДZ ddДZ ddДZddДZddДZ ddДZddДZddДZddДZdd ДZd!d"ДZd#d$ДZd%d&ДZd'd(ДZd)d*ДZd+d,ДZd-d.ДZd/d0ДZd1d2ДZd3d4ДZd5d6ДZd7d8ДZd9d:ДZd;d<ДZ d=d>ДZ!d?d@ДZ"dAdBДZ#dCdDДZ$dEdFДZ%dGdHДZ&dIdJДZ'dKdLДZ(dMdNДZ)dOdPДZ*dQdRДZ+dSdTДZ,dUdVДZ-dWdXДZ.dYdZДZ/d[d\ДZ0d]d^d_d`dadbe1Гgie2Гg Z3dcddДZ4dedfДZ5dgdhДZ6didjДZ7dkdlДZ8dmdnДZ9dodpДZ:GdqdrДdre;ГZdwdxДZ?dydzДZ@d{d|ДZAd}d~ДZBddАДZCdБdВДZDdГdДДZEdЕdЖДZFdЗdИДZGd]S)Й┌MessageTestcCsXtjГdkr|jdГtjdГ}|jtГП}|jj|ГWdQRX|j dt |jГГdS)N┌pythonzКSkipping testBadUtf8String, currently only the python api implementation raises UnicodeDecodeError when a string field contains bad utf-8.Zbad_utf8_stringzTestAllTypes.optional_string)r┌TypeZskipTestr┌GoldenFileData┌assertRaises┌UnicodeDecodeError┌TestAllTypes┌ FromString┌assertIn┌str┌ exception)┌self┌message_moduleZ bad_utf8_data┌contextrrr┌testBadUtf8Stringjs zMessageTest.testBadUtf8StringcCsr|tkrtjdГ}n tjdГ}|jГ}|j|Г|tkrDtj||Г|j||jГГtj |Г}|j||jГГdS)NZ golden_message_oneof_implementedZgolden_message_proto3) r rrr!┌ParseFromStringZExpectAllFieldsSet┌assertEqual┌SerializeToString┌copy┌deepcopy)r&r'┌golden_data┌golden_message┌golden_copyrrr┌testGoldenMessagets zMessageTest.testGoldenMessagecCshtjdГ}|jГ}|j|Г|jГ}tj|Г|j||Г|j||jГГtj|Г}|j||jГГdS)N┌golden_packed_fields_message) rr┌TestPackedTypesr*ZSetAllPackedFieldsr+r,r-r.)r&r'r/r0┌all_setr1rrr┌testGoldenPackedMessageЖs z#MessageTest.testGoldenPackedMessagecCs╝|jГ}|jt|jdГ|jt|jdГtjddГ}tjГdkrv|jt j ГП}|j|ГWdQRX|jdt|j ГГnBtjddНП0}tjd Г|j|Г|jd t|dj ГГWdQRXdS)Nr┌0щщrzUnexpected end-group tag.T)┌record┌alwaysz4Unexpected end-group tag: Not all data was convertedщ )r!r┌ TypeErrorr"┌ ExceptionrZTagBytesrrr┌DecodeErrorr+r$r%┌warnings┌catch_warnings┌simplefilter)r&r'┌msgZend_tagr(┌wrrr┌testParseErrorsСs zMessageTest.testParseErrorsc sиd}|jГ}|jjddddgГ|j||jddНГ|j||jddНГ|j||jddНГGd d Дd tГЙGЗfddДdtГ}|jИГП|j|ГdНWdQRXdS) Ns5т One stringт Two stringт Red stringтBlue stringz One stringz Two stringz Red stringzBlue string)┌ deterministicFTc@seZdZdS)z:MessageTest.testDeterminismParameters..BadArgErrorN)┌__name__┌ __module__┌__qualname__rrrr┌BadArgError╝srJcs$eZdZЗfddДZЗfddДZdS)z5MessageTest.testDeterminismParameters..BadArgcs ИГВdS)Nr)r&)rJrr┌__nonzero__┴szAMessageTest.testDeterminismParameters..BadArg.__nonzero__cs ИГВdS)Nr)r&)rJrr┌__bool__─sz>MessageTest.testDeterminismParameters..BadArg.__bool__N)rGrHrIrKrLr)rJrr┌BadArg┐srM)r!┌repeated_string┌extendr+r,r>┌objectr)r&r'r/r0rMr)rJr┌testDeterminismParametersжs"z%MessageTest.testDeterminismParameterscCs@tjdГ}|jГ}|j|Гtj|Г}tj|Г}|j||ГdS)Nr0)rrr!r*┌pickle┌dumps┌loadsr+)r&r'r/r0┌pickled_message┌unpickled_messagerrr┌testPickleSupport╩s zMessageTest.testPickleSupportcCsА|tkrd}nd}|jГ}|j|Г|jt|jГГ|jt|jГГ|jt|jdГГ|jt|jdГГ|j ||j ГГdS)Ns]АaЁ═А╤Ёs ]АaЁ╩А╥Ёr)r r!r*┌ assertTruer┌optional_float┌optional_double┌repeated_float┌repeated_doubler+r,)r&r'r/r0rrr┌testPositiveInfinity╙s z MessageTest.testPositiveInfinitycCsА|tkrd}nd}|jГ}|j|Г|jt|jГГ|jt|jГГ|jt|jdГГ|jt|jdГГ|j ||j ГГdS)Ns]А aЁ ═А ╤Ё s ]А aЁ ╩А ╥Ё r)r r!r*rXrrYrZr[r\r+r,)r&r'r/r0rrr┌testNegativeInfinityчs z MessageTest.testNegativeInfinitycCs─d}|jГ}|j|Г|jt|jГГ|jt|jГГ|jt|jdГГ|jt|jdГГ|jГ}|jГ}|j|Г|jt|jГГ|jt|jГГ|jt|jdГГ|jt|jdГГdS)Ns]└a°═└╤°r) r!r*rXrrYrZr[r\r,)r&r'r/r0┌ serializedrrrr┌testNotANumber√s zMessageTest.testNotANumbercCsRd}|jГ}|j|Г|jt|jdГГ|jt|jdГГ|j||jГГdS)NsвАкЁr)r4r*rXr┌packed_float┌ packed_doubler+r,)r&r'r/r0rrr┌testPositiveInfinityPackeds z&MessageTest.testPositiveInfinityPackedcCsRd}|jГ}|j|Г|jt|jdГГ|jt|jdГГ|j||jГГdS)NsвА кЁ r)r4r*rXrrarbr+r,)r&r'r/r0rrr┌testNegativeInfinityPackeds z&MessageTest.testNegativeInfinityPackedcCsДd}|jГ}|j|Г|jt|jdГГ|jt|jdГГ|jГ}|jГ}|j|Г|jt|jdГГ|jt|jdГГdS)Nsв└к°r)r4r*rXrrarbr,)r&r'r/r0r_rrrr┌testNotANumberPacked%s z MessageTest.testNotANumberPackedcCst|jГ}tjddГ}||_|j|jГГ|j|j|kГdtjddГ}||_|j|jГГ|j|j|kГ||_|j|jГГ|j|j|kГ||_|j|jГГ|j|j|kГtjddГ}||_|j|jГГ|j|j|kГdtjddГ}||_|j|jГГ|j|j|kГ||_|j|jГГ|j|j|kГ||_|j|jГГ|j|j|kГdS)Nщщg°?iБ iБ )r!┌math┌powrYr*r,rX)r&r'r┌kMostPosExponentNoSigBits┌kMostPosExponentOneSigBit┌kMostNegExponentNoSigBits┌kMostNegExponentOneSigBitrrr┌testExtremeFloatValues3s:z"MessageTest.testExtremeFloatValuescCst|jГ}tjddГ}||_|j|jГГ|j|j|kГdtjddГ}||_|j|jГГ|j|j|kГ||_|j|jГГ|j|j|kГ||_|j|jГГ|j|j|kГtjddГ}||_|j|jГГ|j|j|kГdtjddГ}||_|j|jГГ|j|j|kГ||_|j|jГГ|j|j|kГ||_|j|jГГ|j|j|kГdS)Nrfi g°?i№ i№ )r!rhrirZr*r,rX)r&r'rrjrkrlrmrrr┌testExtremeDoubleValues`s:z#MessageTest.testExtremeDoubleValuescCs"|jГ}d|_|jt|ГdГdS)Ng@zoptional_float: 2.0 )r!rYr+r$)r&r'rrrr┌testFloatPrintingНszMessageTest.testFloatPrintingcCs>|jГ}d|_tjdkr*|jt|ГdГn|jt|ГdГdS)Ng^!┴7▌Ъ┐?щz%optional_double: 0.12345678912345678 z optional_double: 0.123456789123 )rq)r!rZ┌sys┌version_infor+r$)r&r'rrrr┌testHighPrecisionFloatPrintingТs z*MessageTest.testHighPrecisionFloatPrintingcCs<|jГ}tj|Г|jГ}|j|jГГ|jt|ГdГdS)N┌)r!rZSetAllNonLazyFieldsZTestEmptyMessager*r,r+r$)r&r'Z populated┌emptyrrr┌testUnknownFieldPrintingЪs z$MessageTest.testUnknownFieldPrintingcCs▓|jГ}|jjddН|jjddН|jjddН|jjddН|jddddgddД|jDГГ|jddddgddДt|jГDГГ|jddddgd dД|jddd ЕDГГdS)Nr8)┌bbrfrqr9cSsg|] }|jСqSr)rx)┌.0┌mrrr· йsz@MessageTest.testRepeatedNestedFieldIteration..cSsg|] }|jСqSr)rx)ryrzrrrr{лscSsg|] }|jСqSr)rx)ryrzrrrr{нsr<)r!┌repeated_nested_message┌addr+┌reversed)r&r'rCrrr┌ testRepeatedNestedFieldIterationбsz,MessageTest.testRepeatedNestedFieldIterationcCsЁ|jГ}|jjdГ|jjdГ|jjdГ|jjГ|j|jddГ|j|jddГ|j|jddГ|jt|jГtdddgГГ|jjdГ|jjdГ|jjdГ|jjГ|j|jddГ|j|jddГ|j|jddГ|jjdГ|jjd Г|jjd Г|jjГ|j|jddГ|j|jdd Г|j|jdd Г|jt|jГtdd d gГГ|j jdГ|j jdГ|j jd Г|j jГ|j|j ddГ|j|j dd Г|j|j ddГ|jt|j Гtdd dgГГdS)z7Check some different types with the default comparator.r8rqrfrgЪЩЩЩЩЩё?g═╠╠╠╠╠Ї?g333333є?┌a┌c┌bєaєcєbN) r!┌repeated_int32┌append┌sortr+r$r[┌assertAlmostEqualrNZrepeated_bytes)r&r'rrrr┌0testSortingRepeatedScalarFieldsDefaultComparatorпs@ zMessageTest.testSortingRepeatedCompositeFieldsCustomComparatorcCs╪|jГ}d|jjГ_d|jjГ_d|jjГ_d|jjГ_d|jjГ_d|jjГ_d|jjГ_|jjdd Дd Н|jdddddddgddД|jDГГ|jГ}|jГ|j|Г|jdddddddgd dД|jDГГdS)zECheck passing a custom comparator to sort a repeated composite field.щщщ щ!щщщ cSs |jdS)NrЭ)rx)┌zrrrr szFMessageTest.testSortingRepeatedCompositeFieldsStable..)rЛcSsg|] }|jСqSr)rx)ry┌nrrrr{ szHMessageTest.testSortingRepeatedCompositeFieldsStable..cSsg|] }|jСqSr)rx)ryrЯrrrr{sN) r!r|r}rxrИr+r,┌ClearZMergeFromString)r&r'r┌pbrrr┌(testSortingRepeatedCompositeFieldsStable s$ z4MessageTest.testSortingRepeatedCompositeFieldsStablecCs@|jГ}tjdГ}ddД}d|jjГ_d|jjГ_d|jjГ_d|jjГ_d|jjГ_d |jjГ_|jj|d Н|jddД|jDГdddd ddgГ|jj|d dН|jddД|jDГddd dddgГtj dkr╓dS|jj|dН|jddД|jDГdddd ddgГ|jj|d dН|jddД|jDГddd dddgГdS)zECheck sorting a repeated composite field using list.sort() arguments.rxcSst|j|jГS)N)┌cmprx)rАrВrrrrszEMessageTest.testRepeatedCompositeFieldSortArguments..r8rqrfrТrУr9)rЛcSsg|] }|jСqSr)rx)ry┌krrrr{%szGMessageTest.testRepeatedCompositeFieldSortArguments..T)rЛ┌reversecSsg|] }|jСqSr)rx)ryrдrrrr{(sN)┌ sort_functioncSsg|] }|jСqSr)rx)ryrдrrrr{,s)rгrеcSsg|] }|jСqSr)rx)ryrдrrrr{/s)rq) r!rФrХr|r}rxrИr+rrrs)r&r'rZget_bbZcmp_bbrrr┌'testRepeatedCompositeFieldSortArgumentss. z3MessageTest.testRepeatedCompositeFieldSortArgumentscCs▓|jГ}|jjdГ|jjdГ|jjdГ|jjtdН|jt|jГdddgГ|jjtddН|jt|jГdddgГtjdkr┌ddД}|jj|d Н|jt|jГdddgГ|jj|dd Н|jt|jГdddgГ|j jdГ|j jdГ|j jd Г|j jt dН|jt|j Гd ddgГ|j jt ddН|jt|j Гddd gГtjd kРrоddД}|j j|d Н|jt|j Гd ddgГ|j j|dd Н|jt|j Гddd gГdS)!z9Check sorting a scalar field using list.sort() arguments.rqrfr8)rЛT)rЛrеcSstt|Гt|ГГS)N)rгrП)rАrВrrrr>szBMessageTest.testRepeatedScalarFieldSortArguments..)rж)rгrеrМrxrБcSstt|Гt|ГГS)N)rгrР)rАrВrrrrLsNrНrОr<r<rОrНrНrОr<)rqr<rОrНrНrОr<)rq)r!rЖrЗrИrПr+┌listrrrsrNrР)r&r'rZabs_cmpZlen_cmprrr┌$testRepeatedScalarFieldSortArguments2s6 z0MessageTest.testRepeatedScalarFieldSortArgumentscCsJ|jГ}|jГ}|jjdГ|jjdГ|jjdГ|jjdГ|jjdГ|jjdГd|jjГ_d|jjГ_d|jjГ_d|jjГ_d|jjГ_d|jjГ_tjdkrоdS||k}|j|jk}|jt ||ГdГ|jt |j|jГdГ|jt |jdddgГdГ|jt |j|jГdГ|j tГПt |j|jddЕГWdQRXdS)Nrr8rfrq)rq)r!rЖrЗr|r}rxrrrsr+rгrr=)r&r'┌m1┌m2┌_rrr┌testRepeatedFieldsComparableRs2 z(MessageTest.testRepeatedFieldsComparablecCs,|jГ}|j|jtjГ|j|jtjГdS)N)r!┌assertIsInstancerЖ┌collections┌MutableSequencer|)r&r'rzrrr┌testRepeatedFieldsAreSequencesusz*MessageTest.testRepeatedFieldsAreSequencescCsL|jГ}|jtГПt|jГWdQRX|jtГПt|jГWdQRXdS)N)r!rr=┌hashrЖr|)r&r'rzrrr┌testRepeatedFieldsNotHashable{s z)MessageTest.testRepeatedFieldsNotHashablecCs*|jГ}|jjjgГ|j|jdГГdS)N┌payload)┌NestedTestAllTypesr┤rЖrOrX┌HasField)r&r'rzrrr┌$testRepeatedFieldInsideNestedMessageВsz0MessageTest.testRepeatedFieldInsideNestedMessagecCst||Г|j|j|ГГdS)z├Make sure that a nested message object exists. As soon as a nested message attribute is accessed, it will be present in the _fields dict, without being marked as actually being set. N)┌getattr┌assertFalser╢)r&rCZ attributerrr┌ensureNestedMessageExistsЗs z%MessageTest.ensureNestedMessageExistscCs,|jГ}|jt|jdГ|jt|jdГdS)NZno_such_oneof_fieldr)r!r┌ ValueError┌ WhichOneofr>)r&r'rzrrr┌ testOneofGetCaseNonexistingFieldРsz,MessageTest.testOneofGetCaseNonexistingFieldcCsЮ|jГ}|jd|jdГГ|j|jdГГd|_|jd|jdГГ|j|jdГГ|j|jdГГd|_|jd|jdГГ|j|jdГГ|j|jdГГdS)N┌oneof_field┌oneof_uint32r┌oneof_stringru) r!┌assertIsr╝r╣r╢r┐r+rXr└)r&r'rzrrr┌testOneofDefaultValuesХsz"MessageTest.testOneofDefaultValuescCsb|jГ}|jd|jdГГd|_|jd|jdГГ|j|jdГГd|_|jd|jdГГ|j|jdГГ|j|jdГГ|j |jd|jdГГ|j|jdГГ|j|jdГГ|j j |jd|jdГГ|j|jdГГ|j|jdГГd|j _ |jd|jdГГ|j|jdГГ|j|jdГГd|_|jd|jdГГ|j|jdГГ|j|jdГГdS) Nr╛rЫr┐┌foor└┌oneof_nested_messagesbb┌oneof_bytes)r!r┴r╝r┐r+rXr╢r└r╣r─rxr┼)r&r'rzrrr┌testOneofSemanticsеs2zMessageTest.testOneofSemanticscCs>|jГ}d|_|j|dГ|jd|jdГГ|jd|jГdS)NrЫr─r┐r╛)r!r┐r║r+r╝)r&r'rzrrr┌!testOneofCompositeFieldReadAccess╚s z-MessageTest.testOneofCompositeFieldReadAccesscCs▓|jГ}|jd|jdГГ|tkr2|j|jdГГd|_|jd|jdГГ|tkrb|j|jdГГd|_ |jd|jdГГ|j dГ|jd|jdГГ|tkrо|j|jdГГdS)Nr╛rЫr┐sbbr┼)r!r┴r╝r r╣r╢r┐r+rXr┼┌ ClearField)r&r'rzrrr┌testOneofWhichOneof╨s zMessageTest.testOneofWhichOneofcCsV|jГ}d|_|jdГ|tkr0|j|jdГГ|j|jdГГ|jd|jdГГdS)NrЫr╛r┐)r!r┐r╚r r╣r╢r┴r╝)r&r'rzrrr┌testOneofClearFieldуs zMessageTest.testOneofClearFieldcCsV|jГ}d|_|jdГ|tkr0|j|jdГГ|j|jdГГ|jd|jdГГdS)NrЫr┐r╛)r!r┐r╚r r╣r╢r┴r╝)r&r'rzrrr┌testOneofClearSetFieldьs z"MessageTest.testOneofClearSetFieldcCsp|jГ}d|_|j|dГ|jdГ|jd|jГ|tkrJ|j|jdГГ|j|jdГГ|jd|jdГГdS)NrЫr─r╛r┐) r!r┐r║r╚r+r rXr╢r╝)r&r'rzrrr┌testOneofClearUnsetFieldїs z$MessageTest.testOneofClearUnsetFieldcCs:|jГ}d|_|jГ}|j|jГГ|jd|jdГГdS)NrЫr┐r╛)r!r┐r*r,r+r╝)r&r'rzrлrrr┌testOneofDeserializes z MessageTest.testOneofDeserializecCs6|jГ}d|_|jГ}|j|Г|jd|jdГГdS)NrЫr┐r╛)r!r┐┌CopyFromr+r╝)r&r'rzrлrrr┌testOneofCopyFroms zMessageTest.testOneofCopyFromcCsb|jГ}d|j_|jГ}d|j_d|jj_|j|Г|jd|jjdГГ|jd|jjjdГГdS)NrЫsbbr┐r╛r┼)r╡r┤r┐r┼┌child┌ MergeFromr+r╝)r&r'rzrлrrr┌testOneofNestedMergeFroms z$MessageTest.testOneofNestedMergeFromcCsf|jГ}d|jj_d|jjj_|jГ}d|j_|j|Г|jd|jjdГГ|jd|jjjdГГdS)NrЫщrЩr─r╛) r╡r┤r─rxr╨r┐r╤r+r╝)r&r'rzrлrrr┌testOneofMessageMergeFroms z%MessageTest.testOneofMessageMergeFromcCs(|j|jjГdН}|jd|jdГГdS)N)r─r─r╛)r!┌ NestedMessager+r╝)r&r'rzrrr┌testOneofNestedMessageInit$sz&MessageTest.testOneofNestedMessageInitcCsB|jГ}d|_|jГ|j|jdГГd|_|jd|jdГГdS)NrЫr╛sbbr┼)r!r┐rа┌assertIsNoner╝r┼r+)r&r'rzrrr┌testOneofClear)szMessageTest.testOneofClearcCs&|jГ}tdГ|_|j|jtjГdS)zmAssigning a byte string to a string field should result in the value being converted to a Unicode string.ruN)r!r$┌optional_stringrо┌six┌ text_type)r&r'rzrrr┌"testAssignByteStringToUnicodeField1s z.MessageTest.testAssignByteStringToUnicodeFieldcCsИ|jГ}|jjdГ|jtdГtt|jГГЕ}|jt|jГt|ГГd|jjГ_|jtdГtt|jГГЕ}|jt|jГt|ГГdS)z|It should be possible to use long-valued indicies in slices This didn't used to work in the v2 C++ implementation. r8rrqN) r!rЖrЗ┌longrРr+r|r}rx)r&r'rzZslrrr┌testLongValuedSlice8szMessageTest.testLongValuedSlicecCsl|jГ}|jtГП }|jjddДtdГDГГWdQRX|jtГП }|jjddДtdГDГГWdQRXdS)z5This didn't use to work in the v2 C++ implementation.css|] }tVqdS)N)rА)ry┌irrr· MszCMessageTest.testExtendShouldNotSwallowExceptions..rЭNcss|] }tVqdS)N)rА)ryr▀rrrrрPs)r!r┌ NameErrorrЖrO┌range┌repeated_nested_enum)r&r'rzrмrrr┌$testExtendShouldNotSwallowExceptionsIs$z0MessageTest.testExtendShouldNotSwallowExceptionsNFrgєrucCs^|jГ}|jg|jГx(tjD]}|jj|Г|jg|jГqW|jjgГ|jg|jГdS)z,Test no-ops extending repeated int32 fields.N)r!┌assertSequenceEqualrЖr┌FALSY_VALUESrO)r&r'rz┌falsy_valuerrr┌testExtendInt32WithNothingTsz&MessageTest.testExtendInt32WithNothingcCs^|jГ}|jg|jГx(tjD]}|jj|Г|jg|jГqW|jjgГ|jg|jГdS)z,Test no-ops extending repeated float fields.N)r!rцr[rrчrO)r&r'rzrшrrr┌testExtendFloatWithNothingasz&MessageTest.testExtendFloatWithNothingcCs^|jГ}|jg|jГx(tjD]}|jj|Г|jg|jГqW|jjgГ|jg|jГdS)z-Test no-ops extending repeated string fields.N)r!rцrNrrчrO)r&r'rzrшrrr┌testExtendStringWithNothingnsz'MessageTest.testExtendStringWithNothingcCsД|jГ}|jg|jГ|jjdgГ|jdg|jГ|jjddgГ|jdddg|jГ|jjddgГ|jdddddg|jГdS)z7Test extending repeated int32 fields with python lists.rr8rfrqr9N)r!rцrЖrO)r&r'rzrrr┌testExtendInt32WithPythonList{sz)MessageTest.testExtendInt32WithPythonListcCsД|jГ}|jg|jГ|jjdgГ|jdg|jГ|jjddgГ|jdddg|jГ|jjddgГ|jdddddg|jГdS)z7Test extending repeated float fields with python lists.ggЁ?g@g@g@N)r!rцr[rO)r&r'rzrrr┌testExtendFloatWithPythonListЖsz)MessageTest.testExtendFloatWithPythonListcCsД|jГ}|jg|jГ|jjdgГ|jdg|jГ|jjddgГ|jdddg|jГ|jjddgГ|jdddddg|jГdS)z8Test extending repeated string fields with python lists.ruZ11Z22Z33Z44N)r!rцrNrO)r&r'rzrrr┌testExtendStringWithPythonListСsz*MessageTest.testExtendStringWithPythonListcCs:|jГ}|jg|jГ|jjdГ|jdddg|jГdS)zDTest extending repeated string fields with characters from a string.┌abcrАrВrБN)r!rцrNrO)r&r'rzrrr┌testExtendStringWithStringЬsz&MessageTest.testExtendStringWithStringc@s2eZdZdZdddДZddДZddДZd d ДZdS)zMessageTest.TestIterablezРThis iterable object mimics the behavior of numpy.array. __nonzero__ fails for length > 1, and returns bool(item[0]) for length == 1. NcCs|pg|_dS)N)┌_list)r&┌valuesrrr┌__init__кsz!MessageTest.TestIterable.__init__cCs8t|jГ}|dkrdS|dkr,t|jdГStdГВdS)NrFr8zTruth value is ambiguous.)rРrё┌boolr╗)r&┌sizerrrrKнs z$MessageTest.TestIterable.__nonzero__cCs t|jГS)N)rРrё)r&rrr┌__len__╡sz MessageTest.TestIterable.__len__cCs |jjГS)N)rё┌__iter__)r&rrrrў╕sz!MessageTest.TestIterable.__iter__)N)rGrHrI┌__doc__rєrKrЎrўrrrr┌TestIterableгs r∙cCs╢|jГ}|jg|jГ|jjtjgГГ|jg|jГ|jjtjdgГГ|jdg|jГ|jjtjddgГГ|jdddg|jГ|jjtjddgГГ|jdddddg|jГdS)z3Test extending repeated int32 fields with iterable.rr8rfrqr9N)r!rцrЖrOrr∙)r&r'rzrrr┌testExtendInt32WithIterable╗sz'MessageTest.testExtendInt32WithIterablecCs╢|jГ}|jg|jГ|jjtjgГГ|jg|jГ|jjtjdgГГ|jdg|jГ|jjtjddgГГ|jdddg|jГ|jjtjddgГГ|jdddddg|jГdS)z3Test extending repeated float fields with iterable.ggЁ?g@g@g@N)r!rцr[rOrr∙)r&r'rzrrr┌testExtendFloatWithIterable╚sz'MessageTest.testExtendFloatWithIterablecCs╢|jГ}|jg|jГ|jjtjgГГ|jg|jГ|jjtjdgГГ|jdg|jГ|jjtjddgГГ|jdddg|jГ|jjtjddgГГ|jdddddg|jГdS)z4Test extending repeated string fields with iterable.ru┌1┌2┌3┌4N)r!rцrNrOrr∙)r&r'rzrrr┌testExtendStringWithIterable╒sz(MessageTest.testExtendStringWithIterablec CsPtjГdkstjГdkrdS|jГ}|jtjГП}tj|jtj ГWdQRXdS)N┌cpprf) rrZVersionr!rrRZPickleErrorrSrЖZHIGHEST_PROTOCOL)r&r'rzrмrrr┌!testPickleRepeatedScalarContainerтsz-MessageTest.testPickleRepeatedScalarContainercCs|jГ}|jjГdS)z?Exercise a scenario that has led to segfaults in the past. N)r!r|rИ)r&r'rzrrr┌'testSortEmptyRepeatedCompositeContainerЁsz3MessageTest.testSortEmptyRepeatedCompositeContainerc Cs,|jГ}|jtГП}|jdГWdQRXdS)zBUsing HasField on a repeated field should raise an exception. rЖN)r!rr╗r╢)r&r'rzrмrrr┌testHasFieldOnRepeatedFieldЎsz'MessageTest.testHasFieldOnRepeatedFieldcCsИ|jГ}|jtГП}|jjГWdQRX|jjtdГГ|jd|jjГГ|jd|jjdГГ|jd|jjdГГ|jddg|jГdS)NrУr9rrfr8rq)r!r┌ IndexErrorrЖ┌poprOrтr+)r&r'rzrмrrr┌testRepeatedScalarFieldPop¤sz&MessageTest.testRepeatedScalarFieldPopcCs╠|jГ}|jtГП}|jjГWdQRX|jtГП}|jjdГWdQRXx tdГD]}|jjГ}||_qTW|j d|jjГjГ|j d|jjdГjГ|j d|jjdГjГ|j ddgdd Д|jDГГdS) Nr7rУr9rrfr8rqcSsg|] }|jСqSr)rx)ryrЯrrrr{sz=MessageTest.testRepeatedCompositeFieldPop..) r!rrr|rr=rтr}rxr+)r&r'rzrмr▀rЯrrr┌testRepeatedCompositeFieldPops z)MessageTest.testRepeatedCompositeFieldPopcCs\|jГ}x.tdГD]"}|jj||Г|jjГ}||_qW|j|j|jГ|j|j|jГdS)NrУ) r!rтrЖ┌insertr|r}rxrцr+)r&r'rzr▀rЯrrr┌testRepeatedCompareWithSelfs z'MessageTest.testRepeatedCompareWithSelfcCs6|jГ}|jjГ}|j}|jjГ}|j|jjdГdS)zwA case that lead to a segfault when a message detached from its parent container has itself a child container. rN)r╡Zrepeated_childr}r╨r+r┤┌optional_int32)r&r'rzrrr┌testReleasedNestedMessagess z&MessageTest.testReleasedNestedMessagescCsz|jГ}|jtГПg|_WdQRX|jjdГtjГdkrZ|jtГПg|_WdQRXn|jtГПg|_WdQRXdS)Nr8r)r!r┌AttributeErrorrЖrЗrrr=)r&r'rzrrr┌testSetRepeatedComposite(sz$MessageTest.testSetRepeatedComposite)HrGrHrIr)r2r6rErQrWr]r^r`rcrdrernrorprtrwrrКrСrЦrвrзrйrнr▒r│r╖r║r╜r┬r╞r╟r╔r╩r╦r╠r═r╧r╥r╘r╓r╪r▄r▐rф┌ bytearray┌setrчrщrъrыrьrэrюrЁrPr∙r·r√rrrrrrr rrrrrrresД $ --( # # rc@sTeZdZddДZddДZddДZddДZd d ДZddДZd dДZ ddДZ ddДZdS)┌ Proto2TestcCs╩tjГ}|j|jdГГ|j|jdГГ|j|jdГГ|jtГП|jdГWdQRX|jtГП|jdГWdQRX|jtГП|jdГWdQRX|jd|jГ|jd|jГ|jd|j j Гd|_d|_d|j _ |j|jdГГ|j|jdГГ|j|jdГГd |_d |_d|j _ |j|jdГГ|j|jdГГ|j|jdГГ|jdГ|jdГ|jdГ|j|jdГГ|j|jdГГ|j|jdГГ|jd|jГ|jd|jГ|jd|j j ГdS)Nr┌ optional_bool┌optional_nested_message┌field_doesnt_existrЖr|rFrУTщ) r r!r╣r╢rr╗r+rrrrxrXr╚)r&rrrr┌testFieldPresence:sD zProto2Test.testFieldPresencecCs tjГ}|jtГП}d|_WdQRX|jt|jjdГ|jjdГd|jd<|jtГПd|jd<WdQRXtjГ}d|_|jjdГ|jГ}tjГ}|j |Г|j |jdГГ|jd|jГ|jdt |jГГ|jГ|j |jГГ|jd|jГ|jd|jdГdS) z:Assigning an invalid enum number is not allowed in proto2.iЗ╓Nrfri@тi▒╦t┌optional_nested_enumr8)r r!rr╗rrуrЗr r,r*r╣r╢r+rРrа)r&rzrмrлr_Zm3rrr┌testAssignInvalidEnumjs* z Proto2Test.testAssignInvalidEnumc Cs6tjГ}d|jd<|jtГПd|jd<WdQRXdS)Nrщ{r8)rZTestEnumMapZknown_map_fieldrr╗Zunknown_map_field)r&rzrrr┌testUnknownEnumMapЙs zProto2Test.testUnknownEnumMapcCstjГ}|jtt|dГdS)N┌ Extensions)r r!rr r╕)r&rCrrr┌testExtensionsErrorsПszProto2Test.testExtensionsErrorscCshtjdГ}tjГ}|j|ГtjГ}tj|Г|j||Г|j||jГГtj |Г}|j||jГГdS)Nr0) rrr ZTestAllExtensionsr*ZSetAllExtensionsr+r,r-r.)r&r/r0r5r1rrr┌testGoldenExtensionsУs zProto2Test.testGoldenExtensionscCshtjdГ}tjГ}|j|ГtjГ}tj|Г|j||Г|j||jГГtj |Г}|j||jГГdS)Nr3) rrr ZTestPackedExtensionsr*ZSetAllPackedExtensionsr+r,r-r.)r&r/r0r5r1rrr┌testGoldenPackedExtensionsЮs z%Proto2Test.testGoldenPackedExtensionscCsNtjddН}tj|Г}tj|Г}|j||Г|j|jdГ|jtj |j ГdS)Nr8)rА)r ZTestRequiredrRrSrTr+rАrrZEncodeErrorr,)r&r0rUrVrrr┌testPickleIncompleteProtoйs z$Proto2Test.testPickleIncompleteProtocCs└tjГtjГtjГg}d|d_d|d_d|d_d|d_tjГ}d|_d|_d|_tjjГ}|jj|Г|j j|Г|j j|Г|jj|Г|jj|Г|j jГjj|dГ|j jГjj|dГ|j jГjj|dГ|jjГjj|dГ|jjГjj|dГ|jjГjj|dГ|jГ}tjГ}|j|Г|j|j|Г|j|j|Г|j|jj|Г|j|jtjj|Г|jt|jГdГ|jt|jГdГ|jt|jtjjГdГdS)zcCheck the merge behavior when a required or optional field appears multiple times in the input.r8rrfrqZhelloN)r r!r┌optional_int64r┘ZTestParsingMergeZRepeatedFieldsGeneratorZfield1rOZfield2Zfield3Zext1Zext2Zgroup1r}r╤Zgroup2r,r*r+Zrequired_all_typesZoptional_all_types┌ optionalgroupZoptional_group_all_typesrZoptional_extrРZrepeated_all_types┌ repeatedgroupZrepeated_ext)r&ZmessagesZmerged_message┌ generator┌dataZ parsing_mergerrr┌testParsingMerge╖sJ zProto2Test.testParsingMergecCs`tjddddddiddiid dd iddigdtjjgd ddН}|j|tjГ|jd|jГ|jd|jГ|jd|jГ|jd|jГ|jd|j j Г|j|jtjjГ|jd|jj Г|j|jdГГ|j|jtjГГ|jtjj|jГ|jdt|jГГ|jd |jdj Г|jd|jdj Г|jdt|jГГ|jtjj|jdГ|jtjj|jdГ|jd |jГ|jd|jГ|j|jdГГ|jdt|jГГ|jd|jГtjd dН}|jtjj|jГ|jtГПtjddidНWdQRX|jtГПtjddidНWdQRX|jtГПtjddНWdQRX|jtГПtjddНWdQRXdS)Nщdщ╚g╚r@єxrАiРrxiЇ┌BAZiXi╝┌FOOi r)r┌optional_fixed32rY┌optional_bytesr!r┌optional_foreign_messagerr"rу┌ default_int32r└r-rfrr8r щ*)rZINVALID_NESTED_FIELDщ)rZINVALID_VALUE_TYPEZ INVALID_LABEL)rу) r r!ZBARrоr+rr+rYr,r!rАrr╒rxrXr╢r-┌ForeignMessager)rrРr"rуr*r.r└r╣r[Z default_int64rr╗r=)r&rrrr┌testPythonicInitшsl zProto2Test.testPythonicInitN)rGrHrIrrrrrrrr%r2rrrrr8s01rc@seZdZddДZddДZddДZddДZd d ДZddДZd dДZ ddДZ ddДZddДZddДZ ddДZddДZddДZddДZdd ДZd!d"ДZd#d$ДZd%d&ДZd'd(ДZd)d*ДZd+d,ДZd-d.ДZd/d0ДZd1d2ДZd3d4ДZd5d6ДZd7d8ДZd9d:ДZd;d<ДZ d=d>ДZ!d?d@ДZ"dAS)B┌ Proto3TestcCs@t|Г}x&|D]\}}|j|||Г||=qW|ji|ГdS)N)┌dictr+)r&Zmap_iterZ dict_valuerд┌vrrr┌assertMapIterEquals+s zProto3Test.assertMapIterEqualsc4CsЎtjГ}|jtГП|jdГWdQRX|jtГП|jdГWdQRX|jtГП|jdГWdQRX|jtГП|jdГWdQRX|j|jdГГ|jtГП|jdГWdQRX|jtГП|jdГWdQRX|jtГП|jdГWdQRX|jd |jГ|jd |jГ|jd |j Г|jd|j Г|jd |jjГd |j_|j |jdГГd|_d |_d|_ d|_ d|j_|jdГ|jdГ|jdГ|jdГ|jdГ|jd |jГ|jd |jГ|jd |j Г|jd|j Г|jd |jjГdS)NrrYr┘rrrrЖr|rruFrУgЪЩЩЩЩЩё?rяTr)r r!rr╗r╢r╣r+rrYr┘rrrxrXr╚)r&rrrrr5sL zProto3Test.testFieldPresencecCsФtjГ}d|_|jd|jГ|jjdГ|jd|jdГd|jd<|jd|jdГ|jГ}tjГ}|j|Г|jd|jГ|jd|jdГdS)zCAssigning an unknown enum value is allowed and preserves the value.iЗ╓iў╦Tri▒╦tN)r r!rr+rуrЗr,r*)r&rzr_rлrrr┌testAssignUnknownEnumns z Proto3Test.testAssignUnknownEnumcCsNtjГ}|jd|jkГ|jd|jkГ|jd|jkГ|jd|jkГ|jd|jkГ|jd|jkГ|jd|j kГ|jd|j kГ|jd|jkГ|jd|jdГ|jd|jdГ|jd|jdГ|jd|jdГ|jd |jdГ|j t|jdtГГ|jd|jdГ|j t|jdtГГ|jd |j dГ|jd|j dГ|jd|jdГ|j d|jkГ|j d|jkГ|j d|jkГ|j d|jkГ|j d|jkГ|j d|jkГ|j d|j kГ|j d|j kГ|j d|jkГ|j|j dtjГ|jtГП|j dWdQRX|jtГПd|j kWdQRXdS)NrrfrЪFrящoixrgrurхiЕ ll¤ liЕ ll¤ liЕ ll¤ l)r┌TestMapr╣┌map_int32_int32┌map_int64_int64┌map_uint32_uint32┌map_uint64_uint64┌map_int32_double┌ map_bool_bool┌map_string_stringZmap_int32_bytes┌map_int32_enumr+rX┌ isinstance┌floatrЇrоr┌r█rr=)r&rCrrr┌testScalarMapDefaultsЖsFz Proto3Test.testScalarMapDefaultscCs&tjГ}|j|jjdГГ|jd|jjddГГ|j|jjdГГd|jd<|jd|jjdГГ|jd|jjdГГ|jtГП|jjdГWdQRX|j|jjdГГ|jd|jjddГГ|jd}|j ||jjdГГt jГdkРr|jtГП|jjdГWdQRXn|jd|jjdГГdS)NrУrЭrrur)rr9r╫r:┌getr+rr=┌map_int32_foreign_messager┴rr)r&rC┌submsgrrr┌ testMapGet╢s" zProto3Test.testMapGetc!Cs4tjГ}|jdt|jГГ|jd|jkГd|jd<d|jd<d|jd<d|jd<d|j d<d |j d <d|jd<d |jd <d|j d<d|j d<|jg|jГГ|jd t|jГГ|jtГПd|jd<WdQRX|jd t|jГГ|jtГПd|jd<WdQRX|jГ}tjГ}|j|Г|jtГПd|jd<WdQRX|jtГПd|jd<WdQRX|jd|jdГ|jd|jdГ|jd|jdГ|jd|jd Г|jd|j dГ|jd |j d Г|jd|jdГ|jd |jd Г|jd|j dГ|jd|j dГtjГdkРr0|jdt|jГГdS)!NrrУi╚rrfщ"rЪg333333є?gffffff @r8┌123rяTixrz{-123: -456}i8■ iЕ ll¤ ll¤ lli8■ iЕ ll¤ ll¤ ll)rr9r+rРr:r╣r;r<r=Zmap_int32_floatr>r@r?rA┌FindInitializationErrorsrr=r,r*rЙrrr$)r&rCr_┌msg2rrr┌ testScalarMap╤sP zProto3Test.testScalarMapcCs@tjГ}d|jd<d|jd<|j|jГdГ|jd|jГГdS)Nrrur╙s r )rr9r:r@r+┌ByteSizer,)r&rCrrr┌testMapEntryAlwaysSerializeds z'Proto3Test.testMapEntryAlwaysSerializedcCsntjГ}d}|jdГ}||j|<t|jjГГd\}}|j||Г|j||Г|j|tj Г|j|tj ГdS)NuсИ┤┌utf8r) rr9┌encoder@rи┌itemsr+rоr┌r█)r&rCZunicode_objZ bytes_objrЛ┌valuerrr┌ testStringUnicodeConversionInMaps z+Proto3Test.testStringUnicodeConversionInMapcCsZtjГ}|jdt|jГГ|jd|jkГ|jd|jjd Г|jdt|jГГ|jd|jГ|jd|jГ|jdt|jГГ|jt ГП|jdWdQRX|jt ГП|jd|jd<WdQRX|jdt|jГГ|jГ}tjГ}|j|Г|jdt|jГГ|jd|jГ|jd|jГ|jdt|jГГt jГdkРrV|jd tt|jГГГdS) NrrУri╚rfrJiчrri8■ i8■ i8■ )rr9r+rРrFr╣Z get_or_creater#rr=r╗r,r*rrr$)r&rCr_rLrrr┌testMessageMap&s0 zProto3Test.testMessageMapcCsЪtjГ}d|jdj_|jd=d|jdj_|jdt|jГГd|jdj_|jdt|jГГ|jГ}tjГ}|j|Гddg}x|D]}|j|=qЖWdS)Nr8rf) rr9┌map_int32_all_typesrrxr+rРr,r*)r&rCr_rL┌keysrЛrrr┌testNestedMessageMapItemDeleteOs z)Proto3Test.testNestedMessageMapItemDeletecCsptjГ}d|jd<|jГ}d|jd<|j|jГ|dГd|jd_|jГ}d|jd_|j|jГ|dГdS)Nr8щАщ)rr9r:rNr+rFrБ)r&rCrїrrr┌testMapByteSize`s zProto3Test.testMapByteSizec CsюtjГ}d|jd<d|jd<d|jd<d|jd_d |jd _tjГ}d|jd<d|jd <d|jd _d|jd _|jd }|j|Г|jd|jdГ|jd|jdГ|jd|jdГ|jd|jd Г|jd|jdjГ|jd |jd jГ|j |jd j dГГtjГdkРr(|jd|jГi}x.|jD]$}|j ||kГ|j|j||<Рq4W|jdd dЬ|Г|jd=|j d|jkГ|jd =|j d |jkГtjГdkРr╠|j tГП|jd=WdQRXn|j tГП|jd=WdQRXdS)NrIr╙щNщ8rЪщrУr8rЭщ▐щ7щcщXrrШ┌dr)r8r_ru)rr9r:r;rFrБrcr╤r+r╣r╢rrrr=┌KeyError)r&rCrLZ old_map_valueZas_dictrЛrrr┌ testMergeFromlsH zProto3Test.testMergeFromc Cs.tjГ}|jtdГП|jdГWdQRXdS)NzWParameter to MergeFrom must be instance of same class: expected .*TestMap got int\.r8)rr9┌assertRaisesRegexpr=r╤)r&rCrrr┌testMergeFromBadTypeиs zProto3Test.testMergeFromBadTypec Cs.tjГ}|jtdГП|jdГWdQRXdS)Nz[Parameter to [A-Za-z]*From must be instance of same class: expected .*TestMap got int\.r8)rr9rfr=r╬)r&rCrrr┌testCopyFromBadType░s zProto3Test.testCopyFromBadTypecCs╢tjГ}tdГ|jtdГ<td Г|jtdГ<tdГ|jtdГ<tdГ|jtd Г<|jГ}tjГ}|j|Г|j d|jdГ|j d|jdГ|j d|jdГ|j d|jdГdS)Ni╚rrfrIrЪi8■ iЕ ll¤ ll¤ lli8■ iЕ ll¤ ll¤ ll) rr9r▌r:r;r<r=r,r*r+)r&rCr_rLrrr┌testIntegerMapWithLongs╕s z"Proto3Test.testIntegerMapWithLongscCsТtjГ}d|jjd<|jГ}tjГ}|j|Г|j||Гd|jjd<|jГ}|j|Г|j||Г|jjjГ|jГ}|j|Г|j||ГdS)Ni╚riчix)r┌TestMapSubmessage┌test_mapr:r,r*r+┌clear)r&rCr_rLrrr┌testMapAssignmentCausesPresence╚s z*Proto3Test.testMapAssignmentCausesPresencecCs╬tjГ}d|jjd_|jГ}tjГ}|j|Г|j||Гd|jjd_|jГ}|j|Г|j||Г|jjdj|jjdГ|jГ}|j|Г|j||Г|jjj Г|jГ}|j|Г|j||ГdS)NrУrщix) rrjrkrFrБr,r*r+r╤rl)r&rCr_rLrrr┌-testMapAssignmentCausesPresenceForSubmessages▐s& z8Proto3Test.testMapAssignmentCausesPresenceForSubmessagescCs~tjГ}t|jГ}t|jГ}d|jd<d|jd_|jtГПx|D]}qDWWdQRX|jtГПx|D]}qhWWdQRXdS)NrJrяrУ)rr9┌iterr@rFrБr┌RuntimeError)r&rCZstring_string_iterZint32_foreign_iterrЛrrr┌testModifyMapWhileIterating·s z&Proto3Test.testModifyMapWhileIteratingc CsОtjГ}|jd}|j||jdГ|j|tjГd|_|jГ}tjГ}|j |Г|j d|jdjГ|jtГПtjГ|jd<WdQRXdS)Nr8rУrb) rr9rFr┴rоr r1rБr,r*r+rr╗)r&rCrGr_rLrrr┌testSubmessageMaps zProto3Test.testSubmessageMapcCs|tjГ}x |jjГD]\}}|jdГqWd|jd<d|jd<d|jd<|jdt|jГГddddЬ}|j|jjГ|ГdS)NFr9rfrТrqщ)rfrqr9)rr9r:rRrXr+rРr6)r&rCrдr5┌ matching_dictrrr┌testMapIterations zProto3Test.testMapIterationcstjdkРr tjГ}d|jd<d|jd<d|jd<d|jd<|j}Иjdt|ГГtjГ}|j|jГГЗfdd Д}||j Г|j ГГ||jГ|jГГ||j Г|jГГИjd|jdГГИjd|jd ГГИjd|jdГГИjd|jdГГИjdt|ГГ|jГ\}}Иjd||ГИjdt|ГГ|jГИjdt|ГГИjtГП|jГWdQRXИjd|jdГГИjdt|ГГ|j|jГИjdt|ГГИjtГП|j|j|jГWdQRXИjtГП|jdГWdQRXИjtГП|jd dНWdQRXdS)Nrqr9rfrТrtrЭrУcs0Иjt|Г|dГИjt|Г|ddЕГdS)Nrr8)r+┌nextrи)┌seq┌iterator)r&rr┌ CheckItems9sz-Proto3Test.testPython2Map..CheckItemsiчrr8r╙)rS)rq)rrrsrr9r:r+rРr*r,rRZ iteritemsrWZiterkeysrЄZ itervaluesrEr┌popitemrlrrd┌ setdefault┌updater=)r&rCZ map_int32rLrzrЛrSr)r&r┌testPython2Map-sJ zProto3Test.testPython2MapcCs^tjГ}d|jd<d|jd<d|jd<d|jd<d|jd<|jjГ}|jjГ}|j||ГdS)Nru┌ local_init_op┌trainable_variables┌ variables┌init_op┌ summaries)rr9r@rRr+)r&rCZitems1Zitems2rrr┌testMapItems]s zProto3Test.testMapItemscCs~d}tjГ}d|jd<d|jd<d|jd<d|jd <d |jd<d |jd<d |jd<d|jd<d|jd<|j||jddНГdS)NsНr init_opdr item1er item2fr item3gr item4QQr local_init_opar summarieser trainable_variablesbr variablescrАrrВrАrБrБrcrВ┌erГZitem1┌fZitem2┌gZitem3ZQQZitem4T)rF)rr9r@r+r,)r&r/rCrrr┌!testMapDeterministicSerializationls z,Proto3Test.testMapDeterministicSerializationcCsNtjГ}d|jd<d|jd<d|jd<|jjГ}~ddddЬ}|j||ГdS)Nr9rfrТrqrt)rfrqr9)rr9r:rRr6)r&rC┌itrurrr┌testMapIterationClearMessageЙs z'Proto3Test.testMapIterationClearMessagecCsftjdddЬdН}|jd|jdГ|jd|jdГtjdtjddНid Н}|jd|jdjГdS) Nrfr9)r8rq)r:r8rqrУ)rБ)rF)rr9r+r:r r1rFrБ)r&rCrrr┌testMapConstructionЧszProto3Test.testMapConstructioncCs`tjГ}|j}d|d<d|d<d|d<|jdГ|jd|jГГddddЬ}|j|jГ|ГdS) Nr9rfrТrqrtr:rх)rfrqr9)rr9r:r╚r+r,r6rR)r&rCZ int32_maprurrr┌testMapValidAfterFieldClearedаs z(Proto3Test.testMapValidAfterFieldClearedcCsHtjГ}|j}d|d_|jdГ|jd|jГГ|jd|jГkГdS)NrУrfrFrх) rr9rFrБr╚r+r,rXrW)r&rCZint32_foreign_messagerrr┌$testMessageMapValidAfterFieldCleared░s z/Proto3Test.testMessageMapValidAfterFieldClearedcCs|tjГ}t|jГ}|jdГ|jtГПx|D]}q.WWdQRXt|jГ}|jdГ|jtГПx|D]}qfWWdQRXdS)Nr:rF)rr9rpr:r╚rrqrF)r&rCrЙrмrrr┌"testMapIterInvalidatedByClearField╜s z-Proto3Test.testMapIterInvalidatedByClearFieldcCsРtjГ}|jdt|jГГd|jd<|jdt|jГГ|jtГП|jd=WdQRX|jd=|jdt|jГГ|jtГП|jd=WdQRXdS)NrrТr9r8rbщ )rr9r+rРr:rrdrV)r&rCrrr┌ testMapDelete╤s zProto3Test.testMapDeletecCsLtjГ}|j|jtjГ|j|jtjГ|j|jtjГ|j|jtjГdS)N)rr9rоr:rп┌Mapping┌MutableMappingrF)r&rCrrr┌testMapsAreMappingтszProto3Test.testMapsAreMappingcCsDtjГ}d|jd<|j|j|jГ|j|j|jГ|j|jdГdS)Ni╚rri8■ iЕ )rr9r:r+rFZassertNotEqual)r&rCrrr┌testMapsCompareъs zProto3Test.testMapsComparecCs@tjГ}d|jd<d|jd<d|jd_|jdt|jГГГdS)NrJrящ@щ#rУr├r) rr9r@r:Zmap_string_foreign_messagerБr+rРrK)r&rCrrr┌(testMapFindInitializationErrorsSmokeTestЄs z3Proto3Test.testMapFindInitializationErrorsSmokeTestN)#rGrHrIr6rr7rDrHrMrOrTrUrXr[rergrhrirmrorrrsrvr~rДrИrКrЛrМrНrОrРrУrФrЧrrrrr3(s@ 90=)<0 r3c@seZdZddДZddДZdS)┌ValidTypeNamesTestcsВtt|ГГjdГdЙd|d|f}|jtЗfddД|DГГdИ|fГИjdГ}|d }dj|ddЕГ}t||gd НdS)N·'r8zRepeated%sContainerzRepeated%sFieldContainerc3s|]}Иj|ГVqdS)N)┌endswith)ryr5)┌tp_namerrrрsz:ValidTypeNamesTest.assertImportFromName..z%r does end with any of %r┌.)┌fromlistr<r<)r$┌type┌splitrX┌any┌join┌ __import__)r&rCZ base_nameZvalid_names┌parts┌ class_nameZmodule_namer)rЫr┌assertImportFromName¤s z'ValidTypeNamesTest.assertImportFromNamecCs(tjГ}|j|jdГ|j|jdГdS)NZScalarZ Composite)r r!rеrЖr|)r&rбrrr┌testTypeNamesCanBeImported sz-ValidTypeNamesTest.testTypeNamesCanBeImportedN)rGrHrIrеrжrrrrrШ√s rШc@s$eZdZddДZddДZddДZdS)┌PackedFieldTestcCsм|jjdГ|jjdГ|jjdГ|jjdГ|jjdГ|jjdГ|jjdГ|jjdГ|j jdГ|j jdГ|jjdГ|jjdГ|j jdГ|jjdГdS)Nr8gЁ?T)rЖrЗZrepeated_int64Zrepeated_uint32Zrepeated_uint64Zrepeated_sint32Zrepeated_sint64Zrepeated_fixed32Zrepeated_fixed64Zrepeated_sfixed32Zrepeated_sfixed64r[r\Z repeated_boolrу)r&rrrr┌ setMessageszPackedFieldTest.setMessagecCs*tjГ}|j|Гd}|j||jГГdS)NsH "*2:BJRZА?bЁ?jr)rr4rиr+r,)r&rr/rrr┌testPackedFields"s z PackedFieldTest.testPackedFieldscCs*tjГ}|j|Гd}|j||jГГdS)Ns: (0=AMQ]А?aЁ?hp)rZTestUnpackedTypesrиr+r,)r&rr/rrr┌testUnpackedFields5s z"PackedFieldTest.testUnpackedFieldsN)rGrHrIrиrйrкrrrrrзsrзrrfrnz;explicit tests of the C++ implementation for PY27 and abovec@s0eZdZeddДГZddДZddДZddДZd S) ┌OversizeProtosTestcCsFd}tjГ}tjГ}tj||Г|j|Гtj|Гj |j dГГ|_dS)Na╢ name: "f/f.msg2" package: "f" message_type { name: "msg1" field { name: "payload" number: 1 label: LABEL_OPTIONAL type: TYPE_STRING } } message_type { name: "msg2" field { name: "field" number: 1 label: LABEL_OPTIONAL type: TYPE_MESSAGE type_name: "msg1" } } zf.msg2)rZDescriptorPoolrZFileDescriptorProtorZParseZAddr ZMessageFactoryZGetPrototypeZFindMessageTypeByName┌ proto_cls)┌clsZ file_descZpoolZdescrrr┌ setUpClassNs zOversizeProtosTest.setUpClasscCs(|jГ|_dd|jj_|jjГ|_dS)NrБirХr8iii)rм┌p┌fieldr┤r,┌p_serialized)r&rrr┌setUpqs zOversizeProtosTest.setUpcCsfddlm}|dГ|jГ}y|j|jГWn4tjk r`}z|jt|ГdГWYdd}~XnXdS)Nr)┌SetAllowOversizeProtosFzError parsing message) ┌google.protobuf.pyext._messager│rмr*r▒rr?r+r$)r&r│┌qrЕrrr┌testAssertOversizeProtovsz*OversizeProtosTest.testAssertOversizeProtocCsBddlm}|dГ|jГ}|j|jГ|j|jjj|jjГdS)Nr)r│T) r┤r│rмr*r▒r+rпr░r┤)r&r│r╡rrr┌testSucceedOversizeProtos z+OversizeProtosTest.testSucceedOversizeProtoN)rGrHrI┌classmethodrоr▓r╢r╖rrrrrлIs# rл┌__main__)rfrn)4r°┌ __author__rпr-rhrФrRr┌rrr@Z unittest2Zunittest┌ImportErrorrгrсZgoogle.protobufrrr r rrr rZgoogle.protobuf.internalrrrrrrrZPY3┌intr▌rrrrZBaseTestCaseZNamedParametersrrr3rШrзZskipIfrrsrлrG┌mainrrrr┌)sДWqX9: