← Back
Editing: message_test.cpython-36.pyc
3 ���hnV � @ s( d Z dZddlZddlZddlZddlZddlZddlZddlZddl Z y ddl Z W n e k rp ddl Z Y nX ye W n ek r� dd� Z Y nX ddlmZ ddlmZ ddlmZ dd lmZ dd lmZ dd lmZ dd lmZ dd lmZ ddlmZ ddlmZ ddlmZ ddlm Z ddlm Z ddlm Z ddlm Z ej �rXe!Z"dd� Z#dd� Z$dd� Z%dd � Z&e j'Z'e j(d efd ef�G d d � d e'��Z)G d!d"� d"e'�Z*G d#d$� d$e'�Z+G d%d&� d&e'�Z,G d'd(� d(e'�Z-e j.ej/� d)k�p�ej0d0k d,�G d-d.� d.e'��Z1e2d/k�r$e j3� 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)� Nc C s | |k| |k S )N� )�x�yr r �"/usr/lib/python3.6/message_test.py�<lambda>> s r )�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)�_parameterizedc C s | | kS )Nr )�valr r r �isnanV s r c C s t | � ot | d �S )Nr )r )r r r r �isinfY s r c C s t | �o| dkS )Nr )r )r r r r �IsPosInf\ s r c C s t | �o| dk S )Nr )r )r r r r �IsNegInf^ s r Z_proto2Z_proto3c @ s8 e Zd Zdd� Zdd� Zdd� Zdd� Zd d � Zd d � Zd d� Z dd� Z dd� Z dd� Z dd� Z dd� Zdd� Zdd � Zd d � Zd d � Zd!d"� Zd#d$� Zd%d&� Zd'd(� Zd)d*� Zd+d,� Zd-d.� Zd/d0� Zd1d2� Zd3d4� Z d5d6� Z d7d8� Z d9d:� Z d;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� g i e2� g Z3dcdd� Z4dedf� Z5dgdh� Z6didj� Z7dkdl� Z8dmdn� Z9dodp� Z:G dqdr� dre;�Z<dsdt� Z=dudv� Z>dwdx� Z?dydz� Z@d{d|� ZAd}d~� ZBdd�� ZCd�d�� ZDd�d�� ZEd�d�� ZFd�d�� ZGd]S )�� MessageTestc C sX t j� dkr| jd� tjd�}| jt��}|jj|� W d Q R X | j dt |j �� d S )N�pythonz�Skipping testBadUtf8String, currently only the python api implementation raises UnicodeDecodeError when a string field contains bad utf-8.Zbad_utf8_stringz TestAllTypes.optional_string) r �TypeZskipTestr �GoldenFileData� assertRaises�UnicodeDecodeError� TestAllTypes� FromString�assertIn�str� exception)�self�message_moduleZ bad_utf8_data�contextr r r �testBadUtf8Stringj s z MessageTest.testBadUtf8Stringc C sr |t krtjd�}n tjd�}|j� }|j|� |t krDtj| |� | j||j� � tj |�}| j||j� � d S )NZ golden_message_oneof_implementedZgolden_message_proto3) r r r r! �ParseFromStringZExpectAllFieldsSet� assertEqual�SerializeToString�copy�deepcopy)r& r' � golden_data�golden_message� golden_copyr r r �testGoldenMessaget s z MessageTest.testGoldenMessagec C sh t jd�}|j� }|j|� |j� }t j|� | j||� | j||j� � tj|�}| j||j� � d S )N� golden_packed_fields_message) r r �TestPackedTypesr* ZSetAllPackedFieldsr+ r, r- r. )r& r' r/ r0 �all_setr1 r r r �testGoldenPackedMessage� s z#MessageTest.testGoldenPackedMessagec C s� |j � }| jt|jd� | jt|jd� tjdd�}tj� dkrv| jt j ��}|j|� W d Q R X | j dt |j �� nftjdd��T}tjd � |j|� t|�dks�t�t|d jt�s�t�| j d t |d j �� W d Q R X d S ) Nr �0� � r zUnexpected end-group tag.T)�record�alwaysz4Unexpected end-group tag: Not all data was converted���r< )r! r � TypeErrorr" � Exceptionr ZTagBytesr r r � DecodeErrorr+ r$ r% �warnings�catch_warnings� simplefilter�len�AssertionError� issubclass�category�RuntimeWarning)r& r' �msgZend_tagr( �wr r r �testParseErrors� s zMessageTest.testParseErrorsc s� d}|j � }|jjddddg� | j||jd d�� | j||jdd�� | j||jdd�� G d d � d t�� G � fd d �d t�}| j� �� |j|� d� W d Q R X d S ) Ns5 � One string� Two string� Red string� Blue stringz One stringz Two stringz Red stringz Blue string)� deterministicFTc @ s e Zd ZdS )z:MessageTest.testDeterminismParameters.<locals>.BadArgErrorN)�__name__� __module__� __qualname__r r r r � BadArgError� s rO c s$ e Zd Z� fdd�Z� fdd�ZdS )z5MessageTest.testDeterminismParameters.<locals>.BadArgc s � � �d S )Nr )r& )rO r r �__bool__� s z>MessageTest.testDeterminismParameters.<locals>.BadArg.__bool__c s � � �d S )Nr )r& )rO r r rP � s N)rL rM rN rP r )rO r r �BadArg� s rQ )r! �repeated_string�extendr+ r, r> �objectr )r& r' r/ r0 rQ r )rO r �testDeterminismParameters� s" z%MessageTest.testDeterminismParametersc C s@ t jd�}|j� }|j|� tj|�}tj|�}| j||� d S )Nr0 )r r r! r* �pickle�dumps�loadsr+ )r& r' r/ r0 �pickled_message�unpickled_messager r r �testPickleSupport� s z MessageTest.testPickleSupportc C s� |t krd}nd}|j� }|j|� | jt|j�� | jt|j�� | jt|jd �� | jt|jd �� | j ||j � � d S )Ns ] �a �� �� �s ] �a �� �� �r ) r r! r* � assertTruer �optional_float�optional_double�repeated_float�repeated_doubler+ r, )r& r' r/ r0 r r r �testPositiveInfinity� s z MessageTest.testPositiveInfinityc C s� |t krd}nd}|j� }|j|� | jt|j�� | jt|j�� | jt|jd �� | jt|jd �� | j ||j � � d S )Ns ] ��a �� ��� �s ] ��a �� ��� �r ) r r! r* r\ r r] r^ r_ r` r+ r, )r& r' r/ r0 r r r �testNegativeInfinity� s z MessageTest.testNegativeInfinityc C s� d}|j � }|j|� | jt|j�� | jt|j�� | jt|jd �� | jt|jd �� |j� }|j � }|j|� | jt|j�� | jt|j�� | jt|jd �� | jt|jd �� d S )Ns ] �a �� �� �r ) r! r* r\ r r] r^ r_ r` r, )r& r' r/ r0 � serializedr r r r �testNotANumber� s zMessageTest.testNotANumberc C sR d}|j � }|j|� | jt|jd �� | jt|jd �� | j||j� � d S )Ns � �� �r )r4 r* r\ r � packed_float� packed_doubler+ r, )r& r' r/ r0 r r r �testPositiveInfinityPacked s z&MessageTest.testPositiveInfinityPackedc C sR d}|j � }|j|� | jt|jd �� | jt|jd �� | j||j� � d S )Ns � ��� �r )r4 r* r\ r re rf r+ r, )r& r' r/ r0 r r r �testNegativeInfinityPacked s z&MessageTest.testNegativeInfinityPackedc C s� d}|j � }|j|� | jt|jd �� | jt|jd �� |j� }|j � }|j|� | jt|jd �� | jt|jd �� d S )Ns � �� �r )r4 r* r\ r re rf r, )r& r' r/ r0 rc r r r r �testNotANumberPacked% s z MessageTest.testNotANumberPackedc C st |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� d S )N� � g �?i����i����)r! �math�powr] r* r, r\ )r& r' r �kMostPosExponentNoSigBits�kMostPosExponentOneSigBit�kMostNegExponentNoSigBits�kMostNegExponentOneSigBitr r r �testExtremeFloatValues3 s: z"MessageTest.testExtremeFloatValuesc C st |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� d S )Nrj i� g �?i���i���)r! rl rm r^ r* r, r\ )r& r' r rn ro rp rq r r r �testExtremeDoubleValues` s: z#MessageTest.testExtremeDoubleValuesc C s" |j � }d|_| jt|�d� d S )Ng @zoptional_float: 2.0 )r! r] r+ r$ )r& r' r r r r �testFloatPrinting� s z MessageTest.testFloatPrintingc C s> |j � }d|_tjdkr*| jt|�d� n| jt|�d� d S )Ng^!�7ݚ�?� z%optional_double: 0.12345678912345678 z optional_double: 0.123456789123 )ru )r! r^ �sys� version_infor+ r$ )r& r' r r r r � testHighPrecisionFloatPrinting� s z*MessageTest.testHighPrecisionFloatPrintingc C s< |j � }tj|� |j� }|j|j� � | jt|�d� d S )N� )r! r ZSetAllNonLazyFieldsZTestEmptyMessager* r, r+ r$ )r& r' Z populated�emptyr r r �testUnknownFieldPrinting� s z$MessageTest.testUnknownFieldPrintingc C s� |j � }|jjdd� |jjdd� |jjdd� |jjdd� | jddddgdd� |jD �� | jddddgdd� t|j�D �� | jddddgd d� |jd d d � D �� d S ) Nr8 )�bbrj ru r9 c S s g | ] }|j �qS r )r| )�.0�mr r r � <listcomp>� s z@MessageTest.testRepeatedNestedFieldIteration.<locals>.<listcomp>c S s g | ] }|j �qS r )r| )r} r~ r r r r � s c S s g | ] }|j �qS r )r| )r} r~ r r r r � s r< )r! �repeated_nested_message�addr+ �reversed)r& r' rH r r r � testRepeatedNestedFieldIteration� s z,MessageTest.testRepeatedNestedFieldIterationc C s� |j � }|jjd� |jjd� |jjd� |jj� | j|jd d� | j|jd d� | j|jd d� | jt|j�tdddg�� |jjd� |jjd� |jjd� |jj� | j|jd d� | j|jd d� | j|jd d� |jjd� |jjd � |jjd � |jj� | j|jd d� | j|jd d � | j|jd d � | jt|j�tdd d g�� |j jd � |j jd � |j jd � |j j� | j|j d d � | j|j d d � | j|j d d � | jt|j �td d d g�� dS )z7Check some different types with the default comparator.r8 ru rj r g�������?g�������?g333333�?�a�c�b� a� c� bN) r! �repeated_int32�append�sortr+ r$ r_ �assertAlmostEqualrR Zrepeated_bytes)r& r' r r r r �0testSortingRepeatedScalarFieldsDefaultComparator� s@ z<MessageTest.testSortingRepeatedScalarFieldsDefaultComparatorc C s� |j � }|jjd � |jjd � |jjd � |jjtd� | j|jd d � | j|jd d� | j|jd d� |jjd� |jjd� |jjd� |jjtd� | j|jd d� | j|jd d� | j|jd d� d S )z2Check some different types with custom comparator.ru rj r8 )�keyr �aaar| r� N������r< r< r� r� )r! r� r� r� �absr+ rR rC )r& r' r r r r �/testSortingRepeatedScalarFieldsCustomComparator� s z;MessageTest.testSortingRepeatedScalarFieldsCustomComparatorc C s� |j � }d|jj� _d|jj� _d|jj� _d|jj� _d|jj� _d|jj� _|jjtjd�d� | j|jd jd� | j|jd jd� | j|jd jd� | j|jd jd� | j|jd jd� | j|jd jd� | jt|j�d � d S ) zECheck passing a custom comparator to sort a repeated composite field.r8 ru rj � � r9 r| )r� r z0[bb: 1 , bb: 2 , bb: 3 , bb: 4 , bb: 5 , bb: 6 ]N) r! r� r� r| r� �operator� attrgetterr+ r$ )r& r' r r r r �2testSortingRepeatedCompositeFieldsCustomComparator� s z>MessageTest.testSortingRepeatedCompositeFieldsCustomComparatorc C s� |j � }d|jj� _d|jj� _d|jj� _d|jj� _d|jj� _d|jj� _d|jj� _|jjdd � d � | jdddddddgd d � |jD �� |j� }|j� |j|� | jdddddddgd d � |jD �� dS )zECheck passing a custom comparator to sort a repeated composite field.� � � �! � � � c S s | j d S )Nr� )r| )�zr r r r s zFMessageTest.testSortingRepeatedCompositeFieldsStable.<locals>.<lambda>)r� c S s g | ] }|j �qS r )r| )r} �nr r r r s zHMessageTest.testSortingRepeatedCompositeFieldsStable.<locals>.<listcomp>c S s g | ] }|j �qS r )r| )r} r� r r r r s N) r! r� r� r| r� r+ r, �ClearZMergeFromString)r& r' r �pbr r r �(testSortingRepeatedCompositeFieldsStable� s$ z4MessageTest.testSortingRepeatedCompositeFieldsStablec C s@ |j � }tjd�}dd� }d|jj� _d|jj� _d|jj� _d|jj� _d|jj� _d |jj� _|jj|d � | jd d � |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.r| c S s t | j|j�S )N)�cmpr| )r� r� r r r r s zEMessageTest.testRepeatedCompositeFieldSortArguments.<locals>.<lambda>r8 ru rj r� r� r9 )r� c S s g | ] }|j �qS r )r| )r} �kr r r r % s zGMessageTest.testRepeatedCompositeFieldSortArguments.<locals>.<listcomp>T)r� �reversec S s g | ] }|j �qS r )r| )r} r� r r r r ( s N)� sort_functionc S s g | ] }|j �qS r )r| )r} r� r r r r , s )r� r� c S s g | ] }|j �qS r )r| )r} r� r r r r / s )ru ) r! r� r� r� r� r| r� r+ rv rw )r& r' r Zget_bbZcmp_bbr r r �'testRepeatedCompositeFieldSortArguments s. z3MessageTest.testRepeatedCompositeFieldSortArgumentsc C s� |j � }|jjd� |jjd� |jjd� |jjtd� | jt|j�dddg� |jjtdd� | jt|j�dddg� tjdk r�dd� }|jj|d � | jt|j�ddd g� |jj|dd � | jt|j�d d d g� |j jd � |j jd � |j jd � |j jt d� | jt|j �d d d g� |j jt dd� | jt|j �d d d g� tjd k �r�dd� }|j j|d � | jt|j �d d d g� |j j|dd � | jt|j �d d d g� dS )!z9Check sorting a scalar field using list.sort() arguments.ru rj r8 )r� T)r� r� c S s t t| �t|��S )N)r� r� )r� r� r r r r > s zBMessageTest.testRepeatedScalarFieldSortArguments.<locals>.<lambda>)r� )r� r� r� r| r� c S s t t| �t|��S )N)r� rC )r� r� r r r r L s Nr� r� r< r< r� r� r� r� r< )ru r< r� r� r� r� r< )ru ) r! r� r� r� r� r+ �listrv rw rR rC )r& r' r Zabs_cmpZlen_cmpr r r �$testRepeatedScalarFieldSortArguments2 s6 z0MessageTest.testRepeatedScalarFieldSortArgumentsc C sJ |j � }|j � }|jjd� |jjd� |jjd� |jjd� |jjd� |jjd� d|jj� _d|jj� _d|jj� _d|jj� _d|jj� _d|jj� _tjdkr�d S ||k }|j|jk }| jt ||�d� | jt |j|j�d� | jt |jdddg�d� | jt |j|j�d� | j t �� t |j|jd d � � W d Q R X d S )Nr r8 rj ru )ru ) r! r� r� r� r� r| rv rw r+ r� r r= )r& r' �m1�m2�_r r r � testRepeatedFieldsComparableR s2 z(MessageTest.testRepeatedFieldsComparablec C s, |j � }| j|jtj� | j|jtj� d S )N)r! �assertIsInstancer� � collections�MutableSequencer� )r& r' r~ r r r � testRepeatedFieldsAreSequencesu s z*MessageTest.testRepeatedFieldsAreSequencesc C sL |j � }| jt�� t|j� W d Q R X | jt�� t|j� W d Q R X d S )N)r! r r= �hashr� r� )r& r' r~ r r r � testRepeatedFieldsNotHashable{ s z)MessageTest.testRepeatedFieldsNotHashablec C s* |j � }|jjjg � | j|jd�� d S )N�payload)�NestedTestAllTypesr� r� rS r\ �HasField)r& r' r~ r r r �$testRepeatedFieldInsideNestedMessage� s z0MessageTest.testRepeatedFieldInsideNestedMessagec C s t ||� | 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& rH Z attributer r r �ensureNestedMessageExists� s z%MessageTest.ensureNestedMessageExistsc C s, |j � }| jt|jd� | jt|jd� d S )NZno_such_oneof_fieldr )r! r � ValueError� WhichOneofr> )r& r' r~ r r r � testOneofGetCaseNonexistingField� s z,MessageTest.testOneofGetCaseNonexistingFieldc C s� |j � }| jd |jd�� | j|jd�� d|_| jd|jd�� | j|jd�� | j|jd�� d|_| jd|jd�� | j|jd�� | j|jd�� d S )N� oneof_field� oneof_uint32r � oneof_stringry ) r! �assertIsr� r� r� r� r+ r\ r� )r& r' r~ r r r �testOneofDefaultValues� s z"MessageTest.testOneofDefaultValuesc C sb |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�� d S ) Nr� r� r� �foor� �oneof_nested_messages bb� oneof_bytes) r! r� r� r� r+ r\ r� r� r� r� r| r� )r& r' r~ r r r �testOneofSemantics� s2 z MessageTest.testOneofSemanticsc C s> |j � }d|_| j|d� | jd|jd�� | jd|j� d S )Nr� r� r� r� )r! r� r� r+ r� )r& r' r~ r r r �!testOneofCompositeFieldReadAccess� s z-MessageTest.testOneofCompositeFieldReadAccessc C s� |j � }| jd |jd�� |tkr2| j|jd�� d|_| jd|jd�� |tkrb| j|jd�� d|_ | jd|jd�� |j d� | jd |jd�� |tkr�| j|jd�� d S )Nr� r� r� s bbr� ) r! r� r� r r� r� r� r+ r\ r� � ClearField)r& r' r~ r r r �testOneofWhichOneof� s z MessageTest.testOneofWhichOneofc C sV |j � }d|_|jd� |tkr0| j|jd�� | j|jd�� | jd |jd�� d S )Nr� r� r� )r! r� r� r r� r� r� r� )r& r' r~ r r r �testOneofClearField� s z MessageTest.testOneofClearFieldc C sV |j � }d|_|jd� |tkr0| j|jd�� | j|jd�� | jd |jd�� d S )Nr� r� r� )r! r� r� r r� r� r� r� )r& r' r~ r r r �testOneofClearSetField� s z"MessageTest.testOneofClearSetFieldc C sp |j � }d|_| j|d� |jd� | jd|j� |tkrJ| j|jd�� | j|jd�� | jd|jd�� d S )Nr� r� r� r� ) r! r� r� r� r+ r r\ r� r� )r& r' r~ r r r �testOneofClearUnsetField� s z$MessageTest.testOneofClearUnsetFieldc C s: |j � }d|_|j � }|j|j� � | jd|jd�� d S )Nr� r� r� )r! r� r* r, r+ r� )r& r' r~ r� r r r �testOneofDeserialize s z MessageTest.testOneofDeserializec C s6 |j � }d|_|j � }|j|� | jd|jd�� d S )Nr� r� r� )r! r� �CopyFromr+ r� )r& r' r~ r� r r r �testOneofCopyFrom s z MessageTest.testOneofCopyFromc C sb |j � }d|j_|j � }d|j_d|jj_|j|� | jd|jjd�� | jd|jjjd�� d S )Nr� s bbr� r� r� )r� r� r� r� �child� MergeFromr+ r� )r& r' r~ r� r r r �testOneofNestedMergeFrom s z$MessageTest.testOneofNestedMergeFromc C sf |j � }d|jj_d|jjj_|j � }d|j_|j|� | jd|jjd�� | jd|jjjd�� d S )Nr� � r� r� r� ) r� r� r� r| r� r� r� r+ r� )r& r' r~ r� r r r �testOneofMessageMergeFrom s z%MessageTest.testOneofMessageMergeFromc C s( |j |j j� d�}| jd|jd�� d S )N)r� r� r� )r! � NestedMessager+ r� )r& r' r~ r r r �testOneofNestedMessageInit$ s z&MessageTest.testOneofNestedMessageInitc C sB |j � }d|_|j� | j|jd�� d|_| jd|jd�� d S )Nr� r� s bbr� )r! r� r� � assertIsNoner� r� r+ )r& r' r~ r r r �testOneofClear) s zMessageTest.testOneofClearc C s& |j � }td�|_| j|jtj� dS )zmAssigning a byte string to a string field should result in the value being converted to a Unicode string.ry N)r! r$ �optional_stringr� �six� text_type)r& r' r~ r r r �"testAssignByteStringToUnicodeField1 s z.MessageTest.testAssignByteStringToUnicodeFieldc C s� |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. r8 r ru N) r! r� r� �intrC r+ r� r� r| )r& r' r~ Zslr r r �testLongValuedSlice8 s z MessageTest.testLongValuedSlicec C sl |j � }| jt�� }|jjdd� td�D �� W dQ R X | jt�� }|jjdd� td�D �� W dQ R X dS )z5This didn't use to work in the v2 C++ implementation.c s s | ] }t V qd S )N)r� )r} �ir r r � <genexpr>M s zCMessageTest.testExtendShouldNotSwallowExceptions.<locals>.<genexpr>r� Nc s s | ] }t V qd S )N)r� )r} r� r r r r� P s )r! r � NameErrorr� rS �range�repeated_nested_enum)r& r' r~ r� r r r �$testExtendShouldNotSwallowExceptionsI s $ z0MessageTest.testExtendShouldNotSwallowExceptionsNFr g � ry c C s^ |j � }| jg |j� x(tjD ] }|jj|� | jg |j� q W |jjg � | jg |j� dS )z,Test no-ops extending repeated int32 fields.N)r! �assertSequenceEqualr� r � FALSY_VALUESrS )r& r' r~ � falsy_valuer r r �testExtendInt32WithNothingT s z&MessageTest.testExtendInt32WithNothingc C s^ |j � }| jg |j� x(tjD ] }|jj|� | jg |j� q W |jjg � | jg |j� dS )z,Test no-ops extending repeated float fields.N)r! r� r_ r r� rS )r& r' r~ r� r r r �testExtendFloatWithNothinga s z&MessageTest.testExtendFloatWithNothingc C s^ |j � }| jg |j� x(tjD ] }|jj|� | jg |j� q W |jjg � | jg |j� dS )z-Test no-ops extending repeated string fields.N)r! r� rR r r� rS )r& r' r~ r� r r r �testExtendStringWithNothingn s z'MessageTest.testExtendStringWithNothingc C s� |j � }| jg |j� |jjdg� | jdg|j� |jjddg� | jdddg|j� |jjddg� | jdddddg|j� dS )z7Test extending repeated int32 fields with python lists.r r8 rj ru r9 N)r! r� r� rS )r& r' r~ r r r � testExtendInt32WithPythonList{ s z)MessageTest.testExtendInt32WithPythonListc C s� |j � }| jg |j� |jjdg� | jdg|j� |jjddg� | jdddg|j� |jjddg� | jdddddg|j� dS )z7Test extending repeated float fields with python lists.g g �?g @g @g @N)r! r� r_ rS )r& r' r~ r r r � testExtendFloatWithPythonList� s z)MessageTest.testExtendFloatWithPythonListc C s� |j � }| jg |j� |jjdg� | jdg|j� |jjddg� | jdddg|j� |jjddg� | jdddddg|j� dS )z8Test extending repeated string fields with python lists.ry Z11Z22Z33Z44N)r! r� rR rS )r& r' r~ r r r � testExtendStringWithPythonList� s z*MessageTest.testExtendStringWithPythonListc C s: |j � }| jg |j� |jjd� | jdddg|j� dS )zDTest extending repeated string fields with characters from a string.�abcr� r� r� N)r! r� rR rS )r& r' r~ r r r �testExtendStringWithString� s z&MessageTest.testExtendStringWithStringc @ s2 e Zd ZdZd dd�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. Nc C s |pg | _ d S )N)�_list)r& �valuesr r r �__init__� s z!MessageTest.TestIterable.__init__c C s8 t | j�}|dkrdS |dkr,t| jd �S td��d S )Nr Fr8 zTruth value is ambiguous.)rC r� �boolr� )r& �sizer r r rP � s z!MessageTest.TestIterable.__bool__c C s t | j�S )N)rC r� )r& r r r �__len__� s z MessageTest.TestIterable.__len__c C s | j j� S )N)r� �__iter__)r& r r r r� � s z!MessageTest.TestIterable.__iter__)N)rL rM rN �__doc__r� rP r� r� r r r r � TestIterable� s r� c C s� |j � }| jg |j� |jjtjg �� | jg |j� |jjtjdg�� | jdg|j� |jjtjddg�� | jdddg|j� |jjtjddg�� | jdddddg|j� dS )z3Test extending repeated int32 fields with iterable.r r8 rj ru r9 N)r! r� r� rS r r� )r& r' r~ r r r �testExtendInt32WithIterable� s z'MessageTest.testExtendInt32WithIterablec C s� |j � }| jg |j� |jjtjg �� | jg |j� |jjtjdg�� | jdg|j� |jjtjddg�� | jdddg|j� |jjtjddg�� | jdddddg|j� dS )z3Test extending repeated float fields with iterable.g g �?g @g @g @N)r! r� r_ rS r r� )r& r' r~ r r r �testExtendFloatWithIterable� s z'MessageTest.testExtendFloatWithIterablec C s� |j � }| jg |j� |jjtjg �� | jg |j� |jjtjdg�� | jdg|j� |jjtjddg�� | jdddg|j� |jjtjddg�� | jdddddg|j� dS )z4Test extending repeated string fields with iterable.ry �1�2�3�4N)r! r� rR rS r r� )r& r' r~ r r r � testExtendStringWithIterable� s z(MessageTest.testExtendStringWithIterablec C sP t j� dkst j� dkr d S |j� }| jtj��}tj|jtj � W d Q R X d S )N�cpprj ) r r ZVersionr! r rV Z PickleErrorrW r� ZHIGHEST_PROTOCOL)r& r' r~ r� r r r �!testPickleRepeatedScalarContainer� s z-MessageTest.testPickleRepeatedScalarContainerc C s |j � }|jj� dS )z?Exercise a scenario that has led to segfaults in the past. N)r! r� r� )r& r' r~ r r r �'testSortEmptyRepeatedCompositeContainer� s z3MessageTest.testSortEmptyRepeatedCompositeContainerc C s, |j � }| jt��}|jd� W dQ R X dS )zBUsing HasField on a repeated field should raise an exception. r� N)r! r r� r� )r& r' r~ r� r r r �testHasFieldOnRepeatedField� s z'MessageTest.testHasFieldOnRepeatedFieldc C s� |j � }| jt��}|jj� W d Q R X |jjttd��� | jd|jj� � | jd|jjd�� | jd|jjd�� | jddg|j� d S )Nr� r9 r rj r8 ru ) r! r � IndexErrorr� �poprS r� r� r+ )r& r' r~ r� r r r �testRepeatedScalarFieldPop� s z&MessageTest.testRepeatedScalarFieldPopc C s� |j � }| jt��}|jj� W d Q R X | jt��}|jjd� W d Q R X x td�D ]}|jj� }||_qTW | j d|jj� j� | j d|jjd�j� | j d|jjd�j� | j ddgdd � |jD �� d S ) Nr7 r� r9 r rj r8 ru c S s g | ] }|j �qS r )r| )r} r� r r r r s z=MessageTest.testRepeatedCompositeFieldPop.<locals>.<listcomp>) r! r r r� r r= r� r� r| r+ )r& r' r~ r� r� r� r r r � testRepeatedCompositeFieldPop s z)MessageTest.testRepeatedCompositeFieldPopc C s\ |j � }x.td�D ]"}|jj||� |jj� }||_qW | j|j|j� | j|j|j� d S )Nr� ) r! r� r� �insertr� r� r| r� r+ )r& r' r~ r� r� r r r �testRepeatedCompareWithSelf s z'MessageTest.testRepeatedCompareWithSelfc C s6 |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. r N)r� Zrepeated_childr� r� r+ r� �optional_int32)r& r' r~ r r r �testReleasedNestedMessages s z&MessageTest.testReleasedNestedMessagesc C sz |j � }| jt�� g |_W d Q R X |jjd� tj� dkrZ| jt�� g |_W d Q R X n | jt�� g |_W d Q R X d S )Nr8 r )r! r �AttributeErrorr� r� r r r= )r& r' r~ r r r �testSetRepeatedComposite( s z$MessageTest.testSetRepeatedComposite)HrL rM rN r) r2 r6 rJ rU r[ ra rb rd rg rh ri rr rs rt rx 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� r� r� r� � bytearray�setr� r� r� r� r� r� r� r� rT r� r� r� r r r r r r r r r r r r r r e s� $ --( # # r c @ sT e Zd Zdd� Zdd� Zdd� Zdd� Zd d � Zd d � Zd d� Z dd� Z dd� Z dS )� Proto2Testc C s� t j� }| j|jd�� | j|jd�� | j|jd�� | jt�� |jd� W d Q R X | jt�� |jd� W d Q R X | jt�� |jd� W d Q R X | 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�� |j d� |j d� |j d� | j|jd�� | j|jd�� | j|jd�� | jd|j� | jd|j� | jd|j j � d S ) Nr � optional_bool�optional_nested_message�field_doesnt_existr� r� r Fr� T� ) r r! r� r� r r� r+ r r r r| r\ r� )r& r r r r �testFieldPresence: sD z Proto2Test.testFieldPresencec C s t j� }| jt�� }d|_W dQ R X | jt|jjd� |jjd� d|jd< | jt�� d|jd< W dQ R X tj� }d|_|jjd� |j� }t j� }|j |� | j |j d�� | j d|j� | j dt |j�� |j� |j |j� � | j d|j� | j d|jd � dS ) z:Assigning an invalid enum number is not allowed in proto2.i�� Nrj r i@� i��t �optional_nested_enumr8 )r r! r r� r r� r� r r, r* r� r� r+ rC r� )r& r~ r� r� rc Zm3r r r �testAssignInvalidEnumj s* z Proto2Test.testAssignInvalidEnumc C s6 t j� }d|jd< | jt�� d|jd< W d Q R X d S )Nr �{ r8 )r Z TestEnumMapZknown_map_fieldr r� Zunknown_map_field)r& r~ r r r �testUnknownEnumMap� s z Proto2Test.testUnknownEnumMapc C s t j� }| jtt|d� d S )N� Extensions)r r! r r r� )r& rH r r r �testExtensionsErrors� s z Proto2Test.testExtensionsErrorsc C sh t jd�}tj� }|j|� tj� }t j|� | j||� | j||j� � tj |�}| j||j� � d S )Nr0 ) r r r ZTestAllExtensionsr* ZSetAllExtensionsr+ r, r- r. )r& r/ r0 r5 r1 r r r �testGoldenExtensions� s z Proto2Test.testGoldenExtensionsc C sh t jd�}tj� }|j|� tj� }t j|� | j||� | j||j� � tj |�}| j||j� � d S )Nr3 ) r r r ZTestPackedExtensionsr* ZSetAllPackedExtensionsr+ r, r- r. )r& r/ r0 r5 r1 r r r �testGoldenPackedExtensions� s z%Proto2Test.testGoldenPackedExtensionsc C sN t jdd�}tj|�}tj|�}| j||� | j|jd� | jtj |j � d S )Nr8 )r� ) r Z TestRequiredrV rW rX r+ r� r r Z EncodeErrorr, )r& r0 rY rZ r r r �testPickleIncompleteProto� s z$Proto2Test.testPickleIncompleteProtoc C s� t j� t j� t j� g}d|d _d|d _d|d _d|d _t j� }d|_d|_d|_t jj� }|jj|� |j j|� |j j|� |j j|� |j j|� |j j� jj|d � |j j� jj|d � |j j� jj|d � |jj� jj|d � |jj� jj|d � |jj� jj|d � |j� }t j� }|j|� | j|j|� | j|j|� | j|jj|� | j|jt jj |� | jt|j�d� | jt|j �d� | jt|jt jj �d� dS )zcCheck the merge behavior when a required or optional field appears multiple times in the input.r8 r rj ru ZhelloN) r r! r �optional_int64r� ZTestParsingMergeZRepeatedFieldsGeneratorZfield1rS Zfield2Zfield3Zext1Zext2Zgroup1r� r� Zgroup2r, r* r+ Zrequired_all_typesZoptional_all_types� optionalgroupZoptional_group_all_typesr Z optional_extrC Zrepeated_all_types� repeatedgroupZ repeated_ext)r& ZmessagesZmerged_message� generator�dataZ parsing_merger r r �testParsingMerge� sJ zProto2Test.testParsingMergec C s` t jddddddiddii d dd idd igd t jjgd dd� }| j|t j� | jd|j� | jd|j� | jd|j� | jd|j� | jd|j j � | j|j t jj � | jd|j j � | j|jd�� | j|jt j� � | jt jj|j� | jdt|j�� | jd |jd j � | jd |jd j � | jdt|j�� | jt jj|jd � | jt jj|jd � | jd |j� | jd|j� | j|jd�� | jdt|j�� | jd|j � t jd d�}| jt jj|j� | j t �� t jddid� W d Q R X | j t �� t jddid� W d Q R X | j t �� t jdd� W d Q R X | j t �� t jd d � W d Q R X d S ) N�d �� g �r@� xr� i� r| i� �BAZiX i� �FOOi r ) r �optional_fixed32r] �optional_bytesr$ r �optional_foreign_messager r% r� � default_int32r� r0 rj r r8 r# �* )r ZINVALID_NESTED_FIELD� )r ZINVALID_VALUE_TYPEZ INVALID_LABEL)r� ) r r! ZBARr� r+ r r. r] r/ r$ r� r r� r| r\ r� r0 �ForeignMessager, r rC r% r� r- r1 r� r� r_ Z default_int64r r� r= )r& r r r r �testPythonicInit� sl zProto2Test.testPythonicInitN) rL rM rN r r r r r r! r" r( r5 r r r r r 8 s 0 1r c @ s e Zd Zdd� Zdd� Zdd� Zdd� Zd d � Zd d � Zd d� Z dd� Z dd� Z dd� Z dd� Z dd� Zdd� Zdd � Zd d � Zd d � Zd!d"� Zd#d$� Zd%d&� Zd'd(� Zd)d*� Zd+d,� Zd-d.� Zd/d0� Zd1d2� Zd3d4� Z d5d6� Z d7d8� Z d9d:� Z d;d<� Z d=d>� Z!d?d@� Z"dAS )B� Proto3Testc C s@ t |�}x&|D ] \}}| j||| � ||= qW | ji |� d S )N)�dictr+ )r& Zmap_iterZ dict_valuer� �vr r r �assertMapIterEquals+ s z Proto3Test.assertMapIterEqualsc 4 C s� t j� }| jt�� |jd� W d Q R X | jt�� |jd� W d Q R X | jt�� |jd� W d Q R X | jt�� |jd� W d Q R X | j|jd�� | jt�� |jd� W d Q R X | jt�� |jd� W d Q R X | jt�� |jd� W d Q R X | jd |j� | jd |j� | jd |j � | jd |j � | jd |j j � d |j _ | j |jd�� d |_d |_d|_ d|_ d|j _ |jd� |jd� |jd� |jd� |jd� | jd |j� | jd |j� | jd |j � | jd |j � | jd |j j � d S )Nr r] r� r r r r� r� r ry Fr� g�������?r� Tr )r r! r r� r� r� r+ r r] r� r r r| r\ r� )r& r r r r r 5 sL z Proto3Test.testFieldPresencec C s� t j� }d|_| jd|j� |jjd� | jd|jd � d|jd< | jd|jd � |j� }t j� }|j|� | jd|j� | jd|jd � dS )zCAssigning an unknown enum value is allowed and preserves the value.i�� i��Tr i��t N)r r! r r+ r� r� r, r* )r& r~ rc r� r r r �testAssignUnknownEnumn s z Proto3Test.testAssignUnknownEnumc C sN t j� }| jd |jk� | jd|jk� | jd|jk� | jd|jk� | jd|jk� | jd|jk� | jd|j k� | jd|j k� | jd|j k� | j d|jd � | j d|jd � | j d|jd � | j d|jd � | j d |jd � | j t|jd t�� | j d|jd � | j t|jd t�� | j d |j d � | j d |j d � | j d|j d � | 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|j k� | j|j d tj� | jt�� |j d W d Q R X | jt�� d|j k W d Q R X d S )Nr rj r� Fr� �o ix r g ry r� i����l l���� l i����l l���� l i����l l���� 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+ r\ � isinstance�floatr� r� r� r� r r= )r& rH r r r �testScalarMapDefaults� sF z Proto3Test.testScalarMapDefaultsc C s& t j� }| j|jjd�� | jd|jjdd�� | j|jjd�� d|jd< | jd|jjd�� | jd|jjd�� | jt�� |jjd� W d Q R X | j|jjd�� | jd|jjdd�� |jd }| j ||jjd�� t j � dk�r| jt�� |jjd� W d Q R X n| jd |jjd�� d S )Nr� r� r ry r ) r r< r� r= �getr+ r r= �map_int32_foreign_messager� r r )r& rH �submsgr r r � testMapGet� s" zProto3Test.testMapGetc ! C s4 t j� }| jdt|j�� | jd|jk� d|jd< d|jd< d|jd< d|jd< d|j d< d |j d < d |j d < d |j d < d|j d< d|j d< | jg |j� � | jd t|j �� | jt�� d |j d< W d Q R X | jd t|j �� | jt�� d|j d < W d Q R X |j� }t j� }|j|� | jt�� d |j d< W d Q R X | jt�� d|j d < W d Q R X | jd|jd � | jd |jd � | jd|jd � | jd |jd � | jd|j d � | jd |j d � | jd |j d � | jd |j d � | jd|j d � | jd|j d � tj� dk�r0| jdt|j�� d S )!Nr r� i� r rj �" r� g333333�?gffffff @r8 �123r� Tix r z {-123: -456}i8���i����l l���� l l���� l l i8���i����l l���� l l���� l l )r r< r+ rC r= r� r> r? r@ Zmap_int32_floatrA rC rB rD �FindInitializationErrorsr r= r, r* r� r r r$ )r& rH rc �msg2r r r � testScalarMap� sP zProto3Test.testScalarMapc C s@ t j� }d|jd< d|jd< | j|j� d� | jd|j� � d S )Nr ry r� s r )r r< r= rC r+ �ByteSizer, )r& rH r r r � testMapEntryAlwaysSerialized s z'Proto3Test.testMapEntryAlwaysSerializedc C sn t j� }d}|jd�}||j|< t|jj� �d \}}| j||� | j||� | j|tj � | j|tj � d S )Nu ሴ�utf8r ) r r< �encoderC r� �itemsr+ r� r� r� )r& rH Z unicode_objZ bytes_objr� �valuer r r � testStringUnicodeConversionInMap s z+Proto3Test.testStringUnicodeConversionInMapc C sZ t j� }| jdt|j�� | jd|jk� |jd |jjd � | jdt|j�� | jd|j� | jd |j� | jdt|j�� | jt �� |jd W d Q R X | jt �� |jd |jd< W d Q R X | jdt|j�� |j � }t j� }|j |� | jdt|j�� | jd|j� | jd |j� | jdt|j�� t j� dk�rV| jd tt|j��� d S ) Nr r� r i� rj rM i� r r i8���i8���i8���)r r< r+ rC rI r� Z get_or_creater# r r= r� r, r* r r r$ )r& rH rc rO r r r �testMessageMap& s0 zProto3Test.testMessageMapc C s� t j� }d|jd j_|jd= d|jd j_| jdt|j�� d|jd j_| jdt|j�� |j� }t j� }|j|� ddg}x|D ] }|j|= q�W d S )Nr8 rj ) r r< �map_int32_all_typesr r| r+ rC r, r* )r& rH rc rO �keysr� r r r � testNestedMessageMapItemDeleteO s z)Proto3Test.testNestedMessageMapItemDeletec C sp t j� }d|jd< |j� }d|jd< | j|j� |d � d|jd _|j� }d|jd _| j|j� |d � d S )Nr8 � � )r r< r= rQ r+ rI r� )r& rH r� r r r �testMapByteSize` s zProto3Test.testMapByteSizec C s� t j� }d|jd< d|jd< d|jd< d|jd _d |jd _t j� }d |jd< d |jd < d|jd _d|jd _|jd }|j|� | jd|jd � | jd|jd � | jd|jd � | jd |jd � | jd|jd j� | jd |jd j� | j |jd j d�� t j � dk�r(| jd|j� i }x.|jD ]$}| j ||k� |j| j||< �q4W | jdd d�|� |jd= | j d|jk� |jd = | j d |jk� t j � dk�r�| j t�� |jd= W d Q R X n | j t�� |jd= W d Q R X d S )NrL r� �N �8 r� � r� r; r� �� �7 �c �X r r� �dr )r; rb ry )r r< r= r> rI r� rf r� r+ r� r� r r r r= �KeyError)r& rH rO Z old_map_valueZas_dictr� r r r � testMergeFroml sH zProto3Test.testMergeFromc C s. t j� }| jtd�� |jd� W d Q R X d S )NzWParameter to MergeFrom\(\) must be instance of same class: expected .*TestMap got int\.r8 )r r< �assertRaisesRegexr= r� )r& rH r r r �testMergeFromBadType� s z Proto3Test.testMergeFromBadTypec C s. t j� }| jtd�� |jd� W d Q R X d S )Nz[Parameter to [A-Za-z]*From\(\) must be instance of same class: expected .*TestMap got int\.r8 )r r< ri r= r� )r& rH r r r �testCopyFromBadType� s z Proto3Test.testCopyFromBadTypec C s� t j� }td�|jtd�< td �|jtd �< td�|jtd�< td �|jtd �< |j� }t j� }|j|� | j d|jd � | j d|jd � | j d|jd � | j d|jd � d S )Ni� r rj rL r� i8���i����l l���� l l���� l l i8���i����l l���� l l���� l l ) r r< r� r= r> r? r@ r, r* r+ )r& rH rc rO r r r �testIntegerMapWithLongs� s z"Proto3Test.testIntegerMapWithLongsc C s� t j� }d|jjd< |j� }t j� }|j|� | j||� d|jjd< |j� }|j|� | j||� |jjj� |j� }|j|� | j||� d S )Ni� r i� ix )r �TestMapSubmessage�test_mapr= r, r* r+ �clear)r& rH rc rO r r r � testMapAssignmentCausesPresence� s z*Proto3Test.testMapAssignmentCausesPresencec C s� t j� }d|jjd _|j� }t j� }|j|� | j||� d|jjd _|j� }|j|� | j||� |jjd j|jjd � |j� }|j|� | j||� |jjj � |j� }|j|� | j||� d S )Nr� r � ix ) r rm rn rI r� r, r* r+ r� ro )r& rH rc rO r r r �-testMapAssignmentCausesPresenceForSubmessages� s& z8Proto3Test.testMapAssignmentCausesPresenceForSubmessagesc C s~ t j� }t|j�}t|j�}d|jd< d|jd _| jt�� x |D ]}qDW W d Q R X | jt�� x |D ]}qhW W d Q R X d S )NrM r� r� )r r< �iterrC rI r� r � RuntimeError)r& rH Zstring_string_iterZint32_foreign_iterr� r r r �testModifyMapWhileIterating� s z&Proto3Test.testModifyMapWhileIteratingc C s� t j� }|jd }| j||jd � | j|tj� d|_|j� }t j� }|j |� | j d|jd j� | j t �� tj� |jd< W d Q R X d S )Nr; r� re ) r r< rI r� r� r r4 r� r, r* r+ r r� )r& rH rJ rc rO r r r �testSubmessageMap s z Proto3Test.testSubmessageMapc C s� t j� }x$t|jj� �D ]\}}| jd� qW d|jd< d|jd< d|jd< | jdt|j�� dddd�}| jt|jj� �|� d S )NFr9 rj r� ru � )rj ru r9 ) r r< r� r= rU r\ r+ rC r9 )r& rH r� r8 � matching_dictr r r �testMapIteration s zProto3Test.testMapIterationc s& t jdk �r"tj� }d|jd< d|jd< d|jd< d|jd< |j}� jdt|�� tj� }|j|j� � � fdd �}|t |j � �t |j � �� |t |j � �t |j � �� |t |j � �t |j � �� � jd|jd�� � jd |jd �� � jd|jd�� � jd |jd�� � jdt|�� |j� \}}� jd| |� � jdt|�� |j� � jd t|�� � jt�� |j� W d Q R X � jd |jd�� � jd t|�� |j|j� � jdt|�� � jt�� |j|j|j� W d Q R X � jt�� |jd � W d Q R X � jt�� |jd d� W d Q R X d S )Nru r9 rj r� rw r� r� c s0 � j t|�| d � � j t|�| dd � � d S )Nr r8 )r+ �nextr� )�seq�iterator)r& r r � CheckItems9 s z-Proto3Test.testPython2Map.<locals>.CheckItemsi� r r8 r� )rV )ru )rv rw r r< r= r+ rC r* r, r� rU rs rZ r� rH r �popitemro r rg � setdefault�updater= )r& rH Z map_int32rO r} r� rV r )r& r �testPython2Map- sJ zProto3Test.testPython2Mapc C sf t j� }d|jd< d|jd< d|jd< d|jd< d|jd< t|jj� �}t|jj� �}| j||� d S )Nry � local_init_op�trainable_variables� variables�init_op� summaries)r r< rC r� rU r+ )r& rH Zitems1Zitems2r r r � testMapItems] s zProto3Test.testMapItemsc C s~ d}t j� }d|jd< d|jd< d|jd< d|jd < d |jd < d |jd < d |jd< d|jd< d|jd< | j||jdd�� d S )Ns� r init_opdr item1er item2fr item3gr item4QQr local_init_opar summarieser trainable_variablesbr variablescr� r� r� r� r� r� rf r� �er� Zitem1�fZitem2�gZitem3ZQQZitem4T)rK )r r< rC r+ r, )r& r/ rH r r r �!testMapDeterministicSerializationl s z,Proto3Test.testMapDeterministicSerializationc C sR t j� }d|jd< d|jd< d|jd< t|jj� �}~dddd�}| j||� d S )Nr9 rj r� ru rw )rj ru r9 )r r< r= r� rU r9 )r& rH �itrx r r r � testMapIterationClearMessage� s z'Proto3Test.testMapIterationClearMessagec C sf t jddd�d�}| jd|jd � | jd|jd � t jdtjdd�id �}| jd|jd j� d S ) Nrj r9 )r8 ru )r= r8 ru r� )r� )rI )r r< r+ r= r r4 rI r� )r& rH r r r �testMapConstruction� s z Proto3Test.testMapConstructionc C sd t j� }|j}d|d< d|d< d|d< |jd� | jd|j� � dddd�}| jt|j� �|� d S ) Nr9 rj r� ru rw r= r� )rj ru r9 ) r r< r= r� r+ r, r9 r� rU )r& rH Z int32_maprx r r r � testMapValidAfterFieldCleared� s z(Proto3Test.testMapValidAfterFieldClearedc C sL t j� }|j}d|d _|jd� | jd|j� � | jdt|j � �k� d S )Nr� rj rI r� ) r r< rI r� r� r+ r, r\ r� rZ )r& rH Zint32_foreign_messager r r �$testMessageMapValidAfterFieldCleared� s z/Proto3Test.testMessageMapValidAfterFieldClearedc C s| t j� }t|j�}|jd� | jt�� x |D ]}q.W W d Q R X t|j�}|jd� | jt�� x |D ]}qfW W d Q R X d S )Nr= rI )r r< rs r= r� r rt rI )r& rH r� r� r r r �"testMapIterInvalidatedByClearField� s z-Proto3Test.testMapIterInvalidatedByClearFieldc C s� t j� }| jdt|j�� d|jd< | jdt|j�� | jt�� |jd= W d Q R X |jd= | jdt|j�� | jt�� |jd= W d Q R X d S )Nr r� r9 r8 re � )r r< r+ rC r= r rg rY )r& rH r r r � testMapDelete� s zProto3Test.testMapDeletec C sL t j� }| j|jtj� | j|jtj� | j|jtj� | j|jtj� d S )N)r r< r� r= r� �Mapping�MutableMappingrI )r& rH r r r �testMapsAreMapping� s z Proto3Test.testMapsAreMappingc C sD t j� }d|jd< | j|j|j� | j|j|j� | j|jd� d S )Ni� r r i8���i����)r r< r= r+ rI ZassertNotEqual)r& rH r r r �testMapsCompare� s zProto3Test.testMapsComparec C s@ t j� }d|jd< d|jd< d|jd _| jdt|j� �� d S )NrM r� �@ �# r� r� r ) r r< rC r= Zmap_string_foreign_messager� r+ rC rN )r& rH r r r �(testMapFindInitializationErrorsSmokeTest� s z3Proto3Test.testMapFindInitializationErrorsSmokeTestN)#rL rM rN r9 r r: rG rK rP rR rW rX r[ r^ rh rj rk rl rp rr ru rv ry r� r� r� r� r� r� r� r� r� r� r� r� r r r r r6 ( s@ 90=) < 0 r6 c @ s e Zd Zdd� Zdd� ZdS )�ValidTypeNamesTestc s� t t|��jd�d � d| d| f}| jt� fdd�|D ��d� |f � � jd�}|d }dj|d d � �}t||gd � d S ) N�'r8 zRepeated%sContainerzRepeated%sFieldContainerc 3 s | ]}� j |�V qd S )N)�endswith)r} r8 )�tp_namer r r� s z:ValidTypeNamesTest.assertImportFromName.<locals>.<genexpr>z%r does end with any of %r�.)�fromlistr< r< )r$ �type�splitr\ �any�join� __import__)r& rH Z base_nameZ valid_names�parts� class_nameZ module_namer )r� r �assertImportFromName� s z'ValidTypeNamesTest.assertImportFromNamec C s( t j� }| j|jd� | j|jd� d S )NZScalarZ Composite)r r! r� r� r� )r& r� r r r �testTypeNamesCanBeImported s z-ValidTypeNamesTest.testTypeNamesCanBeImportedN)rL rM rN r� r� r r r r r� � s r� c @ s$ e Zd Zdd� Zdd� Zdd� ZdS )�PackedFieldTestc C s� |j jd� |jjd� |jjd� |jjd� |jjd� |jjd� |jjd� |jjd� |j jd� |j jd� |j jd� |j jd� |j jd� |jjd� d S )Nr8 g �?T)r� r� Zrepeated_int64Zrepeated_uint32Zrepeated_uint64Zrepeated_sint32Zrepeated_sint64Zrepeated_fixed32Zrepeated_fixed64Zrepeated_sfixed32Zrepeated_sfixed64r_ r` Z repeated_boolr� )r& r r r r � setMessage s zPackedFieldTest.setMessagec C s* t j� }| j|� d}| j||j� � d S )NsH "*2: B J R Z �?b �?jr)r r4 r� r+ r, )r& r r/ r r r �testPackedFields" s z PackedFieldTest.testPackedFieldsc C s* t j� }| j|� d}| j||j� � d S )Ns: (0= A M Q ] �?a �?hp)r ZTestUnpackedTypesr� r+ r, )r& r r/ r r r �testUnpackedFields5 s z"PackedFieldTest.testUnpackedFieldsN)rL rM rN r� r� r� r r r r r� s r� r rj rq z;explicit tests of the C++ implementation for PY27 and abovec @ s0 e Zd Zedd� �Zdd� Zdd� Zdd� Zd S ) �OversizeProtosTestc C sF d}t j� }tj� }tj||� |j|� tj|�j |j d��| _ d S )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) r ZDescriptorPoolr ZFileDescriptorProtor ZParseZAddr ZMessageFactoryZ GetPrototypeZFindMessageTypeByName� proto_cls)�clsZ file_descZpoolZdescr r r � setUpClassN s z OversizeProtosTest.setUpClassc C s( | j � | _dd | jj_| jj� | _d S )Nr� i r� r8 i i i )r� �p�fieldr� r, � p_serialized)r& r r r �setUpq s zOversizeProtosTest.setUpc C sf ddl m} |d� | j� }y|j| j� W n4 tjk r` } z| jt|�d� W Y d d }~X nX d S )Nr )�SetAllowOversizeProtosFzError parsing message) � google.protobuf.pyext._messager� r� r* r� r r? r+ r$ )r& r� �qr� r r r �testAssertOversizeProtov s z*OversizeProtosTest.testAssertOversizeProtoc C sB ddl m} |d� | j� }|j| j� | j| jjj|jj� d S )Nr )r� T) r� r� r� r* r� r+ r� r� r� )r& r� r� r r r �testSucceedOversizeProto s z+OversizeProtosTest.testSucceedOversizeProtoN)rL rM rN � classmethodr� r� r� r� r r r r r� I s # r� �__main__)rj rq )4r� � __author__r� r- rl r� rV r� rv r@ Z unittest2Zunittest� ImportErrorr� r� Zgoogle.protobufr r r r r r r r Zgoogle.protobuf.internalr r r r r r r ZPY3r� Zlongr r r r Z BaseTestCaseZNamedParametersr r r6 r� r� ZskipIfr rw r� rL �mainr r r r �<module>) s� W q X9:
Save File
Cancel