File: //usr/local/lib/python3.7/test/__pycache__/test_pickle.cpython-37.pyc
B
��gJJ � @ s� d dl mZmZmZmZ d dlZd dlZd dlZd dlZd dl Z d dl
Z
d dl Z d dl Z d dl
mZ d dlmZ d dlmZ d dlmZ d dlmZ d dlmZ d d lmZ d d
lmZ d d lmZ yd dlZd ZW n ek
r� d
ZY nX G dd� de�ZG dd� de�Z G dd� de�Z G dd� deee�Z G dd� de �Z G dd� dee �Z!G dd� dee �Z"G d d � d e�Z#G d d � d e�Z$G d d!� d!e�Z%e�rTG 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*G d,d-� d-e �Z+G d.d/� d/e �Z,G d0d1� d1e�Z-G d2d3� d3e�Z.G d4d5� d5e�Z/ej0G d6d7� d7e j1��Z2d8d9d:d;hZ3d<d=d>d?hZ4d@dA� Z5dBdC� Z6dDdE� Z7dFdG� Z8dHdI� Z9G dJdK� dKe j1�Z:dLdM� Z;e<dNk�r�e;� dS )O� )�IMPORT_MAPPING�REVERSE_IMPORT_MAPPING� NAME_MAPPING�REVERSE_NAME_MAPPINGN)�support)�AbstractUnpickleTests)�AbstractPickleTests)�AbstractPickleModuleTests)� AbstractPersistentPicklerTests)�&AbstractIdentityPersistentPicklerTests)�#AbstractPicklerUnpicklerObjectTests)�AbstractDispatchTableTests)�BigmemPickleTestsTFc @ s@ e Zd Zeej�Zeej�Zeej �Z
eej �Z ej
ZejZdS )�
PyPickleTestsN)�__name__�
__module__� __qualname__� staticmethod�pickleZ_dump�dumpZ_dumps�dumps�_load�loadZ_loads�loads�_Pickler�Pickler�
_Unpickler� Unpickler� r r �,/usr/local/lib/python3.7/test/test_pickle.pyr s
r c @ s6 e Zd ZejZefZeje e
e e j
eefZdd� ZdS )�PyUnpicklerTestsc K s t �|�}| j|f|�}|�� S )N)�io�BytesIO� unpicklerr )�self�buf�kwds�f�ur r r r / s
zPyUnpicklerTests.loadsN)r r r r r r# �
IndexError�bad_stack_errors�UnpicklingError�EOFError�AttributeError�
ValueError�struct�error� ImportError�truncated_errorsr r r r r r ' s r c @ s* e Zd ZejZejZddd�Zdd� Z dS )�PyPicklerTestsNc C s4 t �� }| �||�}|�|� |�d� t|�� �S )Nr )r! r" �picklerr �seek�bytes�read)r$ �arg�protor' �pr r r r : s
zPyPicklerTests.dumpsc K s t �|�}| j|f|�}|�� S )N)r! r" r# r )r$ r% r& r' r( r r r r A s
zPyPicklerTests.loads)N)
r r r r r r4 r r# r r r r r r r3 5 s
r3 c @ sB e Zd ZejefZejeee e
j ee fZ
ddd�Zdd� ZdZdS )�InMemoryPickleTestsNc C s t �||�S )N)r r )r$ r8 Zprotocolr r r r O s zInMemoryPickleTests.dumpsc K s t j|f|�S )N)r r )r$ r% r& r r r r R s zInMemoryPickleTests.loads)N)r r r r r+ r) r* r, r- r. r/ r0 r1 r2 r r Z+test_framed_write_sizes_with_delayed_writerr r r r r; G s
r; c @ s e Zd Zddd�Zdd� ZdS )� PersistentPicklerUnpicklerMixinNc s: G � fdd�d� j �}t�� }|||�}|�|� |�� S )Nc s e Zd Z� fdd�ZdS )z:PersistentPicklerUnpicklerMixin.dumps.<locals>.PersPicklerc s
� � |�S )N)�
persistent_id)�subself�obj)r$ r r r= \ s zHPersistentPicklerUnpicklerMixin.dumps.<locals>.PersPickler.persistent_idN)r r r r= r )r$ r r � PersPickler[ s r@ )r4 r! r" r �getvalue)r$ r8 r9 r@ r' r: r )r$ r r Z s
z%PersistentPicklerUnpicklerMixin.dumpsc s4 G � fdd�d� j �}t�|�}||f|�}|�� S )Nc s e Zd Z� fdd�ZdS )z<PersistentPicklerUnpicklerMixin.loads.<locals>.PersUnpicklerc s
� � |�S )N)�persistent_load)r> r? )r$ r r rB e s zLPersistentPicklerUnpicklerMixin.loads.<locals>.PersUnpickler.persistent_loadN)r r r rB r )r$ r r �
PersUnpicklerd s rC )r# r! r" r )r$ r% r& rC r' r( r )r$ r r c s
z%PersistentPicklerUnpicklerMixin.loads)N)r r r r r r r r r r< X s
r< c @ s e Zd ZejZejZdS )�PyPersPicklerTestsN)r r r r r r4 r r# r r r r rD l s rD c @ s4 e Zd ZejZejZej dd� �Z
ej dd� �Z dS )�PyIdPersPicklerTestsc s^ � fdd�}G dd� d� j �}||� G dd� d� j �}||� G dd� d� j �}||� d S )Nc s� xHt tjd �D ]6}t�� }| ||�}|�d� � �� �|�� �d� qW | t�� �}� �|� d�d� t
� |�}~� � |� � d S )N� �abc�def)
�ranger �HIGHEST_PROTOCOLr! r" r � assertEqualr rA r= �weakref�ref� assertIsNone)r r9 r' r4 �r)r$ r r �check{ s
z@PyIdPersPicklerTests.test_pickler_reference_cycle.<locals>.checkc @ s e Zd Zdd� ZdS )zFPyIdPersPicklerTests.test_pickler_reference_cycle.<locals>.PersPicklerc S s |S )Nr )r> r? r r r r= � s zTPyIdPersPicklerTests.test_pickler_reference_cycle.<locals>.PersPickler.persistent_idN)r r r r= r r r r r@ � s r@ c @ s e Zd Zedd� �ZdS )zFPyIdPersPicklerTests.test_pickler_reference_cycle.<locals>.PersPicklerc S s |S )Nr )�clsr? r r r r= � s zTPyIdPersPicklerTests.test_pickler_reference_cycle.<locals>.PersPickler.persistent_idN)r r r � classmethodr= r r r r r@ � s c @ s e Zd Zedd� �ZdS )zFPyIdPersPicklerTests.test_pickler_reference_cycle.<locals>.PersPicklerc S s | S )Nr )r? r r r r= � s zTPyIdPersPicklerTests.test_pickler_reference_cycle.<locals>.PersPickler.persistent_idN)r r r r r= r r r r r@ � s )r4 )r$ rP r@ r )r$ r � test_pickler_reference_cycley s z1PyIdPersPicklerTests.test_pickler_reference_cyclec s^ � fdd�}G dd� d� j �}||� G dd� d� j �}||� G dd� d� j �}||� d S )Nc sx x<t tjd �D ]*}| t�� �d|���}� �|�� d� qW | t�� �}� �|�d�d� t �
|�}~� � |� � d S )NrF rG rH ) rI r rJ r! r" r rK r rB rL rM rN )r r9 r# rO )r$ r r rP � s
zBPyIdPersPicklerTests.test_unpickler_reference_cycle.<locals>.checkc @ s e Zd Zdd� ZdS )zJPyIdPersPicklerTests.test_unpickler_reference_cycle.<locals>.PersUnpicklerc S s |S )Nr )r> �pidr r r rB � s zZPyIdPersPicklerTests.test_unpickler_reference_cycle.<locals>.PersUnpickler.persistent_loadN)r r r rB r r r r rC � s rC c @ s e Zd Zedd� �ZdS )zJPyIdPersPicklerTests.test_unpickler_reference_cycle.<locals>.PersUnpicklerc S s |S )Nr )rQ rT r r r rB � s zZPyIdPersPicklerTests.test_unpickler_reference_cycle.<locals>.PersUnpickler.persistent_loadN)r r r rR rB r r r r rC � s c @ s e Zd Zedd� �ZdS )zJPyIdPersPicklerTests.test_unpickler_reference_cycle.<locals>.PersUnpicklerc S s | S )Nr )rT r r r rB � s zZPyIdPersPicklerTests.test_unpickler_reference_cycle.<locals>.PersUnpickler.persistent_loadN)r r r r rB r r r r rC � s )r# )r$ rP rC r )r$ r � test_unpickler_reference_cycle� s
z3PyIdPersPicklerTests.test_unpickler_reference_cycleN) r r r r r r4 r r# r � cpython_onlyrS rU r r r r rE s s rE c @ s e Zd ZejZejZdS )� PyPicklerUnpicklerObjectTestsN)r r r r r �
pickler_classr �unpickler_classr r r r rW � s rW c @ s e Zd ZejZdd� ZdS )�PyDispatchTableTestsc C s
t j�� S )N)r �dispatch_table�copy)r$ r r r �get_dispatch_table� s z'PyDispatchTableTests.get_dispatch_tableN)r r r r r rX r] r r r r rZ � s rZ c @ s e Zd ZejZdd� ZdS )�PyChainDispatchTableTestsc C s t �i tj�S )N)� collections�ChainMapr r[ )r$ r r r r] � s z,PyChainDispatchTableTests.get_dispatch_tableN)r r r r r rX r] r r r r r^ � s r^ c @ s, e Zd ZddlmZmZmZmZmZm Z dS )� CPickleTestsr )r r r r r r N)
r r r �_pickler r r r r r r r r r ra � s ra c @ s" e Zd ZejZejfZejfZ dS )�CUnpicklerTestsN)
r r r rb r r# r r+ r* r2 r r r r rc � s rc c @ s e Zd ZejZejZdS )�
CPicklerTestsN)r r r rb r r4 r r# r r r r rd � s rd c @ s e Zd ZejZejZdS )�CPersPicklerTestsN)r r r rb r r4 r r# r r r r re � s re c @ s e Zd ZejZejZdS )�CIdPersPicklerTestsN)r r r rb r r4 r r# r r r r rf � s rf c @ s e Zd ZejZejZdS )�CDumpPickle_LoadPickleN) r r r rb r r4 r r r# r r r r rg � s rg c @ s e Zd ZejZejZdS )�DumpPickle_CLoadPickleN) r r r r r r4 rb r r# r r r r rh � s rh c @ s e Zd ZejZejZdd� ZdS )� CPicklerUnpicklerObjectTestsc C sX | � t�� �}| �t�� t|_W d Q R X | �t�� dd i|_W d Q R X dd i|_d S )N���rF )rY r! r" Z assertRaises� TypeError�objectZmemor. )r$ r# r r r �test_issue18339� s z,CPicklerUnpicklerObjectTests.test_issue18339N) r r r rb r rX r rY rm r r r r ri � s ri c @ s e Zd ZejZdd� ZdS )�CDispatchTableTestsc C s
t j�� S )N)r r[ r\ )r$ r r r r] � s z&CDispatchTableTests.get_dispatch_tableN)r r r r r rX r] r r r r rn � s rn c @ s e Zd ZejZdd� ZdS )�CChainDispatchTableTestsc C s t �i tj�S )N)r_ r` r r[ )r$ r r r r] � s z+CChainDispatchTableTests.get_dispatch_tableN)r r r r r rX r] r r r r ro � s ro c @ s" e Zd ZejZdd� Zdd� ZdS )� SizeofTestsc C s� t �d�}t�t�� �}| �t�|�|� t �
d�}t �
d�}| j }|||| d| t �
d� � x td�D ]}|�t|�� qnW |||| d| d � d S ) NZ 6P2n3i2n3iPZ3nP0nZPn0P� s xxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxx� � r )r � calcobjsizerb r r! r" rK rl �
__sizeof__r/ �calcsize� check_sizeof�sys� getsizeofrI r �chr)r$ �basesizer: ZMT_sizeZME_sizerP �ir r r � test_pickler s
zSizeofTests.test_picklerc s� t �d�}tj�t�d�� t�d��| j�xhdD ]`}xZdD ]R}�t�� ||d�}| � t
� |�|� �||d� t |� d t |� d � q:W q0W |t d � d t d
� d �� ����fd d �}|d
dd
� |d
gd dd� |dd� t
d�D �dd� dd� }||d
�dd
� ||d�dd� ||d�dd� ||d�dd� ||d�dd� �t�t�dd
��d d
d�}|�� �|�d� d d � d S )Nz2P2n2P 2P2n2i5P 2P3n6P2n2i�P�n)�ASCIIzUTF-16zlatin-1)�strict�replace)�encoding�errorsrs rF r� r� c sD t �| �}�t�|�ddd�}|�� �|�|� |� � d S )Nr� r� )r� r� )r r r! r" r )�dataZ memo_sizeZ
marks_sizer r( )r~ rP r �stdsizer# r r �check_unpickler% s
z3SizeofTests.test_unpickler.<locals>.check_unpicklerr �d � c S s g | ] }t |��qS r )rz )�.0r| r r r �
<listcomp>0 s z.SizeofTests.test_unpickler.<locals>.<listcomp>� c S s" d}xt | �D ] }||g}qW |S )Nr )rI )Zdeepr� r| r r r �recurse1 s z+SizeofTests.test_unpickler.<locals>.recurse�: �2 �@ � �a� )r rt rb r r/ rv rw r! r" rK rl ru �lenrI r r r )r$ r{ r� r� r( r� r� r )r~ rP r r� r# r �test_unpickler s6
.
zSizeofTests.test_unpicklerN)r r r r rw r} r� r r r r rp s rp )Z _elementtreezxml.etree.ElementTree)ZcPickler )�StringIOr! )Z cStringIOr! )Z __builtin__Z
basestring�builtins�str)�
exceptions�
StandardErrorr� � Exception)�UserDictr� r_ r� )�socketZ
_socketobjectr� Z
SocketTypec C s6 | |ft kr t | |f \} }n| tkr.t| } | |fS )N)r r )�module�namer r r �mappingP s
r� c C s6 | |ft kr t | |f \} }n| tkr.t| } | |fS )N)r r )r� r� r r r �reverse_mappingW s
r� c
C s� y
t j| S tk
r� y t| � W nl tk
rb } z tjrNtd| |f � t�W d d }~X Y n2 tk
r� } ztjr�t|� � W d d }~X Y nX t j| S X d S )NzCan't import module %r: %s) rx �modules�KeyError�
__import__r- r �verbose�printr1 )r� �excr r r � getmodule^ s
r� c C s* t | �}x |�d�D ]}t||�}qW |S )N�.)r� �split�getattr)r� r� r? r r r r � getattributen s r� c c s> x8t | �D ],}t| |�}t|t�r
t|t�r
||fV q
W d S )N)�dirr� �
isinstance�type�
issubclass�
BaseException)�modr� �attrr r r �get_exceptionst s
r� c @ sD e Zd Zdd� Zdd� Zdd� Zdd� Zd d
� Zd d � Zd
d� Z dS )�CompatPickleTestsc C st t t�� �}|t t�O }|dd� tD �O }|dd� t�� D �O }x.|D ]&}y t|� W qF tk
rj Y qFX qFW d S )Nc S s h | ] \}}|�qS r r )r� r� r� r r r � <setcomp>~ s z0CompatPickleTests.test_import.<locals>.<setcomp>c S s h | ] \}}|�qS r r )r� r� r� r r r r� s )�setr �valuesr r r r� r1 )r$ r� r� r r r � test_import{ s
z CompatPickleTests.test_importc
C s� x|t �� D ]p\}}| �||f��T y t|� W n tk
rB Y nX |d d� dkrp| �|t� | �t| |� W d Q R X q
W d S )NrF �_)r �items�subTestr� r1 ZassertInr rK )r$ �module3�module2r r r �test_import_mapping� s z%CompatPickleTests.test_import_mappingc
C s� x�t �� D ]�\\}}\}}| �||f||ff��� ||fdkrZt||�}| �t|t�� n�||fdkr�t||�}| �t|t�� nht||�\}}|d d� dkr�| � ||f||f� yt||�}W n tk
r� Y nX | � t||�|� W d Q R X q
W d S )N)r� �OSError)r� r1 rF r� )
r r� r� r� Z
assertTruer� r� r1 r� rK )r$ r� �name3r� �name2r� r� r� r r r �test_name_mapping� s
z#CompatPickleTests.test_name_mappingc C s� x�t �� D ]�\}}| �||f��� y t|� W n0 tk
r^ } ztjrNt|� W d d }~X Y nX ||ftkr�t �
|d �|kr�x@t �� D ]"\\}}\}}||f||fkr�P q�W | � d||f � t �
||�}t �
||�}| �
||� W d Q R X q
W d S )Nz No reverse mapping from %r to %r)r r� r� r� r1 r r� r� �ALT_IMPORT_MAPPINGr �getr ZfailrK ) r$ r� r� r� Zm3Zn3Zm2Zn2r� r r r �test_reverse_import_mapping� s" z-CompatPickleTests.test_reverse_import_mappingc
C s� x�t �� D ]�\\}}\}}| �||f||ff��~ yt||�}W n tk
rT Y nX t||�\}}||||ftkr�| �||f||f� t||�\}}| �||f||f� W d Q R X q
W d S )N) r r� r� r� r1 r� �ALT_NAME_MAPPINGrK r� )r$ r� r� r� r� r� r� r� r r r �test_reverse_name_mapping� s z+CompatPickleTests.test_reverse_name_mappingc
C s* | � tdd�d� | � tdd�d� | � tdd�d� | � tdd�d� | � tdd�d � x�tt�D ]�\}}| �|��� |tttt fkr�wd|t
k r�t |t
�r�| � td|�d � nh|t k r�t |t �r�| � td|�d
� | � td|�d|f� n,| � td|�d|f� | � td|�d|f� W d Q R X qdW d S ) Nr� r� )r� r� r� r� )r� r� r� )r� r� )r� r� )r� r1 )
rK r� r� r� r� r� �BlockingIOError�ResourceWarning�StopAsyncIteration�RecursionErrorr� r� r1 )r$ r� r� r r r �test_exceptions� s:
z!CompatPickleTests.test_exceptionsc
C sf t �d�}xVt|�D ]J\}}| �|��2 | �td|�d|f� | �td|�d|f� W d Q R X qW d S )Nzmultiprocessing.contextZmultiprocessing)r �
import_moduler� r� rK r� r� )r$ r� r� r� r r r � test_multiprocessing_exceptions� s
z1CompatPickleTests.test_multiprocessing_exceptionsN)
r r r r� r� r� r� r� r� r� r r r r r� z s
!r� c C sT t tttttttg} tr<| � t
t t t
tttttttttg
� tj| � t�t� d S )N)r r r3 rD rE rZ r^ r� �has_c_implementation�extendra rc rd re rf rg rh rW ri rn ro r; rp r Z run_unittestZ run_doctestr )Ztestsr r r � test_main� s
r� �__main__)=Z_compat_pickler r r r r� r r! r_ r/ rx rL ZunittestZtestr Ztest.pickletesterr r r r
r r r
r rb r� r1 r r r3 r; rl r< rD rE rW rZ r^ ra rc rd re rf rg rh ri rn ro rV ZTestCaserp r� r� r� r� r� r� r� r� r� r r r r r �<module> s�
B
Bw