o hp.@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)selfobjnamemoduleZglobs source_linestestZexample __class__Vc:\Users\shubh\Videos\agent\nova\Lib\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)linenor7rN) expandtabsZ _min_indentjoinsplit _RANDOM_TESTrlist_EXAMPLE_RE_PYfinditerr Example_EXAMPLE_RE_IPr$appendstartcount_parse_exampleZ_IS_BLANK_OR_COMMENTr(groupend)r stringr outputZcharnor8Z random_markerr,Ztermsr@mr*rrexc_msgrr3rparses@      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. r7r*r.ps1ps2r&N cs g|] }|ddqS)r&Nr)r1slr7ps1_lenrrr4s z2IPDocTestParser._parse_example..rz *$zOut\[\d+\]: \s*?\n?r6rcr/r0r)r1Zwl)r7rrr4r5msg) r(rFr;_check_prompt_blankZ _check_prefixr:r,r"matchsubZ _EXCEPTION_REZ _find_options) r rJr r8r,rrMrNr*rZ want_linesrKrrrQrrEs4          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&rOz8line %r of the docstring for %s lacks blank after %s: %rN) enumerater( ValueError) r linesr7r r8rRZ space_idxZmin_lenilinerrrrU s   z#IPDocTestParser._check_prompt_blankN)r-)F)rrrr!Z_PS1_PYZ_PS2_PYZ_PS1_IPZ_PS2_IPZ_RE_TPLr"r# MULTILINEVERBOSEr>rAr<r,rLrErUrrrrr%Rs$      D Ar%r cs"eZdZdZdfdd ZZS)IPDocTestRunnerzKTest runner that synchronizes the IPython namespace with test globals. NTcsHtdddtt|||||WdS1swYdS)N80Z24)COLUMNSLINES)rrr_run)r rZ compileflagsoutZ clear_globsrrrrc's  $zIPDocTestRunner.run)NNT)rrrr!rcrrrrrr_#sr_)r!r loggingr"Ztestpathr getLoggerrlogrrrr@r$Z DocTestParserr%Zregister_optionflagr Z DocTestRunnerobjectr_rrrrs    O