U -e7@sdZddlZddlmZddlZddlZddlZddlZddlm Z ddl m Z ddl m Z mZddlmZddlmZdd lmZdd lmZdd lmZmZdd lmZd dlmZd dlmZej dkrej!nej"Z#e#ddZ$ddZ%de%_&ddZ'ddZ(d7ddZ)d8ddZ*d9d d!Z+Gd"d#d#ej,Z-d$Z.d%d&Z/e Z0e1e2dZ3d'Z4Gd(d)d)e5Z6d*Z7Gd+d,d,e6Z8e d-d.Z9e d/d0Z:d1d2Z;d:d3d4Z N)Path)contextmanager)StringIO)PopenPIPE)patch)Config)get_output_error_code) list_strings) temp_pyfileTee) py3compat) decorators) skipdoctestwin32cs*t|}tj|dfdd|DS)aPMake 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 : string or 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'] If a single file is given, the output is still a list:: >>> full_path('/foo','a.txt') ['/a.txt'] rcsg|]}tj|qS)ospathjoin).0fbaserV/var/www/html/Darija-Ai-Train/env/lib/python3.8/site-packages/IPython/testing/tools.py Hszfull_path..)r rrsplit)Z startPathfilesrrr full_path&s rcCstd|tj}|r.t|d}d}||fStd|tj}|r\d}t|d}||fStd|tj}|rt|d}t|d}||fSdS)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)txtZerr_mZnerrZnfailZfail_mZboth_mrrrparse_test_outputKs$r&FcCsdddddgS)zEReturn a valid default argv for creating testing instances of ipythonz--quickz--colors=NoColorz--no-term-titlez --no-bannerz --autocall=0rrrrr default_argvys r'cCsNt}d|j_d|j_d|j_tjddd}t|j |j _ | d|j _ |S)z6Return a config object with good defaults for testing.ZNoColor)Frztest_hist.sqliteF)suffixdeletei')rTerminalInteractiveShellcolorsZ TerminalTerminalInteractiveShellZ term_titleZautocalltempfileNamedTemporaryFilernameZHistoryManagerZ hist_filecloseZ db_cache_size)configrrrrdefault_configsr1cCstjddg}|rd|}|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-mZIPython )sys executabler) as_string ipython_cmdrrrget_ipython_cmds  r7rcCsd}|dkrg}t|}tjt}t}tj||}||d|g}tj} d| d<| dd| D]\} } t | t spt | | qpt|ttt| d} | jtd|pdd \} }t| t|} }| rtd 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--ignoreZPYTHONWARNINGSZPYCHARM_HOSTED)stdoutstderrstdinenv )inputz \x1b\[[^h]+h)r'rrdirname__file__r7renvironcopypopitems isinstancestrprintrr communicater encodedecoder sub)fnameoptionscommands__tracebackhide__ZcmdargsZtest_dirr6Z full_fnameZfull_cmdr<kvpouterrrrripexecs(       rVr?cCsd}t|||\}}|rV|rFd|d|ksVtntd||fd|d|ks~tdS)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 %r produced error: %rN)rVrstrip splitlinesAssertionError ValueError)rMZ expected_outZ expected_errrNrOrPrTrUrrripexec_validates  r[c@s2eZdZdZd ddZddZddZd d Zd S) TempFileMixinzcUtility class to create temporary Python/IPython files. Meant as a mixin class for test cases..pycCs0t||}t|dsg|_|j|||_dS)zMake a valid python temp file.tmpsN)r hasattrr^appendrM)selfsrcextrMrrrmktmps    zTempFileMixin.mktmpcCs6t|dr2|jD] }zt|WqYqXqdS)Nr^)r_r^runlink)rarMrrrtearDowns   zTempFileMixin.tearDowncCs|SNrrarrr __enter__%szTempFileMixin.__enter__cCs |dSrg)rf)raexc_type exc_value tracebackrrr__exit__(szTempFileMixin.__exit__N)r])__name__ __module__ __qualname____doc__rdrfrirmrrrrr\ s  r\z8Testing {0} In: {1!r} Expected: {2!r} Got: {3!r} cCsNd}t|dt|dd}|D],\}}||}||kstt||||qdS)aUtility function for the common case of checking a function with a sequence of input/output pairs. Parameters ---------- func : callable The function to be tested. Should accept a single argument. pairs : iterable A list of (input, expected_output) tuples. Returns ------- None. Raises an AssertionError if any output does not match the expected value. T func_namernz N)getattrrY pair_fail_msgformat)funcpairsrPr.ZinpexpectedrTrrr check_pairs3s  ryzEDid not find {0!r} in printed output (on {1}): ------- {2!s} ------- c@s*eZdZdZd ddZddZdd Zd S) AssertPrintszContext manager for testing that code prints certain text. Examples -------- >>> with AssertPrints("abc", suppress=False): ... print("abcd") ... print("def") ... abcd def r9TcCs0||_t|jttfr |jg|_||_||_dSrg)srFrG_re_typechannelsuppress)rar{r}r~rrr__init__as  zAssertPrints.__init__cCsHtt|j|_t|_t|j|jd|_tt|j|j r<|jn|jdS)N)r}) rsr3r} orig_stream MyStringIObufferr teesetattrr~rhrrrrihszAssertPrints.__enter__cCsd}z|dk rWdS|jtt|j|j|j}|j D]L}t |t rp| |st t|j|j|q@||ks@t t||j|q@WdS|jXdSNTF)rr/flushrr3r}rrgetvaluer{rFr|r!rYnotprinted_msgrupatternraetypevaluerlrPZprintedr{rrrrmns    "zAssertPrints.__exit__N)r9T)rnrorprqrrirmrrrrrzUs rzz>Found {0!r} in printed output (on {1}): ------- {2!s} ------- c@seZdZdZddZdS)AssertNotPrintszcContext manager for checking that certain output *isn't* produced. Counterpart of AssertPrintscCsd}z|dk r |jWdS|jtt|j|j|j}|j D]L}t |t rz| |rt t|j|j|qJ||ksJt t||j|qJWdS|jXdSr)rr/rrr3r}rrrr{rFr|r!rY printed_msgrurrrrrrms,     zAssertNotPrints.__exit__N)rnrorprqrmrrrrrsrccs4ddlm}|j}dd|_z dVW5||_XdS)Nr)warnc_sdSrgr)akwrrrzmute_warn..) IPython.utilsr)rZ save_warnrrr mute_warns    rc cs.t|dddz dVW5t|XdS)zGCreate an empty, named, temporary file for the duration of the context.wzutf-8)encodingN)openr/rre)r.rrr make_tempfiles rcs t|dfdd }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. r?c s<z tWStk r6}ztd|W5d}~XYnXdS)NzNo more inputs given)next StopIterationEOFError)prompteitrr mock_inputs zfake_input..mock_inputzbuiltins.input)r?)iterr)inputsrrrr fake_inputs rcCsXt|dg}t|\}}}|dks,t|d|ks8td|ksDtd|ksPt||fS)z)test that `ipython [subcommand] -h` worksz-hr TracebackOptions --help-allr7r rY subcommandcmdrTrUrcrrrhelp_output_tests   rcCsXt|dg}t|\}}}|dks,t|d|ks8td|ksDtd|ksPt||fS)z1test that `ipython [subcommand] --help-all` worksrrrrClassrrrrrhelp_all_output_tests   r)F)Nr)r?Nr)r?)r?)>rqrpathlibrr r3r,Zunittest contextlibrior subprocessrrZ unittest.mockrZtraitlets.config.loaderrZIPython.utils.processr ZIPython.utils.textr ZIPython.utils.ior r rr r?rdecrplatformZ skip_doctestZ null_decoZ doctest_decorr&Z__test__r'r1r7rVr[ZTestCaser\rtryrtypecompiler|robjectrzrrrrrrrrrrrs\            $+   : /",