| Server IP : 127.0.0.1 / Your IP : 216.73.216.109 Web Server : Apache/2.4.54 (Win64) OpenSSL/1.1.1q PHP/8.1.10 System : Windows NT DESKTOP-E5T4RUN 10.0 build 19045 (Windows 10) AMD64 User : SERVERWEB ( 0) PHP Version : 8.1.10 Disable Function : NONE MySQL : OFF | cURL : ON | WGET : OFF | Perl : OFF | Python : OFF | Sudo : OFF | Pkexec : OFF Directory : C:/cygwin64/lib/python3.9/site-packages/_pytest/assertion/__pycache__/ |
Upload File : |
a
05d� � . @ sz d Z ddlZddlZddlZddlZddlZddlZddlZddl Z ddl
Z
ddl Z ddl Z ddl
Z
ddlZddlZddlmZ ddlmZ ddlmZ ddlmZ ddlmZ ddlmZ dd lmZ dd
lmZ dd lmZ dd lmZ dd
lm Z ddlm Z ddlm Z ddlm Z ddl m!Z! ddl m"Z" ddl#m$Z$ ddl%m&Z& ddl'm(Z) ddl*m+Z+ ddl,m-Z- ddl.m/Z/ ddl.m0Z0 ddl1m2Z2 e �r�ddl%m3Z3 e
j4d k�r�ej5Z6nej7Z6e2d � Z8e
j9j:� d e$� �Z;d d �r�d!�p�d" Z<d#e; e< Z=G d$d%� d%ej>j?ej>j@�ZAeeB e jCejDdd&�d'd(�ZEd ejDe jCeeFd)�d*d+�ZGee+e e jCejDf d,�d-d.�ZHd/d0� feeeeIgdf eejD d1�d2d3�ZJdxejKeBeeI ee+ dd4�d5d6�ZLeMeId7�d8d9�ZNee+ eeO d:�d;d<�ZPeMeId7�d=d>�ZQeMeFd7�d?d@�ZReeI eFeIdA�dBdC�ZSeeI eeF eeI eeM eIdD�dEdF�ZTeOeIeIddG�dHdI�ZUeFdJ�dKdL�ZVejWdMejXdNejYdOejZdPiZ[ej\dQej]dRej^dSej_dTej`dUejadVejbdWejcdXejddYejedZejfd[ejgd\ejhd]ejid^ejjd_ejkd`ejldaejmdbejndcejoddejpdeejqdfejrdgiZsejteejt dh�didj�Zuejvdkdl�eBeeOeIf dm�dndo��ZwG dpdq� dqejx�ZyeeFdr�dsdt�Zzeedu�dvdw�Z{dS )yz5Rewrite assertion AST to produce nice error messages.� N)�Path)�PurePath)�Callable)�Dict)�IO)�Iterable)�Iterator)�List)�Optional)�Sequence)�Set)�Tuple)�
TYPE_CHECKING)�Union)�DEFAULT_REPR_MAX_SIZE)�saferepr)�version)�util)�format_explanation)�Config)�Session)� absolutepath)�
fnmatch_ex)�StashKey)�AssertionState�� � r z-pytest-�.pyT�c�o�.c @ sP e Zd ZdZedd�dd�Zee dd�dd�Ze j
j j Z
d*eeeeeef eej ee j
j d �d
d �Z e j
jeej d �d
d�Zejdd�dd�Zeded�dd�Zeeded�dd�Zeded�dd�Zedd�d d �Zedd �d d �Z eeef ed!�d"d#�Z e j d$k�rLe j d%k�r0d&d'l m!Z! n d&d'l"m!Z! ee!d �d(d)�Z#dS )+�AssertionRewritingHookz1PEP302/PEP451 import hook which rewrites asserts.N��config�returnc C sf || _ z|�d�| _W n ty2 ddg| _Y n0 d | _i | _t� | _d| _dh| _ i | _
d| _ d S )NZ python_filesz test_*.pyz *_test.pyFZconftest) r$ �getini�fnpats�
ValueError�session�_rewritten_names�set�
_must_rewrite� _writing_pyc�_basenames_to_check_rewrite�_marked_for_rewrite_cache�_session_paths_checked)�selfr$ � r2 �=/usr/lib/python3.9/site-packages/_pytest/assertion/rewrite.py�__init__@ s z AssertionRewritingHook.__init__)r) r% c C s || _ d| _d S )NF)r) r0 )r1 r) r2 r2 r3 � set_sessionP s z"AssertionRewritingHook.set_session)�name�path�targetr% c C s� | j r
d S | jjt }| �||�r&d S |�d| � | �||�}|d u sp|jd u spt|j t
j j �rpt
j�|j�std S |j}| �|||�s�d S t
jj||| |jd�S )Nzfind_module called for: %s)�loader�submodule_search_locations)r- r$ �stash�assertstate_key�_early_rewrite_bailout�trace�
_find_spec�origin�
isinstancer9 � importlib� machinery�SourceFileLoader�osr7 �exists�_should_rewriter �spec_from_file_locationr: )r1 r6 r7 r8 �state�spec�fnr2 r2 r3 � find_specW s2 ��� � �z AssertionRewritingHook.find_spec)rJ r% c C s d S �Nr2 )r1 rJ r2 r2 r3 �
create_module s z$AssertionRewritingHook.create_module)�moduler% c C s |j d usJ �|j jd us J �t|j j�}| jjt }|| j|j< tj }t
|�}|rvt |�}|svd}|� d|� �� |j
d d� t }|| }t|||j �} | d u r�|� d|��� t|| j�\}
} |r�d| _zt|| |
|� W d| _nd| _0 n|� d|� �� t| |j� d S )NFzread only directory: ���z
rewriting Tz found cached rewritten pyc for )�__spec__r@ r r$ r; r< r* �__name__�sys�dont_write_bytecode�
get_cache_dir� try_makedirsr> r6 �PYC_TAIL� _read_pyc�
_rewrite_testr- �
_write_pyc�exec�__dict__) r1 rO rK rI �write� cache_dir�okZ
cache_name�pyc�co� source_statr2 r2 r3 � exec_module� s2
z"AssertionRewritingHook.exec_moduler )r6 rI r% c C s� | j durP| jsPd| _| j jD ]0}t|��tj�}| j�tj �
|d �d � q |�d�}|d | jv rldS t |� � d�}| j
D ]&}tj �|�r� dS t||�r� dS q�| �||�r�dS |�d|� �� dS ) a' A fast way to get out of rewriting modules.
Profiling has shown that the call to PathFinder.find_spec (inside of
the find_spec from this class) is a major slowdown, so, this method
tries to filter what we're sure won't be rewritten before getting to
it.
NT���r r! Fr z early skip of rewriting module: )r) r0 Z
_initialpaths�str�splitrE �sepr. �addr7 �splitextr � with_suffixr' �dirnamer �_is_marked_for_rewriter> )r1 r6 rI Z initial_path�partsr7 �patr2 r2 r3 r= � s$
z-AssertionRewritingHook._early_rewrite_bailout)r6 rK rI r% c C s� t j�|�dkr$|�d|��� dS | jd urR| j�t|��rR|�d|��� dS t|�}| jD ]$}t ||�r`|�d|��� dS q`| �
||�S )Nz conftest.pyzrewriting conftest file: Tz.matched test file (was specified on cmdline): zmatched test file ) rE r7 �basenamer> r) Z
isinitpathr r r' r rl )r1 r6 rK rI Zfn_pathrn r2 r2 r3 rG � s
z&AssertionRewritingHook._should_rewritec C s| z | j | W S tyv | jD ]D}||ks:|�|d �r |�d|�d|�d�� d| j |< Y dS q d| j |< Y dS 0 d S )Nr! zmatched marked file z (from �)TF)r/ �KeyErrorr, �
startswithr> )r1 r6 rI Zmarkedr2 r2 r3 rl � s
z-AssertionRewritingHook._is_marked_for_rewrite)�namesr% c G sp t |��tj��| j�}|D ]8}tj| }t�|jp6d�s t |j
t | ��s | � |� q | j
�|� | j�� dS )z�Mark import names as needing to be rewritten.
The named module or package as well as any nested modules will
be rewritten on import.
� N)r+ � intersectionrS �modules�
differencer* �AssertionRewriter�is_rewrite_disabled�__doc__rA �
__loader__�type�_warn_already_importedr, �updater/ �clear)r1 rs Zalready_importedr6 �modr2 r2 r3 � mark_rewrite� s �
�� z#AssertionRewritingHook.mark_rewrite�r6 r% c C s( ddl m} | jj|d| �dd� d S )Nr ��PytestAssertRewriteWarningz2Module already imported so cannot be rewritten: %s� )�
stacklevel)�_pytest.warning_typesr� r$ Zissue_config_time_warning)r1 r6 r� r2 r2 r3 r} s ��z-AssertionRewritingHook._warn_already_imported)�pathnamer% c C s6 t |d��}|�� W d � S 1 s(0 Y dS )z Optional PEP302 get_data API.�rbN)�open�read)r1 r� �fr2 r2 r3 �get_data s z AssertionRewritingHook.get_data�r �
)r � r )�TraversableResourcesc C s: t jdk rddlm} n ddlm} |tj| j| d��S )N)r � r )�
FileReader)r7 )rS � version_infoZimportlib.readersr� Zimportlib.resources.readers�types�SimpleNamespacer* )r1 r6 r� r2 r2 r3 �get_resource_reader s
�z*AssertionRewritingHook.get_resource_reader)NN)$rR �
__module__� __qualname__rz r r4 r
r r5 rB rC �
PathFinderrL r? re r r �bytesr� �
ModuleType�
ModuleSpecrN rc �boolr= rG rl r� r} r� rS r� Zimportlib.resources.abcr� �
importlib.abcr� r2 r2 r2 r3 r" = s6
�
� )� *'
r" )�fprb ra r% c C s\ | � tjj� d}| � |� t|j�d@ }|jd@ }| � t�d||�� | � t �
|�� d S )N� � �� z<LL) r] rB r � MAGIC_NUMBER�int�st_mtime�st_size�struct�pack�marshal�dumps)r� rb ra �flags�mtime�sizer2 r2 r3 �
_write_pyc_fp) s
r� )rI ra rb r` r% c
C s� |� dt �� � �}z:t|d�� }t|||� W d � n1 s@0 Y W n> ty� } z&| �d|� d|j� �� W Y d }~dS d }~0 0 zt �||� W n< ty� } z$| �d|� d|� �� W Y d }~dS d }~0 0 dS )Nr! �wbzerror writing pyc file at z: errno=Fz: T)rE �getpidr� r� �OSErrorr> �errno�replace)rI ra rb r` Zproc_pycr� �er2 r2 r3 rZ ; s .rZ )rK r$ r% c C sN t �| �}| �� }t| �}tj||d�}t||||� t||ddd�}||fS )z1Read and rewrite *fn* and return the code object.)�filenamer[ T)� dont_inherit)rE �stat�
read_bytesre �ast�parse�rewrite_asserts�compile)rK r$ r� �sourceZstrfnZtreera r2 r2 r3 rY T s
rY c C s d S rM r2 )�xr2 r2 r3 �<lambda>` � r� )r� r` r> r% c C sJ zt |d�}W n ty" Y dS 0 |��
z(t�| �}t|j�}|j}|�d�}W nF ty� } z.|d| � d|� �� W Y d}~W d � dS d}~0 0 t|�dkr�|d| � W d � dS |dd� t j
j k� r�|d| � W d � dS |dd � d
k�r$|d | � W d � dS |d d � } t� | d
�|d@ k�rb|d| � W d � dS |d d� }
t� |
d
�|d@ k�r�|d| � W d � dS zt
�|�} W nH t�y� } z.|d| � d|� �� W Y d}~W d � dS d}~0 0 t| tj��s"|d| � W d � dS | W d � S 1 �s<0 Y dS )zsPossibly read a pytest pyc containing rewritten code.
Return rewritten code if successful or None if not.
r� N� z
_read_pyc(z ): OSError z&_read_pyc(%s): invalid pyc (too short)� z-_read_pyc(%s): invalid pyc (bad magic number)r r� z._read_pyc(%s): invalid pyc (unsupported flags)r� �littler� z_read_pyc(%s): out of datez+_read_pyc(%s): invalid pyc (incorrect size)z): marshal.load error z _read_pyc(%s): not a code object)r� r� rE r� r� r� r� r� �lenrB r r� �
from_bytesr� �load� ExceptionrA r� �CodeType) r� r` r> r� � stat_resultr� r� �datar� Z
mtime_dataZ size_datara r2 r2 r3 rX _ sN
$ $ rX )r� r� � module_pathr$ r% c C s t |||��| � dS )z%Rewrite the assert statements in mod.N)rx �run)r� r� r� r$ r2 r2 r3 r� � s r� )�objr% c C s t tj�}t| |d��dd�S )a� Get a safe repr of an object for assertion error messages.
The assertion formatting (util.format_explanation()) requires
newlines to be escaped since they are a special character for it.
Normally assertion.util.format_explanation() does this but for a
custom repr it is possible to contain one of the special escape
sequences, especially '\n{' and '\n}' are likely to be present in
JSON reprs.
��maxsize�
�\n)�_get_maxsize_for_safereprr Z_configr r� )r� r� r2 r2 r3 � _saferepr� s
r� r# c C s6 | dur| � d�nd}|dkr"dS |dkr2td S tS )zJGet `maxsize` configuration for saferepr based on the given config object.N�verboser � � r� )Z getoptionr )r$ � verbosityr2 r2 r3 r� � s r� c C sB ddg}t | t�s$t| �} |�d� |D ]\}}| �||�} q(| S )z�Format the custom assertion message given.
For strings this simply replaces newlines with '\n~' so that
util.format_explanation() will preserve them instead of escaping
newlines. For other objects saferepr() is used first.
)r� �
~)�%�%%)r� r� )rA re r �appendr� )r� ZreplacesZr1Zr2r2 r2 r3 �_format_assertmsg� s
r� c C s4 t | �r dS zt| d� W S ty. Y dS 0 d S )NFrR T)�callable�hasattrr� )r� r2 r2 r3 �_should_repr_global_name� s r� )� explanations�is_orr% c C s& d|r
dp d� | � d }|�dd�S )N�(z or z and rp r� r� )�joinr� )r� r� � explanationr2 r2 r3 �_format_boolop� s r� )�ops�results�expls�each_objr% c C s� t tt| ��||�D ]4\}}}z
| }W n ty> d}Y n0 |r qJqtjd ur~t�| | || ||d �}|d ur~|S |S )NTr� )�zip�ranger� r� r Z _reprcompare) r� r� r� r� �i�res�explZdoneZcustomr2 r2 r3 �_call_reprcompare� s
r� )�lineno�origr� r% c C s t jd urt �| ||� d S rM �r Z_assertion_pass)r� r� r� r2 r2 r3 �_call_assertion_pass� s
r� �r% c C s t jr
dS dS )z�Check if any plugins implement the pytest_assertion_pass hook
in order not to generate explanation unnecessarily (might be expensive).TFr� r2 r2 r2 r3 � _check_if_assertion_pass_impl� s r� znot %sz~%sz-%sz+%s�|�^�&z<<z>>�+�-�*�/z//r� z==z!=�<z<=�>z>=�**�iszis not�inznot in�@��noder% c c s( | V t �| �D ]}t|�E dH qdS )zARecursively yield node and all its children in depth-first order.N)r� Ziter_child_nodes�
traverse_node)r �childr2 r2 r3 r s r r� r� )�srcr% c
s� i �d�g �d� t � �dd�� ����fdd� }t�t�| �j�}|D �]d\}}\}}}}|tjkrr|dkrr|� qF� durF|tjkr�|dv r��d7 �n|tjkr�|d v r��d8 ��s҈�||d� � ��|� qF�dk�rr|tjk�rr|d
k�rr|�v �r8t ��dk�r8|t �d � t |� } �d d| � �d < n2|�v �rX�d d|� �d < n��|d|� � |� qF|tj
tj hv �r�|� qF�rF|�vrF��|� ��|� qF�S ) z<Return a mapping from {lineno: "assertion test expression"}.r Nr� c s: � d us J �d� ���� �d��� < d�g �d � t� �d S )Nrt �\r )r� �rstripr+ r2 �Z
assert_lineno�depth�lines�retZ
seen_linesr2 r3 �_write_and_reset$ s z._get_assertion_exprs.<locals>._write_and_reset�assertz([{r� z)]}�,rd ) r+ �tokenize�io�BytesIO�readline�NAME�OPr� rh r� �NEWLINE� ENDMARKER)
r r �tokens�tpr� r� �offset�_�lineZoffset_in_trimmedr2 r
r3 �_get_assertion_exprs s@
r c s e Zd ZdZee ee edd�� fdd� Ze j
dd�dd�Z e ee
d �d
d ��Zed �d
d�Ze je jd�dd�Ze je jd�dd�Zee je jd�dd�Zee jd�dd�Ze jed�dd�Zdd �d d �Ze je jd �d d �Ze jee jef d!�d"d#�Z e j e e j d$�d%d&�Z e!ee!ef d�d'd(�Z"e jee jef d�d)d*�Z#e j$ee jef d+�d,d-�Z%e j&ee jef d.�d/d0�Z'e j(ee jef d1�d2d3�Z)e j*ee jef d4�d5d6�Z+e j,ee j,ef d7�d8d9�Z-e jee jef d:�d;d<�Z.e j/ee jef d=�d>d?�Z0� Z1S )@rx a�
Assertion rewriting implementation.
The main entrypoint is to call .run() with an ast.Module instance,
this will then find all the assert statements and rewrite them to
provide intermediate values and a detailed assertion error. See
http://pybites.blogspot.be/2011/07/behind-scenes-of-pytests-new-assertion.html
for an overview of how this works.
The entry point here is .run() which will iterate over all the
statements in an ast.Module and for each ast.Assert statement it
finds call .visit() with it. Then .visit_Assert() takes over and
is responsible for creating new ast statements to replace the
original assert statement: it rewrites the test of an assertion
to provide intermediate values and replace it with an if statement
which raises an assertion error with a detailed explanation in
case the expression is false and calls pytest_assertion_pass hook
if expression is true.
For this .visit_Assert() uses the visitor pattern to visit all the
AST nodes of the ast.Assert.test field, each visit call returning
an AST node and the corresponding explanation string. During this
state is kept in several instance attributes:
:statements: All the AST statements which will replace the assert
statement.
:variables: This is populated by .variable() with each variable
used by the statements so that they can all be set to None at
the end of the statements.
:variable_counter: Counter to create new unique variables needed
by statements. Variables are created using .variable() and
have the form of "@py_assert0".
:expl_stmts: The AST statements which will be executed to get
data from the assertion. This is the code which will construct
the detailed assertion message that is used in the AssertionError
or for the pytest_assertion_pass hook.
:explanation_specifiers: A dict filled by .explanation_param()
with %-formatting placeholders and their corresponding
expressions to use in the building of an assertion message.
This is used by .pop_format_context() to build a message.
:stack: A stack of the explanation_specifiers dicts maintained by
.push_format_context() and .pop_format_context() which allows
to build another %-formatted string while already building one.
:variables_overwrite: A dict filled with references to variables
that change value within an assert. This happens when a variable is
reassigned with the walrus operator
This state, except the variables_overwrite, is reset on every new assert
statement visited and used by the other visitors.
N)r� r$ r� r% c sB t � �� || _|| _|d ur,|�d�| _nd| _|| _i | _d S )N�enable_assertion_pass_hookF)�superr4 r� r$ r& r r� �variables_overwrite)r1 r� r$ r� �� __class__r2 r3 r4 � s
�zAssertionRewriter.__init__)r� r% c s |j s
dS t|dd�}|du }|dur4| �|�r4dS d}d}|j D ]n}|r�t|tj�r�t|jtj�r�|jj}| �|�r| dS d}n&t|tj �r�|j
dkr�|j dkr�n q�|d7 }qBt|tj �r�|j
r�|j
d j� n|j� tjdk�r
tjdd � dd
�tjd d � dd
�g}nt�dd �t�d d �g}� fd
d�|D �}||j ||�< |g}|�r
|�� } t�| �D ]�\}
} t| t��r�g } t| �D ]L\}
}t|tj��r�| �| �|�� n"| �|� t|tj��r�|�|� �q�t| |
| � n&t| tj��r`t| tj ��s`|�| � �q`�qHdS )z5Find all assert statements in *mod* and rewrite them.N� docstringr FZ
__future__r� r� �builtins� @py_builtins�r� �
col_offsetz_pytest.assertion.rewrite�
@pytest_arc s g | ]}t j|g� d d��qS )r r' )r� ZImport)�.0�alias�r� r2 r3 �
<listcomp>� s z)AssertionRewriter.run.<locals>.<listcomp>) �body�getattrry rA r� �Expr�value�Str�sZ
ImportFrom�levelrO Z FunctionDef�decorator_listr� rS r� r+ �popZ iter_fields�list� enumerate�Assert�extend�visitr� �AST�setattr�expr)r1 r� �docZexpect_docstring�pos�item�aliasesZimportsZnodesr r6 Zfield�newr� r r2 r, r3 r� � s�
�
� �
���
��
�
�
�
�zAssertionRewriter.run)r$ r% c C s d| v S )NZPYTEST_DONT_REWRITEr2 )r$ r2 r2 r3 ry � s z%AssertionRewriter.is_rewrite_disabledr� c C s" dt t| j�� }| j�|� |S )zGet a new variable.z
@py_assert)re �next�variable_counter� variablesr� )r1 r6 r2 r2 r3 �variable� s zAssertionRewriter.variable)r> r% c C s: | � � }| j�t�t�|t�� �g|�� t�|t�� �S )zGive *expr* a name.)rG �
statementsr� r� �Assign�Name�Store�Load)r1 r> r6 r2 r2 r3 �assign� s "zAssertionRewriter.assignc C s | � d|�S )z Call saferepr on the expression.r� )�helper)r1 r> r2 r2 r3 �display� s zAssertionRewriter.display)r6 �argsr% c G s4 t �dt �� �}t �||t �� �}t �|t|�g �S )z Call a helper in this module.r) )r� rJ rL � Attribute�Callr7 )r1 r6 rP Zpy_name�attrr2 r2 r3 rN s zAssertionRewriter.helperr� c C s" t �dt �� �}t �||t �� �S )z!Return the builtin called *name*.r& )r� rJ rL rQ )r1 r6 � builtin_namer2 r2 r3 �builtin s zAssertionRewriter.builtinc C s( dt t| j�� }|| j|< d| d S )aJ Return a new named %-formatting placeholder for expr.
This creates a %-formatting placeholder for expr in the
current formatting context, e.g. ``%(py0)s``. The placeholder
and expr are placed in the current format context so that it
can be used on the next call to .pop_format_context().
�pyz%(z)s)re rD rE �explanation_specifiers)r1 r> Z specifierr2 r2 r3 �explanation_param
s
z#AssertionRewriter.explanation_paramc C s i | _ | j�| j � dS )a� Create a new formatting context.
The format context is used for when an explanation wants to
have a variable value formatted in the assertion message. In
this case the value required can be added using
.explanation_param(). Finally .pop_format_context() is used
to format a string of %-formatted values as added by
.explanation_param().
N)rW �stackr� �r1 r2 r2 r3 �push_format_context s
z%AssertionRewriter.push_format_context)� expl_exprr% c C s� | j �� }| j r | j d | _dd� |�� D �}t�|t|�� ��}t�|t� � |�}dt
t | j �� }| j
rx| j�|� | j�t�t�|t�� �g|�� t�|t�� �S )aZ Format the %-formatted string with current format context.
The expl_expr should be an str ast.expr instance constructed from
the %-placeholders created by .explanation_param(). This will
add the required code to format said string to .expl_stmts and
return the ast.Name instance of the formatted string.
rd c S s g | ]}t �|��qS r2 )r� r2 )r* �keyr2 r2 r3 r- 1 r� z8AssertionRewriter.pop_format_context.<locals>.<listcomp>z
@py_format)rY r6 rW �keysr� r r7 �values�BinOp�Modre rD rE r �format_variablesr� �
expl_stmtsrI rJ rK rL )r1 r\ �currentr^ Z format_dictZformr6 r2 r2 r3 �pop_format_context& s
"z$AssertionRewriter.pop_format_contextr c C s. t |tj�sJ �| �|�}|| �| �|��fS )z1Handle expressions we don't have custom code for.)rA r� r> rM rX rO )r1 r r� r2 r2 r3 �
generic_visit: s
z AssertionRewriter.generic_visit)�assert_r% c C s8 t |jtj�rZt|jj�dkrZddlm} ddl}| j dus@J �|j
|d�d| j |j d� g | _ g | _
t�� | _| jr|g | _g | _g | _| �� | �|j�\}}t�t�� |�}| j�r | �t�|��}|jr�| � d|j�}d} nt�d �}d
} t� t�| �t� � |�}
t� |t� � |
�} t� d t� � �} | � d | �}
t�!| |
gg �}t�"|d�}g }|�#| j� |�$|� | � d |�}t%| j&�|j }t�'| � d
t�(|j �t�|�|��}t�)| � d�| j|g g �}|g}t�)|||�}| j �$|� | j�r�dd� | jD �}t�*|t�+d��}| j �$|� n�| j}| j �$t�)||g �� |j�r\| � d|j�}d| }nt�d �}d
| }t� |t� � t�|��}| �|�}| � d |�}
t� d t� � �} t�!| |
gg �}t�"|d�}|�$|� | j
�rdd� | j
D �}t�*|t�+d��}| j �$|� | j D ]"} t,| �D ]} t�-| |� �q�q| j S )a. Return the AST statements to replace the ast.Assert instance.
This rewrites the test of an assertion to provide
intermediate values and replace it with an if statement which
raises an assertion error with a detailed explanation in case
the expression is false.
r� r r� Nz5assertion is always true, perhaps remove parentheses?)�categoryr� r� r� z
>assert rt zassert �AssertionError�_format_explanationr� r� c S s g | ]}t �|t �� ��qS r2 �r� rJ rK �r* r6 r2 r2 r3 r- � s z2AssertionRewriter.visit_Assert.<locals>.<listcomp>c S s g | ]}t �|t �� ��qS r2 rk rl r2 r2 r3 r- � r� ).rA �testr� r
r� �eltsr� r� �warningsr� �
warn_explicitr� rH rF � itertools�countrE r rb rY rc r[ r; �UnaryOp�Notre r2 �msgrN r` �AddrJ rL rR ZRaiser: r� r r� r0 �Num�IfrI Z NameConstantr Z
copy_location) r1 rg r� ro Z
top_conditionr� Znegationru Z assertmsgZgluestrZerr_explanation�err_msgZerr_nameZfmt�excZraise_Zstatements_failZfmt_passr� Zhook_call_passZhook_impl_testZstatements_passZ main_testrF Z clear_formatr. �templater �stmtr r2 r2 r3 � visit_Assert@ s� ��
��
� �
z AssertionRewriter.visit_Assertc C s� t �| �d�g g �}|jj}t �t �|�t �� g|g�}| �d|�}t � t �
� ||g�}t � || � |�t �|��}|| �
|�fS �N�localsr� )r� rR rU r8 �id�Comparer2 �InrN �BoolOp�Or�IfExprO rX )r1 r6 �locsZ target_id�inlocs�doreprrm r> r2 r2 r3 �visit_NamedExpr� s z!AssertionRewriter.visit_NamedExprc C s| t �| �d�g g �}t �t �|j�t �� g|g�}| �d|�}t �t � � ||g�}t �
|| � |�t �|j��}|| � |�fS r~ )
r� rR rU r� r2 r� r� rN r� r� r� rO rX )r1 r6 r� r� r� rm r> r2 r2 r3 �
visit_Name� s z AssertionRewriter.visit_Name)�boolopr% c
C s� | � � }| �t�g t�� ��}t�|dt�� �}tt|jtj ��}| j
}}| j }t |j
�d } | �� t|j
�D �]6\}
} |
r�g } | j �t�|
| g �� | | _ t| tj�r�t| jt�r�| jjjdd� |j
d |
� D �v r�| � � }|| j| jjj< || jj_| �� | �| �\}}|�t�t�|t�� �g|�� | � t� |��}t� ||gg �}| j �t� |�� |
| k rr|}
|�r�t� t�!� |
�}
g }| j
�t�|
|g �� | | _
}qr|| _
|| _ | �"d|t�#|��}| � |�}t�|t�� �| �$|�fS )Nr� r� c S s g | ]}t |d �r|j�qS )r� )r� r� )r* Zast_exprr2 r2 r3 r- � s
�z2AssertionRewriter.visit_BoolOp.<locals>.<listcomp>r� )%rG rM r� r rL rQ r� rA �opr� rH rc r� r_ r[ r8 r� rx r� �left� namedExprr8 r� r! r; rI rJ rK re r2 rR r0 rs rt rN rw rX )r1 r� Zres_varZ expl_listZappr� r. ZsaveZ fail_saveZlevelsr� �vZ
fail_innerZcondZ pytest_tempr� r� Z expl_format�call�innerZ
expl_templater2 r2 r3 � visit_BoolOp� sX
�
� ���
z AssertionRewriter.visit_BoolOp)�unaryr% c C s> t |jj }| �|j�\}}| �t�|j|��}|||f fS rM )� UNARY_MAPr� r# r; �operandrM r� rs )r1 r� �patternZ operand_resZ operand_explr� r2 r2 r3 �
visit_UnaryOp� s z AssertionRewriter.visit_UnaryOp)�binopr% c C sb t |jj }| �|j�\}}| �|j�\}}d|� d|� d|� d�}| �t�||j|��}||fS )Nr� � rp ) � BINOP_MAPr� r# r; r� �rightrM r� r` ) r1 r� �symbolZ left_expr� left_explZ
right_exprZ
right_explr� r� r2 r2 r3 � visit_BinOp s z AssertionRewriter.visit_BinOp)r� r% c C s� | � |j�\}}g }g }g }|jD ]&}| � |�\}} |�| � |�|� q"|jD ]R}
| � |
j�\}} |�t�|
j|�� |
jr�|�|
jd | � qP|�d| � qPd� |d�
|��} t� |||�} | � | �}| �
| �|��} | � d| � d| � d�}
||
fS )N�=r� z{}({})z, z
{z = z
})r; �funcrP r� �keywordsr1 r� �keyword�arg�formatr� rR rM rX rO )r1 r� �new_funcZ func_explZ arg_expls�new_argsZ
new_kwargsr� r� r� r� Znew_call�res_explZ
outer_explr2 r2 r3 �
visit_Call
s(
z AssertionRewriter.visit_Call)�starredr% c C s* | � |j�\}}t�||j�}|d| fS )Nr� )r; r1 r� �Starred�ctx)r1 r� r� r� Z new_starredr2 r2 r3 �
visit_Starred" s z AssertionRewriter.visit_Starred)rS r% c C sp t |jtj�s| �|�S | �|j�\}}| �t�||j t�� ��}| �
| � |��}d}|||||j f }||fS )Nz%s
{%s = %s.%s
}) rA r� r� rL rf r; r1 rM rQ rS rX rO )r1 rS r1 Z
value_explr� r� rn r� r2 r2 r3 �visit_Attribute( s
z!AssertionRewriter.visit_Attribute)�compr% c sH � � � t|jtj�r6|jj� jv r6� j|jj |j_� �|j�\}}t|jtjtj f�rfd|� d�}� fdd�t
t |j ��D �}dd� |D �}dd� |D �}t
t
t |j ��|j |j�}g }g } |g}
|D ]�\} } }
t|
t��rt|tj��r|
jj|jk�r� �� |
j_|
jj� j|j< � �|
�\}}t|
tjtj f��rHd|� d�}|
�|� t| j }| �t�|�� |� d|� d|� �}|�t�|�� t�|| g|g�}� j�t�|| g|�� || }}qʈ �dt�| t�� �t�|t�� �t�|t�� �t�|
t�� ��}t |j �d k�r,t� t�� |�}n|d
}|� � � � |��fS ) Nr� rp c s g | ] }� � � �qS r2 )rG )r* r� rZ r2 r3 r- : r� z3AssertionRewriter.visit_Compare.<locals>.<listcomp>c S s g | ]}t �|t �� ��qS r2 )r� rJ rL �r* r� r2 r2 r3 r- ; r� c S s g | ]}t �|t �� ��qS r2 rk r� r2 r2 r3 r- < r� r� r� r� r ) r[ rA r� r� rJ r� r! r; r� r� r� r� r� r� � comparatorsr� r8 rG r� r� r# r2 rH rI rN r
rL ZAndrX re )r1 r� Zleft_resr� Z
res_variablesZ
load_namesZ store_names�itr� Zsymsr� r� r� Z next_operandZnext_resZ next_explZsymr� Zres_exprZ expl_callr� r2 rZ r3 �
visit_Compare2 sX �
� �
�z AssertionRewriter.visit_Compare)2rR r� r� rz r
re r r� r4 r� �Moduler� � staticmethodr� ry rG r> rJ rM rO rN rQ rU rX r[ re r< r
rf r9 r r| r} r� r� r� r� r� rs r� r` r� rR r� r� r� r� r� r� �
__classcell__r2 r2 r" r3 rx Q s4 9�T
u
1
rx )r^ r% c
C s� zt j| dd� W nj tttfy, Y dS ty> Y dS ty| } z(|jtjkrfW Y d}~dS � W Y d}~n
d}~0 0 dS )z~Attempt to create the given directory and sub-directories exist.
Returns True if successful or if it already exists.
T)�exist_okFN) rE �makedirs�FileNotFoundError�NotADirectoryError�FileExistsError�PermissionErrorr� r� ZEROFS)r^ r� r2 r2 r3 rV d s rV )� file_pathr% c C s: t jdkr,t jr,tt j�t| jdd� � S | jd S dS )zKReturn the cache directory to write .pyc files for the given .py file path.r r� rd � __pycache__N)rS r� �pycache_prefixr rm �parent)r� r2 r2 r3 rU z s rU )NN)|rz r� r� � functoolsr� rB Zimportlib.machinery�importlib.utilr rq r� rE r� rS r r� �pathlibr r �typingr r r r r r r
r r r
r r Z_pytest._io.safereprr r Z_pytest._versionr Z_pytest.assertionr Z_pytest.assertion.utilr rj Z_pytest.configr Z _pytest.mainr Z_pytest.pathlibr r Z
_pytest.stashr r r� Z NamedExprr� r0 r< �implementation� cache_tagZ
PYTEST_TAGZPYC_EXTrW �abc�MetaPathFinder�Loaderr" r� r� r� r� r� rZ rY re rX r� r� �objectr� r� r� r� r� r� r� r� r� rt ZInvertZUSubZUAddr� ZBitOrZBitXorZBitAndZLShiftZRShiftrv ZSubZMultZDivZFloorDivra ZEqZNotEqZLtZLtEZGtZGtEZPowZIsZIsNotr� ZNotInZMatMultr� r< r � lru_cacher Z NodeVisitorrx rV rU r2 r2 r2 r3 �<module> s�
n� � �� 4 ��
� �
6