Your IP : 216.73.216.85
a
XC?h�K � @ s@ d dl mZmZmZmZ d dlZd dlZd dlZd dlZd dl Z d dl
Z
d dl Z d dl Z d dl
Z
d dlZd dlmZ d dlmZ d dlmZ d dlmZ d dlmZ d dlmZ d d lmZ d d
lmZ d d lmZ d d lmZ d d
lmZ zd dl Z dZ W n e �y dZ Y n0 G dd� deej �Z G dd� deej �Z!G dd� deej �Z"G dd� deeeej �Z#G dd� de$�Z%G dd� dee%ej �Z&G d d � d ee%ej �Z'G d d � d eej �Z(G d d!� d!eej �Z)G d"d#� d#eej �Z*G d$d%� d%eej �Z+e �r�G d&d'� d'eej �Z,G d(d)� d)e!�Z-G d*d+� d+e"�Z.G d,d-� d-e&�Z/G d.d/� d/e'�Z0G d0d1� d1e"�Z1G d2d3� d3e"�Z2G d4d5� d5eej �Z3G d6d7� d7eej �Z4G d8d9� d9eej �Z5G d:d;� d;eej �Z6ej7G d<d=� d=ej ��Z8h d>�Z9h d?�Z:d@dA� Z;dBdC� Z<dDdE� Z=dFdG� Z>dHdI� Z?G dJdK� dKej �Z@dLdM� ZAeBdNk�r<e�C� dS )O� )�IMPORT_MAPPING�REVERSE_IMPORT_MAPPING� NAME_MAPPING�REVERSE_NAME_MAPPINGN)�support)�AbstractHookTests)�AbstractUnpickleTests)�AbstractPickleTests)�AbstractPickleModuleTests)� AbstractPersistentPicklerTests)�&AbstractIdentityPersistentPicklerTests)�#AbstractPicklerUnpicklerObjectTests)�AbstractDispatchTableTests)�AbstractCustomPicklerClass)�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 �5/opt/alt/python39/lib64/python3.9/test/test_pickle.pyr " s
r c @ s6 e Zd ZejZefZeje e
e e j
eefZdd� ZdS )�PyUnpicklerTestsc K s$ t �|�}| j|fi |��}|�� S �N��io�BytesIO� unpicklerr ��self�buf�kwds�f�ur r r! r 3 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 K s< t �� }| j||fi |��}|�|� |�d� t|�� �S �Nr )r% r& �picklerr �seek�bytes�read)r) �arg�proto�kwargsr, �pr r r! r > s
zPyPicklerTests.dumpsc K s$ t �|�}| j|fi |��}|�� S r# r$ r( r r r! r E s
zPyPicklerTests.loads)N)
r r r r r r: r r' r r r r r r! r8 9 s
r8 c @ sB e Zd ZejefZejeee e
j ee fZ
ddd�Zdd� ZdZdS )�InMemoryPickleTestsNc K s t j||fi |��S r# )r r )r) r> Zprotocolr@ r r r! r S s zInMemoryPickleTests.dumpsc K s t j|fi |��S r# )r r )r) r* r+ r r r! r V s zInMemoryPickleTests.loads)N)r r r r r0 r. r/ r1 r2 r3 r4 r5 r6 r7 r r Z+test_framed_write_sizes_with_delayed_writerr r r r! rB K s
�
rB 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 r# )�
persistent_id��subself�obj�r) r r! rD ` s zHPersistentPicklerUnpicklerMixin.dumps.<locals>.PersPickler.persistent_idN�r r r rD r rH r r! � PersPickler_ s rJ )r: r% r& r �getvalue)r) r> r? rJ r, rA r rH r! r ^ s
z%PersistentPicklerUnpicklerMixin.dumpsc s8 G � fdd�d� j �}t�|�}||fi |��}|�� S )Nc s e Zd Z� fdd�ZdS )z<PersistentPicklerUnpicklerMixin.loads.<locals>.PersUnpicklerc s
� � |�S r# )�persistent_loadrE rH r r! rL i s zLPersistentPicklerUnpicklerMixin.loads.<locals>.PersUnpickler.persistent_loadN�r r r rL r rH r r! �
PersUnpicklerh s rN )r' r% r& r )r) r* r+ rN r, r- r rH r! r g s
z%PersistentPicklerUnpicklerMixin.loads)N)r r r r r r r r r! rC \ s
rC c @ s e Zd ZejZejZdS )�PyPersPicklerTestsN)r r r r r r: r r' r r r r! rO p s rO 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� t tjd �D ]6}t�� }| ||�}|�d� � �� �|�� �d� q| t�� �}� �|� d�d� t
� |�}~� � |� � d S �N� �abc�def)
�ranger �HIGHEST_PROTOCOLr% r& r � assertEqualr rK rD �weakref�ref� assertIsNone)r r? r, r: �rrH r r! �check s
z@PyIdPersPicklerTests.test_pickler_reference_cycle.<locals>.checkc @ s e Zd Zdd� ZdS )�FPyIdPersPicklerTests.test_pickler_reference_cycle.<locals>.PersPicklerc S s |S r# r rE r r r! rD � s �TPyIdPersPicklerTests.test_pickler_reference_cycle.<locals>.PersPickler.persistent_idNrI r r r r! rJ � s rJ c @ s e Zd Zedd� �ZdS )r] c S s |S r# r )�clsrG r r r! rD � s r^ N)r r r � classmethodrD r r r r! rJ � s c @ s e Zd Zedd� �ZdS )r] c S s | S r# r )rG r r r! rD � s r^ N)r r r r rD r r r r! rJ � s )r: )r) r\ rJ r rH r! � test_pickler_reference_cycle} s z1PyIdPersPicklerTests.test_pickler_reference_cyclec s^ � fdd�}G dd� d� j �}||� G dd� d� j �}||� G dd� d� j �}||� d S )Nc st t tjd �D ]*}| t�� �d|���}� �|�� d� q| t�� �}� �|�d�d� t �
|�}~� � |� � d S rQ ) rU r rV r% r& r rW r rL rX rY rZ )r r? r' r[ rH r r! r\ � s
zBPyIdPersPicklerTests.test_unpickler_reference_cycle.<locals>.checkc @ s e Zd Zdd� ZdS )�JPyIdPersPicklerTests.test_unpickler_reference_cycle.<locals>.PersUnpicklerc S s |S r# r )rF �pidr r r! rL � s �ZPyIdPersPicklerTests.test_unpickler_reference_cycle.<locals>.PersUnpickler.persistent_loadNrM r r r r! rN � s rN c @ s e Zd Zedd� �ZdS )rb c S s |S r# r )r_ rc r r r! rL � s rd N)r r r r` rL r r r r! rN � s c @ s e Zd Zedd� �ZdS )rb c S s | S r# r )rc r r r! rL � s rd N)r r r r rL r r r r! rN � s )r' )r) r\ rN r rH r! � test_unpickler_reference_cycle� s
z3PyIdPersPicklerTests.test_unpickler_reference_cycleN) r r r r r r: r r' r � cpython_onlyra re r r r r! rP w s
rP c @ s e Zd ZejZejZdS )� PyPicklerUnpicklerObjectTestsN)r r r r r �
pickler_classr �unpickler_classr r r r! rg � s rg c @ s e Zd ZejZdd� ZdS )�PyDispatchTableTestsc C s
t j�� S r# �r �dispatch_table�copyrH r r r! �get_dispatch_table� s z'PyDispatchTableTests.get_dispatch_tableN�r r r r r rh rn r r r r! rj � s rj c @ s e Zd ZejZdd� ZdS )�PyChainDispatchTableTestsc C s t �i tj�S r# �� collections�ChainMapr rl rH r r r! rn � s z,PyChainDispatchTableTests.get_dispatch_tableNro r r r r! rp � s rp c @ s$ e Zd ZG dd� deje�ZeZdS )�PyPicklerHookTestsc @ s e Zd ZdS )z'PyPicklerHookTests.CustomPyPicklerClassN�r r r r r r r! �CustomPyPicklerClass� s rv N)r r r r r r rv rh r r r r! rt � s �rt 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! rw � s rw c @ s" e Zd ZejZejfZejfZ dS )�CUnpicklerTestsN)
r r r rx r r' r r0 r/ r7 r r r r! ry � s ry c @ s e Zd ZejZejZdS )�
CPicklerTestsN�r r r rx r r: r r' r r r r! rz � s rz c @ s e Zd ZejZejZdS )�CPersPicklerTestsNr{ r r r r! r| � s r| c @ s e Zd ZejZejZdS )�CIdPersPicklerTestsNr{ r r r r! r} � s r} c @ s e Zd ZejZejZdS )�CDumpPickle_LoadPickleN) r r r rx r r: r r r' r r r r! r~ � s r~ c @ s e Zd ZejZejZdS )�DumpPickle_CLoadPickleN) r r r r r r: rx r r' r r r r! r � s r c @ s e Zd ZejZejZdd� ZdS )� CPicklerUnpicklerObjectTestsc C s� | � t�� �}| �t�� t|_W d � n1 s40 Y | �t�� dd i|_W d � n1 sh0 Y dd i|_d S )N���rR )ri r% r& Z assertRaises� TypeError�object�memor3 )r) r' r r r! �test_issue18339� s $ (z,CPicklerUnpicklerObjectTests.test_issue18339N) r r r rx r rh r ri r� r r r r! r� � s r� c @ s e Zd ZejZdd� ZdS )�CDispatchTableTestsc C s
t j�� S r# rk rH r r r! rn s z&CDispatchTableTests.get_dispatch_tableN�r r r r r rh rn r r r r! r� s r� c @ s e Zd ZejZdd� ZdS )�CChainDispatchTableTestsc C s t �i tj�S r# rq rH r r r! rn s z+CChainDispatchTableTests.get_dispatch_tableNr� r r r r! r� s r� c @ s$ e Zd ZG dd� deje�ZeZdS )�CPicklerHookTestsc @ s e Zd ZdS )z%CPicklerHookTests.CustomCPicklerClassNru r r r r! �CustomCPicklerClass s r� N)r r r rx r r r� rh r r r r! r� s r� c @ s" e Zd ZejZdd� Zdd� ZdS )� SizeofTestsc C s� t �d�}t�t�� �}| �t�|�|� t �
d�}t �
d�}| j }|||| d| t �
d� � td�D ]}|�t|�� ql|||| d| d � d S ) NZ 7P2n3i2n3i2PZ3nP0nZPn0P� s xxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxx� � r )r � calcobjsizerx r r% r& rW r� �
__sizeof__r4 �calcsize� check_sizeof�sys� getsizeofrU r �chr)r) �basesizerA ZMT_sizeZME_sizer\ �ir r r! � test_pickler s,
��� ���zSizeofTests.test_picklerc s� t �d�}tj�t�d�� t�d��| j�dD ]\}dD ]R}�t�� ||d�}| � t
� |�|� �||d� t |� d t |� d � q6q.|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 2P3n8P2n2i�P�n)�ASCIIzUTF-16zlatin-1)�strict�replace��encoding�errorsr� rR r� r� c sD t �| �}�t�|�ddd�}|�� �|�|� |� � d S )Nr� r� r� )r r r% r& r )�dataZ memo_sizeZ
marks_sizer r- �r� r\ r� Zstdsizer' r r! �check_unpickler5 s
�z3SizeofTests.test_unpickler.<locals>.check_unpicklerr �d � c S s g | ] }t |��qS r )r� )�.0r� r r r! �
<listcomp>@ � z.SizeofTests.test_unpickler.<locals>.<listcomp>� c S s d}t | �D ] }||g}q |S r9 )rU )Zdeepr� r� r r r! �recurseA s
z+SizeofTests.test_unpickler.<locals>.recurse�2 �@ �< � �a� )r r� rx r r4 r� r� r% r& rW r� r� �lenrU r r r )r) r� r� r� r- r� r� r r� r! �test_unpickler% sL
������
�zSizeofTests.test_unpicklerN)r r r r r� r� r� r r r r! r� s r� > )Z cStringIOr% )�StringIOr% )Z _elementtreezxml.etree.ElementTree)ZcPickler > )�socketZ
_socketobjectr� Z
SocketType)�UserDictr� rr r� )Z __builtin__Z
basestring�builtins�str)�
exceptions�
StandardErrorr� � Exceptionc C s6 | |ft v r t | |f \} }n| tv r.t| } | |fS r# )r r ��module�namer r r! �mapping` s
r� c C s6 | |ft v r t | |f \} }n| tv r.t| } | |fS r# )r r r� r r r! �reverse_mappingg s
r� c
C s� z t j| W S ty� zNt�� �2 tjr0dnd}t�|t� t | � W d � n1 s\0 Y W np t
y� } z&tjr�t d| |f � t �W Y d }~n<d }~0 t y� } z tjr�t |� � W Y d }~n
d }~0 0 t j| Y S 0 d S )N�always�ignorezCan't import module %r: %s)
r� �modules�KeyError�warnings�catch_warningsr �verbose� simplefilter�DeprecationWarning�
__import__r2 �printr6 )r� �action�excr r r! � getmodulen s"
*r� c C s& t | �}|�d�D ]}t||�}q|S )N�.)r� �split�getattr)r� r� rG r� r r r! � getattribute� s r� c c s: t | �D ],}t| |�}t|t�rt|t�r||fV qd S r# )�dirr� �
isinstance�type�
issubclass�
BaseException)�modr� �attrr r r! �get_exceptions� s
r� c @ sD e Zd Zdd� Zdd� Zdd� Zdd� Zd d
� Zd d � Zd
d� Z dS )�CompatPickleTestsc C sn t t�� �}|t t�O }|dd� tD �O }|dd� t�� D �O }|D ]$}z t|� W qD tyf Y qD0 qDd S )Nc S s h | ] \}}|�qS r r �r� r� r� r r r! � <setcomp>� r� z0CompatPickleTests.test_import.<locals>.<setcomp>c S s h | ] \}}|�qS r r r� r r r! r� � r� )�setr �valuesr r r r� r6 )r) r� r� r r r! � test_import� s z CompatPickleTests.test_importc
C s� t �� D ]�\}}| �||f��\ z t|� W n ty> Y n0 |d d� dkrl| �|t� | �t| |� W d � q1 s�0 Y qd S )NrR �_)r �items�subTestr� r6 ZassertInr rW )r) �module3�module2r r r! �test_import_mapping� s z%CompatPickleTests.test_import_mappingc
C s
t �� D ]�\\}}\}}| �||f||ff��� ||fdkrXt||�}| �t|t�� n�||fdkr�t||�}| �t|t�� nft||�\}}|d d� dkr�| � ||f||f� zt||�}W n ty� Y n0 | � t||�|� W d � q1 s�0 Y qd S )N�r� �OSError�r� r6 rR r� )
r r� r� r� Z
assertTruer� r� r6 r� rW )r) r� �name3r� �name2r� r� r� r r r! �test_name_mapping� s
z#CompatPickleTests.test_name_mappingc C s t �� D ]�\}}| �||f��� z t|� W n2 ty^ } ztjrJt|� W Y d }~n
d }~0 0 ||ftvr�t �
|d �|kr�t �� D ]$\\}}\}}||f||fkr� q�q�| � d||f � t �
||�}t �
||�}| �
||� W d � q1 s�0 Y qd S )Nz No reverse mapping from %r to %r)r r� r� r� r6 r r� r� �ALT_IMPORT_MAPPINGr �getr ZfailrW ) r) r� r� r� Zm3Zn3Zm2Zn2r� r r r! �test_reverse_import_mapping� s& �� z-CompatPickleTests.test_reverse_import_mappingc
C s� t �� D ]�\\}}\}}| �||f||ff��� zt||�}W n tyP Y n0 t||�\}}||||ftvr�| �||f||f� t||�\}}| �||f||f� W d � q1 s�0 Y qd S r# ) r r� r� r� r6 r� �ALT_NAME_MAPPINGrW r� )r) r� r� r� r� r� r� r� r r r! �test_reverse_name_mapping� s z+CompatPickleTests.test_reverse_name_mappingc C sH | � tdd�d� | � tdd�d� | � tdd�d� | � tdd�d� | � tdd�d � tt�D ]�\}}| �|��� |tttt fv r�W d � qb|t
ur�t |t
�r�| � td|�d � nh|t ur�t |t �r�| � td|�d
� | � td|�d|f� n,| � td|�d|f� | � td|�d|f� W d � qb1 �s80 Y qbd S ) Nr� r� )r� r� r� r� )r� r� r� )r� r� r� r� )
rW r� r� r� r� r� �BlockingIOError�ResourceWarning�StopAsyncIteration�RecursionErrorr� r� r6 )r) r� r� r r r! �test_exceptions� sP � � � � � � � � � � �z!CompatPickleTests.test_exceptionsc C sv t �d�}t|�D ]^\}}| �|��< | �td|�d|f� | �td|�d|f� W d � q1 sf0 Y qd S )Nzmultiprocessing.contextZmultiprocessing)r �
import_moduler� r� rW 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� � s
!r� c C s |� t�� � |S r# )ZaddTest�doctestZ DocTestSuite)�loaderZtests�patternr r r! �
load_tests s r� �__main__)DZ_compat_pickler r r r r� r r% rr r4 r� r� rX r� Zunittest�testr Ztest.pickletesterr r r r
r r r
r r r rx Zhas_c_implementationr6 ZTestCaser r" r8 rB r� rC rO rP rg rj rp rt rw ry rz r| r} r~ r r� r� r� r� rf r� r� r� r� r� r� r� r� r� r� r �mainr r r r! �<module> s�
� � �C
Aw