kKj3 dZddlZddlmZddlZddlZddlZddlZddlm Z ddl m Z ddl m Z mZddlmZddlmZdd lmZdd lmZdd lmZmZdd lmZd dlmZd dlmZddl m!Z!ddl"m#Z#m$Z$m%Z%m&Z&ejNdk(r ejPn ejRZ*e*de+de,e+de,e+fdZ-de+de%e.e.ffdZ/de/_0de#e+fdZ1defdZ2d3de3de#e+fdZ4d4de+de$e#e+d e%e+d!fde%e+e+ffd"Z5 d5de+d$e+d%e+de$e#e+d e%e+d!ff d&Z6Gd'd(ejnZ8e Z9e:ejvd#Zd,Z?Gd-d.e>Z@e d/ZAd0ZBd6d1ZCd6d2ZDy)7zWGeneric testing tools. Authors ------- - Fernando Perez N)Path)contextmanager)StringIO)PopenPIPE)patch)Config)get_output_error_code) list_strings) temp_pyfileTee) py3compat) decorators) skipdoctest) TracebackType)ListOptionalTupleTypewin32 startPathfilesreturnct|tsJtjj |d}|Dcgc]"}tjj ||$c}Scc}w)aMake full paths for all the listed files, based on startPath. Only the base part of startPath is kept, since this routine is typically used with a script's ``__file__`` variable as startPath. The base of startPath is then prepended to all the listed files, forming the output list. Parameters ---------- startPath : string Initial path to use as the base for the results. This path is split using os.path.split() and only its first component is kept. files : list One or more files. Examples -------- >>> full_path('/foo/bar.py',['a.txt','b.txt']) ['/foo/a.txt', '/foo/b.txt'] >>> full_path('/foo',['a.txt','b.txt']) ['/a.txt', '/b.txt'] r) isinstancelistospathsplitjoin)rrbasefs f/Users/ahmed/devFolder/Ultron/claude-voice/.venv/lib/python3.12/site-packages/IPython/testing/tools.py full_pathr%(sR6 eT "" " 77== #A &D+0 25aRWW\\$q !5 22 2s'A#txtctjd|tj}|r t|j d}d}||fStjd|tj}|r d}t|j d}||fStjd|tj}|r8t|j d}t|j d}||fSy)aParse the output of a test run and return errors, failures. Parameters ---------- txt : str Text output of a test run, assumed to contain a line of one of the following forms:: 'FAILED (errors=1)' 'FAILED (failures=1)' 'FAILED (errors=1, failures=1)' Returns ------- nerr, nfail number of errors and failures. z^FAILED \(errors=(\d+)\)rrz^FAILED \(failures=(\d+)\)z(^FAILED \(errors=(\d+), failures=(\d+)\))rr)research MULTILINEintgroup)r&err_mnerrnfailfail_mboth_ms r$parse_test_outputr3Hs& II13 EE 5;;q>"e  YY4c2<< HF FLLO$e  YYBC||%F 6<<?#FLLO$e  Fc gdS)zEReturn a valid default argv for creating testing instances of ipython)z--quickz--colors=nocolorz--no-term-titlez --no-bannerz --autocall=0r6r4r$ default_argvr7vs  r4c8t}d|j_d|j_d|j_t jdd}t|j|j_ |jd|j_ |S)z6Return a config object with good defaults for testing.nocolorFrztest_hist.sqliteF)suffixdeletei')r TerminalInteractiveShellcolors TerminalTerminalInteractiveShell term_titleautocalltempfileNamedTemporaryFilernameHistoryManager hist_fileclose db_cache_size)configr#s r$default_configrJsx XF-6F##*9AF++6/0F##,##+=eLA&*166lF#GGI*/F' Mr4 as_stringcRtjddg}|rdj|}|S)aW Return appropriate IPython command line name. By default, this will return a list that can be used with subprocess.Popen, for example, but passing `as_string=True` allows for returning the IPython command as a string. Parameters ---------- as_string: bool Flag to allow to return the command as a string. z-mIPython )sys executabler!)rK ipython_cmds r$get_ipython_cmdrRs+>>43Khh{+ r4fnameoptionscommands.cd}|g}t|z}tjjt}t }tjj ||}||zd|gz}tjj} d| d<| jdd| jD]"\} } t| trt| | $t|ttt| } | j!t#j$dj |xsd \} }t#j&| t#j&|}} | rt)j*d d | } | |fS) aUtility to call 'ipython filename'. Starts IPython with a minimal and safe configuration to make startup as fast as possible. Note that this starts IPython in a subprocess! Parameters ---------- fname : str, Path Name of file to be executed (should have .py or .ipy extension). options : optional, list Extra command-line flags to be passed to IPython. commands : optional, list Commands to send in on stdin Returns ------- ``(stdout, stderr)`` of ipython subprocess. TNz--ignorePYTHONWARNINGSPYCHARM_HOSTED)stdoutstderrstdinenv )inputz \x1b\[[^h]+h)r7rrdirname__file__rRr!environcopypopitemsrstrprintrr communicaterencodedecoder)sub)rSrTrU__tracebackhide__cmdargstest_dirrQ full_fnamefull_cmdr]kvpouterrs r$ipexecrws9.nw&Gwwx(H!#Kh.JW$j'99H **// C%CGG d# 1!S! !QK  htD#FA}}9#3#3DIIh4G#H#PD}QHC$i&6&6s&;C ff_b#. 8Or4r` expected_out expected_errcd}t|||\}}|rp|r]dj|jjdj|jjk(sJt d|d|dj|jjdj|jjk(sJy)a9Utility to call 'ipython filename' and validate output/error. This function raises an AssertionError if the validation fails. Note that this starts IPython in a subprocess! Parameters ---------- fname : str, Path Name of the file to be executed (should have .py or .ipy extension). expected_out : str Expected stdout of the process. expected_err : optional, str Expected stderr of the process. options : optional, list Extra command-line flags to be passed to IPython. Returns ------- None Tr^z Running file z produced error: N)rwr!strip splitlines ValueError)rSrxryrTrUrmrurvs r$ipexec_validater~s4eWh/HC  99SYY[3356$))""$//1; #S*+ + 99SYY[++- .$))'')3  r4c4eZdZdZd dedefdZdZdZdZy) TempFileMixinzcUtility class to create temporary Python/IPython files. Meant as a mixin class for test cases.srcextct||}t|dsg|_|jj|||_y)zMake a valid python temp file.tmpsN)r hasattrrappendrS)selfrrrSs r$mktmpzTempFileMixin.mktmps8C%tV$DI  r4czt|dr(|jD]} tj|yy#Y!xYw)Nr)rrrunlink)rrSs r$tearDownzTempFileMixin.tearDowns> 4 IIe$ # ! s6:c|SNr6rs r$ __enter__zTempFileMixin.__enter__&s r4c$|jyr)r)rexc_type exc_value tracebacks r$__exit__zTempFileMixin.__exit__)s  r4N)z.py) __name__ __module__ __qualname____doc__rgrrrrr6r4r$rr s).3 r4rzEDid not find {0!r} in printed output (on {1}): ------- {2!s} ------- cXeZdZdZd dededefdZdZdee e dee d ee fd Z y ) AssertPrintszContext manager for testing that code prints certain text. Examples -------- >>> with AssertPrints("abc", suppress=False): ... print("abcd") ... print("def") ... abcd def schannelsuppressc||_t|jttfr|jg|_||_||_yr)rrrg_re_typerr)rrrrs r$__init__zAssertPrints.__init__Cs6 dffsHo .ffXDF   r4c0tt|j|_t |_t |j |j|_tt|j|jr|j y|jy)N)r) getattrrOr orig_stream MyStringIObufferr teesetattrrrs r$rzAssertPrints.__enter__JsW"3 5 l t{{DLL9T\\$--4;;NTXXNr4etypevaluercd} | |jjy|jjtt|j |j |jj}|jD]~}t|trB|j|r%Jtj|j|j |||vrZJtj||j | |jjy#|jjwxYwNTF)rrGflushrrOrrrgetvaluerrrr*notprinted_msgformatpatternrrrrrmprintedrs r$rzAssertPrints.__exit__Ps    HHNN  HHNN  Ct'7'7 8kk**,GVVa*88G,en.C.CAIIt||]d.ee,<X)>)>q$,,PW)XX<   HHNN DHHNN sD$B D$,4D$!'D$$EN)rZT) rrrrrgboolrrrr BaseExceptionrrr6r4r$rr7sW !#!!!O htM':;H]D[hpq~hr4rz>Found {0!r} in printed output (on {1}): ------- {2!s} ------- ceZdZdZdZy)AssertNotPrintszcContext manager for checking that certain output *isn't* produced. Counterpart of AssertPrintscd} |6|jj |jjy|jjtt|j |j |jj}|jD]~}t|trB|j|s%Jtj|j|j |||vrZJtj||j | |jjy#|jjwxYwr)rrGrrrOrrrrrrrr* printed_msgrrrs r$rzAssertNotPrints.__exit__ms      HHNN  HHNN  Ct'7'7 8kk**,GVVa* xx0:1C1C 4<<2::0G+2[-?-?4<<.22+  HHNN DHHNN sD>B D>4D>;'D>>EN)rrrrrr6r4r$rris #r4rc#Kt|ddj dtj|y#tj|wxYww)zGCreate an empty, named, temporary file for the duration of the context.wzutf-8)encodingN)openrGrr)rDs r$ make_tempfilers; sW%++-  $ $sA:AAAc>t|dfd }td|S)aTemporarily replace the input() function to return the given values Use as a context manager: with fake_input(['result1', 'result2']): ... Values are returned in order. If input() is called again after the last value was used, EOFError is raised. cV tS#t$r}td|d}~wwxYw)NzNo more inputs given)next StopIterationEOFError)prompteits r$ mock_inputzfake_input..mock_inputs/ :8O :12 9 :s  ( #(zbuiltins.inputr`)iterr)inputsrrs @r$ fake_inputrs# fB: !: ..r4ct|dgz}t|\}}}|dk(sJ|d|vsJd|vsJd|vsJ||fS)z)test that `ipython [subcommand] -h` worksz-hr TracebackOptions --help-allrRr  subcommandcmdrurvrcs r$help_output_testrsh  z40 0C(-LCb 7C7 c !! !    3   8Or4ct|dgz}t|\}}}|dk(sJ|d|vsJd|vsJd|vsJ||fS)z1test that `ipython [subcommand] --help-all` worksrrrrClassrrs r$help_all_output_testrsf  z<8 8C(-LCb 7C7 c !! !    c>> 8Or4r:)Nr6)r`Nr6r)Errpathlibrr)rOrBunittest contextlibrior subprocessrr unittest.mockrtraitlets.config.loaderr IPython.utils.processr IPython.utils.textr IPython.utils.ior r IPython.utilsrr`rdecrtypesrtypingrrrrplatform skip_doctest null_deco doctest_decorgrr%r,r3__test__r7rJrrRrwr~TestCaserrtypecompilerrrrrrrrrr6r4r$rs  %"*7+-#.. ,/<<7+B{''  33T#Y34933>'3'5c?'V# d3i   td3i$7#7c 37E#s(O7[`adfiai[j7tFHQS,3,c,,%d3i0,AFsCx,^H%%D   3  **X l4/(r4