File: //usr/local/lib/python3.7/test/__pycache__/test_pickletools.cpython-37.opt-1.pyc
B
��g� � @ 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 �1/usr/local/lib/python3.7/test/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 �}|�|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"