Your IP : 216.73.216.196
o
0�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�pickleZ_dump�dumpZ_dumps�dumps�_load�loadZ_loads�loads�_Pickler�Pickler�
_Unpickler� Unpickler� r! r! �7/opt/alt/python310/lib64/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�kwargsr. �pr! r! r" r ? s
zPyPicklerTests.dumpsc K r$ r% r& r* r! r! r" r F r0 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+ rA ZprotocolrC r! r! r" r T s zInMemoryPickleTests.dumpsc K s t j|fi |��S r% )r r )r+ r, r- r! r! r" r W s zInMemoryPickleTests.loadsr% )r r r r r3 r1 r2 r4 r5 r6 r7 r8 r9 r: r r Z+test_framed_write_sizes_with_delayed_writerr! r! r! r" rE L s
�
rE 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" rI a �
zHPersistentPicklerUnpicklerMixin.dumps.<locals>.PersPickler.persistent_idN�r r r rI r! rM r! r" � PersPickler` � rP )r= r' r( r �getvalue)r+ rA rB rP r. rD r! rM r" r _ s
z%PersistentPicklerUnpicklerMixin.dumpsc s8 G � fdd�d� j �}t�|�}||fi |��}|�� S )Nc rG )z<PersistentPicklerUnpicklerMixin.loads.<locals>.PersUnpicklerc rH r% )�persistent_loadrJ rM r! r" rS j rN zLPersistentPicklerUnpicklerMixin.loads.<locals>.PersUnpickler.persistent_loadN�r r r rS r! rM r! r" �
PersUnpickleri rQ rU )r) r' r( r )r+ r, r- rU r. r/ r! rM r" r h s
z%PersistentPicklerUnpicklerMixin.loadsr% )r r r r r r! r! r! r" rF ] s
rF c @ � e Zd ZejZejZdS )�PyPersPicklerTestsN)r r r r r r= r r) r! r! r! r" rW q s
rW 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 rR rI �weakref�ref� assertIsNone)r rB r. r= �rrM 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! rJ r! r! r" rI � � �TPyIdPersPicklerTests.test_pickler_reference_cycle.<locals>.PersPickler.persistent_idNrO r! r! r! r" rP � � rP c @ � e Zd Zedd� �ZdS )rg c S rh r% r! )�clsrL r! r! r" rI � � rj N)r r r � classmethodrI r! r! r! r" rP � � c @ rl )rg c S � | S r% r! )rL r! r! r" rI � rn rj N)r r r r rI r! r! r! r" rP � rp )r= )r+ re rP r! rM 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+ �argsrC )� __class__�tabler! r" ru � s zYPyIdPersPicklerTests.test_custom_pickler_dispatch_table_memleak.<locals>.Pickler.__init__)r r r ru �
__classcell__r! �rx )rw 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( ZassertIsrs ra rb ZassertIsNotNoner Z
gc_collectrc )r+ r r} r= Z table_refr! rz r" �*test_custom_pickler_dispatch_table_memleak� s
z?PyIdPersPicklerTests.test_custom_pickler_dispatch_table_memleakc rY )Nc st t tjd �D ]}| t�� �d|���}� �|�� d� q| t�� �}� �|�d�d� t �
|�}~� � |� � d S rZ ) r^ r r_ r' r( r r` r rS ra rb rc )r rB r) rd rM r! r" re � s
zBPyIdPersPicklerTests.test_unpickler_reference_cycle.<locals>.checkc @ rf )�JPyIdPersPicklerTests.test_unpickler_reference_cycle.<locals>.PersUnpicklerc S rh r% r! )rK �pidr! r! r" rS � ri �ZPyIdPersPicklerTests.test_unpickler_reference_cycle.<locals>.PersUnpickler.persistent_loadNrT r! r! r! r" rU � rk rU c @ rl )r� c S rh r% r! )rm r� r! r! r" rS � rn r� N)r r r ro rS r! r! r! r" rU � rp c @ rl )r� c S rq r% r! )r� r! r! r" rS � rn r� N)r r r r rS r! r! r! r" rU � rp )r) )r+ re rU r! rM r" � test_unpickler_reference_cycle� s
z3PyIdPersPicklerTests.test_unpickler_reference_cycleN)
r r r r r r= r r) r � cpython_onlyrr r r� r! r! r! r" rX x s
rX c @ rV )� 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 rs �copyrM r! r! r" �get_dispatch_table� rN 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 rs rM 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 r3 r2 r: r! r! r! r" r� � s r� c @ rV )�
CPicklerTestsN�r r r r� r r= r r) r! r! r! r" r� � �
r� c @ rV )�CPersPicklerTestsNr� r! r! r! r" r� � r� r� c @ rV )�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�objectZmemor6 )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� rM r! r! r" r� rN 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� rM 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( r` r� �
__sizeof__r7 �calcsize� check_sizeof�sys� getsizeofr^ r �chr)r+ �basesizerD ZMT_sizeZME_sizere �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 r/ �r� re 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< )r^ )Zdeepr� r� r! r! r" �recurseY s
z+SizeofTests.test_unpickler.<locals>.recurse�2 �@ �< � �a� )r r� r� r r7 r� r� r' r( r` r� r� �lenr^ r r r )r+ r� r� r� r/ 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� > )Z cStringIOr' )Z _elementtreezxml.etree.ElementTree)�StringIOr' )ZcPickler > )�
exceptions�
StandardError�builtins� Exception)�socketZ
_socketobjectr� Z
SocketType)�UserDictr� r� r� )Z __builtin__Z
basestringr� �strc 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__r5 �printr9 )r� �action�excr! r! r" � getmodule� s.
������r� c C s& t | �}|�d�D ]}t||�}q |S )N�.)r� �split�getattr)r� r� rL 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� r9 )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� r9 ZassertInr r` )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� r9 r[ r )
r r r r� Z
assertTruer� r r9 r� r` )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� r9 r r� r� �ALT_IMPORT_MAPPINGr �getr Zfailr` ) 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� r9 r� �ALT_NAME_MAPPINGr` 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 )r` r� r� r� r� r �BlockingIOError�ResourceWarning�StopAsyncIteration�RecursionError�EncodingWarningr r� r9 )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.contextZmultiprocessing)r �
import_moduler� r r` 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__)FZ_compat_pickler r r r r� r r' r� r7 r� r� ra r Zunittest�testr Z test.supportr Ztest.pickletesterr r r
r r r
r r r r r� Zhas_c_implementationr9 ZTestCaser r# r; rE r� rF rW rX 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 �mainr! r! r! r" �<module> s� � � � �Z
Ax
�