Format: application/x-python-code
1 o
2
3 ��h�1 � @ s� d Z ddlZddlZddlZddlZddlZddlZddlZddl m
4 Z
5 ddl m Z ddl
6 mZmZ ddlmZ ddlmZmZ ddlmZ d Zd
7 Zd d � ZG d
8 d� dej�Z dZ G dd� dej�Z G dd� dej�Z dZ G dd� dej�Z!ed�dd� �Z"dZ#dZ$G d d � d ej�Z%ddlZe�&� d kr� G d d � d ej�Z'G d!d"� d"ej�Z(G d#d$� d$ej�Z)G d%d&� d&ej�Z*d'd(� Z+dS ))z Tests for IPython.core.ultratb
9 � N)�dedent)�TemporaryDirectory)�ColorTB� VerboseTB)�tools)�onlyif_unicode_paths� skip_without)�prepended_to_syspathz1
10 2
11 3
12 def f():
13 1/0
14 zdef f():
15 1/0
16 c s � fdd�}|S )z:
17 decorator to set the recursion limit temporarily
18 c s �� fdd�}|S )Nc s: t �� }t �� � z
19 �| i |��W t �|� S t �|� w �N)�sys�getrecursionlimit�setrecursionlimit)�args�kwargsZrl)�frames�
20 test_function� �Uc:\Users\shubh\Videos\agent\nova\Lib\site-packages\IPython/core/tests/test_ultratb.py�wrapper& s
21
22 z.recursionlimit.<locals>.inner.<locals>.wrapperr )r r �r )r r �inner% s z recursionlimit.<locals>.innerr )r r r r r �recursionlimit s r c @ � e Zd Zdd� ZdS )�ChangedPyFileTestc
23 C s� t � ��}tj�|d�}t|ddd��
24 }|�t� W d � n1 s"w Y t|��
25 t� d� W d � n1 s;w Y t
26 � d��
27 t� d� W d � n1 sUw Y t|ddd��
28 }|�t � W d � n1 sqw Y t
29 j
30 d d
31 d ��< t
32 � d��
33 t� d� W d � n1 s�w Y t
34 � d��
35 t� d� W d � n1 s�w Y W d � n1 s�w Y W d � dS W d � dS 1 s�w Y dS ) z�Traceback produced if the line where the error occurred is missing?
36
37 https://github.com/ipython/ipython/issues/1456
38 �foo.py�w�utf-8��encodingN�
39 import foo�ZeroDivisionError�foo.f()zInternal Python error�stderr)Zchannel)r �os�path�join�open�write�file_1r �ip�run_cell�tt� AssertPrints�file_2�AssertNotPrints��self�td�fname�fr r r �test_changing_py_file4 s4 �
40 � � � � �� ��"�z'ChangedPyFileTest.test_changing_py_fileN)�__name__�
41 __module__� __qualname__r4 r r r r r 3 � r uL # coding: iso-8859-5
42
43 def fail():
44 """дбИЖ"""
45 1/0 # дбИЖ
46 c @ s( e Zd Zedd� �Zdd� Zdd� ZdS )� NonAsciiTestc C s� t dd��g}tj�|d�}t|ddd��
47 }|�t� W d � n1 s$w Y t|��
48 t� d� W d � n1 s=w Y t
49 � d��
50 t� d � W d � n1 sWw Y W d � d S W d � d S 1 sow Y d S )
51 N� é)�suffixu fooé.pyr r r r r r! ) r r# r$ r% r& r' r( r r) r* r+ r, r/ r r r �test_nonascii_pathX s �
52 � ��"�z NonAsciiTest.test_nonascii_pathc
53 C s t � �}tj�|d�}tj|ddd��
54 }|�t� W d � n1 s#w Y t|��
55 t �
56 d� W d � n1 s<w Y t � d��$ t j ddd ��
57 t �
58 d
59 � W d � n1 s^w Y W d � n1 smw Y W d � d S W d � d S 1 s�w Y d S ) Nz
60 dfghjkl.pyr z
61 iso-8859-5r zfrom dfghjkl import failr u дбИЖF��suppresszfail())
62 r r# r$ r% �ior& r' �iso_8859_5_filer r) r* r+ r, r/ r r r �test_iso8859_5f s" �
63 � �� �� "�zNonAsciiTest.test_iso8859_5c C s d}d}t �d� t�|��
64 t �|� W d � n1 s w Y t �d� t�|��
65 t �|� W d � n1 s=w Y t �d� t�|��
66 t �|� W d � n1 s\w Y t �d� t�d��
67 t �|� W d � n1 s{w Y t �d� d S )Nu raise Exception('é')u Exception('é')z %xmode plainz%xmode verbosez%xmode contextz%xmode minimalu
68 Exception: é)r) r* r+ r, )r0 �cell�expectedr r r �test_nonascii_msgt s&
69 �
70 �
71 �
72 �z NonAsciiTest.test_nonascii_msgN)r5 r6 r7 r r< rA rD r r r r r9 W s
73
74
75 r9 c @ s e Zd ZdZdd� ZdS )�NestedGenExprTestCasez�
76 Regression test for the following issues:
77 https://github.com/ipython/ipython/issues/8293
78 https://github.com/ipython/ipython/issues/8205
79 c C sF t d�}tjddd�� t�|� W d � d S 1 s w Y d S )Nz� class SpecificException(Exception):
80 pass
81
82 def foo(x):
83 raise SpecificException("Success!")
84
85 sum(sum(foo(x) for _ in [0]) for x in [0])
86 zSpecificException: Success!Fr= )r r+ r, r) r* )r0 �coder r r �test_nested_genexpr� s � "�z)NestedGenExprTestCase.test_nested_genexprN)r5 r6 r7 �__doc__rG r r r r rE � s rE zif True:
87 zoom()
88 c @ r )�IndentationErrorTestc
89 C sH t �d��$ t jddd��
90 t�t� W d � n1 s w Y W d � n1 s,w Y t� �f}tj�|d�}t |ddd��
91 }|�
92 t� W d � n1 sSw Y t �d��% t jddd�� t� d |� W d � n1 svw Y W d � n1 s�w Y W d � d S W d � d S 1 s�w Y d S )
93 N�IndentationErrorzzoom()Fr= r r r r �run) r+ r, r) r* �indentationerror_filer r# r$ r% r& r' �run_line_magicr/ r r r � test_indentationerror_shows_line� s( �� � � �� ��"�z5IndentationErrorTest.test_indentationerror_shows_lineN)r5 r6 r7 rN r r r r rI � r8 rI Zpandasc C s> d} t �d�� t�| � W d � d S 1 sw Y d S )Nz�
94 import pandas
95 df = pandas.DataFrame([])
96
97 # Important: only fails inside of an "exec" call:
98 exec("df.foobarbaz()")
99 zCould not get source)r+ r, r) r* )rF r r r �test_dynamic_code� s "�rO z1
100 2
101 7/
102 z7/
103 c @ s, e Zd Zdd� Zdd� Zdd� Zdd� Zd S )
104 �SyntaxErrorTestc C sr d}t �d��
105 t�|� W d � n1 sw Y t �d�� t�|� W d � d S 1 s2w Y d S )Nz
106 def foo():
107 ..
108 � SyntaxError�foo())r+ r, r) r* r. )r0 Z syntax_error_at_compile_timer r r �.test_syntaxerror_no_stacktrace_at_compile_time� s � "�z>SyntaxErrorTest.test_syntaxerror_no_stacktrace_at_compile_timec C s� d}t �d��
109 t�|� W d � n1 sw Y t �ddg��
110 t�|� W d � n1 s3w Y tjd= tjd= d S )Nz8
111 def foo():
112 eval("..")
113
114 def bar():
115 foo()
116
117 bar()
118 rQ rR zbar()�bar�foo)r+ r, r) r* Zuser_ns)r0 Zsyntax_error_at_runtimer r r �6test_syntaxerror_stacktrace_when_running_compiled_code� s � � zFSyntaxErrorTest.test_syntaxerror_stacktrace_when_running_compiled_codec C s. t � ��}tj�|d�}t|ddd��
119 }|�t� W d � n1 s"w Y t�ddg�� t �
120 d|� W d � n1 s?w Y t|ddd��
121 }|�t � W d � n1 s[w Y t�ddg�� t �
122 d|� W d � n1 sxw Y W d � d S W d � d S 1 s�w Y d S )Nr r r r z7/rQ rK ) r r# r$ r% r& r' � se_file_1r+ r, r) rM � se_file_2r/ r r r r4 � s" � � � ��
123 "�z%SyntaxErrorTest.test_changing_py_filec C sT zt d�� t y) t�d�� t�� W d � Y d S 1 s!w Y Y d S w )NZQWERTY)�
124 ValueErrorr+ r, r) �showsyntaxerror�r0 r r r �test_non_syntaxerror� s
125 &��z$SyntaxErrorTest.test_non_syntaxerrorN)r5 r6 r7 rS rV r4 r\ r r r r rP � s
126 rP �PyPyc @ r )�MemoryErrorTestc C s d}t �|� d S )Na� (((((((((((((((((((((((((((((((((((((((((((((((((((((((((((((((((((((((((((((((((((((((((((((((((((((((((((((((((((((((((((((((((((((((((((((((((((((((((((((((((((((((((((((((((((((((((((((((((((((((()))))))))))))))))))))))))))))))))))))))))))))))))))))))))))))))))))))))))))))))))))))))))))))))))))))))))))))))))))))))))))))))))))))))))))))))))))))))))))))))))))))))))))))))))))))))))))))))))))))))))r) r* )r0 Zmemoryerror_coder r r �test_memoryerror s z MemoryErrorTest.test_memoryerrorN)r5 r6 r7 r_ r r r r r^
127 r8 r^ c @ sT e Zd ZdZdZdZdZdd� Zdd� Zd d
128 � Z d d � Z
129 d
130 d� Z dd� Z dd� Z
131 dS )� Python3ChainedExceptionsTestz�
132 try:
133 x = 1 + 2
134 print(not_defined_here)
135 except Exception as e:
136 x += 55
137 x - 1
138 y = {}
139 raise KeyError('uh') from e
140 z�
141 try:
142 x = 1 + 2
143 print(not_defined_here)
144 except Exception as e:
145 x += 55
146 x - 1
147 y = {}
148 raise KeyError('uh')
149 zL
150 try:
151 1/0
152 except Exception:
153 raise ValueError("Yikes") from None
154 zA
155 try:
156 1/0
157 except Exception as e:
158 raise SystemExit(1)
159 c C �@ t �g d��� t�| j� W d � d S 1 sw Y d S )N��KeyError� NameErrorz direct cause�r+ r, r) r* �DIRECT_CAUSE_ERROR_CODEr[ r r r �test_direct_cause_error5 � "�z4Python3ChainedExceptionsTest.test_direct_cause_errorc C ra )N�rc rd zDuring handling�r+ r, r) r* � EXCEPTION_DURING_HANDLING_CODEr[ r r r �$test_exception_during_handling_error9 rh zAPython3ChainedExceptionsTest.test_exception_during_handling_errorc C s� t �ddg��/ t jdgdd�� t�| j� W d � n1 s!w Y W d � d S W d � d S 1 s9w Y d S )N�
160 SystemExitzto see the full tracebackzanother exceptionFr= )r+ r, r. r) r* �SYS_EXIT_WITH_CONTEXT_CODEr[ r r r �!test_sysexit_while_handling_error= s ��"�z>Python3ChainedExceptionsTest.test_sysexit_while_handling_errorc C sz t �d��. t jddd�� t�| j� W d � n1 s w Y W d � d S W d � d S 1 s6w Y d S )Nr rY Fr= �r+ r. r, r) r* �SUPPRESS_CHAINING_CODEr[ r r r � test_suppress_exception_chainingB s
161 �P�z=Python3ChainedExceptionsTest.test_suppress_exception_chainingc C �T t �g d��� t�d� t�| j� t�d� W d � d S 1 s#w Y d S )Nrb � %xmode Plain�%xmode Verbosere r[ r r r � test_plain_direct_cause_errorG �
162
163 "�z:Python3ChainedExceptionsTest.test_plain_direct_cause_errorc C rs )Nri rt ru rj r[ r r r �*test_plain_exception_during_handling_errorM rw zGPython3ChainedExceptionsTest.test_plain_exception_during_handling_errorc C s� t �d��8 t jddd�� t�d� t�| j� t�d� W d � n1 s(w Y W d � d S W d � d S 1 s@w Y d S )Nr rY Fr= rt ru rp r[ r r r �&test_plain_suppress_exception_chainingS s �
164 P�zCPython3ChainedExceptionsTest.test_plain_suppress_exception_chainingN)r5 r6 r7 rf rk rq rn rg rl ro rr rv rx ry r r r r r` s r` c @ s@ e Zd ZdZdd� Zdd� Zed�dd� �Zed �d
165 d � �Zd S )
166 �
167 RecursionTestz�
168 def non_recurs():
169 1/0
170
171 def r1():
172 r1()
173
174 def r3a():
175 r3b()
176
177 def r3b():
178 r3c()
179
180 def r3c():
181 r3a()
182
183 def r3o1():
184 r3a()
185
186 def r3o2():
187 r3o1()
188 c C s t �| j� d S r
189 )r) r* � DEFINITIONSr[ r r r �setUpr s zRecursionTest.setUpc C s: t �d�� t�d� W d � d S 1 sw Y d S )Nzskipping similar framesz non_recurs())r+ r. r) r* r[ r r r �test_no_recursionu s "�z RecursionTest.test_no_recursion�� c C s@ t �t�d��� t�d� W d � d S 1 sw Y d S )NzB\[\.\.\. skipping similar frames: r1 at line 5 \(\d{2,3} times\)\]zr1()�r+ r, �re�compiler) r* r[ r r r �test_recursion_one_framey s
190 � "�z&RecursionTest.test_recursion_one_frame� c C s� t �d��d t jt�d�dd��A t jt�d�dd��' t jt�d�dd��
191 t�d� W d � n1 s6w Y W d � n1 sEw Y W d � n1 sTw Y W d � d S W d � d S 1 slw Y d S )Nz [... skipping similar frames: z r3a at line 8 \(\d{2} times\)Fr= z r3b at line 11 \(\d{2} times\)z r3c at line 14 \(\d{2} times\)zr3o2()r r[ r r r �test_recursion_three_frames� s ��� �� �P z)RecursionTest.test_recursion_three_framesN) r5 r6 r7 r{ r| r} r r� r� r r r r rz [ s
192 rz c @ s e Zd ZdZdd� Zdd� ZdS )�PEP678NotesReportingTestz�
193 try:
194 raise AssertionError("Message")
195 except Exception as e:
196 try:
197 e.add_note("This is a PEP-678 note.")
198 except AttributeError: # Python <= 3.10
199 e.__notes__ = ("This is a PEP-678 note.",)
200 raise
201 c C ra )N��AssertionError�MessagezThis is a PEP-678 note.�r+ r, r) r* �ERROR_WITH_NOTEr[ r r r �test_verbose_reports_notes� rh z3PEP678NotesReportingTest.test_verbose_reports_notesc C rs )Nr� rt ru r� r[ r r r �test_plain_reports_notes� rw z1PEP678NotesReportingTest.test_plain_reports_notesN)r5 r6 r7 r� r� r� r r r r r� � s r� c s �fdd�} d� fdd� �t � fdd�� t�� }|�d� |�d � z
202 |�| dd
203 �� W n
204 tj|d � Y t|d �}|�d
205 � z
206 |�| dd
207 �� W n
208 |t�� � Y |�d� t |d �}|�d� z
209 |�| dd
210 �� W n
211 |t�� � Y |�d� d S )Nc s& |\}}| | }| | }� ||� d S r
212 r )�cZd_e�d�e�x�y)rU r r �spam� s ztest_handlers.<locals>.spam� c s � | || � d S r
213 r )�a�brT )�eggsr r rU � s ztest_handlers.<locals>.fooc S s | | }| | }|| S r
214 r )r3 �g�z�h�ir r r r� � s ztest_handlers.<locals>.eggs� z*** Before ***)� � )�file)Zostreamz*** ColorTB ***z*** VerboseTB ***)r� )
215 �globalsr? �StringIOr' � traceback� print_excr r �exc_infor )r� Zbuff�handlerr )r� rU r �
216 test_handlers� s0
217
218
219
220
221
222
223 r� ),rH r? Zos.pathr# �platformr� r r� Zunittest�textwrapr �tempfiler ZIPython.core.ultratbr r ZIPython.testingr r+ ZIPython.testing.decoratorsr r Z IPython.utils.syspathcontextr r( r- r ZTestCaser r@ r9 rE rL rI rO rW rX rP �python_implementationr^ r` rz r� r� r r r r �<module> sH 3
224
225 9 K.