File: //usr/local/lib/python3.7/test/test_importlib/frozen/__pycache__/test_loader.cpython-37.pyc
B
��g{$ � @ s� d dl mZ d dl mZ e�d�ZddlmZ ddlZddlZddl Z G dd� dej
�Z ej e ed �\Z
ZG d
d � d ej
�Z
ej e
ed �\ZZG d d
� d
�Zej eed �\ZZedkr�e�� dS )� )�abc)�utilzimportlib.machinery� )�captured_stdoutNc @ sH e Zd Zdd� Zdd� Zdd� Zdd� Zd d
� Zd d � Zd
Z dd� Z
d
S )�ExecModuleTestsc
C s� t �|��� t� ��}| jj|| jjd| jj�|�d�}t�|�}||_ t
|d�rTt �| jj� |� | �
|j� | �
t
|d�� | �|j jd� ||�� fS Q R X W d Q R X d S )N�frozen)�origin�
is_package� initialized�__spec__)r �uncacher � machinery�
ModuleSpec�FrozenImporterr �types�
ModuleTyper �hasattr�AssertionError� exec_module�
assertTruer
� assertEqualr �getvalue)�self�name�stdout�spec�module� r �B/usr/local/lib/python3.7/test/test_importlib/frozen/test_loader.pyr s
zExecModuleTests.exec_modulec C sb d}| � |�\}}d|i}x&|�� D ]\}}| �t||�|� q$W | �|d� | �t|d�� d S )N� __hello__�__name__z
Hello world!
r )r �itemsr �getattrr r )r r r �output�check�attr�valuer r r � test_module s zExecModuleTests.test_modulec C sf d}| � |�\}}d|i}x:|�� D ].\}}t||�}| �||dj||||d�� q$W | �|d� d S )N�
__phello__r z,for {name}.{attr}, {given!r} != {expected!r})r r% �given�expectedz
Hello world!
)r r! r" r �format)r r r r# r$ r% r& �
attr_valuer r r � test_package% s
z ExecModuleTests.test_packagec C s| d}t �d��d | �|�\}}d|i}x:|�� D ].\}}t||�}| �||dj||||d�� q0W | �|d� W d Q R X d S )Nz__phello__.spamr( r z*for {name}.{attr}, {given} != {expected!r})r r% r) r* z
Hello world!
)r r r r! r" r r+ )r r r r# r$ r% r& r, r r r �test_lacking_parent1 s
z#ExecModuleTests.test_lacking_parentc C sP d}| � |�\}}t�� � t�dt� | jj�|�}W d Q R X | �|d� d S )Nr �ignorez <module '__hello__' (frozen)>) r �warnings�catch_warnings� simplefilter�DeprecationWarningr
r � module_reprr )r r r r# �repr_strr r r �test_module_repr> s
z ExecModuleTests.test_module_reprc C s& d}| � |�\}}| �t|�d� d S )Nr z <module '__hello__' (frozen)>)r r �repr)r r r r# r r r �test_module_repr_indirectG s
z)ExecModuleTests.test_module_repr_indirectNc C sJ | j j�d�d kst�| �t��}| �d� W d Q R X | �|jj d� d S )N� _not_real)
r
r � find_moduler � assertRaises� ImportErrorr r � exceptionr )r �cmr r r �test_unloadableP s z ExecModuleTests.test_unloadable) r �
__module__� __qualname__r r' r- r. r6 r8 �test_state_after_failurer? r r r r r s
r )r
c @ sH e Zd Zdd� Zdd� Zdd� Zdd� Zd d
� Zd d � Zd
Z dd� Z
d
S )� LoaderTestsc C s� t �d��� t� ��}t�� � t�dt� | jj� d�}W d Q R X dd| jjd�}x&|�
� D ]\}}| � t ||�|� q\W | � |�
� d� | �t|d�� W d Q R X W d Q R X d S )Nr r/ � )r � __package__�
__loader__z
Hello world!
�__file__)r r r r0 r1 r2 r3 r
r � load_moduler! r r" r � assertFalser )r r r r$ r% r& r r r r' ^ s
zLoaderTests.test_modulec C s� t �d��� t� ��}t�� � t�dt� | jj� d�}W d Q R X ddg | jjd�}x6|�
� D ]*\}}t ||�}| � ||d|||f � q^W | � |�
� d� | �t|d�� W d Q R X W d Q R X d S )Nr( r/ )r rE �__path__rF zfor __phello__.%s, %r != %rz
Hello world!
rG )r r r r0 r1 r2 r3 r
r rH r! r" r r rI r )r r r r$ r% r& r, r r r r- l s
zLoaderTests.test_packagec C s� t �dd��� t� ��}t�� � t�dt� | jj� d�}W d Q R X dd| jjd�}x6|�
� D ]*\}}t ||�}| � ||d|||f � q^W | � |�
� d� | �t|d�� W d Q R X W d Q R X d S )Nr( z__phello__.spamr/ )r rE rF z for __phello__.spam.%s, %r != %rz
Hello world!
rG )r r r r0 r1 r2 r3 r
r rH r! r" r r rI r )r r r r$ r% r& r, r r r r. ~ s
z LoaderTests.test_lacking_parentc C s� t �d��p t� �^}t�� �. t�dt� | jj� d�}| jj� d�}W d Q R X | �
||� | � |� � d� W d Q R X W d Q R X d S )Nr r/ zHello world!
Hello world!
)
r r r r0 r1 r2 r3 r
r rH ZassertIsr r )r r Zmodule1Zmodule2r r r �test_module_reuse� s
z LoaderTests.test_module_reusec C st t �d��` t� �N t�� �. t�dt� | jj� d�}| jj�
|�}W d Q R X | � |d� W d Q R X W d Q R X d S )Nr r/ z <module '__hello__' (frozen)>) r r r r0 r1 r2 r3 r
r rH r4 r )r r r5 r r r r6 � s
z LoaderTests.test_module_reprc
C sJ t �d��& t� � | jj�d�}W d Q R X W d Q R X | �t|�d� d S )Nr z <module '__hello__' (frozen)>)r r r r
r rH r r7 )r r r r r r8 � s "
z%LoaderTests.test_module_repr_indirectNc C sN | j j�d�d kst�| �t��}| j j�d� W d Q R X | �|jj d� d S )Nr9 )
r
r r: r r; r< rH r r= r )r r> r r r r? � s zLoaderTests.test_unloadable) r r@ rA r' r- r. rK r6 r8 rB r? r r r r rC \ s
rC c @ s0 e Zd ZdZdd� Zdd� Zdd� Zdd � Zd
S ) �InspectLoaderTestsz7Tests for the InspectLoader methods for FrozenImporter.c C s^ d}t � �J}| jj�|�}t�|�}t||j� | �t |d�� | �
|� � d� W d Q R X d S )Nr r
z
Hello world!
) r r
r �get_coder r �exec�__dict__r r r r )r r r �code�modr r r �
test_get_code� s
z InspectLoaderTests.test_get_codec C s | j j�d�}| �|� d S )Nr )r
r �
get_sourceZ assertIsNone)r �resultr r r �test_get_source� s z"InspectLoaderTests.test_get_sourcec C s8 d}x.|D ]&\}}| j j�|�}| �t|�|� q
W d S )N))r F)r( T)z__phello__.spamF)r
r r r �bool)r Ztest_forr r rT r r r �test_is_package� s z"InspectLoaderTests.test_is_packagec
C sN xHdD ]@}t | jj|�}| �t��}|d� W d Q R X | �|jjd� qW d S )N)rM rS r � importlib)r" r
r r; r< r r= r )r Z meth_name�methodr> r r r � test_failure� s
z InspectLoaderTests.test_failureN)r r@ rA �__doc__rR rU rW rZ r r r r rL � s
rL �__main__)rD r r Zimport_importlibr
Z test.supportr r Zunittestr0 rC r Z test_bothZFrozen_ExecModuleTestsZSource_ExecModuleTestsZFrozen_LoaderTestsZSource_LoaderTestsrL ZFrozen_ILTestsZSource_ILTestsr �mainr r r r �<module> s
MY%