}eZfdZddlZddlZddlZejed5ddlmZdddddl m Z ddl m Z m Z ddlmZd Zd Zd Zd Zd ZGddej,ZGddej,ZGddej,ZGddeZedk(rddlmZedcZZdZ eZ ejBrSdejDzZ#ere#e$e e#zzZ%ne$e Z%ejLe%dZ ejBrSejNZ(ejRZ*e+de*e+de(dZ-d Z.Gd!d"e/Z0Gd#d$e0ej,Z1Gd%d&e0ej,Z2gd'Z3d(Z4y#1swYkxYw#e,$r e+dYawxYw))z#Tests for the inputsplitter module.N inputsplitter)match)r)InputTransformer)syntax syntax_ml)toolscddlm}|}|jrHd|jz}d|z}|||z}|j||jrH|j }|S)zMinimal example of the logic of an interactive interpreter loop. This serves as an example, and it is used by the test system with a fake raw_input that simulates interactive input.r) InputSplitter z>>> )IPython.core.inputsplitterr push_accepts_moreget_indent_spacespush source_reset) input_funcr ispindentpromptlinesrcs G/usr/lib/python3/dist-packages/IPython/core/tests/test_inputsplitter.pymini_interactive_looprs 9 /C #   !*S**,,& 6** #   ! #   C Jc(t|fd}|S)zDReturn a function that acts like raw_input but feeds the input list.c: tS#t$rYywxYw)N)next StopIteration)riliness rraw_inzpseudo_input..raw_in9s$ <   s  )iter)linesr rs @r pseudo_inputr#6s %[F Mrcgd}tjt5tjt j |dddy#1swYyxYw)N) rr)r ) r)z r&)xr)z xr&)z x)z x)z xr&)z xr))pytestwarnsPendingDeprecationWarningtt check_pairsrnum_ini_spacestestss r test_spacesr3CsA E / 02 s))51222s %A  AcTgd}tjtj|y)N))textr5)ztext # commentztext )ztext # comment text )ztext # comment r6)zline # c line z line line )z"line # c line#c2 line line #c zline line line line )r.r/rremove_commentsr1s rtest_remove_commentsr8Ts ENN3&&.rc|tj}t|tsJdj |dk(sJy)Nteststest)rget_input_encoding isinstancestrencode)encodings rtest_get_input_encodingr@`s9%%'H h $$ $ == "g -- -rceZdZdZdZdZy)NoInputEncodingTestCasecftj|_Gdd}|}|t_y)Nc eZdZy)(NoInputEncodingTestCase.setUp..XN)__name__ __module__ __qualname__rrXrEksrrJ)sysstdin old_stdin)selfrJ fake_stdins rsetUpzNoInputEncodingTestCase.setUpis  S  rcPtj}|j|dy)Nascii)rr; assertEqual)rNencs rr:zNoInputEncodingTestCase.testos"$$& g&rc.|jt_yN)rMrKrLrNs rtearDownz NoInputEncodingTestCase.tearDownus NN rN)rFrGrHrPr:rXrIrrrBrBhs ' #rrBceZdZdZdZdZdZdZdZdZ dZ d Z d Z d Z d Zd ZdZdZdZdZdZdZdZdZdZdZej6j9dej<dveddZ dZ!y)InputSplitterTestCasec6tj|_yrV)rr rWs rrPzInputSplitterTestCase.setUpzs$$&rc|j}|jd|j|j|jg|j|j d|j|j d|j|jd|j|jdy)Nx=1rrF) rrresetrS_bufferrsourcecode _is_completerNrs r test_resetz InputSplitterTestCase.test_reset}shh  b) ...0!4 R( 4( ))51rc |jjd|jjd|j|jjd|jt |jj dkDd|j|jj d|j|jj g|j|jjdy)N12z1 2 rTr)r_storerSr`lenr_rrWs r test_sourcez!InputSplitterTestCase.test_sources   (3 TXX--.q0$7 ..0(; ))2. "-rcN|j}|jd|j|jd|jd|j|jd|jd|j|jdy)Nr]rz if 1: x=1r*y=2 rrrSrrcs r test_indentz!InputSplitterTestCase.test_indentshh ...0!4!" ...0!4 ...0!4rcN|j}|jd|j|jd|jd|j|jd|jd|j|jdy)Nif 1:r*z x=1z rmrcs r test_indent2z"InputSplitterTestCase.test_indent2shh ...0!4 ...0!4 ...0!4rc|j}|jd|jd|j|jdy)Nrpz x = (1+ 2)r*rmrcs r test_indent3z"InputSplitterTestCase.test_indent3sFhh &' ...0!4rc|j}|jd|j|jd|jd|j|jd|jd|j|jd|jd|j|jdy)Nzif 1: x=1r*rlrzif 1: x=1rmrcs r test_indent4z"InputSplitterTestCase.test_indent4shh"# ...0!4 ...0!4#$ ...0!4 ...0!4rcN|j}|jd|j|jd|jd|j|jd|jd|j|jdy)Nzif 1: passes = 5r*zif 1: passrzif 1: pass rmrcs rtest_dedent_passz&InputSplitterTestCase.test_dedent_passshh() ...0!4#$ ...0!4&' ...0!4rcN|j}|jd|j|jd|jd|j|jd|jd|j|jdy)Nzwhile 1: breaks = 5r*zwhile 1: breakrzwhile 1: break rmrcs rtest_dedent_breakz'InputSplitterTestCase.test_dedent_breakshh+, ...0!4'( ...0!4*+ ...0!4rcN|j}|jd|j|jd|jd|j|jd|jd|j|jdy)Nzwhile 1: continues = 5r*zwhile 1: continuerzwhile 1: continue rmrcs rtest_dedent_continuez*InputSplitterTestCase.test_dedent_continueshh./ ...0!4*+ ...0!4-. ...0!4rc|j}|jd|j|jd|jd|j|jd|jd|j|jd|jd|j|jdy)Nzif 1: raised = 4r*zif 1: raise TypeError()rzif 1: raisezif 1: raise rmrcs rtest_dedent_raisez'InputSplitterTestCase.test_dedent_raiseshh() ...0!401 ...0!4$% ...0!4*+ ...0!4rc|j}|jd|j|jd|jd|j|jd|jd|j|jd|jd|j|jd|jd|j|jdy)Nzif 1: returning = 4r*zif 1: return 5 + 493rzif 1: returnzif 1: return zif 1: return(0)rmrcs rtest_dedent_returnz(InputSplitterTestCase.test_dedent_returnshh+, ...0!4-. ...0!4%& ...0!4+, ...0!4() ...0!4rc`|j}|j|jddy)Nr]T)rrSrrcs r test_pushzInputSplitterTestCase.test_pushs%hh %$/rc|j}|j|jdddD]/}t||j|j|d1y)NrpF) x=1z # a commentz y=2T)rrSrprint)rNrrs r test_push2z InputSplitterTestCase.test_push2sVhh '*E25 3D $K   XSXXd^T 2 3rc|j}|jd|jd|j|jddy)Nzif True:z a = 1zb = [1,F)rrrSrcs r test_push3z InputSplitterTestCase.test_push3sBhh ),e4rc|j}|jd|j|jdy)Nr]FrrrSr rcs rtest_push_accepts_morez,InputSplitterTestCase.test_push_accepts_mores4hh ...0%8rcN|j}|jd|j|jd|jd|j|jd|jd|j|jdy)NrpTrrFrrcs rtest_push_accepts_more2z-InputSplitterTestCase.test_push_accepts_more2shh ...0$7 ...0$7  ...0%8rc|j}|jd|j|jdy)Nz x = (2+ 3)Frrcs rtest_push_accepts_more3z-InputSplitterTestCase.test_push_accepts_more3s5hh ...0%8rc|j}|jd|jd|jd|j|jd|jd|j|jd|jd|j|jdy)Nrpz x = (2+z 3)Tz y = 3rFrrcs rtest_push_accepts_more4z-InputSplitterTestCase.test_push_accepts_more4shh  ...0$7 ...0$7  ...0%8rc|j}|jd|jd|jd|jd|j|jdy)Nztry:z a = 5zexcept:z raiseTrrcs rtest_push_accepts_more5z-InputSplitterTestCase.test_push_accepts_more5%sahh ...0$7rc|j}|jd|j|jd|jd|j|jdy)Nz import os, \TrKFrrcs rtest_continuationz'InputSplitterTestCase.test_continuation/s^hh! ...0$7 ...0%8rc|j}|jd|j|jdy)Nzrun fooFrrcs rtest_syntax_errorz'InputSplitterTestCase.test_syntax_error6s7hh  ...0%8rc|jjd|jjd|jjdy)NuPérezuéuu'é')rrrWs r test_unicodez"InputSplitterTestCase.test_unicode>s2  i   k"  n%ru"Bug in python 3.9.8 – bpo 45738)) ralphar)T)reason conditionraisesstrictcN|j}|jd|j|jd|jd|j|jd|jd|j|jdy)z Test issue #2108.z1 \ Fz1 \ z(1 \ Nrrcs rtest_line_continuationz,InputSplitterTestCase.test_line_continuationCshh ...0%8  ...0%8  ...0%8rcp|j}|j|jdd|j|jdd|j|jdd|j|jdd|j|jd dy) Na = 1)completeNzfor a in range(5):) incompleter*z raise = 2)invalidNz a = [1, 2,)rrzdef a(): x=1 global x)rrScheck_completercs rtest_check_completez)InputSplitterTestCase.test_check_completeYshh +++G46HI +++,@ACTU +++K8:KL +++M:>  9  9 ]rrZc4eZdZdZdZdZdZdZdZdZ y) InteractiveLoopTestCasez7Tests for an interactive loop like a python shell. ctt|}i}t|||jD]\}}|j |||y)a$Validate that the given input lines produce the resulting namespace. Note: the input lines are given exactly as they would be typed in an auto-indenting environment, as mini_interactive_loop above already does auto-indenting and prepends spaces to the input. N)rr#execitemsrS)rNr"nsrtest_nskvs rcheck_nsz InteractiveLoopTestCase.check_nsdsQ$L$78 S'88: ,CAa   WQZ + ,rc>|jdgtdy)Nr]r&r(rdictrWs r test_simplez#InteractiveLoopTestCase.test_simplets ugtay)rc@|jddgtdy)Nrpzx=2r)rrrWs r test_simple2z$InteractiveLoopTestCase.test_simple2ws w&q 2rc@|jdgtddy)Nzx=1; y=2r&r))r(yrrWs rtest_xyzInteractiveLoopTestCase.test_xyzs zlD1N3rcD|jgdtdddy)N)rpza=1zb=2zc=3r&r)r)abcrrWs rtest_abcz InteractiveLoopTestCase.test_abc}s 14!qA3FGrc@|jgdtdy)N)zx =(1+z1+z2)r*rrrWs r test_multiz"InteractiveLoopTestCase.test_multis *D1I6rN) rFrGrH__doc__rrrrrrrIrrrras&, *34H7rrc4eZdZdZdZdZdZdZdZdZ y) IPythonInputTestCaseaSBy just creating a new class whose .isp is a different instance, we re-run the same test battery on the new input splitter. In addition, this runs the tests over the syntax and syntax_ml dicts that were tested by individual functions, as part of the OO interface. It also makes some checks on the raw buffer storage. c6tj|_yrV)rIPythonInputSplitterrWs rrPzIPythonInputTestCase.setUps++-rc |j}tjD]}|D]\}}|jdr|j|dz|j }|j }|j|j|tjjd||||j|j|jy)z6Call all single-line syntax tests from the main objectr r'rN) rrvalues startswithr source_rawrrSrstripr. pair_fail_msgformat)rNrexamplerawout_tout_rawouts r test_syntaxz IPythonInputTestCase.test_syntaxshh}} AG% A U>>#&T"..&c&&(  u((//UCPR  !13::<@ A Arc |j}tjD]}|D]}g}g}|D]>\}}||j|||j||j|@|j }|j } dj|j} dj|j} |j| j| |j|j| yNr') rrrappendrrrjoinrrS) rNrr line_pairs out_t_parts raw_partslraw out_t_partrrrrs rtest_syntax_multilinez*IPythonInputTestCase.test_syntax_multilineshh '') 8G% 8   (2/$D*!-#**:6' !((. /..&c&&( +.557ii *113  u5  !137! 8 8rc@|j}tjD]{}g}|D]r}djd|D}djd|D}|j|}|j |j |j t}y)Nr'c3,K|] \}}| |ywrVrI).0r_s r zBIPythonInputTestCase.test_syntax_multiline_cell..sJdaAMJ c3,K|] \}}| |ywrVrI)rrts rrzBIPythonInputTestCase.test_syntax_multiline_cell..s!K!Q]!!Kr)rrrrtransform_cellrSr)rNrrrrrrrs rtest_syntax_multiline_cellz/IPythonInputTestCase.test_syntax_multiline_cellshh '') ?GK% ? iiJjJJ !Kz!KK(c((-  u||~>  ? ?rc|j}dD]S\}}}}d|d|d|d}|j|}|j|j|jUy)N))z%%cellm a In[1]:cellmrzIn[1]:)z%%cellm line >>> hirrz line >>> hi)z>>> %%cellm line >>> hirrzline hi)z%%cellm >>> hirrz>>> hi)z%%cellm line1 line2rrz line1 line2)z%%cellm line1\\ line2rrz line1\\ line2zget_ipython().run_cell_magic(z, ))rrrSr)rNrrnamercellexpectedrs rtest_cellmagic_preemptz+IPythonInputTestCase.test_cellmagic_preemptsfhh& > !CtTdDH%#$$S)C   SZZ\8??+< = >rc6|j}Gddt}|jjd|ddddzfdd d zfd d dzffD]E\}}|j|}|j |j |j Gy) NceZdZdZdZdZy)KIPythonInputTestCase.test_multiline_passthrough..CommentTransformercg|_yrV)_linesrWs r__init__zTIPythonInputTestCase.test_multiline_passthrough..CommentTransformer.__init__s   rc@|jj|dzy)N#)rr)rNrs rrzPIPythonInputTestCase.test_multiline_passthrough..CommentTransformer.pushs ""4#:.rcJdj|j}g|_|Sr)rr)rNr5s rr^zQIPythonInputTestCase.test_multiline_passthrough..CommentTransformer.resets yy-   rN)rFrGrHrrr^rIrrCommentTransformerrs ! / rrr)za=5za=5#z%ls fooz$get_ipython().run_line_magic(%r, %r))lszfoo#z!ls foo %ls barz=get_ipython().system(%r) get_ipython().run_line_magic(%r, %r))zls foo#rzbar#z1 2 3 %ls foo 4 5z31# 2# 3# get_ipython().run_line_magic(%r, %r) 4# 5#)rrphysical_line_transformsinsertrrSr)rNrrrrrs rtest_multiline_passthroughz/IPythonInputTestCase.test_multiline_passthroughshh !1  $$++A/A/CD  >AQQ R !ae" &'adt't u  >MC%#$$S)C   SZZ\8??+< = >rN) rFrGrHrrPrrrrrrIrrrrs&. A8* ?> >rr__main__)rzIn> Tr z... zInput source was: zRaw source was: Byec*tjddusJtjddusJtjddusJtjddusJtjddusJtjddusJtjd dusJtjd dusJtjd dusJtjd dusJtjd dusJy)NrFabcabc zabc ar'Tz zabc abc zabc d z abc d e z abc )r last_blankrIrrtest_last_blankr s >>"  && & >>% E )) ) >>' "e ++ + >>( #u ,, , >>$ 4 '' ' >>% D (( ( >>( #t ++ + >>) $ ,, , >>, '4 // / >>/ *d 22 2 >>* +t 33 3rc$tjddusJtjddusJtjddusJtjddusJtjddusJtjddusJtjdd usJtjd d usJtjd d usJtjd d usJtjd d usJtjdd usJtjdd usJtjdd usJtjdd usJtjdd usJy)NrFrrzabc azabc r z Tz z zabc zabc zabc zabc z abc zabc d z abc d e f )rlast_two_blanksrIrrtest_last_two_blanksr#s   r "e ++ +   u % .. .   w '5 00 0   z *e 33 3   z *e 33 3   y )U 22 2   v &$ .. .   w '4 // /   w '4 // /   z *d 22 2   { +t 33 3   | , 44 4   } - 55 5    /4 77 7   ~ .$ 66 6   3 4 << eZdZejdZdZdZy)CellModeCellMagicsFline_input_checkerc|j}|jd|jdusJ|jd|jdusJ|jd|jdusJy)Nz%%cellm firstline Tzline2 r'rrNrs rtest_incrementalz#CellModeCellMagics.test_incrementalJst WW %&##%---  ##%---  ##%---rcldjgd}|jj|}d|vsJy)Nr')z%%writefile foo.py# coding: utf-8uprint(u"üñîçø∂é")r!)rrr)rNrrs rtest_no_strip_codingz'CellModeCellMagics.test_no_strip_codingTs9ii   gg$$S) C'''rN)rFrGrHrrrrr"rIrrrrGs ! ! !U ;B.(rrc8eZdZejdZdZy)LineModeCellMagicsTrc|j}|jd|jdusJ|jd|jdusJy)Nz%%cellm line2 Tr'Frrs rrz#LineModeCellMagics.test_incrementalasQ WW !"##%---  ##%...rN)rFrGrHrrrrrIrrr$r$^s ! ! !T :B/rr$))rr)z for a in b:r*)zdef f():r*)zdef f(): #commentr*)za = ":#not a comment"r)zdef f(): a = 1r*)zdef f(): return 1r)z'for a in b: if a < 0: continuer)za = {r*)z a = { 1,)z b = """123rr%)zdef f(): passr)z$class Bar: def f(): passr*)z%class Bar: def f(): raiser*ctD]9\}}tj|}dj|||}||k(r4J|y)Nz#{!r} != {!r} (expected) Code: {!r})indentation_samplesrfind_next_indentr)raexpresmsgs rtest_find_next_indentr-sJ( c""4(4;;CdKcz3zr)5runittestr+rKr,DeprecationWarning IPython.corerrIPython.core.inputtransformerr(IPython.core.tests.test_inputtransformerrrIPython.testingrr.rr#r3r8r@TestCaserBrZrrrFr r start_prompt autoindentrr rrinputrrrrrrrEOFErrorr robjectrrr$r(r-rIrrr:s)  V\\$O<212:F':2" /.#h//#"f]H--f]P 7h// 7Ff>0f>X z@-.CJ!F'#'')2S2244!E&-$88D =D(#'')..C"#""$C ' - %s +#. 4=( v  ()8+<+<(. /)8+<+< /*g22F  e s$F7A&F 0F F F0/F0