Format: application/x-python-code
1 o
2
3 ��h�Z � @ s� d Z ddlZddlZddlZejedd�� ddlmZ W d � n1 s&w Y ddl m
4 Z
5 ddl m Z m
6 Z
7 ddlmZ d d
8 � Zd d � Zd
9 d� Zdd� Zdd� ZG dd� dej�ZG dd� dej�ZG dd� dej�ZG dd� de�Zedkr�dd l m Z e � d ZZ d Z z9 e Z e�!� r�d e�"� Z#e r�e#e$e e# � Z%ne$e �Z%e�&e%� d Z e�!� s�ej'Z(e�)� Z*e+d!e*� e+d"e(� q� e,y� e+d#� Y nw d$d%� Z-d&d'� Z.G d(d)� d)e/�Z0G d*d+� d+e0ej�Z1G d,d-� d-e0ej�Z2g d.�Z3d/d0� Z4dS )1z#Tests for the inputsplitter module.� N�
10 inputsplitter)�match)r )�InputTransformer)�syntax� syntax_ml)�toolsc C sX ddl m} |� }|�� r&d|�� }d| }|| |� }|�|� |�� s
11 |�� }|S )z�Minimal example of the logic of an interactive interpreter loop.
12
13 This serves as an example, and it is used by the test system with a fake
14 raw_input that simulates interactive input.r )�
15 InputSplitter� z>>> )�IPython.core.inputsplitterr �push_accepts_more�get_indent_spaces�push� source_reset)Z
16 input_funcr �isp�indent�prompt�line�src� r �[c:\Users\shubh\Videos\agent\nova\Lib\site-packages\IPython/core/tests/test_inputsplitter.py�mini_interactive_loop s
17 �r c s t | �� � fdd�}|S )zDReturn a function that acts like raw_input but feeds the input list.c s zt � �W S ty Y dS w )N� )�next�
18 StopIteration)r �Zilinesr r �raw_in9 s
19
20 �z pseudo_input.<locals>.raw_in)�iter)�linesr r r r � pseudo_input6 s r c C sF g d�} t �t�� t�tj| � W d � d S 1 s w Y d S )N)
21 �r r )r � )�
22 r )z
23 r )�xr )z xr )z x� )z x� )z xr )z xr# )�pytest�warns�PendingDeprecationWarning�tt� check_pairsr Znum_ini_spaces��testsr r r � test_spacesC s "�r, c C s g d�} t �tj| � d S )N))�textr- )ztext # commentztext )ztext # comment
24 �text
25 )ztext # comment
26 r. )zline # c
27 line
28 z line
29 line
30 )z"line # c
31 line#c2
32 line
33 line #c
34
35 zline
36 line
37 line
38 line
39
40 )r( r) r Zremove_commentsr* r r r �test_remove_commentsT s r/ c C s, t �� } t| t�s J �d�| �dksJ �d S )N�tests test)r �get_input_encoding�
41 isinstance�str�encode)�encodingr r r �test_get_input_encoding` s r6 c @ �$ e Zd Zdd� Zdd� Zdd� ZdS )�NoInputEncodingTestCasec C s& t j| _G dd� d�}|� }|t _d S )Nc @ s e Zd ZdS )z(NoInputEncodingTestCase.setUp.<locals>.XN)�__name__�
42 __module__� __qualname__r r r r �Xk s r< )�sys�stdin� old_stdin)�selfr< Z
43 fake_stdinr r r �setUpi s
44 z NoInputEncodingTestCase.setUpc C s t �� }| �|d� d S )N�ascii)r r1 � assertEqual)r@ �encr r r r0 o s z NoInputEncodingTestCase.testc C s | j t_d S �N)r? r= r> �r@ r r r �tearDownu s z NoInputEncodingTestCase.tearDownN)r9 r: r; rA r0 rG r r r r r8 h s r8 c @ s� e Zd Zdd� Zdd� Zdd� Zdd� Zd d
45 � Zd d � Zd
46 d� Z dd� Z
47 dd� Z dd� Z dd� Z
48 dd� Zdd� Zdd � Zd d � Zd d � Zd!d"� Zd#d$� Zd%d&� Zd'd(� Zd)d*� Zd+d,� Zd-d.� Zejj d/e j d0v e d1d2�d3d4� �Z d5d6� Z!d7S )8�InputSplitterTestCasec C � t �� | _ d S rE )r r rF r r r rA z � zInputSplitterTestCase.setUpc C sd | j }|�d� |�� | �|jg � | �|�� d� | �|jd� | �|jd � | �|jd� d S )N�x=1r r F) r r
49 �resetrC �_bufferr �source�codeZ _is_complete�r@ r r r r �
50 test_reset} s
51 z InputSplitterTestCase.test_resetc C sv | j �d� | j �d� | �| j jd� | �t| j j�dkd� | �| j �� d� | �| j jg � | �| j jd� d S )N�1�2z1
52 2
53 r Tr )r �_storerC rN �lenrM r rF r r r � test_source� s z!InputSplitterTestCase.test_sourcec C sX | j }|�d� | �|�� d� |�d� | �|�� d� |�d� | �|�� d� d S )NrK r z
54 if 1:
55 x=1r$ �y=2
56 �r r
57 rC r rP r r r � test_indent� �
58
59
60 z!InputSplitterTestCase.test_indentc C sX | j }|�d� | �|�� d� |�d� | �|�� d� |�d� | �|�� d� d S )N�if 1:r$ z x=1z rX rP r r r � test_indent2� s
61
62
63 z"InputSplitterTestCase.test_indent2c C s. | j }|�d� |�d� | �|�� d� d S )Nr[ z x = (1+
64 2)r$ rX rP r r r � test_indent3� s
65
66 z"InputSplitterTestCase.test_indent3c C sr | j }|�d� | �|�� d� |�d� | �|�� d� |�d� | �|�� d� |�d� | �|�� d� d S )Nzif 1:
67 x=1r$ rW r zif 1:
68 x=1rX rP r r r � test_indent4� �
69
70
71
72 z"InputSplitterTestCase.test_indent4c C �X | j }|�d� | �|�� d� |�d� | �|�� d� |�d� | �|�� d� d S )Nzif 1:
73 passes = 5r$ zif 1:
74 passr zif 1:
75 pass rX rP r r r �test_dedent_pass� �
76
77
78 z&InputSplitterTestCase.test_dedent_passc C r` )Nzwhile 1:
79 breaks = 5r$ zwhile 1:
80 breakr zwhile 1:
81 break rX rP r r r �test_dedent_break� rb z'InputSplitterTestCase.test_dedent_breakc C r` )Nzwhile 1:
82 continues = 5r$ zwhile 1:
83 continuer zwhile 1:
84 continue rX rP r r r �test_dedent_continue� rb z*InputSplitterTestCase.test_dedent_continuec C sr | j }|�d� | �|�� d� |�d� | �|�� d� |�d� | �|�� d� |�d� | �|�� d� d S )Nzif 1:
85 raised = 4r$ z if 1:
86 raise TypeError()r zif 1:
87 raisezif 1:
88 raise rX rP r r r �test_dedent_raise� r_ z'InputSplitterTestCase.test_dedent_raisec C s� | j }|�d� | �|�� d� |�d� | �|�� d� |�d� | �|�� d� |�d� | �|�� d� |�d� | �|�� d� d S )Nzif 1:
89 returning = 4r$ zif 1:
90 return 5 + 493r zif 1:
91 returnzif 1:
92 return zif 1:
93 return(0)rX rP r r r �test_dedent_return� s
94
95
96
97
98 z(InputSplitterTestCase.test_dedent_returnc C s | j }| �|�d�d� d S )NrK T)r rC r
99 rP r r r � test_push� s z InputSplitterTestCase.test_pushc C s@ | j }| �|�d�d� dD ]}t|� | �|�|�d� qd S )Nr[ F)� x=1z # a commentz y=2T)r rC r
100 �print)r@ r r r r r �
101 test_push2� s �z InputSplitterTestCase.test_push2c C s0 | j }|�d� |�d� | �|�d�d� d S )Nzif True:z a = 1zb = [1,F)r r
102 rC rP r r r �
103 test_push3� s
104
105 z InputSplitterTestCase.test_push3c C �$ | j }|�d� | �|�� d� d S )NrK F�r r
106 rC r rP r r r �test_push_accepts_more �
107 z,InputSplitterTestCase.test_push_accepts_morec C sX | j }|�d� | �|�� d� |�d� | �|�� d� |�d� | �|�� d� d S )Nr[ Trh r Frm rP r r r �test_push_accepts_more2 rZ z-InputSplitterTestCase.test_push_accepts_more2c C rl )Nz
108 x = (2+
109 3)Frm rP r r r �test_push_accepts_more3 ro z-InputSplitterTestCase.test_push_accepts_more3c C sl | j }|�d� |�d� |�d� | �|�� d� |�d� | �|�� d� |�d� | �|�� d� d S )Nr[ z x = (2+z 3)Tz y = 3r Frm rP r r r �test_push_accepts_more4 s
110
111
112
113
114 z-InputSplitterTestCase.test_push_accepts_more4c C sB | j }|�d� |�d� |�d� |�d� | �|�� d� d S )Nztry:z a = 5zexcept:z raiseTrm rP r r r �test_push_accepts_more5% s
115
116
117
118 z-InputSplitterTestCase.test_push_accepts_more5c C s> | j }|�d� | �|�� d� |�d� | �|�� d� d S )Nz import os, \Tr= Frm rP r r r �test_continuation/ s
119
120
121 z'InputSplitterTestCase.test_continuationc C rl )Nzrun fooFrm rP r r r �test_syntax_error6 s
122 z'InputSplitterTestCase.test_syntax_errorc C s( | j �d� | j �d� | j �d� d S )Nu Pérezu éu u'é')r r
123 rF r r r � test_unicode> s z"InputSplitterTestCase.test_unicodeu" Bug in python 3.9.8 – bpo 45738))� � r �alphar# T)�reason� conditionZraises�strictc C sX | j }|�d� | �|�� d� |�d� | �|�� d� |�d� | �|�� d� dS )z Test issue #2108.z1 \
124
125 Fz1 \ z(1 \ Nrm rP r r r �test_line_continuationC s
126
127
128 z,InputSplitterTestCase.test_line_continuationc C sd | j }| �|�d�d� | �|�d�d� | �|�d�d� | �|�d�d� | �|�d �d� d S )
129 N�a = 1)ZcompleteNzfor a in range(5):)�
130 incompleter$ z raise = 2)�invalidNz
131 a = [1,
132 2,)r r zdef a():
133 x=1
134 global x)r rC Zcheck_completerP r r r �test_check_completeY s z)InputSplitterTestCase.test_check_completeN)"r9 r: r; rA rQ rV rY r\ r] r^ ra rc rd re rf rg rj rk rn rp rq rr rs rt ru rv r% �markZxfailr= � version_info� SystemErrorr} r� r r r r rH y s@
135
136
137
138
139
140 �
141 rH c @ �@ e Zd ZdZdd� Zdd� Zdd� Zdd � Zd
142 d � Zd d
143 � Z dS )�InteractiveLoopTestCasez7Tests for an interactive loop like a python shell.
144 c C s@ t t|��}i }t||� |�� D ] \}}| �|| |� qdS )a$ Validate that the given input lines produce the resulting namespace.
145
146 Note: the input lines are given exactly as they would be typed in an
147 auto-indenting environment, as mini_interactive_loop above already does
148 auto-indenting and prepends spaces to the input.
149 N)r r �exec�itemsrC )r@ r �nsr Ztest_ns�k�vr r r �check_nsd s
150 �z InteractiveLoopTestCase.check_nsc C s | � dgtdd�� d S )NrK r �r" �r� �dictrF r r r � test_simplet s z#InteractiveLoopTestCase.test_simplec C s | � ddgtdd�� d S )Nr[ zx=2r# r� r� rF r r r � test_simple2w � z$InteractiveLoopTestCase.test_simple2c C s | � dgtddd�� d S )Nzx=1; y=2r r# )r" �yr� rF r r r �test_xyz r� z InteractiveLoopTestCase.test_xyc C s | � g d�tdddd�� d S )N)r[ za=1zb=2zc=3r r# rw )�a�b�cr� rF r r r �test_abc} s z InteractiveLoopTestCase.test_abcc C s | � g d�tdd�� d S )N)zx =(1+z1+z2)r$ r� r� rF r r r �
151 test_multi� r� z"InteractiveLoopTestCase.test_multiN)
152 r9 r: r; �__doc__r� r� r� r� r� r� r r r r r� a s r� c @ r� )�IPythonInputTestCaseaS By just creating a new class whose .isp is a different instance, we
153 re-run the same test battery on the new input splitter.
154
155 In addition, this runs the tests over the syntax and syntax_ml dicts that
156 were tested by individual functions, as part of the OO interface.
157
158 It also makes some checks on the raw buffer storage.
159 c C rI rE )r �IPythonInputSplitterrF r r r rA � rJ zIPythonInputTestCase.setUpc C s� | j }t�� D ]7}|D ]2\}}|�d�rq |�|d � |j}|�� }| �|�� |t j
160 � d|||�� | �|�� |�� � q qdS )z6Call all single-line syntax tests from the main objectr r! r N) r r �values�
161 startswithr
162 �
163 source_rawr rC �rstripr( Z
164 pair_fail_msg�format)r@ r �example�raw�out_t�out_raw�outr r r � test_syntax� s
165 ���z IPythonInputTestCase.test_syntaxc C s� | j }t�� D ]N}|D ]I}g }g }|D ]\}}|d ur |�|� |d ur.|�|� |�|� q|j}|�� } d�|��� }
166 d�|��� } | � | �� |
167 � | � |�� | � q qd S �Nr! )
168 r r r� �appendr
169 r� r �joinr� rC ) r@ r r� �
170 line_pairs� out_t_partsZ raw_partsZlrawZ
171 out_t_partr� r� r� r� r r r �test_syntax_multiline� s(
172
173
174 ���z*IPythonInputTestCase.test_syntax_multilinec C sl | j }t�� D ],}g }|D ]%}d�dd� |D ��}d�dd� |D ��}|�|�}| �|�� |�� � q
175 qd S )Nr! c s s � | ] \}}|d ur|V qd S rE r )�.0�r�_r r r � <genexpr>� � � zBIPythonInputTestCase.test_syntax_multiline_cell.<locals>.<genexpr>c s s � | ] \}}|d ur|V qd S rE r )r� r� �tr r r r� � r� )r r r� r� �transform_cellrC r� )r@ r r� r� r� r� r� r� r r r �test_syntax_multiline_cell� s
176 ��z/IPythonInputTestCase.test_syntax_multiline_cellc C sH | j }dD ] \}}}}d|||f }|�|�}| �|�� |�� � qd S )N))z%%cellm a
177 In[1]:�cellmr� zIn[1]:)z%%cellm
178 line
179 >>> hir� r z line
180 >>> hi)z>>> %%cellm
181 line
182 >>> hir� r zline
183 hi)z%%cellm
184 >>> hir� r z>>> hi)z%%cellm
185 line1
186 line2r� r z line1
187 line2)z%%cellm
188 line1\\
189 line2r� r z
190 line1\\
191 line2z(get_ipython().run_cell_magic(%r, %r, %r))r r� rC r� )r@ r r� �namer �cell�expectedr� r r r �test_cellmagic_preempt� s �
192 �z+IPythonInputTestCase.test_cellmagic_preemptc C sv | j }G dd� dt�}|j�d|� � dddd fdd d
193 fd d d ffD ]\}}|�|�}| �|�� |�� � q%d S )
194 Nc @ r7 )zKIPythonInputTestCase.test_multiline_passthrough.<locals>.CommentTransformerc S s
195 g | _ d S rE )�_linesrF r r r �__init__� s
196 zTIPythonInputTestCase.test_multiline_passthrough.<locals>.CommentTransformer.__init__c S s | j �|d � d S )N�#)r� r� )r@ r r r r r
197 � s zPIPythonInputTestCase.test_multiline_passthrough.<locals>.CommentTransformer.pushc S s d� | j�}g | _|S r� )r� r� )r@ r- r r r rL � s zQIPythonInputTestCase.test_multiline_passthrough.<locals>.CommentTransformer.resetN)r9 r: r; r� r
198 rL r r r r �CommentTransformer� s r� r )za=5za=5#z%ls fooz$get_ipython().run_line_magic(%r, %r))�lszfoo#z!ls foo
199 %ls barz=get_ipython().system(%r)
200 get_ipython().run_line_magic(%r, %r))zls foo#r� zbar#z1
201 2
202 3
203 %ls foo
204 4
205 5z31#
206 2#
207 3#
208 get_ipython().run_line_magic(%r, %r)
209 4#
210 5#)r r Zphysical_line_transforms�insertr� rC r� )r@ r r� r� r� r� r r r �test_multiline_passthrough� s
211
212
213 �
214 �z/IPythonInputTestCase.test_multiline_passthroughN)
215 r9 r: r; r� rA r� r� r� r� r� r r r r r� � s r� �__main__)r� zIn> Tr z... zInput source was:
216 zRaw source was:
217 ZByec C s� t �d�du s J �t �d�du sJ �t �d�du sJ �t �d�du s$J �t �d�du s-J �t �d�du s6J �t �d �du s?J �t �d
218 �du sHJ �t �d �du sQJ �t �d �du sZJ �t �d
219 �du scJ �d S )Nr F�abc�abc
220 zabc
221 ar! Tz
222 zabc
223 �abc
224
225 zabc
226 d
227
228 z abc
229 d
230 e
231
232 z
233 abc
234
235
236
237 )r Z
238 last_blankr r r r �test_last_blank s r� c C s$ t �d�du s J �t �d�du sJ �t �d�du sJ �t �d�du s$J �t �d�du s-J �t �d�du s6J �t �d�d u s?J �t �d
239 �d u sHJ �t �d �d u sQJ �t �d �d u sZJ �t �d
240 �d u scJ �t �d�d u slJ �t �d�d u suJ �t �d�d u s~J �t �d�d u s�J �t �d�d u s�J �d S )Nr Fr� r� zabc
241
242 azabc
243
244 r� z
245
246 Tz
247
248 z
249
250 zabc
251
252 zabc
253
254
255 zabc
256
257
258 zabc
259
260
261 z abc
262
263
264
265 zabc
266 d
267
268
269 z abc
270 d
271 e
272 f
273
274
275 )r Zlast_two_blanksr r r r �test_last_two_blanks# s r� c @ r7 )�CellMagicsCommonc C s$ d}| j �|�}d}||ksJ �d S )Nz%%cellm line
276 body
277 z6get_ipython().run_cell_magic('cellm', 'line', 'body')
278 )�spr� )r@ r r� �refr r r �test_whole_cell9 s z CellMagicsCommon.test_whole_cellc C s" | j �d� | j �� du sJ �d S )Nz%%cellm?F�r� r
279 r rF r r r �test_cellmagic_help? s z$CellMagicsCommon.test_cellmagic_helpc C s | j �� d S rE )r� rL rF r r r rG C rJ zCellMagicsCommon.tearDownN)r9 r: r; r� r� rG r r r r r� 7 s r� c @ s( e Zd Zejdd�Zdd� Zdd� ZdS )�CellModeCellMagicsF�Zline_input_checkerc C sX | j }|�d� |�� du sJ �|�d� |�� du s J �|�d� |�� du s*J �d S )Nz%%cellm firstline
280 Tzline2
281 r! r� �r@ r� r r r �test_incrementalJ s
282
283
284 z#CellModeCellMagics.test_incrementalc C s* d� g d��}| j�|�}d|v sJ �d S )Nr! )z%%writefile foo.py�# coding: utf-8u print(u"üñîçø∂é")r� )r� r� r� )r@ r r� r r r �test_no_strip_codingT s z'CellModeCellMagics.test_no_strip_codingN)r9 r: r; r r� r� r� r� r r r r r� G s
285 r� c @ s e Zd Zejdd�Zdd� ZdS )�LineModeCellMagicsTr� c C s> | j }|�d� |�� du sJ �|�d� |�� du s J �d S )Nz%%cellm line2
286 Tr! Fr� r� r r r r� a s
287
288
289 z#LineModeCellMagics.test_incrementalN)r9 r: r; r r� r� r� r r r r r� ^ s r� ))r~ r )z for a in b:r$ )zdef f():r$ )zdef f(): #commentr$ )za = ":#not a comment"r )zdef f():
290 a = 1r$ )zdef f():
291 return 1r )z'for a in b:
292 if a < 0: continuerw )za = {r$ )z
293 a = {
294 1,� )z
295 b = """123r r )zdef f():
296 passr )z$class Bar:
297 def f():
298 passr$ )z%class Bar:
299 def f():
300 raiser$ c C s: t D ]\} }t�| �}d�||| �}||ksJ |��qd S )Nz#{!r} != {!r} (expected)
301 Code: {!r})�indentation_samplesr Zfind_next_indentr� )rO �exp�res�msgr r r �test_find_next_indent s
302
303 �r� )5r� Zunittestr% r= r& �DeprecationWarningZ IPython.corer r Z IPython.core.inputtransformerr Z(IPython.core.tests.test_inputtransformerr r ZIPython.testingr r( r r r, r/ r6 ZTestCaser8 rH r� r� r9 r
304 r� Z start_promptZ
305 autoindentr r r r �inputr r
306 r� r� r r ri �EOFErrorr� r� �objectr� r� r� r� r� r r r r �<module> sd �
307 i#l
308 �
309
310 � �