File: //usr/local/lib/python3.10/test/test_import/__pycache__/__init__.cpython-310.pyc
o
�i� � @ sP d dl Z d dlZd dlZd dlZd dlZd dlmZ d dlZd dl Z d dl
Z
d dl Z d dl Z d dl
Z
d dlZd dlZd dlZd dlZd dlZd dlZd dlmZ d dlmZ d dlmZmZmZmZ d dlm Z m Z m Z m Z m Z d dl!m"Z"m#Z#m$Z$m%Z%m&Z& d dlm'Z' d d lm(Z( d d
l)m*Z* d d l+m,Z, e�-ej.d �Z/d
d� Z0ej1d+dd��Z2G dd� dej3�Z4e/G dd� dej3��Z5G dd� dej3�Z6G dd� dej3�Z7G dd� dej3�Z8G d d � d ej3�Z9G d d � d ej3�Z:G d d!� d!ej3�Z;eG d"d#� d#ej3��Z<eG d$d%� d%ej3��Z=G d&d'� d'ej3�Z>G d(d)� d)ej3�Z?e@d*k�r&e�A� dS dS ),� N)�_get_sourcefile)�mock)� os_helper)� is_jython� swap_attr� swap_item� cpython_only)�forget�make_legacy_pyc�unlink�unload�
DirsOnSysPath)�TESTFN�rmtree�
temp_umask�TESTFN_UNENCODABLE�temp_dir)�
script_helper)�threading_helper)�uncache)�
ModuleTypez*test meaningful only when writing bytecodec C s6 | d | d | d | d fD ]}t |� qtd� d S )N�.py�.pyc�.pyw� $py.class� __pycache__)r r )�name�f� r �6/usr/local/lib/python3.10/test/test_import/__init__.py� remove_files% s �
r � c
c s� � | pd} t � �[}t�|| |�}tj�| d �}z'tj�d|� | |fV tj�|� W |d ur5|tj| < n&| tjv r>tj| = n|d urI|tj| < w | tjv rRtj| = w W d � d S W d � d S 1 sfw Y d S )N�spamr ) r r � make_script�sys�modules�pop�path�insert�remove)r �source�tempdirr' Z
old_moduler r r �_ready_to_import. s. �
��
�
�� "�r, c @ s$ e Zd Zdd� Zdd� Zdd� Zdd� Zd d
� Zd d � Ze d
d� �Z
dd� Z dd� Z dd� Z
dd� Zdd� Zdd� ZdAd d �Zd d � Zd d!� Zd"d#� Zd$d%� Zed&d'� �Zd(d)� Zd*d+� Zed,d-� �Zd.d/� Ze d0d1� �Z d2d3� Z d4d5� Z d6d7� Z e d8d9� �Z d:d;� Z!e"�#e$j%d<kd=�d>d?� �Z&d@S )B� ImportTestsc C s t t� t�� d S �N)r r � importlib�invalidate_caches��selfr r r �setUpE � zImportTests.setUpc C s t t� d S r. )r r r1 r r r �tearDownI � zImportTests.tearDownc C �8 | � t��
dd l}W d � d S 1 sw Y d S �Nr )� assertRaises�ModuleNotFoundError�(something_that_should_not_exist_anywhere�r2 r; r r r �&test_import_raises_ModuleNotFoundErrorL s
"�z2ImportTests.test_import_raises_ModuleNotFoundErrorc C �< | � t�� ddlm} W d � d S 1 sw Y d S )Nr )�blah)r9 r: r; r? )r2 r? r r r �:test_from_import_missing_module_raises_ModuleNotFoundErrorP � "�zFImportTests.test_from_import_missing_module_raises_ModuleNotFoundErrorc C r> )Nr )r; )r9 � ImportErrorr/ r; r< r r r �0test_from_import_missing_attr_raises_ImportErrorT rA z<ImportTests.test_from_import_missing_attr_raises_ImportErrorc C sn | � t��}ddlm} W d � n1 sw Y | �|jjd� | �|jjtj� | � t
|j�d� d S )Nr �� i_dont_exist�osz7cannot import name 'i_dont_exist' from 'os' \(.*os.py\)) r9 rB rF rE � assertEqual� exceptionr r' �__file__� assertRegex�str�r2 �cmrE r r r �/test_from_import_missing_attr_has_name_and_pathX s �z;ImportTests.test_from_import_missing_attr_has_name_and_pathc C sv dd l }| �t��}ddl m} W d � n1 sw Y | �|jjd� | �|jj|j� | � t
|j�d� d S )Nr rD � _testcapizCcannot import name 'i_dont_exist' from '_testcapi' \(.*\.(so|pyd)\)) rO r9 rB rE rG rH r r' rI rJ rK )r2 rO rM rE r r r �2test_from_import_missing_attr_has_name_and_so_path_ s �z>ImportTests.test_from_import_missing_attr_has_name_and_so_pathc C sX | � t��}ddlm} W d � n1 sw Y | �|jjd� | �|jj� d S )Nr rD �_warning) r9 rB rQ rE rG rH r Z assertIsNoner' rL r r r �&test_from_import_missing_attr_has_nameh s
�z2ImportTests.test_from_import_missing_attr_has_namec C sZ | � t��}ddlm} W d � n1 sw Y | �|jjddh� | �|j� d S )Nr rD � posixpath�ntpath)r9 rB Zos.pathrE �assertInrH r �assertIsNotNonerL r r r �/test_from_import_missing_attr_path_is_canonicalo s
�z;ImportTests.test_from_import_missing_attr_path_is_canonicalc C s� dd l }t� �S\}}t|ddd��
}|�d� W d � n1 s!w Y i }| �t|�|�� d��� td|� d�|� W d � n1 sHw Y | �d |� W d � n1 s]w Y t� �T\}}t|ddd��
}|�d
� W d � n1 sw Y i }| �t|�|�� d ��� td|� d�|� W d � n1 s�w Y | �d |� W d � d S 1 s�w Y d S ) Nr �w�utf-8��encodingz__all__ = [b'invalid_type']z\.__all__ must be strzfrom z import *s invalid_typez%globals()[b'invalid_type'] = object()z\.__dict__ must be str) �rer, �open�write�assertRaisesRegex� TypeError�escape�exec� assertNotIn)r2 r\ r r' r �globalsr r r �"test_from_import_star_invalid_typeu s2 �� � � �� �"�z.ImportTests.test_from_import_star_invalid_typec C r7 r8 )r9 rB �RAnDoM)r2 rf r r r �test_case_sensitivity� s
"�z!ImportTests.test_case_sensitivityc C s ddl m} d S )Nr )� double_const)�testrh )r2 rh r r r �test_double_const� s z ImportTests.test_double_constc sj � fdd�}t j�dtj� z!|d� t j�d�r"dD ]
}||� qW t jd= d S W t jd= d S t jd= w )Nc
sL t | }tr t d }nt d }t|ddd��(}td| d|d� t�d �}t�d �}td
||d� td ||d� W d � n1 sAw Y t tjv rOtjt = t� � zEzt
t �}W n t yu } z� � d | |f � W Y d }~nd }~ww � �
|j|d
| � � �
|j|d
| � W tt � t|� t|� d S tt � t|� t|� w )Nr r rX rY rZ z)# This tests Python's ability to import azfile.��filei� za =zb =zimport from %s failed: %sz'module loaded (%s) but contents invalid)r r r] �print�random� randranger$ r% r/ r0 �
__import__rB �failrG �a�br r )�extr* Zpycr rr rs �mod�errr1 r r �test_with_extension� sF
�
�
��
�
� �
z4ImportTests.test_import.<locals>.test_with_extensionr r �win)z.PYz.Pyz.pYr z.PYWz.pYw)r$ r' r( rF �curdir�platform�
startswith)r2 rw rt r r1 r � test_import� s "
� �zImportTests.test_import�longlistc C sL |d }t |ddd�� }|�d� td�D ]}|�d� q|�d� W d � n1 s,w Y z t�|� W t|� nt|� w tj�d � t �
� i }z3t |� t d
| d |� W tjd = t|d � t|d
� |�
� ztj|= W d S ty� Y d S w tjd = t|d � t|d
� |�
� ztj|= W w ty� Y w w )Nr rX rY rZ zd = [
i� z"",
�]r! zimport ����c�o)r] r^ �range�
py_compile�compiler r$ r' �appendr/ r0 r
rb �clearr% �KeyError)r2 �module�filenamer �i� namespacer r r � test_module_with_large_stack� sD
� �� �z(ImportTests.test_module_with_large_stackc C s� t d }t|ddd��}td|d� W d � n1 s w Y tj�dtj� t� � t tj
v r6tj
t = z!dD ]}| � t t
t � | �t tj
d | � q9W tjd= tt � d S tjd= tt � w )
Nr rX rY rZ za = 1/0rk r )� � � z'damaged module in sys.modules on %i try)r r] rm r$ r' r( rF ry r/ r0 r% r9 �ZeroDivisionErrorrp rc r )r2 r* r r� r r r �test_failing_import_sticks� s& �
�� �
z&ImportTests.test_failing_import_sticksc C sT dd l }dd l}| �|||j� | �t|jd�� dd lm} | �||j|j� d S )Nr rI )ri � test.support�assertIs�__name__�
assertTrue�hasattr�support)r2 �xri �yr r r �test_import_name_binding� s z$ImportTests.test_import_name_bindingc C s� z
d}W dd l m m} n
dd l m m} w d}|dk r/dd l m m} |d7 }|dk s td�D ] }dd l m m} q3d S )N.r �
r� r� )Ztest.support.script_helperr� r r� )r2 r� r� r r r �test_issue31286 s (� �zImportTests.test_issue31286c C sN t tj d }t|ddd��
}|�d� W d � n1 s w Y tj�dtj� znt t �}| �
t tj � | � |j
dd� | � |jd d� tt � t|ddd��
}|�d
� W d � n1 sbw Y | �ttj|� tj �t �}| �|d � | � |j
d d� | � |jd d� W tjd= tt � tt � d S tjd= tt � tt � w )
N�pyrX rY rZ z
a = 1
b=2
r r� z!module has wrong attribute valuesr� za = 10
b=20//0
z$expected module to be in sys.modulesr� )r rF �extsepr] r^ r$ r' r( ry rp rU r% rG rr rs r r9 r� r/ �reload�getrV r )r2 r* r ru r r r �test_failing_reload s2 � � �
z ImportTests.test_failing_reloadc C s t d }t|ddd��
}|�d� W d � n1 sw Y tj�dtj� zGtt �}| � |j
� d�� t� |� tj
t = t|� t�� tt �}tj�|j
�\}}| �|d� W tjd= tt � t tj
v rntj
t = d S d S tjd= tt � t tj
v r�tj
t = w )Nr rX rY rZ z test = None
r r )r r] r^ r$ r' r( rF ry rp r� rI �endswithr) r% r
r/ r0 �splitextrG r )r2 r* r ru �basert r r r �test_file_to_source? s0 �
��
z ImportTests.test_file_to_sourcec C s~ t j�t�}t�� }z|�|� W n ty" | �d� |�� Y nw | �
t ��
}t |� W d � d S 1 s8w Y d S )Nzpath is not encodable to {})
rF r' �abspathr r$ �getfilesystemencoding�encode�UnicodeEncodeError�skipTest�formatr9 rB rp )r2 r' r[ r� r r r �test_import_by_filenameW s �
"�z#ImportTests.test_import_by_filenamec C s" t �dtt�d��}t �|� d S )Nr! z� import sys
class C:
def __del__(self):
import importlib
sys.argv.insert(0, C())
)r r# r �textwrap�dedent�assert_python_ok)r2 Ztestfnr r r �!test_import_in_del_does_not_crasha s z-ImportTests.test_import_in_del_does_not_crashc
C s t j�dtj� zutd }tj�|�}t |ddd��}W d � n1 s%w Y zt�
|d� W n7 t y@ | � d� Y n* t
yi } z |jttdd �ttd d �fvrW� | � d
�|�� W Y d }~nd }~ww tt� t�|� W t jd= tt� d S t jd= tt� w ) Nr r rX rY rZ )� �� r� z-cannot set modification time to large integerZ EOVERFLOW�EINVALz2cannot set modification time to large integer ({}))r$ r' r( rF ry r r/ �util�cache_from_sourcer] �utime�
OverflowErrorr� �OSError�errno�getattrr� rp �statr )r2 r* Zcompiledr �er r r �test_timestamp_overflowl s2 �
� �� �
z#ImportTests.test_timestamp_overflowc C s4 z
t ddgd� W d S ty | �d� Y d S w )N�httpr? )�fromlistz fromlist must allow bogus names)rp rB rq r1 r r r �test_bogus_fromlist� s
�z ImportTests.test_bogus_fromlistc C s. ddg}t j|� }|�� \}}| �d|� d S )N�-cz&del __builtins__.__import__; import oss ImportError)r Z spawn_python� communicaterU )r2 �args�popen�stdout�stderrr r r �test_delete_builtins_import� s
z'ImportTests.test_delete_builtins_importc C �> | � td�� ddlm} W d � d S 1 sw Y d S )Nz^No module named 'bogus'r )�foo)r_ rB �bogusr� )r2 r� r r r �/test_from_import_message_for_nonexistent_module� � "�z;ImportTests.test_from_import_message_for_nonexistent_modulec C r� )Nz^cannot import name 'bogus'r )r� )r_ rB r\ r� )r2 r� r r r �,test_from_import_message_for_existing_module� r� z8ImportTests.test_from_import_message_for_existing_modulec C sv G dd� d�}d}| � t|� |� tj|< | �t��}ddlm} W d � n1 s+w Y | �t |j
�d� d S )Nc @ � e Zd Zdd� ZdS )zIImportTests.test_from_import_AttributeError.<locals>.AlwaysAttributeErrorc S s t �r. )�AttributeError)r2 �_r r r � __getattr__� s zUImportTests.test_from_import_AttributeError.<locals>.AlwaysAttributeError.__getattr__N)r� �
__module__� __qualname__r� r r r r �AlwaysAttributeError� � r� � test_from_import_AttributeErrorr ��does_not_existzScannot import name 'does_not_exist' from '<unknown module name>' (unknown location)) �
addCleanupr r$ r% r9 rB r� r� rG rK rH )r2 r� � module_namerM r� r r r r� � s � �z+ImportTests.test_from_import_AttributeErrorc C s� t tdd ��D | �t�� ddlm} W d � n1 s w Y | �t�� tj W d � n1 s5w Y W d � d S W d � d S 1 sMw Y d S )Nr� r r� )r rF r9 rB r� r� )r2 r� r r r �test_issue31492� s � ��"�zImportTests.test_issue31492c s dd� � t j�dtj�tj�t�d�� zid �� ��fdd��td�D ]S}t� � ��fdd �td
�D �}z,t
� |�j ��
t
�d� W d � n1 sLw Y W t j�d d � t j�d d � nt j�d d � t j�d d � w �d urv��q#W t jd= d S t jd= w )
Nc S s* |dkr| j jdkrt�d� d S d S d S )N�call� has_deadlockg�������?)�f_code�co_name�time�sleep)�frame�event�argr r r �delay_has_deadlock� s �z8ImportTests.test_concurrency.<locals>.delay_has_deadlockr �datac
sV t �� � ��� zdd l} W n ty# } z|�W Y d }~nd }~ww t �d � d S r8 )r$ �settrace�wait�package�
BaseException)r� r� )r� r� �excr r �run� s
��z)ImportTests.test_concurrency.<locals>.runr� c s g | ]}t j� d ��qS ))�target)� threading�Thread)�.0r� )r� r r �
<listcomp>� s z0ImportTests.test_concurrency.<locals>.<listcomp>r� r� zpackage.submodule)r$ r' r( rF �join�dirnamerI r� r� �Eventr Z
start_threads�setr� r� r% r& )r2 r� �threadsr )r� r� r� r� r �test_concurrency� s,
����� z ImportTests.test_concurrency�win32�Windows-specificc
C s� ddl m} |tj�}tj�d�j}tj �
tj � |�d� d|v r"dnd��}t
�� ��}tj �
|d�}t�|� tj �
|tj �tj��}t�tj|� t�||� t�tj �
t�tj�d��D ]}t�||� q^t�||� d } d d
� tjD �} |d tj � tj� | d < tj|d
d �
dd� tj � |��ddg�gtj| tj � |�d� t�||� tj|d
dgtj| tj � |�d� W d � d S 1 s�w Y d S )Nr )�GetModuleFileNameZ_sqlite3z
sqlite3{}.dllZ_dr! ZDLLszvcruntime*.dllc S s i | ]
}|� � tj| �qS r )�upperrF �environ)r� �kr r r �
<dictcomp>� s z:ImportTests.test_dll_dependency_import.<locals>.<dictcomp>�;Z
PYTHONPATHz-Scz import osz p = os.add_dll_directory({!r})zimport _sqlite3zp.close)r� �env�cwd) �_winapir� r$ Z dllhandler/ r� � find_spec�originrF r' r� r� r� r r �mkdir�basename�
executable�shutil�copy�globra �prefixr� rI �
subprocess�
check_call�STDOUT)
r2 r� ZdllnameZpydnameZdepname�tmpZtmp2Zpyexer r� r r r �test_dll_dependency_import� sP
�
���
�
�"�z&ImportTests.test_dll_dependency_importN)r} )'r� r� r� r3 r5 r= r@ rC rN r rP rR rW re rg rj r| r� r� r� r� r� �skip_if_dont_write_bytecoder� r� r� r� r� r� r� r� r� r� r� �unittest�
skipUnlessr$ rz r r r r r r- C sJ
,( &
"r- c @ sb e Zd Ze�ejdkd�dd� �Ze�ejdkd�dd� �Ze�ejdkd�dd� �Z d d
� Z
d S ) �FilePermissionTests�posix�%test meaningful only on posix systemsc C s� d}t |��7 t� �$\}}tj�|�}t|�}tj�|�s"| � d� t�
|�}W d � n1 s1w Y W d � n1 s@w Y | � t t
�
|j��t d| @ �� d S )N� �4__import__ did not result in creation of a .pyc filei� )r r, r/ r� r� rp rF r' �existsrq r� rG �oct�S_IMODE�st_mode)r2 �maskr r' � cached_pathr� � stat_infor r r �test_creation_mode s
��
�z&FilePermissionTests.test_creation_modec C s� d}t d��= t� �*\}}tj�|�}t�||� t|� tj� |�s(| �
d� t� |�}W d � n1 s7w Y W d � n1 sFw Y | � t
t �|j��t
|�� d S )N� r r �r r, r/ r� r� rF �chmodrp r' r rq r� rG r r r )r2 �moder r' r r r r r �test_cached_mode_issue_2051 s
�� z/FilePermissionTests.test_cached_mode_issue_2051c C s� d}t d��= t� �*\}}tj�|�}t�||� t|� tj� |�s(| �
d� t� |�}W d � n1 s7w Y W d � n1 sFw Y |dB }| � t
t �|j��t
|�� d S )N� r r � r! )r2 r# r r' r r Zexpectedr r r �test_cached_readonly, s
�� z(FilePermissionTests.test_cached_readonlyc C s` t � ��\}}t|ddd��
}|�d� W d � n1 s w Y t�|�}t�||j|jd f� t�|d� t |�}| �
|j d� t�|d� t|ddd��
}|�d � W d � n1 saw Y t |� t
�� t |�}| �
|j d
� t|� t |� t
�� |d }t�t
j�|�|� t |�}| �
|j d
� W d � d S 1 s�w Y d S ) NrX rY rZ zx = 'original'
i �r% Zoriginalr zx = 'rewritten'
Z rewrittenr� )r, r] r^ rF r� r� �st_atime�st_mtimer" rp rG r� r r/ r0 r �renamer� r� )r2 r r' r �s�mZ
bytecode_onlyr r r �test_pyc_always_writable< s2 �
�"�z,FilePermissionTests.test_pyc_always_writableN) r� r� r� r r rF r r r$ r' r- r r r r r s �
�
�
r c @ s| e Zd ZdZdZej�e�Z ej�
e e�ej d Z e
j�e �Zdd� Zdd� Zdd � Zd
d � Zd d
� Zdd� Zdd� ZdS )�PycRewritingTestsZunlikely_module_namez�
import sys
code_filename = sys._getframe().f_code.co_filename
module_filename = __file__
constant = 1
def func():
pass
func_filename = func.__code__.co_filename
r� c C s� t jd d � | _t j�| jd �| _t�| j � t
| j ddd��}|� | j
� W d � n1 s0w Y t j�d| j � t�� d S )NrX rY rZ r )r$ r' �sys_pathr% r& r� � orig_modulerF r �dir_namer] � file_namer^ �
module_sourcer( r/ r0 )r2 r r r r r3 n s � zPycRewritingTests.setUpc C sV | j tjd d �< | jd ur| jtj| j< nt| j� t| j� t| j � t
| j � d S r. ) r/ r$ r' r0 r% r� r r r2 �
compiled_namer r1 r1 r r r r5 w s
zPycRewritingTests.tearDownc C s t � }t| j||� tj| j S r. )rd rp r� r$ r% )r2 �nsr r r �
import_module� s z PycRewritingTests.import_modulec C s~ | � � }| �|j| j� | �|j| j� | �|j| j� tj| j= | � � }| �|j| j� | �|j| j� | �|j| j� d S r. ) r6 rG �module_filenamer2 �
code_filename�
func_filenamer$ r% r� �r2 ru r r r � test_basics� s
z PycRewritingTests.test_basicsc C sL t j| jdd� | �� }| �|j| j� | �|j| j� | �|j| j� d S �Nzanother_module.py)�dfile)r� r� r2 r6 rG r7 r8 r9 r: r r r �test_incorrect_code_name� s
z*PycRewritingTests.test_incorrect_code_namec C sh d}t j| j|d� t�| j� t| j�}t�� | �� }| � |j
|� | � |j |� | � |j |� d S r< )
r� r� r2 rF r) r
r/ r0 r6 rG r7 r8 r9 )r2 r� �pyc_fileru r r r �test_module_without_source� s
z,PycRewritingTests.test_module_without_sourcec C s� t �| j� t| jd��}|�d�}t�|�}W d � n1 s!w Y t|j �}t
j j }|�
d�}|||< |jt|�d�}t| jd��}|�|� t�||� W d � n1 s\w Y | � � }| �|jj|j� d S )N�rb� r� )� co_consts�wb)r� r� r2 r] r4 �read�marshal�load�listrC r/ r6 �__code__�index�replace�tupler^ �dumprG Zconstant� co_filename)r2 r �header�codeZ constantsZ foreign_code�posru r r r �test_foreign_code� s
�
�z#PycRewritingTests.test_foreign_codeN)r� r� r� r� r3 rF r' r� r r1 r� r� r2 r/ r� r� r4 r3 r5 r6 r; r> r@ rR r r r r r. [ s
r. c @ sF e Zd ZdZeZdd� Zdd� Zdd� Ze �
e j dkd �d
d � �Z
d S )
�
PathsTests)ri u testäöüßu testéèu
test°³²c C s t �| j� tjd d � | _d S r. )rF r r' r$ �syspathr1 r r r r3 � s zPathsTests.setUpc C s t | j� | jtjd d �< d S r. )r r' rT r$ r1 r r r r5 � s
zPathsTests.tearDownc C sx t tj�| jd�ddd��
}|�d� W d � n1 s w Y tj�| jd � td�}| �|j d� t
d� d S )Nztest_trailing_slash.pyrX rY rZ z testdata = 'test_trailing_slash'�/�test_trailing_slash) r] rF r' r� r^ r$ r� rp rG �testdatar )r2 r ru r r r rV � s � � z PathsTests.test_trailing_slashr� r� c
C s\ t tj�| jd�d��
}|�d� W d � n1 sw Y t�� tj�| j�}dd l}|� � }|d }d||f }||dd � 7 }zt�
|� W n t yl } z|j t j
t jt jfv rg| �d|f � � d }~ww tj�d|� ztd�}W n ty� } z| �d ||f � W Y d }~nd }~ww | �|jd� | �|j�|�|j� td� d S )
Nztest_unc_path.pyrX ztestdata = 'test_unc_path'r z\\%s\%s$r� z%cannot access administrative share %rZ
test_unc_pathz,could not import 'test_unc_path' from %r: %r) r] rF r' r� r^ r/ r0 r� �socket� gethostname�listdirr� r� �EPERMZEACCES�ENOENTr� r$ r( rp rB rq rG rW r� rI r{ r ) r2 r r' rX Zhn�driveZuncr� ru r r r �
test_UNC_path� s< � �� ��� zPathsTests.test_UNC_pathN)r� r� r� ZSAMPLESr r' r3 r5 rV r r r$ rz r^ r r r r rS � s
rS c @ sH e Zd Zdd� ZeZdd� Zdd� Zdd� Zd d
� Zd d � Z d
d� Z
dS )�RelativeImportTestsc C s t d� d S )Nztest.relimport)r r1 r r r r5 � r6 z RelativeImportTests.tearDownc C s ddl m} | �t|d�� d S )Nr� )� relimportr_ )r! r` r� r� )r2 r` r r r �test_relimport_star� s z'RelativeImportTests.test_relimport_starc s| � fdd�}t ddd�� |� t ddd�� |� t ddd�� | �t|� t ddd�� | �t|� t t� d�� | �t|� d S ) Nc s t d� � d S )Nzfrom . import relimport)rb r �r5 r r �check_relative� s z:RelativeImportTests.test_issue3221.<locals>.check_relativeri ztest.notarealmodule)� __package__r� znotarealpkg.notarealmoduler� )rd )�dictr9 r: �objectr` )r2 rc r rb r �test_issue3221� s z"RelativeImportTests.test_issue3221c C s t �dddd� d S )Nz-W�ignorer� zfoo = 1; from . import foo)r Zassert_python_failurer1 r r r �)test_parentless_import_shadowed_by_global s
�z=RelativeImportTests.test_parentless_import_shadowed_by_globalc C sF | � t�� ddlm} | �d� W d � d S 1 s w Y d S )Nr� )�sepz>explicit relative import triggered an implicit absolute import)r9 rB rF rj rq )r2 rj r r r �#test_absolute_import_without_future s "�z7RelativeImportTests.test_absolute_import_without_futurec
C s� t j�t j�t�dd�}tdd��L t|��/ | �t�� dd l }W d � n1 s+w Y | �
dt j � | �
dt j � W d � n1 sHw Y W d � d S W d � d S 1 s`w Y d S )Nr� �package2�
submodule1�
submodule2r )
rF r' r� r� rI r r
r9 rB rm rc r$ r% )r2 r' rm r r r � test_import_from_non_package s
�P�z0RelativeImportTests.test_import_from_non_packagec C s� t ddd��6 ttj�tj�t�d��� dd l}|jj W d � n1 s'w Y W d � d S W d � d S 1 s?w Y d S )Nrl zpackage2.submodule1zpackage2.submodule2r� r )
r r
rF r' r� r� rI Zpackage2.submodule1rm rn )r2 rl r r r �!test_import_from_unloaded_package( s �
P�z5RelativeImportTests.test_import_from_unloaded_packageN) r� r� r� r5 r3 ra rg ri rk ro rp r r r r r_ � s r_ c @ r� )� OverridingImportBuiltinTestsc C s� dd l }dd� }| �|� |� ttddd� �� | �|� d� W d � n1 s)w Y tt� ddd� �� | �|� |� W d � d S 1 sJw Y d S ) Nr c S s dd l } | S r8 �rF rr r r r r� 4 s z?OverridingImportBuiltinTests.test_override_builtin.<locals>.foorp c W � dS �N� r �r� r r r �<lambda>9 � zDOverridingImportBuiltinTests.test_override_builtin.<locals>.<lambda>ru c W rs rt r rv r r r rw ? rx )rF rG r �builtinsr rd )r2 rF r� r r r �test_override_builtin0 s �"�z2OverridingImportBuiltinTests.test_override_builtinN)r� r� r� rz r r r r rq / r� rq c @ s� e Zd Zdd� Zdd� Zdd� Zedd� �Ze� e
j d kd
�e� e
e
d �o*e
�� d kd
�edd� ���Zedd� �Zedd� �Zdd� Zedd� �Zedd� �Zdd� Zd d � Zd S ) � PycacheTestsc C s t t� td� t| j� d S )Nr )r r r r r* r1 r r r �_cleanF s zPycacheTests._cleanc C sj t d | _| �� t| jddd��}td|d� W d � n1 s"w Y tj�dtj � t
� � d S )Nr rX rY rZ z/# This is a test file written by test_import.pyrk r ) r r* r| r] rm r$ r' r( rF ry r/ r0 )r2 �fpr r r r3 K s
� zPycacheTests.setUpc C s, t jd tjks J d��t jd= | �� d S )Nr zUnexpected sys.path[0])r$ r' rF ry r| r1 r r r r5 S s zPycacheTests.tearDownc C sZ | � tj�d�� tt� | �tj�d�� tj� | j
�}| �tj�|�d� |t�� d S )Nr z*bytecode file {!r} for {!r} does not exist) � assertFalserF r' r rp r r� r/ r� r� r* r� �r2 Zpyc_pathr r r �test_import_pyc_pathX s �z!PycacheTests.test_import_pyc_pathr r �geteuidr z=due to varying filesystem permission semantics (issue #11956)c C sp t d�� tt� W d � n1 sw Y | �tj�d�� tj� | j
�}| � tj�|�d� |t�� d S )N� r z"bytecode file {!r} for {!r} exists)
r rp r r� rF r' r r/ r� r� r* r~ r� r r r r �test_unwritable_directoryb s
��z&PycacheTests.test_unwritable_directoryc C sV t t� tj�| j�}| �tj� |�� t�
| j� t t� t� � | �
tt t� d S r. )rp r r/ r� r� r* r� rF r' r r) r r0 r9 rB )r2 r? r r r �test_missing_sourcer s z PycacheTests.test_missing_sourcec
C s| t t� t| j�}t�| j� tt� t�� t t�}z | � |j
tj � t�
� tjtj �|��� W t�|� d S t�|� w r. )rp r r
r* rF r) r r/ r0 rG rI r' r� �getcwdry �relpath�r2 r? r, r r r �test_missing_source_legacy~ s
�z'PycacheTests.test_missing_source_legacyc C s< t t�}tj�td �}| �|jtj� t�
� tj |�� d S )Nr ) rp r r/ r� r� rG �
__cached__rF r' r� r� ry )r2 r, r? r r r �test___cached__� s $z PycacheTests.test___cached__c
C sb t t� t| j�}t�| j� tt� t�� t t�}| � |j
tj � t�
� tjtj �|��� d S r. )rp r r
r* rF r) r r/ r0 rG r� r' r� r� ry r� r� r r r �test___cached___legacy_pyc� s
�z'PycacheTests.test___cached___legacy_pycc C s
dd� }t �d� | �|� tt j�dd�d�� W d � n1 s#w Y tt j�dd�d�� W d � n1 s=w Y t�� td�}tj �
t j�dd��}| � |j t j�t �
� t j|�� tj �
t j�dd��}| � tjd j t j�t �
� t j|�� d S )Nc S � t d� td� td� d S �N�pep3147� pep3147.foo�r r r r r r �cleanup� � z5PycacheTests.test_package___cached__.<locals>.cleanupr� � __init__.pyrD �foo.pyr� )rF r r� r] r' r� r/ r0 rp r� r� rG r� r� ry r$ r% �r2 r� r, Zinit_pycZfoo_pycr r r �test_package___cached__� s&
� � � �z$PycacheTests.test_package___cached__c C s* dd� }t �d� | �|� tt j�dd�d�� W d � n1 s#w Y tt j�dd�d�� W d � n1 s=w Y t�� td�}t d� t d� t�� td�}tj
� t j�dd��}| � |j
t j�t �� t j|�� tj
� t j�dd��}| � tjd j
t j�t �� t j|�� d S )Nc S r� r� r� r r r r r� � r� z>PycacheTests.test_package___cached___from_pyc.<locals>.cleanupr� r� rD r� r� )rF r r� r] r' r� r/ r0 rp r r� r� rG r� r� ry r$ r% r� r r r � test_package___cached___from_pyc� s.
� � � �z-PycacheTests.test_package___cached___from_pycc C sf t t� tt� t| jddd��}td|d� W d � n1 s!w Y t t�}| �|jd� d S )Nrr rY rZ zx = 5rk ru )rp r r r] r* rm rG r� )r2 r} r, r r r � test_recompute_pyc_same_second� s �z+PycacheTests.test_recompute_pyc_same_secondN)r� r� r� r| r3 r5 r r� r r rF r �skipIfr� r� r� r� r� r� r� r� r� r� r r r r r{ C s2
��
r{ c @ sV e Zd ZdZed Zdd� Zdd� Ze�e e
d� p e
� � dkd �e j
d
d � ��Zd S )
� TestSymbolicallyLinkedPackage�samplez-taggedc C s� t �| j� t �| j� tjd d � | _t�| j� | � t j| j� tj�
| jd�}t � |� tj� |�s7J �tj
| j| jdd� | � t j| j� t�� | �tj�| j�d� tj�tj�
| jd��sfJ �d S )Nr� T)�target_is_directory)r r �tagged� package_namer$ r' �
orig_sys_pathrF r r� r� Zcreate_empty_filer �symlinkr r/ r0 rG �isdir�isfile)r2 Z init_filer r r r3 � s
z#TestSymbolicallyLinkedPackage.setUpc C s | j tjd d �< d S r. )r� r$ r' r1 r r r r5 � s z&TestSymbolicallyLinkedPackage.tearDown�getwindowsversion)� r z Windows Vista or later requiredc C sN dgt jd d �< tj�| j�sJ �tj�tj�| jd��s J �t�| j� d S )N�.r� )r$ r' rF r r� r� r/ r6 r1 r r r � test_symlinked_dir_importable s z;TestSymbolicallyLinkedPackage.test_symlinked_dir_importableN)r� r� r� r� r� r3 r5 r r r� r$ r� r Zskip_unless_symlinkr� r r r r r� � s
�r� c @ s, e Zd Zdd� Zdd� Zdd� Zdd� Zd S )
�ImportlibBootstrapTestsc C s t jd }| �|� d S )N�_frozen_importlib)r$ r% r� r: r r r �test_frozen_importlib s
z-ImportlibBootstrapTests.test_frozen_importlibc C �X ddl m} tjd }| �||� | �|jd� | �|jd� | �|j �
d�|j � d S )Nr )�
_bootstrapr� zimportlib._bootstrapr/ z
_bootstrap.py) r/ r� r$ r% r� rG r� rd r� rI r� )r2 r� ru r r r �"test_frozen_importlib_is_bootstrap �
z:ImportlibBootstrapTests.test_frozen_importlib_is_bootstrapc C r� )Nr )�_bootstrap_external�_frozen_importlib_externalz importlib._bootstrap_externalr/ z_bootstrap_external.py) r/ r� r$ r% r� rG r� rd r� rI r� )r2 r� ru r r r �4test_frozen_importlib_external_is_bootstrap_external" r� zLImportlibBootstrapTests.test_frozen_importlib_external_is_bootstrap_externalc C s* ddl m} tjd }| �|j|j� d S )Nr )� machineryr� )r/ r� r$ r% r� �
ModuleSpec)r2 r� ru r r r �test_there_can_be_only_one* s
z2ImportlibBootstrapTests.test_there_can_be_only_oneN)r� r� r� r� r� r� r� r r r r r� s
r� c @ s( e Zd ZdZdd� Zdd� Zdd� ZdS ) �GetSourcefileTestsz�Test importlib._bootstrap_external._get_sourcefile() as used by the C API.
Because of the peculiarities of the need of this function, the tests are
knowingly whitebox tests.
c C sV t �d�� }d|_td }td }| �t|�|� W d � d S 1 s$w Y d S )N�*importlib._bootstrap_external._path_isfileTr r �r ZpatchZ return_valuer rG r )r2 � _path_isfiler' �expectr r r �test_get_sourcefile= s "�z&GetSourcefileTests.test_get_sourcefilec C sN t �d��}d|_td }| �t|�|� W d � d S 1 s w Y d S )Nr� Fr r� )r2 r� r' r r r � test_get_sourcefile_no_sourceF s
"�z0GetSourcefileTests.test_get_sourcefile_no_sourcec C s t d }| �t|�|� d S )Nz.bad_ext)r rG r )r2 r' r r r �test_get_sourcefile_bad_extN s z.GetSourcefileTests.test_get_sourcefile_bad_extN)r� r� r� �__doc__r� r� r� r r r r r� 3 s
r� c @ s� e Zd Zdd� Zdd� Zd$dd�Zdd � Zd
d � Zd d
� Zdd� Z dd� Z
dd� Z dd� Z dd� Z
dd� Zdd� Zd d � Zed d � �Ze�ed �d!d"� �Zd#S )%�ImportTracebackTestsc C s, t �t� tjd d � | _tj�dt� d S r8 )rF r r r$ r' �old_pathr( r1 r r r r3 W s
zImportTracebackTests.setUpc C s | j tjd d �< tt� d S r. )r� r$ r' r r r1 r r r r5 \ s z ImportTracebackTests.tearDownr c C sb t j�t|| �}t|ddd��
}|�|� W d � n1 s w Y | �t|� t� � |S )NrX rY rZ )
rF r' r� r r] r^ r� r r/ r0 )r2 ru �contentsrt Zfnamer r r r �
create_module` s � z"ImportTracebackTests.create_modulec C st g }|r |j j}|j}|r||d kr|�|� |j}|s| �t|�t|�|� t||�D ]
\}}| �||� q-d S )Nr ) �tb_framer� rN r� �tb_nextrG �len�ziprU )r2 �tb�filesZ
deduped_filesrP �fn�patr r r �assert_tracebackh s
��z%ImportTracebackTests.assert_tracebackc
C sT zdd l }W n ty } z |j}W Y d }~n
d }~ww | �d� | �|tg� d S )Nr �#ImportError should have been raised)�nonexistent_xyzzyrB �
__traceback__rq r� rI )r2 r� r� r� r r r �test_nonexistent_modulet s ��
z,ImportTracebackTests.test_nonexistent_modulec
C �b | � dd� zdd l}W n ty! } z |j}W Y d }~n
d }~ww | �d� | �|tdg� d S )Nr� zimport nonexistent_xyzzyr r� r� )r� r� rB r� rq r� rI �r2 r� r� r� r r r � test_nonexistent_module_nested~ � ��
z3ImportTracebackTests.test_nonexistent_module_nestedc
C r� )Nr� �1/0r �)ZeroDivisionError should have been raisedr� �r� r� r� r� rq r� rI r� r r r �test_exec_failure� r� z&ImportTracebackTests.test_exec_failurec
C sp | � dd� | � dd� zdd l}W n ty' } z |j}W Y d }~n
d }~ww | �d� | �|tddg� d S ) Nr� z
import bar�barr� r r� r� �bar.pyr� r� r r r �test_exec_failure_nested� s ��
z-ImportTracebackTests.test_exec_failure_nestedc
C s` | � dd� zdd l}W n ty! } z |j}W Y d }~n
d }~ww | �d� | �|tg� d S )Nr� zinvalid syntax is invalidr z#SyntaxError should have been raised)r� r� � SyntaxErrorr� rq r� rI r� r r r �test_syntax_error� s ��
z&ImportTracebackTests.test_syntax_errorc C s� d}| � t|� tj�t|�}t�|� tj�|d�}t|ddd��
}|�|� W d � n1 s2w Y tj�|d�}t|ddd��
}|�|� W d � n1 sUw Y t �
� ||fS )N� _parent_foor� rX rY rZ r� ) r� r rF r' r� r r r] r^ r/ r0 )r2 �parent�childZpkg_nameZpkg_path� init_pathr �bar_pathr r r �_setup_broken_package� s
� �z*ImportTracebackTests._setup_broken_packagec
C sf | � dd�\}}zdd l}W n ty# } z |j}W Y d }~n
d }~ww | �d� | �|t|g� d S )Nr! r� r r� �r� Z_parent_foo.barr� r� rq r� rI �r2 r� r� r� r� r� r r r �test_broken_submodule� � ��
z*ImportTracebackTests.test_broken_submodulec
C sj | � dd�\}}zddlm} W n ty% } z |j}W Y d }~n
d }~ww | �d� | �|t|g� d S )Nr! r� r �r� r� �r� r� r� r� r� rq r� rI �r2 r� r� r� r� r� r r r �test_broken_from� � ��
z%ImportTracebackTests.test_broken_fromc
C sf | � dd�\}}zdd l}W n ty# } z |j}W Y d }~n
d }~ww | �d� | �|t|g� d S )Nr� r! r r� r� r� r r r �test_broken_parent� r� z'ImportTracebackTests.test_broken_parentc
C sj | � dd�\}}zddlm} W n ty% } z |j}W Y d }~n
d }~ww | �d� | �|t|g� d S )Nr� r! r r� r� r� r� r r r �test_broken_parent_from� r� z,ImportTracebackTests.test_broken_parent_fromc
C s� | � dd� tjd }dt|j�v r|jj}nd }zBdd� }||j_zdd l}W n ty= } z |j}W Y d }~n
d }~ww | � d� | �
|t d t g� W |d u rV|j`d S ||j_d S |d u rd|j`w ||j_w )
Nr� r! r� � load_modulec W s dd d S )Nr� r r )r� r r r � exec_module� r6 z9ImportTracebackTests.test_import_bug.<locals>.exec_moduler r� z<frozen importlib) r� r$ r% �vars� SourceLoaderr� r� r� r� rq r� rI )r2 r/ Zold_exec_moduler� r� r� r� r r r �test_import_bug� s,
��
� z$ImportTracebackTests.test_import_bugzneed TESTFN_UNENCODABLEc C s> t �dtd�}| �t|� |d d� }t jdd| dd� d S )Nr! �pass���r� zmod = __import__(%a)F)Z
__isolated)r r# r r� r r� )r2 Zpynamer r r r �test_unencodable_filename� s
�z.ImportTracebackTests.test_unencodable_filenameN)r )r� r� r� r3 r5 r� r� r� r� r� r� r� r� r� r� r� r� r r� r r r r� r r r r r� U s&
r� c @ s` e Zd ZdZdd� Zdd� Zdd� Zdd � Zd
d � Zd d
� Z dd� Z
dd� Z dd� Z dd� Z
dS )�CircularImportTestszQSee the docstrings of the modules being imported for the purpose of the
test.c C s* t tj�� �D ] }|�d�rtj|= qdS )zSMake sure no modules pre-exist in sys.modules which are being used to
test.z&test.test_import.data.circular_importsN)rH r$ r% �keysr{ )r2 �keyr r r r5
s
��z CircularImportTests.tearDownc C �. zdd l }W d S ty | �d� Y d S w )Nr z/circular import through relative imports failed)Z,test.test_import.data.circular_imports.basicrB rq �r2 ri r r r � test_direct �
�z CircularImportTests.test_directc C r� )Nr z@relative import in module contributing to circular import failed)Z/test.test_import.data.circular_imports.indirectrB rq r� r r r �
test_indirect r� z!CircularImportTests.test_indirectc C r� )Nr z-circular import involving a subpackage failed)Z1test.test_import.data.circular_imports.subpackagerB rq r� r r r �test_subpackage r� z#CircularImportTests.test_subpackagec C s` zdd l m m m m} W n ty | �d� Y nw ddlm} | � |j|j� d S )Nr z9circular import with rebinding of module attribute failed)r� )
Z0test.test_import.data.circular_imports.rebindingr| r� Zcircular_imports� rebindingrB rq Z-test.test_import.data.circular_imports.subpkgr� r� )r2 r� r� r r r �test_rebinding$ s " � z"CircularImportTests.test_rebindingc C r� )Nr z9circular import with binding a submodule to a name failed)Z.test.test_import.data.circular_imports.bindingrB rq r� r r r � test_binding, r� z CircularImportTests.test_bindingc C s dd l }dd l}d S r8 )Z*test.test_import.data.circular_imports.use�-test.test_import.data.circular_imports.sourcer� r r r �test_crossreference12 r4 z(CircularImportTests.test_crossreference1c C sp | � t�� }dd l}W d � n1 sw Y t|j�}| �d|� | �d|� | �d|� | �d|� d S )Nr z-test.test_import.data.circular_imports.sourcer" z partially initialized modulezcircular import)r9 r� r rK rH rU )r2 rM ri �errmsgr r r �test_crossreference26 s
�
z(CircularImportTests.test_crossreference2c C sH | � t�� }dd l}W d � n1 sw Y | �dt|j�� d S )Nr z�cannot import name 'b' from partially initialized module 'test.test_import.data.circular_imports.from_cycle1' (most likely due to a circular import))r9 rB Z2test.test_import.data.circular_imports.from_cycle1rU rK rH )r2 rM ri r r r �test_circular_from_import? s
��z-CircularImportTests.test_circular_from_importc C s� | � td� | � td� dd lm m m} | �t�� ddlm} W d � n1 s.w Y | � t
|�t � | � t
|�t � | �
t�� d|_W d � d S 1 sWw Y d S )Nz test.test_import.data.unwritablez"test.test_import.data.unwritable.xr rv �* )r� r Z test.test_import.data.unwritabler| r� �
unwritableZ assertWarns�
ImportWarningr� ZassertNotEqual�typer rG r9 r� )r2 r r� r r r �test_unwritable_moduleI s � "�z*CircularImportTests.test_unwritable_moduleN)r� r� r� r� r5 r� r� r� r� r� r r r r r r r r r� s
r� �__main__)Nr! )Bry �
contextlibr� r �importlib.utilr/ Z importlib._bootstrap_externalr rF rF r� rn r r� r r$ r� r� r� r r r� r r r r r Ztest.support.import_helperr r
r r r
Ztest.support.os_helperr r r r r r r Ztest.test_importlib.utilr �typesr r� �dont_write_bytecoder r �contextmanagerr, ZTestCaser- r r. rS r_ rq r{ r� r� r� r� r� r� �mainr r r r �<module> sr � GSY6E ". ! 1
R �