Your IP : 216.73.216.196
B
� f� � @ sl d dl Z d dlZd dlmZ d dlmZ d dlZG dd� de�ZG dd� dej�Z dd � Z
ed
krhe
� dS )� N)�support)�AbstractPickleTestsc @ s6 e Zd Zd
dd�Zdd� ZdZdZdd� Zdd � ZdS )�OptimizedPickleTestsNc C s t �t�||��S )N)�pickletools�optimize�pickle�dumps)�self�arg�proto� r �:/opt/alt/python37/lib64/python3.7/test/test_pickletools.pyr s zOptimizedPickleTests.dumpsc K s t j|f|�S )N)r �loads)r Zbuf�kwdsr r r
r s zOptimizedPickleTests.loadsc C s� dd� t d�D �}|�|d � x�t tjd �D ]�}t�||�}t�|�}| �||� | �|d |d � t� |�}t�|�}| �||� | �|d |d � | �
tj|� | �
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 ���� ���)
�range�appendr ZHIGHEST_PROTOCOLr r �assertEqual�assertIsr r �assertNotInZLONG_BINGETZLONG_BINPUT)r �datar �pickled� unpickled�pickled2�
unpickled2r r r
�test_optimize_long_binget s
z.OptimizedPickleTests.test_optimize_long_bingetc C s� d}| � tj|� t�|�}| �|dddg� | �|d |d � t�|�}t�|�}| �|dddg� | �|d |d � | �tj|� d S )Ns �� ]�(�spamq�ham�he.ZspamZhamr � ) 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_2xZ+test_framed_write_sizes_with_delayed_writerr! r# r r r r
r s
r c @ s e Zd Zdd� ZdS )�MiscTestCasec K C s� ddddddddd d
ddd
ddddddddddddddddddd 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 )MNZbytes_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_pairZread_string1Zread_string4Zread_bytes1Zread_bytes4Zread_bytes8Zread_unicodestringnlZread_unicodestring1Zread_unicodestring4Zread_unicodestring8Zread_decimalnl_shortZread_decimalnl_longZread_floatnlZread_float8Z
read_long1Z
read_long4Zuint1Zuint2Zint4Zuint4Zuint8ZstringnlZstringnl_noescapeZstringnl_noescape_pairZstring1Zstring4Zbytes1Zbytes4Zbytes8ZunicodestringnlZunicodestring1Zunicodestring4Zunicodestring8Zdecimalnl_shortZdecimalnl_longZfloatnlZfloat8Zlong1Zlong4ZStackObjectZpyintZpylongZpyinteger_or_boolZpyboolZpyfloatZpybytes_or_strZpystringZpybytesZ pyunicodeZpynoneZpytupleZpylistZpydictZpysetZpyfrozensetZ anyobjectZ
markobjectZ
stacksliceZ
OpcodeInfoZopcodesZcode2op)� blacklist)r Zcheck__all__r )r r( r r r
�test__all__A s4
zMiscTestCase.test__all__N)r$ r% r&