B \7@sdZddlZddlZddlZddlZddlmZddlmZddl m Z m Z ddl m Z yddlmZdZWnek rdZYnXdd lmZdd lmZdd lmZdd lmZmZdd lmZddlmZ ddlm!Z!ej"dkre!j#ne j$Z%e%ddZ&ddZ'de'_(ddZ)ddZ*d7ddZ+d8ddZ,d9d d!Z-Gd"d#d#e.Z/d$Z0d%d&Z1eZ2e3e4dZ5d'Z6Gd(d)d)e.Z7d*Z8Gd+d,d,e7Z9ed-d.Z:ed/d0Z;d1d2ZdS) N)contextmanager)StringIO)PopenPIPE)patchTF)Config)get_output_error_code) list_strings) temp_pyfileTee) py3compat) decorators) skipdoctestZwin32cs*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).0f)baser4lib/python3.7/site-packages/IPython/testing/tools.py Oszfull_path..)r rrsplit)Z startPathfilesr)rr 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+)\)r rz^FAILED \(failures=(\d+)\)z(^FAILED \(errors=(\d+), failures=(\d+)\))rr)research MULTILINEintgroup)ZtxtZerr_mZnerrZnfailZfail_mZboth_mrrrparse_test_outputRs"r"cCsdddddgS)zEReturn a valid default argv for creating testing instances of ipythonz--quickz--colors=NoColorz--no-term-titlez --no-bannerz --autocall=0rrrrr default_argvsr#cCsJt}d|j_d|j_d|j_tjddd}|j|j _ | d|j _ |S)z6Return a config object with good defaults for testing.ZNoColor)Frztest_hist.sqliteF)suffixdeletei') rZTerminalInteractiveShellZcolorsZ TerminalTerminalInteractiveShellZ term_titleZautocalltempfileZNamedTemporaryFilenameZHistoryManagerZ hist_filecloseZ db_cache_size)Zconfigrrrrdefault_configs r)cCstjddg}|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)Z as_string ipython_cmdrrrget_ipython_cmds  r.rcCs|dkr g}t|}tjt}t}tj||}|||g}tj}d|d<x(| D]\} } t | t s`t | | q`Wt |ttt|d} | jtd|pdd\} } t| t| } } | rtdd| } | | 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 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. NignoreZPYTHONWARNINGS)stdoutstderrstdinenv )inputz \x1b\[[^h]+h)r#rrdirname__file__r.renvironcopyitems isinstancestrprintrrZ communicater encodedecodersub)fnameoptionscommandsZcmdargsZtest_dirr-Z full_fnameZfull_cmdr3kvpouterrrrripexecs$     rJr6cCsddlm}t|||\}}|r^|rN|d|d|ntd||f|d|d|dS)a3Utility 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 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 rNr4z"Running file %r produced error: %r) nose.toolstoolsrJ assert_equalrstrip splitlines ValueError)rBZ expected_outZ expected_errrCrDntrHrIrrripexec_validates * rRc@s2eZdZdZd ddZddZddZd d Zd S) TempFileMixinzcUtility class to create temporary Python/IPython files. Meant as a mixin class for test cases..pycCs8t||\}}t|dsg|_|j||f||_dS)zMake a valid python temp file.tmpsN)r hasattrrUappendrB)selfsrcZextrBrrrrmktmps  zTempFileMixin.mktmpcCsFt|drBx6|jD],\}}|yt|WqYqXqWdS)NrU)rVrUr(runlink)rXrrBrrrtearDowns zTempFileMixin.tearDowncCs|S)Nr)rXrrr __enter__%szTempFileMixin.__enter__cCs |dS)N)r\)rXexc_type exc_value tracebackrrr__exit__(szTempFileMixin.__exit__N)rT)__name__ __module__ __qualname____doc__rZr\r]rarrrrrS s  rSz8Testing {0} In: {1!r} Expected: {2!r} Got: {3!r} cCsNt|dt|dd}x4|D],\}}||}||kstt||||qWdS)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. Z func_namerbz N)getattrAssertionError pair_fail_msgformat)funcZpairsr'ZinpZexpectedrHrrr check_pairs3srkzEDid 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 r0TcCs0||_t|jttfr |jg|_||_||_dS)N)sr<r=_re_typechannelsuppress)rXrmrorprrr__init___s  zAssertPrints.__init__cCsHtt|j|_t|_t|j|jd|_tt|j|j r<|jn|jdS)N)ro) rfr+ro orig_stream MyStringIObufferr teesetattrrp)rXrrrr]fszAssertPrints.__enter__cCsz|dk rdS|jtt|j|j|j}xV|jD]L}t |t rj| |st t |j|j|q:||ks:t t ||j|q:WdS|jXdS)NF)ruflushrvr+rorrrtgetvaluermr<rnrrgnotprinted_msgripatternr()rXetypevaluer`printedrmrrrrals    " zAssertPrints.__exit__N)r0T)rbrcrdrerqr]rarrrrrlSs  rlz>Found {0!r} in printed output (on {1}): ------- {2!s} ------- c@seZdZdZddZdS)AssertNotPrintszcContext manager for checking that certain output *isn't* produced. Counterpart of AssertPrintscCsz|dk r|jdS|jtt|j|j|j}xV|j D]L}t |t rt| |rt t|j|j|qD||ksDt t||j|qDWdS|jXdS)NF)rur(rwrvr+rorrrtrxrmr<rnrrg printed_msgrirz)rXr{r|r`r}rmrrrras     zAssertNotPrints.__exit__N)rbrcrdrerarrrrr~sr~ccs4ddlm}|j}dd|_z dVWd||_XdS)Nr)warnc_sdS)Nr)akwrrrszmute_warn..) IPython.utilsr)rZ save_warnrrr mute_warns    rc cs*t|dz dVWdt|XdS)zM Create an empty, named, temporary file for the duration of the context. wN)openr(rr[)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. r6cs*ytStk r$tdYnXdS)NzNo more inputs given)next StopIterationEOFError)prompt)itrr mock_inputszfake_input..mock_inputzbuiltins.input)r6)iterr)Zinputsrr)rr fake_inputs rcCsVt|dg}t|\}}}t|d|td|td|td|||fS)z)test that `ipython [subcommand] -h` worksz-hr TracebackOptionsz --help-all)r.rrQrM assert_not_in assert_in) subcommandcmdrHrIrcrrrhelp_output_tests   rcCsVt|dg}t|\}}}t|d|td|td|td|||fS)z1test that `ipython [subcommand] --help-all` worksz --help-allrrrZClass)r.rrQrMrr)rrrHrIrrrrhelp_all_output_tests   r)F)Nr)r6Nr)r6)r6)?rerrr+r& contextlibrior subprocessrrZ unittest.mockrrKrLrQZhas_nose ImportErrorZtraitlets.config.loaderrZIPython.utils.processrZIPython.utils.textr ZIPython.utils.ior r rr r6rZdecrplatformZ skip_doctestZ null_decoZ doctest_decorr"Z__test__r#r)r.rJrRobjectrSrhrkrstypecompilernryrlrr~rrrrrrrrrsX            %+   5 +#.