[Fie >UdZddlZddlmZddlZddlZddlZddlmZddl Z ddl Z ddl Z ddl Z ddl mZddl mZddl mZddl mZdd l mZdd l mZdd l mZdd l mZdd l mZddl mZddl mZddl mZddl mZddlZddlmZddlmZddlm Z ddlm!Z!ddl"m#Z#ddl$m%Z%ddl&m'Z'ddl(m)Z)ddl*m+Z+ddl*m,Z,ddl-m.Z.ddl-m/Z/ddl0m1Z1ddl0m2Z2dd l3m4Z4dd!l5m6Z6dd"l7m8Z8dd#l9m:Z:er ddl;Z;dd$l md'Z?d(Z@d)ZAe=e>e?e@eAfZBdaCdaDeed*eEd+<d,e)d-dfd.ZFd\d/ZGd0ed1e.d-eed2fd3ZHd4ed-eIfd5ZJd6e'd4ed1e.d-eIfd7ZKd4ed-eIfd8ZLGd9d:e!ZMGd;dZP d]d?ed*d@eeIdAeQdBeId-d=f dCZRGdDdEe/ZSd-eeTeQffdFZUd6e'd-eQfdGZVd6e'd-eIfdHZWGdIdJe6ZXd^dKZYdLeZd-eIfdMZ[ed-edNfdOZ\GdPdQe6Z]d-ed*fdRZ^d_dSZ_d-eQfdTZ`d-eQfdUZad-eQfdVZbdWeTd-eQfdXZce+dYZd-eeTeffd[Zdy)`z4Discover and run doctests in modules and test files.N)contextmanager)Path)Any)Callable)Dict) Generator)Iterable)List)Optional)Pattern)Sequence)Tuple)Type) TYPE_CHECKING)Union)outcomes) ExceptionInfo)ReprFileLocation) TerminalRepr)TerminalWriter safe_getattr)Config)Parser)fixture) TopRequest) Collector)Item)OutcomeException)skip fnmatch_ex)Module)approx) PytestWarning)Selfnonecdiffndiffudiffonly_first_failuredoctest.OutputChecker CHECKER_CLASSparserreturnc~|jddddg|jddd |jd }|jd d d dd|jdtjddt d|jddgddd|jdd d dd|jdd d d d!y)"Ndoctest_optionflagszOption flags for doctestsargsELLIPSIS)typedefaultdoctest_encodingzEncoding used for doctest fileszutf-8)r5collectz--doctest-modules store_trueFzRun doctests in all .py modulesdoctestmodules)actionr5helpdestz--doctest-reportr*z9Choose another output format for diffs on doctest failure doctestreport)r4r5r;choicesr<z--doctest-globappendpatz2Doctests file matching pattern, default: test*.txt doctestglob)r:r5metavarr;r<z--doctest-ignore-import-errorsz Ignore doctest collection errorsdoctest_ignore_import_errorsz--doctest-continue-on-failurezz_is_doctest..s8$z$%8s)r\session isinitpath getoptionr^)r[rZrUglobss ` rLrbrbsL {{&&6>>+D+DT+J   ] + < }E 8%8 88rNc |jdk(S)Nz __main__.py)rerYs rLr`r`s 99 %%rNcBeZdZdeeeeefddfdZdeddfdZ y)ReprFailDoctestreprlocation_linesr/Nc||_yrP)rt)selfrts rL__init__zReprFailDoctest.__init__s #5rNtwc~|jD].\}}|D]}|j||j|0yrP)rtline toterminal)rvrx reprlocationlinesrzs rLr{zReprFailDoctest.toterminalsD#'#:#: ( L%     # #B ' (rN) __name__ __module__ __qualname__r rrrHrwrr{rRrNrLrsrss>5"*51A8C=1P+Q"R5 5 (^((rNrsc.eZdZdedddffd ZxZS)MultipleDoctestFailuresfailureszdoctest.DocTestFailurer/Nc0t|||_yrP)superrwr)rvr __class__s rLrwz MultipleDoctestFailures.__init__s   rN)r~rrr rw __classcell__rs@rLrrs!!*B!C!!!rNrdoctest.DocTestRunnerc>ddlGfddj}|S)Nrc eZdZdZ ddeddeedededdf fd Zd d d d deddffd Z d d d d de e e e e jfddffd ZxZS)/_init_runner_class..PytestDoctestRunnerzRunner to collect failures. Note that the out variable in this case is a list instead of a stdout-like object. Ncheckerr,verbose optionflagscontinue_on_failurer/c8t||||||_y)N)rrr)rrwr)rvrrrrrs rLrwz8_init_runner_class..PytestDoctestRunner.__init__s! G Wg;  W':D $rNtestdoctest.DocTestexamplezdoctest.Examplegotclj|||}|jr|j|y|rP)DocTestFailurerr?)rvoutrrrfailuredoctests rLreport_failurez>_init_runner_class..PytestDoctestRunner.report_failures5-g,,T7C@G'' 7# rNexc_infoct|dtr|dt|dtjrt j dj |||}|jr|j|y|)NzQuitting debugger) isinstancerbdbBdbQuitrexitUnexpectedExceptionrr?)rvrrrrrrs rLreport_unexpected_exceptionzK_init_runner_class..PytestDoctestRunner.report_unexpected_exceptionsm(1+'78qk!(1+s{{3 121g11$JG'' 7# rNNNrT)r~rr__doc__r boolintrwrHrrr BaseExceptiontypes TracebackTyperr)rrs@rLPytestDoctestRunnerrs :>&* (,  ;56 ;d^ ;  ; "&  ;   ; $ '      $ '   D/@S@SST      rNr)r DebugRunner)rrs @rL_init_runner_classrs!-g11-^ rNrrrrc@t tat||||S)Nrrrr)rQrrs rL _get_runnerrs-)+  /  rNc eZdZdedddddddd f fd Zedddedddddd f fd Zdd ZddZddZ ddZ de e de eefffd Zdee defeeeffdZxZS) DoctestItemrerUz%Union[DoctestTextfile, DoctestModule]runnerrdtestrr/Nct|||||_||_d|_|j j }|j|dd}||_|j|_ |jy)N)nodefunccls) rrwrrobjrm_fixturemanagergetfixtureinfo _fixtureinfo names_closure fixturenames _initrequest)rvrerUrrfm fixtureinfors rLrwzDoctestItem.__init__ss v&   \\ ) )''T$'G ''55 rNr&c*t|||||S)zThe public named constructor.)rerUrr)rra)rrUrerrrs rLrazDoctestItem.from_parentsw"VFRW"XXrNc6i|_t|d|_y)NT) _ispytest)funcargsr_requestrvs rLrzDoctestItem._initrequests+- "448 rNc.|jjt|jj}|jjdj D] \}}|||< |j j j|y)N) getfixturedoctest_namespace)r _fillfixturesdictgetfixturevalueitemsrrpupdate)rvrprevalues rLsetupzDoctestItem.setup#ss ##% = =>==889LMSSU KD%E$K  &rNct|j|jg}|jj |j||r t |y)N)r)_check_all_skippedr$_disable_output_capturing_for_darwinrrunr)rvrs rLruntestzDoctestItem.runtest*sL4::& 11335  1 )(3 3 rNcJtjdk7ry|jjj d}|rd|j d|j \}}tjj|tjj|yy)zFDisable output capturing. Otherwise, stdout is lost to doctest (#985).DarwinNcapturemanagerT)in_) platformsystemr[ pluginmanager getpluginsuspend_global_captureread_global_capturesysstdoutwritestderr)rvcapmanrerrs rLrz0DoctestItem._disable_output_capturing_for_darwin4s~ ??  ( **445EF   ) )d ) 3113HC JJ  S ! JJ  S ! rNexcinfocddl}d}t|j|j|jfr|jg}n0t|jt r|jj }|t|!|Sg}|D]X}|j}|j}|j}|jd} n|j|jzdz} t|j} t|| | } t!} t#|j$j'd} | |jj(J|jj(j+d}|jJt-|Dcgc]\}}d||jzdz|fz}}}|t/|jdz d|jdz}n=dg}d}|j0j+D]}|j3d |d |d }t||jr0|| j5||j6| j9d z }nwt;j<|j>}|d tA|jzgz }|tCjD|j>Dcgc]}|jGd c}z }|j3| |f[tI|Scc}}wcc}w)Nrrr=Fz%03d %s z?EXAMPLE LOCATION UNKNOWN, not showing all tests of that examplez>>>z???  z... zUNEXPECTED EXCEPTION: %s)%rrrrrrrr repr_failurerrfilenamelinenor4r~r _get_checker_get_report_choicer[ro docstring splitlines enumeratemaxsourcer?output_differencersplitr from_exc_inforrepr tracebackformat_exceptionstriprs)rvrrrrtrrrrrmessager|r report_choicer}ixindentrz inner_excinfors rLrzDoctestItem.repr_failure@s     MMG22G4O4OP  H  '> ?}}--H  7'0 0) =GooG<>A;MNV#NN557#DLL4xq!78"F#'7#9#9:22W[[-%+!. ; ;G3}->3 sL !>3@HE%(:C(?"@(3-QT"TUHrNrc ddl}t|j|j|j|j |j |jttt S)Nr) DONT_ACCEPT_TRUE_FOR_1DONT_ACCEPT_BLANKLINENORMALIZE_WHITESPACEr3IGNORE_EXCEPTION_DETAILCOMPARISON_FLAGS ALLOW_UNICODE ALLOW_BYTESNUMBER) rrr r r r3r r _get_allow_unicode_flag_get_allow_bytes_flag_get_number_flagrs rL_get_flag_lookuprsZ &==%;;$99!! ' ? ? 11-/)+!  rNc^|jd}t}d}|D] }|||z} |S)Nr1r)getinir)r[optionflags_strflag_lookup_tableflag_accflags rLget_optionflagsrsCmm$9:O(*H,%d++, OrNcR|jd}|r|jdrd}|S)NrDusepdbF)getvalue)r[rs rL_get_continue_on_failurer s/ &0M N ??8 $"'  rNc"eZdZdZdeefdZy)rWNr/c# Kddl}|jjd}|jj |}t |j}|jj }ddi}t|j}td|tt|j}|j} | j||||d} | jr&tj|| j || yyw)Nrr6r~__main__Frrrrrerr)rr[rrZ read_textrHrerrrr  DocTestParser get_doctestexamplesrra) rvrencodingtextrrerprrr.rs rLr7zDoctestTextfile.collects;;%%&89yy""8,tyy>yy~~Z(%dkk2 # N 8 E  '&&(!!$tXqA ==))499V4*  sDD)r~rrrr rr7rRrNrLrWrWs C+.rNrWcfddltfd|jD}|r tdyy)zVRaise pytest.skip() if all examples in the given DocTest have the SKIP option set.rNc3jK|]*}|jjjd,yw)FN)optionsgetSKIP)rjrrs rLrlz%_check_all_skipped..s$PQaiimmGLL%8Ps03z!all tests skipped by +SKIP option)rallr)r )r all_skippedrs @rLrrs-P$--PPK 01rNrc t|ddduS)zmReturn if an object is possibly a mock object by checking the existence of a highly improbable attribute.1pytest_mock_example_attribute_that_shouldnt_existNr)rs rL _is_mockedr5s SMtT rN)NNNc#Ktjdddtdtfdtttgtfdtffd}|t_ dt_y#t_wxYww)zContext manager which replaces ``inspect.unwrap`` with a version that's aware of mock objects and doesn't recurse into them.Nstopr.r8r/c ||turtS|fdS#t$r'}tjd|ddtd}~wwxYw)Nr7c.t|xsSrP)r5)r_stoprs rLzF_patch_unwrap_mock_aware.._mock_aware_unwrap..sjo6TtrNzGot z when unwrapping z. This is usually caused by a violation of Python's object protocol; see e.g. https://github.com/pytest-dev/pytest/issues/5080)r5 Exceptionwarningswarnr%)rr8er; real_unwraps` @rL_mock_aware_unwrapz4_patch_unwrap_mock_aware.._mock_aware_unwrapst |tz1"4j99Et*TU U  MMqe,TH5CC    s-- A"AA)inspectunwraprrr )rBrAs @rL_patch_unwrap_mock_awarerEsn..KMQsCx +3HcUCZ4H+I "(GN% $sAA9A) A9) A66A9ceZdZdeefdZy)rVr/c#rKddl}Gdd|j} |j}|jjj||}t|j }td|tt|j }|j!|j"D]4}|j$st&j)||j*||6y#tj$r9|j j drtd|jznY wxYww) NrcbeZdZdZfdZ dfd Zejdkr fdZxZ SxZ S)5DoctestModule.collect..MockAwareDocTestFinderzA hackish doctest finder that overrides stdlib internals to fix a stdlib bug. https://github.com/pytest-dev/pytest/issues/3456 https://bugs.python.org/issue25532 ct|tr t|d|}t|drt j |}t |||S)a3Doctest code does not take into account `@property`, this is a hackish way to fix it. https://bugs.python.org/issue17446 Wrapped Doctests will need to be unwrapped so the correct line number is returned. This will be reported upstream. #8796 fget __wrapped__)rpropertygetattrhasattrrCrDr _find_lineno)rvr source_linesrs rLrPzBDoctestModule.collect..MockAwareDocTestFinder._find_linenosOc8,!#vs3C3 .!..-Cw+ rNc t|ryt5t| |||||||dddy#1swYyxYwrP)r5rEr_find) rvtestsrremodulerQrpseenrs rLrSz;DoctestModule.collect..MockAwareDocTestFinder._findsEc?-/GMsD&,ts 7A) cpt|tjr |j}t|||S)a>`cached_property` objects are never considered a part of the 'current module'. As such they are skipped by doctest. Here we override `_from_module` to check the underlying function instead. https://github.com/python/cpython/issues/107995 )r functoolscached_propertyrr _from_module)rvrUobjectrs rLr\zBDoctestModule.collect..MockAwareDocTestFinder._from_module#s2 "&)*C*CD!'!7/??rNr) r~rrrrPrSr version_infor\rrs@rLMockAwareDocTestFinderrIs;   (  ') @ @rNr_rCzunable to import module %rFr$r%)r DocTestFinderrr CollectErrorr[rr rZrmrparsefactoriesrrrr findr~r)rrare)rvrr_rUfinderrrrs rLr7zDoctestModule.collects 5 W%:%:5 n XXF $$33D9()%dkk2 # N 8 E  KK8 D}}!--tyyt. )%% {{##$BC1DII=>? s/D7 C(BD7(D7(AD40D73D44D7N)r~rrr rr7rRrNrLrVrVsT+.TrNrVcFddl}ddlGfdd|j}|S)NrceZdZWjdWjZWjdWjZWjdWjZde de de de ffd Z de de de fd Z xZS) 2_init_checker_class..LiteralsOutputCheckerz(\W|^)[uU]([rR]?[\'\"])z(\W|^)[bB]([rR]?[\'\"])a (?P (?P (?P [+-]?\d*)\.(?P\d+) | (?P [+-]?\d+)\. ) (?: [Ee] (?P [+-]?\d+) )? | (?P [+-]?\d+) (?: [Ee] (?P [+-]?\d+) ) ) wantrrr/ct||||ry|tz}|tz}|t z}|s|s|sydt t dt dt f fd }|r&||j|}||j|}|r&||j|}||j|}|r|j||}t||||S)NTFregextxtr/c*j|d|S)Nz\1\2)sub)rjrkres rLremove_prefixeszX_init_checker_class..LiteralsOutputChecker.check_output..remove_prefixes|svveWc22rN) r check_outputrrrr rH_unicode_literal_re_bytes_literal_re_remove_unwanted_precision) rvrhrr allow_unicode allow_bytes allow_numberrorrns rLrpz?_init_checker_class..LiteralsOutputChecker.check_outputqsw#D#{;'*A*CCM%(=(??K&)9);;L \ 3ws| 3# 3# 3&t'?'?F%d&>&>D&t'='=tD%d&<&.LiteralsOutputChecker._remove_unwanted_precisionsc11$78E0056D5zSY& FE4( J1*+''**=*+''+*># ww{3H!)!1As8} 'X.I#veAGGI.>B N'SS 0aggi&01AGGI=AEEGfDTDV@WWaeeg 1QUUWqwwy5HIIF J JrN)r~rrcompileUNICODErqrrVERBOSEr~rHrrrprsr)rrns@rLLiteralsOutputCheckerrgSs !jj)CRZZPJJ'A2::NZZ & JJ) . @S @s @ @ @6 3 S S rNr)rrn OutputChecker)rrrns @rL_init_checker_classrOs& O 5 5Ob ! rNc6t tatS)aReturn a doctest.OutputChecker subclass that supports some additional options: * ALLOW_UNICODE and ALLOW_BYTES options to ignore u'' and b'' prefixes (respectively) in string literals. Useful when the same doctest should run in Python 2 and Python 3. * NUMBER to ignore floating-point differences smaller than the precision of the literal number in the doctest. An inner class is used to avoid importing "doctest" at the module level. )r-rrRrNrLrrs+- ?rNc.ddl}|jdS)z+Register and return the ALLOW_UNICODE flag.rNrrregister_optionflagrs rLrrs &7 & & 77rNc.ddl}|jdS)z)Register and return the ALLOW_BYTES flag.rNrrrs rLrrs &7 & &} 55rNc.ddl}|jdS)z$Register and return the NUMBER flag.rNrrrs rLrrs &7 & &x 00rNkeyc ddl}t|jt|jt |j t|jtdi|S)zReturn the actual `doctest` module flag value. We want to do it as late as possible to avoid importing `doctest` and all its dependencies when parsing options, as it adds overhead and breaks tests. rN) rDOCTEST_REPORT_CHOICE_UDIFF REPORT_UDIFFDOCTEST_REPORT_CHOICE_CDIFF REPORT_CDIFFDOCTEST_REPORT_CHOICE_NDIFF REPORT_NDIFF(DOCTEST_REPORT_CHOICE_ONLY_FIRST_FAILUREREPORT_ONLY_FIRST_FAILUREDOCTEST_REPORT_CHOICE_NONE)rrs rLrrsN  $W%9%9#W%9%9#W%9%90'2S2S"A      rNrm)scopectS)aFixture that returns a :py:class:`dict` that will be injected into the namespace of doctests. Usually this fixture is used in conjunction with another ``autouse`` fixture: .. code-block:: python @pytest.fixture(autouse=True) def add_np(doctest_namespace): doctest_namespace["np"] = numpy For more details: :ref:`doctest_namespace`. )rrRrNrLrrs  6MrNrr)rrr/N)r/r,)err contextlibrrZrCospathlibrrrrrtypingrrrrr r r r r rrrrr>_pytestr_pytest._code.coderrr _pytest._ior_pytest.compatr_pytest.configr_pytest.config.argparsingr_pytest.fixturesrr _pytest.nodesrr_pytest.outcomesrr _pytest.pathlibr"_pytest.pythonr#_pytest.python_apir$_pytest.warning_typesr%rr&rrrrrrJrQr-__annotations__rMrSrcrr_rbr`rsr=rrrrrrHrrr rWrr]r5rErVrrrrrrrrRrNrLrs: %    ,/+&'!,$'#-!&!%/#%%%+?(,  9= x456=/V//d    e678 AtAA99d9I9$9&d&t& (l (!i! 2D!892l26" $  - . d^   (CH$CHL $sCx.  FsVf>2Ft%),<"=%%:UFUpU!T"9:U!p(886s61#1 C C " y4S>rN