File: //usr/local/lib/python3.10/test/__pycache__/test_pickle.cpython-310.pyc
o
�i�N � @ sN 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
lm Z d dlm Z zd dl Z dZ W n e y� dZ Y nw G dd� deej!�Z"G dd� deej!�Z#G dd� deej!�Z$G dd� deee ej!�Z%G dd� de&�Z'G dd � d ee'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 �rqG d'd(� d(eej!�Z.G d)d*� d*e#�Z/G d+d,� d,e$�Z0G d-d.� d.e(�Z1G d/d0� d0e)�Z2G d1d2� d2e$�Z3G d3d4� d4e$�Z4G d5d6� d6eej!�Z5G d7d8� d8eej!�Z6G d9d:� d:eej!�Z7G d;d<� d<eej!�Z8ej9G d=d>� d>ej!��Z:h d?�Z;h d@�Z<dAdB� Z=dCdD� Z>dEdF� Z?dGdH� Z@dIdJ� ZAG dKdL� dLej!�ZBdMdN� ZCeDdOk�r�e�E� dS dS )P� )�IMPORT_MAPPING�REVERSE_IMPORT_MAPPING� NAME_MAPPING�REVERSE_NAME_MAPPINGN)�support)�
import_helper)�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�pickle�_dump�dump�_dumps�dumps�_load�load�_loads�loads�_Pickler�Pickler�
_Unpickler� Unpickler� r$ r$ �-/usr/local/lib/python3.10/test/test_pickle.pyr # s
r c @ s6 e Zd ZejZefZeje e
e e j
eefZdd� ZdS )�PyUnpicklerTestsc K �$ t �|�}| j|fi |��}|�� S �N��io�BytesIO� unpicklerr ��self�buf�kwds�f�ur$ r$ r% r 4 �
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�kwargsr1 �pr$ r$ r% r ? s
zPyPicklerTests.dumpsc K r' r( r) r- r$ r$ r% r F r3 zPyPicklerTests.loadsr( )
r r r r r r@ r" r, r r r$ r$ r$ r% r> : s
r> c @ sB e Zd ZejefZejeee e
j ee fZ
ddd�Zdd� ZdZdS )�InMemoryPickleTestsNc K s t j||fi |��S r( )r r )r. rD �protocolrF r$ r$ r% r T s zInMemoryPickleTests.dumpsc K s t j|fi |��S r( )r r )r. r/ r0 r$ r$ r% r W s zInMemoryPickleTests.loadsr( )r r r r r6 r4 r5 r7 r8 r9 r: r; r< r= r r Z+test_framed_write_sizes_with_delayed_writerr$ r$ r$ r% rH L s
�
rH c @ s e Zd Zddd�Zdd� ZdS )� PersistentPicklerUnpicklerMixinNc s: G � fdd�d� j �}t�� }|||�}|�|� |�� S )Nc � e Zd Z� fdd�ZdS )z:PersistentPicklerUnpicklerMixin.dumps.<locals>.PersPicklerc �
� � |�S r( )�
persistent_id��subself�obj�r. r$ r% rM a �
zHPersistentPicklerUnpicklerMixin.dumps.<locals>.PersPickler.persistent_idN�r r r rM r$ rQ r$ r% � PersPickler` � rT )r@ r* r+ r �getvalue)r. rD rE rT r1 rG r$ rQ r% r _ s
z%PersistentPicklerUnpicklerMixin.dumpsc s8 G � fdd�d� j �}t�|�}||fi |��}|�� S )Nc rK )z<PersistentPicklerUnpicklerMixin.loads.<locals>.PersUnpicklerc rL r( )�persistent_loadrN rQ r$ r% rW j rR zLPersistentPicklerUnpicklerMixin.loads.<locals>.PersUnpickler.persistent_loadN�r r r rW r$ rQ r$ r% �
PersUnpickleri rU rY )r, r* r+ r )r. r/ r0 rY r1 r2 r$ rQ r% r h s
z%PersistentPicklerUnpicklerMixin.loadsr( )r r r r r r$ r$ r$ r% rJ ] s
rJ c @ � e Zd ZejZejZdS )�PyPersPicklerTestsN)r r r r r r@ r" r, r$ r$ r$ r% r[ q s
r[ c @ sB e Zd ZejZejZej dd� �Z
ej dd� �Z ej dd� �Z dS )�PyIdPersPicklerTestsc �^ � fdd�}G dd� d� j �}||� G dd� d� j �}||� G dd� d� j �}||� d S )Nc s� t tjd �D ]}t�� }| ||�}|�d� � �� �|�� �d� q| t�� �}� �|� d�d� t
� |�}~� � |� � d S �N� �abc�def)
�ranger �HIGHEST_PROTOCOLr* r+ r � assertEqualr rV rM �weakref�ref� assertIsNone)r! rE r1 r@ �rrQ r$ r% �check� s
z@PyIdPersPicklerTests.test_pickler_reference_cycle.<locals>.checkc @ � e Zd Zdd� ZdS )�FPyIdPersPicklerTests.test_pickler_reference_cycle.<locals>.PersPicklerc S � |S r( r$ rN r$ r$ r% rM � � �TPyIdPersPicklerTests.test_pickler_reference_cycle.<locals>.PersPickler.persistent_idNrS r$ r$ r$ r% rT � � rT c @ � e Zd Zedd� �ZdS )rk c S rl r( r$ )�clsrP r$ r$ r% rM � � rn N)r r r � classmethodrM r$ r$ r$ r% rT � � c @ rp )rk c S � | S r( r$ )rP r$ r$ r% rM � rr rn N)r r r r rM r$ r$ r$ r% rT � rt )r@ )r. ri rT r$ rQ r% � test_pickler_reference_cycle~ s z1PyIdPersPicklerTests.test_pickler_reference_cyclec sv G � fdd�d| j �}G dd� d�}|� � |t�� �}| �|j� � t�� �}| �|� � ~� t� � | �
|� � d S )Nc s e Zd Z� �fdd�Z� ZS )zPPyIdPersPicklerTests.test_custom_pickler_dispatch_table_memleak.<locals>.Picklerc s �| _ t� j|i |�� d S r( )�dispatch_table�super�__init__)r. �argsrF )� __class__�tabler$ r% ry � s zYPyIdPersPicklerTests.test_custom_pickler_dispatch_table_memleak.<locals>.Pickler.__init__)r r r ry �
__classcell__r$ �r| )r{ r% r! � s r! c @ � e Zd ZdS )zVPyIdPersPicklerTests.test_custom_pickler_dispatch_table_memleak.<locals>.DispatchTableN�r r r r$ r$ r$ r% �
DispatchTable� � r� ) r@ r* r+ ZassertIsrw re rf ZassertIsNotNoner �
gc_collectrg )r. r! r� r@ Z table_refr$ r~ r% �*test_custom_pickler_dispatch_table_memleak� s
z?PyIdPersPicklerTests.test_custom_pickler_dispatch_table_memleakc r] )Nc st t tjd �D ]}| t�� �d|���}� �|�� d� q| t�� �}� �|�d�d� t �
|�}~� � |� � d S r^ ) rb r rc r* r+ r rd r rW re rf rg )r# rE r, rh rQ r$ r% ri � s
zBPyIdPersPicklerTests.test_unpickler_reference_cycle.<locals>.checkc @ rj )�JPyIdPersPicklerTests.test_unpickler_reference_cycle.<locals>.PersUnpicklerc S rl r( r$ )rO �pidr$ r$ r% rW � rm �ZPyIdPersPicklerTests.test_unpickler_reference_cycle.<locals>.PersUnpickler.persistent_loadNrX r$ r$ r$ r% rY � ro rY c @ rp )r� c S rl r( r$ )rq r� r$ r$ r% rW � rr r� N)r r r rs rW r$ r$ r$ r% rY � rt c @ rp )r� c S ru r( r$ )r� r$ r$ r% rW � rr r� N)r r r r rW r$ r$ r$ r% rY � rt )r, )r. ri rY r$ rQ r% � test_unpickler_reference_cycle� s
z3PyIdPersPicklerTests.test_unpickler_reference_cycleN)
r r r r r r@ r" r, r � cpython_onlyrv r� r� r$ r$ r$ r% r\ x s
r\ c @ rZ )� PyPicklerUnpicklerObjectTestsN)r r r r r �
pickler_classr" �unpickler_classr$ r$ r$ r% r� � s
r� c @ � e Zd ZejZdd� ZdS )�PyDispatchTableTestsc C �
t j�� S r( �r rw �copyrQ r$ r$ r% �get_dispatch_table� rR z'PyDispatchTableTests.get_dispatch_tableN�r r r r r r� r� r$ r$ r$ r% r� � � r� c @ r� )�PyChainDispatchTableTestsc C � t �i tj�S r( �� collections�ChainMapr rw rQ r$ r$ r% r� � � z,PyChainDispatchTableTests.get_dispatch_tableNr� r$ r$ r$ r% r� � r� r� c @ �$ e Zd ZG dd� deje�ZeZdS )�PyPicklerHookTestsc @ r )z'PyPicklerHookTests.CustomPyPicklerClassNr� r$ r$ r$ r% �CustomPyPicklerClass� s r� N)r r r r r r r� 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% r� � s $r� c @ s" e Zd ZejZejfZejfZ dS )�CUnpicklerTestsN)
r r r r� r# r, r r6 r5 r= r$ r$ r$ r% r� � s r� c @ rZ )�
CPicklerTestsN�r r r r� r! r@ r# r, r$ r$ r$ r% r� � �
r� c @ rZ )�CPersPicklerTestsNr� r$ r$ r$ r% r� � r� r� c @ rZ )�CIdPersPicklerTestsNr� r$ r$ r$ r% r� r� r� c @ � e Zd ZejZejZdS )�CDumpPickle_LoadPickleN) r r r r� r! r@ r r" r, r$ r$ r$ r% r� r� r� c @ r� )�DumpPickle_CLoadPickleN) r r r r r r@ r� r# r, r$ r$ r$ r% r� r� r� c @ s e Zd ZejZejZdd� ZdS )� CPicklerUnpicklerObjectTestsc C s� | � t�� �}| �t�� t|_W d � n1 sw Y | �t��
dd i|_W d � n1 s4w Y dd i|_d S )N���r_ )r� r* r+ Z assertRaises� TypeError�object�memor9 )r. r, r$ r$ r% �test_issue18339 s � �z,CPicklerUnpicklerObjectTests.test_issue18339N) r r r r� r! r� r# r� r� r$ r$ r$ r% r� s r� c @ r� )�CDispatchTableTestsc C r� r( r� rQ r$ r$ r% r� rR z&CDispatchTableTests.get_dispatch_tableN�r r r r r! r� r� r$ r$ r$ r% r� � r� c @ r� )�CChainDispatchTableTestsc C r� r( r� rQ r$ r$ r% r� r� z+CChainDispatchTableTests.get_dispatch_tableNr� r$ r$ r$ r% r� r� r� c @ r� )�CPicklerHookTestsc @ r )z%CPicklerHookTests.CustomCPicklerClassNr� r$ r$ r$ r% �CustomCPicklerClass$ r� r� N)r r r r� r! r r� r� 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|�� q6|||| d| d � d S ) NZ 7P2n3i2n3i2PZ3nP0nZPn0P� s xxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxx� � r )r � calcobjsizer� r! r* r+ rd r� �
__sizeof__r: �calcsize� check_sizeof�sys� getsizeofrb r �chr)r. �basesizerG ZMT_sizeZME_sizeri �ir$ r$ r% � test_pickler, s,
��� ��
�zSizeofTests.test_picklerc s� t �d�}tj�t�d�� t�d��| j�dD ].}dD ])}�t�� ||d�}| � t
� |�|� �||d� t |� d t |� d � qq|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� r_ r� r� c sD t �| �}�t�|�ddd�}|�� �|�|� |� � d S )Nr� r� r� )r r r* r+ r )�dataZ memo_sizeZ
marks_sizer r2 �r� ri r� Zstdsizer, r$ r% �check_unpicklerM s
� z3SizeofTests.test_unpickler.<locals>.check_unpicklerr �d � c S s g | ]}t |��qS r$ )r� )�.0r� r$ r$ r% �
<listcomp>X � z.SizeofTests.test_unpickler.<locals>.<listcomp>� c S s d}t | �D ]}||g}q|S r? )rb )Zdeepr� r� r$ r$ r% �recurseY s
z+SizeofTests.test_unpickler.<locals>.recurse�2 �@ �< � �a� )r r� r� r# r: r� r� r* r+ rd r� r� �lenrb r r r )r. r� r� r� r2 r� r� r$ r� r% �test_unpickler= sN
������� � zSizeofTests.test_unpicklerN)r r r r r� r� r� r$ r$ r$ r% r� ( s r� > )� _elementtreezxml.etree.ElementTree)�cPickler )� cStringIOr* )�StringIOr* > )�UserDictr� r� r� )�socket�
_socketobjectr� �
SocketType)� __builtin__�
basestring�builtins�str)�
exceptions�
StandardErrorr� � Exceptionc C �< | |ft v rt | |f \} }| |fS | tv rt| } | |fS r( )r r ��module�namer$ r$ r% �mappingx � �r� c C r� r( )r r r� r$ r$ r% �reverse_mapping r� r� c
C s� zt j| W S tyh z't�� � tjrdnd}t�|t� t | � W d � n1 s.w Y W n, t
yM } z
tjrGt d| |f � t �d }~w t y` } ztjr[t |� � d }~ww t j| Y S w )N�always�ignorezCan't import module %r: %s)
r� �modules�KeyError�warnings�catch_warningsr �verbose� simplefilter�DeprecationWarning�
__import__r8 �printr< )r� �action�excr$ r$ r% � getmodule� s.
������r c C s& t | �}|�d�D ]}t||�}q |S )N�.)r �split�getattr)r� r� rP 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 sl t t�� �}|t t�O }|dd� tD �O }|dd� t�� D �O }|D ]}zt|� W q" ty3 Y q"w d S )Nc S � h | ]\}}|�qS r$ r$ �r� r� r� r$ r$ r% � <setcomp>� r� z0CompatPickleTests.test_import.<locals>.<setcomp>c S r r$ r$ r r$ r$ r% r � r� )�setr �valuesr r r r r< )r. r� r� r$ r$ r% � test_import� s ��z CompatPickleTests.test_importc
C s� t �� D ]A\}}| �||f��. zt|� W n ty Y nw |d d� dkr6| �|t� | �t| |� W d � n1 s@w Y qd S )Nr_ �_)r �items�subTestr r< ZassertInr rd )r. �module3�module2r$ r$ r% �test_import_mapping� s � � ���z%CompatPickleTests.test_import_mappingc
C s
t �� D ]~\\}}\}}| �||f||ff��c ||fdkr,t||�}| �t|t�� nG||fdkr@t||�}| �t|t�� n3t||�\}}|d d� dkrY| � ||f||f� zt||�}W n tyi Y n
w | � t||�|� W d � n1 s}w Y qd S )N�r� �OSError�r� r< r_ r )
r r r r Z
assertTruer r r< r� rd )r. r �name3r �name2r r� r� r$ r$ r% �test_name_mapping� s*
�� ���z#CompatPickleTests.test_name_mappingc C s t �� D ]{\}}| �||f��h zt|� W n ty/ } z
tjr%t|� W Y d }~nd }~ww ||ftvr^t �
|d �|kr^t �� D ]\\}}\}}||f||fkrT n
qB| � d||f � t �
||�}t �
||�}| �
||� W d � n1 szw Y qd S )Nz No reverse mapping from %r to %r)r r r r r< r r� r� �ALT_IMPORT_MAPPINGr �getr Zfailrd ) r. r r r� Zm3Zn3Zm2Zn2r� r$ r$ r% �test_reverse_import_mapping� s0 �� �� ���z-CompatPickleTests.test_reverse_import_mappingc
C s� t �� D ]^\\}}\}}| �||f||ff��C zt||�}W n ty( Y nw t||�\}}||||ftvrB| �||f||f� t||�\}}| �||f||f� W d � n1 s]w Y qd S r( ) r r r r r< r� �ALT_NAME_MAPPINGrd r� )r. r r r r r r� r� r$ r$ r% �test_reverse_name_mapping� s � ���z+CompatPickleTests.test_reverse_name_mappingc C sJ | � tdd�d� | � tdd�d� | � tdd�d� | � tdd�d� | � tdd�d � tt�D ]q\}}| �|��` |tttt t
fv rL W d � q1|t ur_t |t �r_| � td|�d � n4|t
ur}t |t
�r}| � td|�d
� | � td|�d|f� n| � td|�d|f� | � td|�d|f� W d � n1 s�w Y q1d S ) Nr� r� )r� r� r� r� )r� r� r )r� r r r )rd r� r� r r� r �BlockingIOError�ResourceWarning�StopAsyncIteration�RecursionError�EncodingWarningr r r< )r. r� r� r$ r$ r% �test_exceptions� s\ � � � � � �� � � � � �� ���z!CompatPickleTests.test_exceptionsc C sv t �d�}t|�D ]/\}}| �|�� | �td|�d|f� | �td|�d|f� W d � n1 s3w Y q d S )Nzmultiprocessing.context�multiprocessing)r �
import_moduler r rd 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 r1 �__main__)F�_compat_pickler r r r r� r r* r� r: r� r� re r. Zunittest�testr Z test.supportr Ztest.pickletesterr r r
r r r
r r r r r� Zhas_c_implementationr< ZTestCaser r& r> rH r� rJ r[ r\ r� r� r� r� r� r� r� r� r� r� r� r� r� r� r� r� r� r r# r� r� r r r r
r1 r �mainr$ r$ r$ r% �<module> s� � � � �Z
Ax
�