Your IP : 216.73.217.129
3
��2az � @ sl d dl Z d dlZd dlmZ d dlmZ d dlZG dd� de�ZG dd� dej�Z dd � Z
e d
krhe
� dS ) � N)�support)�AbstractPickleTestsc @ s2 e Zd Zd
dd�Zdd� ZdZdd� Zdd � ZdS ) �OptimizedPickleTestsNc C s t jtj||��S )N)� pickletools�optimize�pickle�dumps)�self�arg�proto� r �5/opt/alt/python36/lib64/python3.6/test_pickletools.pyr s zOptimizedPickleTests.dumpsc K s t j|f|�S )N)r �loads)r �buf�kwdsr r r
r s zOptimizedPickleTests.loadsc C s� dd� t d�D �}|j|d � x�t tjd �D ]�}tj||�}tj|�}| j||� | j|d |d � tj |�}tj|�}| j||� | j|d |d
� | j
tj |� | j
tj |� q0W d S ) Nc S s g | ] }t |��qS r )�str)�.0�ir r r
�
<listcomp> s zBOptimizedPickleTests.test_optimize_long_binget.<locals>.<listcomp>i � � ���r ���r r )
�range�appendr ZHIGHEST_PROTOCOLr r � assertEqual�assertIsr r � assertNotInZ LONG_BINGETZ LONG_BINPUT)r �datar �pickled� unpickled�pickled2�
unpickled2r r r
�test_optimize_long_binget s
z.OptimizedPickleTests.test_optimize_long_bingetc C s� d}| j tj|� tj|�}| j|dddg� | j|d |d � tj|�}tj|�}| j|dddg� | j|d |d � | jtj|� d S )Ns �� ]�(�spamq�ham�he.ZspamZhamr r ) ZassertInr ZBINPUTr r r r r r )r r r r! r" r r r
� test_optimize_binput_and_memoize" s
z5OptimizedPickleTests.test_optimize_binput_and_memoize)N)�__name__�
__module__� __qualname__r r Ztest_pickle_to_2xr# r$ r r r r
r s
r c @ s e Zd Zdd� ZdS )� MiscTestCasec K C s� ddddddddd d
d d d
ddddddddddddddd d d d d d!d"d#d$d%d&d'd(d)d*d+d,d-d.d/d0d1d2d3d4d5d6d7d8d9d:d;d<d=d>d?d@dAdBdCdDdEdFdGdHdIdJdKhK}t j| t|dL� d S )MNZ bytes_typesZ
UP_TO_NEWLINEZTAKEN_FROM_ARGUMENT1ZTAKEN_FROM_ARGUMENT4ZTAKEN_FROM_ARGUMENT4UZTAKEN_FROM_ARGUMENT8UZArgumentDescriptorZ
read_uint1Z
read_uint2Z read_int4Z
read_uint4Z
read_uint8Z
read_stringnlZread_stringnl_noescapeZread_stringnl_noescape_pairZ read_string1Z read_string4Z read_bytes1Z read_bytes4Z read_bytes8Zread_unicodestringnlZread_unicodestring1Zread_unicodestring4Zread_unicodestring8Zread_decimalnl_shortZread_decimalnl_longZ read_floatnlZ read_float8Z
read_long1Z
read_long4Zuint1Zuint2Zint4Zuint4Zuint8ZstringnlZstringnl_noescapeZstringnl_noescape_pairZstring1Zstring4Zbytes1Zbytes4Zbytes8ZunicodestringnlZunicodestring1Zunicodestring4Zunicodestring8Zdecimalnl_shortZdecimalnl_longZfloatnlZfloat8Zlong1Zlong4Z StackObjectZpyintZpylongZpyinteger_or_boolZpyboolZpyfloatZpybytes_or_strZpystringZpybytesZ pyunicodeZpynoneZpytupleZpylistZpydictZpysetZ pyfrozensetZ anyobjectZ
markobjectZ
stacksliceZ
OpcodeInfoZopcodesZcode2op)� blacklist)r Z check__all__r )r r) r r r
� test__all__> s4
zMiscTestCase.test__all__N)r% r& r' r* r r r r
r( = s r( c C s" t jt� t jt� t jt� d S )N)r Z run_unittestr r( Z run_doctestr r r r r
� test_main\ s
r+ �__main__) r r Ztestr Ztest.pickletesterr Zunittestr ZTestCaser( r+ r% r r r r
�<module> s 6