o Zhp.@sdZddlZddlZddlZddlmZeeZGdddej Z Gdddej Z Gdd d ej Z Gd d d ejZed ZGd ddejeZdS)aNose Plugin that supports IPython doctests. Limitations: - When generating examples for use as doctests, make sure that you have pretty-printing OFF. This can be done either by setting the ``PlainTextFormatter.pprint`` option in your configuration file to False, or by interactively disabling it with %Pprint. This is required so that IPython output matches that of normal Python, which is used by doctest for internal execution. - Do not rely on specific prompt numbers for results (such as using '_34==True', for example). For IPython tests run via an external process the prompt numbers may be different, and IPython tests run as normal python code won't even have these special _NN variables set at all. N) modified_envcseZdZfddZZS) DocTestFindercsHt|||||}tt|ddr"|dur"|jD]}d|jtj<q|S)NZ__skip_doctest__FT)super _get_testboolgetattrZexamplesoptionsdoctestSKIP)selfobjnamemoduleglobs source_linestestZexample __class__W/var/www/html/lang_env/lib/python3.10/site-packages/IPython/testing/plugin/ipdoctest.pyr)s  zDocTestFinder._get_test)__name__ __module__ __qualname__r __classcell__rrrrr(src@s"eZdZdZedZddZdS)IPDoctestOutputCheckerzSecond-chance checker with support for random tests. If the default comparison doesn't pass, this checker looks in the expected output string for flags that tell us to ignore the output. z #\s*random\s+cCs*tj||||}|s|j|rdS|S)zCheck output, accepting special markers embedded in the output. If the output didn't pass the default validation but the special string '#random' is included, we accept it.T)r OutputChecker check_output random_research)r wantgotZ optionflagsretrrrr<s z#IPDoctestOutputChecker.check_outputN)rrr__doc__recompilerrrrrrr3s  rc@s eZdZdS) IPExampleN)rrrrrrrr%Os r%c@seZdZdZdZdZdZdZdZe eeeeefe j e j BZ e eeeeefe j e j BZe dZdd Zdd d ZdddZddZdS)IPDocTestParserz A class used to parse strings containing doctest examples. Note: This is a version modified to properly recognize IPython input and convert any IPython examples into valid Python ones. z>>>z\.\.\.z In\ \[\d+\]:z\ \ \ \.\.\.+:a] # Source consists of a PS1 line followed by zero or more PS2 lines. (?P (?:^(?P [ ]*) (?P %s) .*) # PS1 line (?:\n [ ]* (?P %s) .*)*) # PS2 lines \n? # a newline # Want consists of any non-blank lines that do not start with PS1. (?P (?:(?![ ]*$) # Not a blank line (?![ ]*%s) # Not a line starting with PS1 (?![ ]*%s) # Not a line starting with PS2 .*$\n? # But any other line )*) z#\s*all-random\s+cCs*tj|}t|dkrt|S|S)z/Convert input IPython source into valid Python.)_ipZinput_transformer_managerZtransform_celllen splitlinesZ prefilter)r sourceblockrrrip2pys  zIPDocTestParser.ip2pyc sR|}||dkrdfdd|dD}g}d\}}|j|r+d}nd}d}t|j|}|r=t j } n t|j |}t } d }|D]R} | ||| ||d|| 7}|| |||\} } } }| |7} || s| | | | ||t| d | d ||d| | 7}| }qK| ||d |S) a= Divide the given string into examples and intervening text, and return them as a list of alternating Examples and strings. Line numbers for the Examples are 0-based. The optional argument `name` is a name identifying this string, and is only used for error messages. r cg|]}|dqSNr).0lZ min_indentrr z)IPDocTestParser.parse..)rrz # randomFTindent)linenor8rN) expandtabsZ _min_indentjoinsplit _RANDOM_TESTrlist_EXAMPLE_RE_PYfinditerr Example_EXAMPLE_RE_IPr%appendstartcount_parse_exampleZ_IS_BLANK_OR_COMMENTr)groupend)r stringr outputZcharnor9Z random_markerr-ZtermsrAmr+rrexc_msgrr4rparses@      zIPDocTestParser.parseFc sRt|d|dd}|d}|d}t||||||r8||ddd|||dfd d |D}|rL||}|d } | d} t| dkrgtd | d rg| d =|| d||t|t dd| d| d<dfdd | D} |j | }|r|d} nd} | |||} || | | fS)a7 Given a regular expression match from `_EXAMPLE_RE` (`m`), return a pair `(source, want)`, where `source` is the matched example's source code (with prompts and indentation stripped); and `want` is the example's expected output (with indentation stripped). `name` is the string's name, and `lineno` is the line number where the example starts; both are used for error messages. Optional: `ip2py`: if true, filter the input via IPython to convert the syntax into valid python. r8r+r/ps1ps2r'N cs g|] }|ddqS)r'Nr)r2slr8ps1_lenrrr5s z2IPDocTestParser._parse_example..rz *$zOut\[\d+\]: \s*?\n?r7rcr0r1r)r2Zwl)r8rrr5r6msg) r)rGr<_check_prompt_blankZ _check_prefixr;r-r#matchsubZ _EXCEPTION_REZ _find_options) r rKr r9r-rrNrOr+rZ want_linesrLrrrRrrFs4          zIPDocTestParser._parse_examplec Csb||}|d}t|D]"\}} t| |kr.| |dkr.td||d|| ||| fq dS)a Given the lines of a source string (including prompts and leading indentation), check to make sure that every prompt is followed by a space character. If any line is not followed by a space character, then raise ValueError. Note: IPython-modified version which takes the input prompt length as a parameter, so that prompts of variable length can be dealt with. r'rPz8line %r of the docstring for %s lacks blank after %s: %rN) enumerater) ValueError) r linesr8r r9rSZ space_idxmin_lenilinerrrrV s   z#IPDocTestParser._check_prompt_blankN)r.)F)rrrr"Z_PS1_PYZ_PS2_PYZ_PS1_IPZ_PS2_IPZ_RE_TPLr#r$ MULTILINEVERBOSEr?rBr=r-rMrFrVrrrrr&Rs$      D Ar&r cs"eZdZdZdfdd ZZS)IPDocTestRunnerzKTest runner that synchronizes the IPython namespace with test globals. NTcsHtdddtt|||||WdS1swYdS)N80Z24)COLUMNSLINES)rrrarun)r rZ compileflagsoutZ clear_globsrrrre's  $zIPDocTestRunner.run)NNT)rrrr"rerrrrrra#sra)r"r loggingr#Ztestpathr getLoggerrlogrrrrAr%Z DocTestParserr&Zregister_optionflagr Z DocTestRunnerobjectrarrrrs    O