Your IP : 216.73.216.52
o
0�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 �@/opt/alt/python310/lib64/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 �sourceZtempdirr' 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 r0 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�r1 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)r8 r9 r: r> )r1 r> r r r �:test_from_import_missing_module_raises_ModuleNotFoundErrorP � "�zFImportTests.test_from_import_missing_module_raises_ModuleNotFoundErrorc C r= )Nr )r: )r8 � ImportErrorr. r: r; r r r �0test_from_import_missing_attr_raises_ImportErrorT r@ 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\)) r8 rA rE rD � assertEqual� exceptionr r' �__file__� assertRegex�str�r1 �cmrD 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 rC � _testcapizCcannot import name 'i_dont_exist' from '_testcapi' \(.*\.(so|pyd)\)) rN r8 rA rD rF rG r r' rH rI rJ )r1 rN rL rD 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 rC �_warning) r8 rA rP rD rF rG r Z assertIsNoner' rK 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 rC � posixpathZntpath)r8 rA Zos.pathrD �assertInrG r �assertIsNotNonerK 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)r1 rZ r r' r �globalsr r r �"test_from_import_star_invalid_typeu s2 �� � � �� �"�z.ImportTests.test_from_import_star_invalid_typec C r6 r7 )r8 rA �RAnDoM)r1 rd r r r �test_case_sensitivity� s
"�z!ImportTests.test_case_sensitivityc C s ddl m} d S )Nr )� double_const)�testrf )r1 rf 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 rV rW rX 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�randomZ randranger$ r% r. r/ �
__import__rA �failrF �a�br r )�extr* Zpycr ro rp �mod�errr0 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( rE �curdir�platform�
startswith)r1 rt rq r r0 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 rV rW rX zd = [
i� z"",
�]r! zimport ����c�o)r[ r\ �range�
py_compile�compiler r$ r' �appendr. r/ r
r` �clearr% �KeyError)r1 �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 rV rW rX za = 1/0ri r )� � � z'damaged module in sys.modules on %i try)r r[ rk r$ r' r( rE rv r. r/ r% r8 �ZeroDivisionErrorrm ra r )r1 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 rH )rg � test.support�assertIs�__name__�
assertTrue�hasattr�support)r1 �xrg �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 )r1 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�pyrV rW rX 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 rE �extsepr[ r\ r$ r' r( rv rm rS r% rF ro rp r r8 r� r. �reload�getrT r )r1 r* r rr 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 rV rW rX z test = None
r r )r r[ r\ r$ r' r( rE rv rm r� rH �endswithr) r% r
r. r/ �splitextrF r )r1 r* r rr �baserq 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 {})
rE r' �abspathr r$ �getfilesystemencoding�encode�UnicodeEncodeError�skipTest�formatr8 rA rm )r1 r' rY 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)r1 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 rV rW rX )� �� r� z-cannot set modification time to large integerZ EOVERFLOWZEINVALz2cannot set modification time to large integer ({}))r$ r' r( rE rv r r. �util�cache_from_sourcer[ �utime�
OverflowErrorr� �OSError�errno�getattrr� rm �statr )r1 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 )NZhttpr> )�fromlistz fromlist must allow bogus names)rm rA rn r0 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_pythonZ communicaterS )r1 �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] rA �bogusr� )r1 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] rA rZ r� )r1 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)r1 �_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% r8 rA r� r� rF rJ rG )r1 r� � module_namerL 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 rE r8 rA r� r� )r1 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 )NZcall� 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 r7 )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)� threadingZThread)�.0r� )r� r r �
<listcomp>� s z0ImportTests.test_concurrency.<locals>.<listcomp>r� r� zpackage.submodule)r$ r' r( rE �join�dirnamerH r r� ZEventr Z
start_threads�setr� r� r% r& )r1 r� Zthreadsr )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 )�upperrE �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) Z_winapir� r$ Z dllhandler. r� � find_spec�originrE r' r� r� r� r r �mkdir�basename�
executable�shutil�copy�globr_ �prefixr� rH �
subprocessZ
check_callZSTDOUT)
r1 r� ZdllnameZpydnameZdepname�tmpZtmp2Zpyexer r� r r r �test_dll_dependency_import� sP
�
���
�
�"�z&ImportTests.test_dll_dependency_importN)rz )'r� r� r� r2 r4 r< r? rB rM r rO rQ rU rc re rh ry r� r� r� r� r� �skip_if_dont_write_bytecoder� r� r� r� r� r� r� r� r� r� r� �unittest�
skipUnlessr$ rw 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� rm rE r' �existsrn r� rF �oct�S_IMODE�st_mode)r1 �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� rE �chmodrm r' r rn r� rF r r
r )r1 �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 )r1 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 ) NrV rW rX zx = 'original'
i �r Zoriginalr zx = 'rewritten'
Z rewrittenr} )r+ r[ r\ rE r� r� �st_atime�st_mtimer rm rF r� r r. r/ r �renamer� r� )r1 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 rE 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 )NrV rW rX r )r$ r' �sys_pathr% r& r� � orig_modulerE r� �dir_namer[ � file_namer\ �
module_sourcer( r. r/ )r1 r r r r r2 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' r# r% r� r r r% �
compiled_namer r$ r0 r r r r4 w s
zPycRewritingTests.tearDownc C s t � }t| j||� tj| j S r- )rb rm r� r$ r% )r1 �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- ) r) rF �module_filenamer% �
code_filename�
func_filenamer$ r% r� �r1 rr 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� r% r) rF r* r+ r, 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� r% rE r) r
r. r/ r) rF r* r+ r, )r1 r� �pyc_filerr 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� r% r[ r' �read�marshal�load�listr6 r. r) �__code__�index�replace�tupler\ �dumprF Zconstant� co_filename)r1 r �header�codeZ constantsZ foreign_code�posrr r r r �test_foreign_code� s
�
�z#PycRewritingTests.test_foreign_codeN)r� r� r� r� r& rE r' r� r r$ r� r� r% r. r� r� r' r2 r4 r) r. r1 r3 rE r r r r r! [ s
r! c @ sF e Zd ZdZeZdd� Zdd� Zdd� Ze �
e j dkd �d
d � �Z
d S )
�
PathsTests)rg u testäöüßu testéèu
test°³²c C s t �| j� tjd d � | _d S r- )rE r� r' r$ �syspathr0 r r r r2 � s zPathsTests.setUpc C s t | j� | jtjd d �< d S r- )r r' rG r$ r0 r r r r4 � 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.pyrV rW rX z testdata = 'test_trailing_slash'�/�test_trailing_slash) r[ rE r' r� r\ r$ r� rm rF �testdatar )r1 r rr r r r rI � 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.pyrV 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[ rE r' r� r\ r. r/ r� �socketZ gethostname�listdirr� r� ZEPERMZEACCESZENOENTr� r$ r( rm rA rn rF rJ r� rH rx r ) r1 r r' rK ZhnZdriveZuncr� rr r r r �
test_UNC_path� s< � �� ��� zPathsTests.test_UNC_pathN)r� r� r� ZSAMPLESr r' r2 r4 rI r r r$ rw rM r r r r rF � s
rF 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 r0 r r r r4 � r5 z RelativeImportTests.tearDownc C s ddl m} | �t|d�� d S )Nr� )� relimportrN )r! rO r� r� )r1 rO 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)r` r �r( r r �check_relative� s z:RelativeImportTests.test_issue3221.<locals>.check_relativerg ztest.notarealmodule)� __package__r� znotarealpkg.notarealmoduler� )rS )�dictr8 r9 �objectr^ )r1 rR r rQ 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_failurer0 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)r8 rA rE rY rn )r1 rY 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 )
rE r' r� r� rH r r
r8 rA r\ ra r$ r% )r1 r' r\ 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 )Nr[ zpackage2.submodule1zpackage2.submodule2r� r )
r r
rE r' r� r� rH Zpackage2.submodule1r\ r] )r1 r[ r r r �!test_import_from_unloaded_package( s �
P�z5RelativeImportTests.test_import_from_unloaded_packageN) r� r� r� r4 r2 rP rV rX rZ r^ r_ r r r r rN � s rN 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 r7 �rE ra r r r r� 4 s z?OverridingImportBuiltinTests.test_override_builtin.<locals>.foorm c W � dS �N� r �r� r r r �<lambda>9 � zDOverridingImportBuiltinTests.test_override_builtin.<locals>.<lambda>rd c W rb rc r re r r r rf ? rg )rE rF r �builtinsr rb )r1 rE r� r r r �test_override_builtin0 s �"�z2OverridingImportBuiltinTests.test_override_builtinN)r� r� r� ri r r r r r` / r� r` 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* r0 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 rV rW rX z/# This is a test file written by test_import.pyri r ) r r* rk r[ rk r$ r' r( rE rv r. r/ )r1 �fpr r r r2 K s
� zPycacheTests.setUpc C s, t jd tjks J d��t jd= | �� d S )Nr zUnexpected sys.path[0])r$ r' rE rv rk r0 r r r r4 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) � assertFalserE r' r rm r r� r. r� r� r* r� �r1 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 rm r r� rE r' r r. r� r� r* rm r� rn 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- )rm r r. r� r� r* r� rE r' r r) r r/ r8 rA )r1 r2 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- )rm r r
r* rE r) r r. r/ rF rH r' r� �getcwdrv �relpath�r1 r2 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 ) rm r r. r� r� rF �
__cached__rE r' r� rt rv )r1 r r2 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- )rm r r
r* rE r) r r. r/ rF rx r' r� rt rv ru rv 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__.pyr7 �foo.pyr~ )rE r� r� r[ r' r� r. r/ rm r� r� rF rx rt rv r$ r% �r1 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� r7 r� r~ )rE r� r� r[ r' r� r. r/ rm r r� r� rF rx rt rv 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 )Nro rW rX zx = 5ri rd )rm r r r[ r* rk rF r� )r1 rl r r r r � test_recompute_pyc_same_second� s �z+PycacheTests.test_recompute_pyc_same_secondN)r� r� r� rk r2 r4 r ro r r rE r �skipIfr� rp rr rs rw ry rz r� r� r� r r r r rj C s2
��
rj c @ sV e Zd ZdZed Zdd� Zdd� Ze�e e
d� p e
� � dkd �e j
d
d � ��Zd S )
� TestSymbolicallyLinkedPackageZsamplez-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)Ztarget_is_directory)r r �tagged� package_namer$ r' �
orig_sys_pathrE r� r� r� Zcreate_empty_filer �symlinkr r. r/ rF �isdir�isfile)r1 Z init_filer r r r2 � s
z#TestSymbolicallyLinkedPackage.setUpc C s | j tjd d �< d S r- )r� r$ r' r0 r r r r4 � 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' rE r r� r� r. r) r0 r r r � test_symlinked_dir_importable s z;TestSymbolicallyLinkedPackage.test_symlinked_dir_importableN)r� r� r� r� r� r2 r4 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� rF r� rS r� rH r� )r1 r� rr 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� rF r� rS r� rH r� )r1 r� rr 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)r1 r� rr 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 rF r )r1 � _path_isfiler' Zexpectr 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� )r1 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 rF r )r1 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 r7 )rE r� r r$ r' �old_pathr( r0 r r r r2 W s
zImportTracebackTests.setUpc C s | j tjd d �< tt� d S r- )r� r$ r' r r r0 r r r r4 \ s z ImportTracebackTests.tearDownr c C sb t j�t|| �}t|ddd��
}|�|� W d � n1 s w Y | �t|� t� � |S )NrV rW rX )
rE r' r� r r[ r\ r� r r. r/ )r1 rr �contentsrq 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� rA r� �tb_nextrF �len�ziprS )r1 �tb�filesZ
deduped_filesrC �fnZpatr 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_xyzzyrA �
__traceback__rn r� rH )r1 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� rA r� rn r� rH �r1 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� rn r� rH 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� rn r� rH 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� rV rW rX r� ) r� r rE r' r� r r� r[ r\ r. r/ )r1 �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� rn r� rH �r1 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� rn r� rH �r1 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� r5 z9ImportTracebackTests.test_import_bug.<locals>.exec_moduler r� z<frozen importlib) r� r$ r% �vars� SourceLoaderr� r� r� r� rn r� rH )r1 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� )r1 Zpynamer r r r �test_unencodable_filename� s
�z.ImportTracebackTests.test_unencodable_filenameN)r )r� r� r� r2 r4 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)r; r$ r% �keysrx )r1 �keyr r r r4
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.basicrA rn �r1 rg 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.indirectrA rn 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.subpackagerA rn 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.rebindingry r� Zcircular_imports� rebindingrA rn Z-test.test_import.data.circular_imports.subpkgr� r� )r1 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.bindingrA rn r� r r r � test_binding, r� z CircularImportTests.test_bindingc C s dd l }dd l}d S r7 )Z*test.test_import.data.circular_imports.use�-test.test_import.data.circular_imports.sourcer� r r r �test_crossreference12 r3 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)r8 r� r� rJ rG rS )r1 rL rg �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))r8 rA Z2test.test_import.data.circular_imports.from_cycle1rS rJ rG )r1 rL rg 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 re �* )r� r Z test.test_import.data.unwritablery r� �
unwritableZ assertWarns�
ImportWarningr� ZassertNotEqual�typer rF r8 r� )r1 r� r� r r r �test_unwritable_moduleI s � "�z*CircularImportTests.test_unwritable_moduleN)r� r� r� r� r4 r� r� r� r� r� r� r� r� r� r r r r r� s
r� �__main__)Nr! )Brh �
contextlibr� r� �importlib.utilr. Z importlib._bootstrap_externalr r9 rE r� rl 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! rF rN r` rj r� r� r� r� r� r� �mainr r r r �<module> sr � GSY6E ". ! 1
R �