← Back
Editing: wire_format_test.cpython-36.pyc
3 ���h�* � @ sp d Z dZy ddlZW n ek r0 ddlZY nX ddlmZ ddlmZ G dd� dej �Z e dkrlej � dS ) z.Test for google.protobuf.internal.wire_format.z#robinson@google.com (Will Robinson)� N)�message)� wire_formatc @ s< e Zd Zdd� Zdd� Zdd� Zdd� Zd d � Zd d � Zd S )�WireFormatTestc C sR d}d}| j |d> |B tj||�� tj}| jtj||d� | jtj||d� d S )Ni� � � � � ���)� assertEqualr �PackTag� assertRaisesr � EncodeError)�self� field_numberZtag_typer � r �&/usr/lib/python3.6/wire_format_test.py� testPackTag/ s zWireFormatTest.testPackTagc C s� xLd D ]D}x>t d�D ]2}tjtj||��\}}| j||� | j||� qW qW | jttjd � | jttjd� | jttjd� | jttjt� � d S ) Nr � � � � r �abcg )r r r r r )�ranger Z UnpackTagr r r � TypeError�object)r Zexpected_field_numberZexpected_wire_typer Z wire_typer r r � testUnpackTag: s z WireFormatTest.testUnpackTagc C s� t j}| jd|d�� | jd|d�� | jd|d�� | jd|d�� | jd|d�� | jd|d�� | jd|d�� | jd |d �� | jd |d�� | jt|d � | jt|d� | jt|d� | jt|t� � d S )Nr r r r � l �� i���l �� l l ���� l ���� l ���� l �abcdg r ���i �l���� )r Z ZigZagEncoder r r r )r �Zr r r �testZigZagEncodeH s z WireFormatTest.testZigZagEncodec C s� t j}| jd|d�� | jd|d�� | jd|d�� | jd|d�� | jd|d�� | jd|d�� | jd|d �� | jd |d �� | jd|d �� | jt|d � | jt|d� | jt|d� | jt|t� � d S )Nr r r r r i���l �� l l �� l ���� l ���� l l ���� r g r r i �l���� )r Z ZigZagDecoder r r r )r r r r r �testZigZagDecodeY s z WireFormatTest.testZigZagDecodec C sD x>d D ]6\}}|| }|||�}| j ||d|||||f � qW d S )Nr r r r � � r zFbyte_size_fn: %s, field_number: %d, value: %r Expected: %d, Actual: %d�r r �r r �r" r �r# r )r$ r% r& r' )r )r � byte_size_fn�valueZexpected_value_sizer Z tag_bytesZ expected_sizeZ actual_sizer r r �NumericByteSizeTestHelperj s z(WireFormatTest.NumericByteSizeTestHelperc 6 C s� t jddgt jddgt jddgt jd dgt jddgt jddgt jddgt jd dgt jddgt jddgt jddgt jt jdgt jddgt jddgt jddgt jt jdgt jddgt jd dgt jddgt jd dgt jddgt jd dgt jd dgt jddgt jd!dgt jddgt jd"dgt jddgt jd#dgt jd dgt j dd gt j t jd gt j dd gt j t jd gt j dd gt j t j d gt j t j d gt jdd gt jt jd gt jt jd gt jd d gt jd d gt jd$d gt jd d gt jd d gt jd%d gt jddgt jddgt jddgt jddgt jddgt jt jdgg4}x|D ]}| j|� �q$W xPt jt jfD ]@}| jd|dd�� | jd|dd�� | jd|ddd �� �qFW | jdt jddjd��� G dd� dt�}d}||d�}| jd| t jd|�� | jd | t jd|�� | jd|j t j d|�� | jd|j t j d|�� d|_ | jd |j t j d|�� d|_ | j|j d t j d|�� d|_ | j|j d t j d|�� | j|j d t j d|�� | j t j!t jddd> � d S )&Nr r � � r � � �? �@ r � g g e��AFTr r r � �as Тестzutf-8c @ s e Zd Zdd� Zdd� ZdS )z9WireFormatTest.testByteSizeFunctions.<locals>.MockMessagec S s || _ d S )N)� byte_size)r r4 r r r �__init__� s zBWireFormatTest.testByteSizeFunctions.<locals>.MockMessage.__init__c S s | j S )N)r4 )r r r r �ByteSize� s zBWireFormatTest.testByteSizeFunctions.<locals>.MockMessage.ByteSizeN)�__name__� __module__� __qualname__r5 r6 r r r r � MockMessage� s r: )r4 r � r r r i����i����r i����i����g e��g e��)"r Z Int32ByteSizeZ Int64ByteSizeZUInt32ByteSizeZ UINT32_MAXZUInt64ByteSizeZ UINT64_MAXZSInt32ByteSizeZSInt64ByteSizeZFixed32ByteSizeZFixed64ByteSizeZSFixed32ByteSizeZ INT32_MINZ INT32_MAXZSFixed64ByteSizeZ INT64_MINZ INT64_MAXZ FloatByteSizeZDoubleByteSizeZ BoolByteSizeZ EnumByteSizer* ZStringByteSizeZ BytesByteSizer �decoder Z GroupByteSizer4 ZMessageByteSizeZMessageSetItemByteSizer r r )r Z NUMERIC_ARGS�argsr( r: Zmessage_byte_sizeZ mock_messager r r �testByteSizeFunctionst s� z$WireFormatTest.testByteSizeFunctionsN) r7 r8 r9 r r r r! r* r> r r r r r - s r �__main__) �__doc__� __author__Z unittest2Zunittest� ImportErrorZgoogle.protobufr Zgoogle.protobuf.internalr ZTestCaser r7 �mainr r r r �<module> s S
Save File
Cancel