F-Phf^UdZddlmZddlZddlmZddlmZddlmZddlmZddl m Z ddl Z ddl Z ddl Z dd lmZddlZddlZddlZddlZddlZdd lmZdd lmZddlZdd lmZdd lmZddlmZddlmZddlm 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.ddl-m/Z/ddl0m1Z1ddl2m3Z3ddl4m5Z5ddl6m7Z7er ddl8Z8ddl9m:Z:dZ;d Zd#Z?e;ee?fZ@daAdaBd$eCd%<did*ZDdjd+ZEdkd1ZFdld4ZGdmd7ZHdld8ZIGd9d:eZJGd;dZM dodpdHZNGdIdJe,ZOdqdLZPdrdMZQdsdNZRGdOdPe3ZSdtdSZTdudVZUe dvdXZVGdYdZe3ZWdwd\ZXdxd^ZYdyd_ZZdyd`Z[dydaZ\dzddZ]e(defd{dhZ^dS)|z4Discover and run doctests in modules and test files.) annotationsN)Callable) Generator)Iterable)Sequence)contextmanager)Path)Any) TYPE_CHECKING)outcomes) ExceptionInfo)ReprFileLocation) TerminalRepr)TerminalWriter safe_getattr)Config)Parser)fixture) TopRequest) Collector)Item)OutcomeException)skip fnmatch_ex)Module)approx) PytestWarning)Selfnonecdiffndiffudiffonly_first_failurez"type[doctest.OutputChecker] | None CHECKER_CLASSparserrreturnNonec|ddddg|ddd |d }|d d d dd|dtjddt d|ddgddd|dd d dd|dd d d d!dS)"Ndoctest_optionflagszOption flags for doctestsargsELLIPSIS)typedefaultdoctest_encodingzEncoding used for doctest fileszutf-8)r/collectz--doctest-modules store_trueFzRun doctests in all .py modulesdoctestmodules)actionr/helpdestz--doctest-reportr$z9Choose another output format for diffs on doctest failure doctestreport)r.r/r5choicesr6z--doctest-globappendpatz2Doctests file matching pattern, default: test*.txt doctestglob)r4r/metavarr5r6z--doctest-ignore-import-errorsz Ignore doctest collection errorsdoctest_ignore_import_errorsz--doctest-continue-on-failurezz_is_doctest..s-88$z$%%888888rH)rUsession isinitpath getoptionrW)rTrSrOglobss ` rFr\r\sh {&&&6>+D+DT+J+J&t   ] + + < }E 8888%888 8 88rHc|jdkS)Nz __main__.py)rarRs rFrYrYs 9 %%rHceZdZd dZd dZd S) ReprFailDoctestreprlocation_lines0Sequence[tuple[ReprFileLocation, Sequence[str]]]r(r)c||_dSrJ)rp)selfrps rF__init__zReprFailDoctest.__init__s#5rHtwrc~|jD]4\}}|D]}||||5dSrJ)rpline toterminal)rsru reprlocationlinesrws rFrxzReprFailDoctest.toterminalsZ#'#: ( ( L%    # #B ' ' ' ' ( (rHN)rprqr(r))rurr(r))__name__ __module__ __qualname__rtrxrLrHrFroros<5555 ((((((rHroc eZdZdfd ZxZS)MultipleDoctestFailuresfailures Sequence[doctest.DocTestFailure]r(r)cVt||_dSrJ)superrtr)rsr __class__s rFrtz MultipleDoctestFailures.__init__s$   rH)rrr(r))r{r|r}rt __classcell__rs@rFrrs=!!!!!!!!!!rHrtype[doctest.DocTestRunner]c:ddlGfddj}|S)NrcHeZdZdZ ddfd Zdfd Zdfd 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. NrTcheckerdoctest.OutputChecker | Noneverbose bool | None optionflagsintcontinue_on_failurer_r(r)c^t|||||_dS)N)rrr)rrtr)rsrrrrrs rFrtz8_init_runner_class..PytestDoctestRunner.__init__s2 GG  Wg;  W W W':D $ $ $rHtestdoctest.DocTestexampledoctest.ExamplegotrBchj|||}|jr||dS|rJ)DocTestFailurerr9)rsoutrrrfailuredoctests rFreport_failurez>_init_runner_class..PytestDoctestRunner.report_failuresC-g,T7C@@G'  7##### rHexc_info>tuple[type[BaseException], BaseException, types.TracebackType]ct|dtr|dt|dtjrt jdj|||}|jr||dS|)NzQuitting debugger) isinstancerbdbBdbQuitr exitUnexpectedExceptionrr9)rsrrrrrrs rFreport_unexpected_exceptionzK_init_runner_class..PytestDoctestRunner.report_unexpected_exceptions(1+'788 "qk!(1+s{33 3 12221g1$JJG'  7##### rHNNrT) rrrrrrrr_r(r))rrrrrrBr(r))rrrrrrr(r))r{r|r}__doc__rtrrr)rrs@rFPytestDoctestRunnerrs  59#' (,  ; ; ; ; ; ; ;                rHr)r DebugRunner)rrs @rF_init_runner_classrsLNNN-------g1---^ rHTrrrrrrrdoctest.DocTestRunnercRttat||||S)Nrrrr)rKrrs rF _get_runnerrs:)++  /    rHcpeZdZdfd Zedfd ZddZddZddZddZ dfd Z ddZ xZ S) DoctestItemrarBrODoctestTextfile | DoctestModulerunnerrdtestrr(r)c t||||_||_d|_|jj}||dd}||_|j |_ | dS)N)nodefunccls) rrtrrobjri_fixturemanagergetfixtureinfo _fixtureinfo names_closure fixturenames _initrequest)rsrarOrrfm fixtureinfors rFrtzDoctestItem.__init__s v&&&   \ )''T$'GG ''5 rHr cNt||||S)zThe public named constructor.)rarOrr)rr[)rrOrarrrs rFr[zDoctestItem.from_parents'ww""VFRW"XXXrHc@i|_t|d|_dS)NT) _ispytest)funcargsr_requestrss rFrzDoctestItem._initrequests!+- "44888 rHc|jt|jj}|jdD] \}}|||< |jj|dS)N) getfixturedoctest_namespace)r _fillfixturesdictgetfixturevalueitemsrrlupdate)rsrlravalues rFsetupzDoctestItem.setup s ##%%% =>>>=889LMMSSUU  KD%E$KK &&&&&rHct|j|g}|j|j||rt |dS)N)r)_check_all_skippedr$_disable_output_capturing_for_darwinrrunr)rsrs rFruntestzDoctestItem.runtest'se4:&&& 1133313  111  4)(33 3 4 4rHcTtjdkrdS|jjd}|rm|d|\}}tj |tj |dSdS)zFDisable output capturing. Otherwise, stdout is lost to doctest (#985).DarwinNcapturemanagerT)in_) platformsystemrT pluginmanager getpluginsuspend_global_captureread_global_capturesysstdoutwritestderr)rscapmanrerrs rFrz0DoctestItem._disable_output_capturing_for_darwin1s ?   ( ( F*445EFF  "  ) )d ) 3 3 31133HC J  S ! ! ! J  S ! ! ! ! !  " "rHexcinfoExceptionInfo[BaseException]str | TerminalReprcddl}d}t|j|j|jfr |jg}n&t|jt r |jj}|!t|Sg}|D]}|j }|j j }j d}nj |j zdz}t|j} t||| } t!} t#|jd} |z|j jJ|j jd} j Jfdt-| D} | t/|j dz d|j dz} n>dg} d}|jD]}| d |d |d } t||jr3| | ||j| d z } nGt;j|j}| d |jgz } | dtAj!|jDz } || | ftE|S)Nrrr7Fc<g|]\}}|jzdzdd|S)r03d )lineno)rfixrs rF z,DoctestItem.repr_failure..csF9?!Qq4;*44444rH z?EXAMPLE LOCATION UNKNOWN, not showing all tests of that examplez>>>z??? rz... zUNEXPECTED EXCEPTION: c8g|]}|dS)r)strip)rfrs rFrz,DoctestItem.repr_failure..ws/&'AGGDMMrH)#rrrrrrrr repr_failurerrfilenamerr.r{r _get_checker_get_report_choicerTrk docstring splitlines enumeratemaxsourcer9output_differencersplitr from_exc_infor tracebackformat_exceptionro)rsrrrrprrrrmessageryr report_choicerzindentrw inner_excinforrs @rFrzDoctestItem.repr_failure=s     MG2G4OP   . HH  '> ? ? .}-H  77''00 0) =) =GoGA;MMNV#N5577##DLL!7!7!7!7!7888"FF'7#9:: 22W[-%++!. ;G3>3>3>3>3>3@FFFFFFFFrHrdict[str, int]c ddl}t|j|j|j|j|j|jttt S)Nr) DONT_ACCEPT_TRUE_FOR_1DONT_ACCEPT_BLANKLINENORMALIZE_WHITESPACEr-IGNORE_EXCEPTION_DETAILCOMPARISON_FLAGS ALLOW_UNICODE ALLOW_BYTESNUMBER) rrrrrr-rr_get_allow_unicode_flag_get_allow_bytes_flag_get_number_flagrs rF_get_flag_lookuprsaNNN &=%;$9! ' ? 1-//)++!!    rHcp|d}t}d}|D] }|||z}|S)Nr+r)getinir)rToptionflags_strflag_lookup_tableflag_accflags rFget_optionflagsr#sKmm$9::O(**H,,%d++ OrHcb|d}|r|drd}|S)Nr>usepdbF)getvalue)rTrs rF_get_continue_on_failurer's? &0M N N( ??8 $ $ ("'  rHceZdZdZddZdS)r]Nr(Iterable[DoctestItem]c#Kddl}|jd}|j|}t |j}|jj}ddi}t|j}td|tt|j}|j } | ||||d} | j r't|| j|| VdSdS)Nrr0r{__main__Frrrrrarr)rrTrrS read_textrBrar#rrr' DocTestParser get_doctestexamplesrr[) rsrencodingtextrrarlrrr'rs rFr1zDoctestTextfile.collects;%%&899y""8,,ty>>y~Z(%dk22 # NN 8 E E    '&((!!$tXqAA = ))49V4*       rHr(r))r{r|r}rr1rLrHrFr]r]s- CrHr]rrcvddltfd|jD}|rtddSdS)zVRaise pytest.skip() if all examples in the given DocTest have the SKIP option set.rNc3XK|]$}|jjdV%dS)FN)optionsgetSKIP)rfrrs rFrhz%_check_all_skipped..s5PPQaimmGL%88PPPPPPrHz!all tests skipped by +SKIP option)rallr1r)r all_skippedrs @rFrrsYNNNPPPP$-PPPPPK2 01111122rHrobjectc(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 rF _is_mockedr?s! SMtTT rHGenerator[None]c#Ktjddd fd }|t_ dVt_dS#t_wxYw) zContext manager which replaces ``inspect.unwrap`` with a version that's aware of mock objects and doesn't recurse into them.NstoprCallable[..., Any]rCCallable[[Any], Any] | Noner(r c | |turtS|fdS#t$r'}tjd|ddtd}~wwxYw)NrBc8t|p SrJ)r?)r_stoprs rFzF_patch_unwrap_mock_aware.._mock_aware_unwrap..sjoo6TtrHzGot 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)r? Exceptionwarningswarnr)rrCerH real_unwraps` @rF_mock_aware_unwrapz4_patch_unwrap_mock_aware.._mock_aware_unwraps |tz11"{4j9999E;t*T*T*T*T*TUUU U    MCqCCTCCC      s55 A&"A!!A&)rrDrCrEr(r )inspectunwrap)rOrNs @rF_patch_unwrap_mock_awarerRst.KJN"(GN% $$$$$s 8AceZdZddZdS)rZr(r)c#JKddl}Gdd|j} |j}nH#tj$r6|jdrtd|jnYnwxYw|j j ||}t|j}td|tt|j}|||jD].}|jr%t&||j||V/dS) NrceZdZejddZedkp+edkoejjdkpedkoejjdkZerfdZejdkrdfd Z ejd kr fd Z xZ SxZ S)5DoctestModule.collect..MockAwareDocTestFinderN) r)rX rXct|trt|d|}t|drt j|}t ||S)a%On older Pythons, doctest code does not take into account `@property`. https://github.com/python/cpython/issues/61648 Moreover, wrapped Doctests need to be unwrapped so the correct line number is returned. #8796 fget __wrapped__)rpropertygetattrhasattrrPrQr _find_lineno)rsr source_linesrs rFrazBDoctestModule.collect..MockAwareDocTestFinder._find_linenosm"#x008%c6377sM222%nS11!77//$rH)rX r(r)c t|rdSt5t|||||||ddddS#1swxYwYdS)zOverride _find to work around issue in stdlib. https://github.com/pytest-dev/pytest/issues/3456 https://github.com/python/cpython/issues/69718 N)r?rRr_find) rstestsrramodulerbrlseenrs rFrez;DoctestModule.collect..MockAwareDocTestFinder._finds"#133 !3flE4s(AAA)rX ct|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)rsrgr<rs rFrmzBDoctestModule.collect..MockAwareDocTestFinder._from_module)s= "&)*CDD-!'!77//???rHr ) r{r|r}r version_infopy_ver_info_minormicrois_find_lineno_brokenrarermrrs@rFMockAwareDocTestFinderrVs  # 0! 4 !G+Q%0OS5E5Ka5OQ%0OS5E5Ka5O " % (')) ')) @ @ @ @ @ @ @ @ @*)))rHrrr=zunable to import module Fr,r-)r DocTestFinderrr CollectErrorrTr&rrSrirparsefactoriesr#rrr'findr{r1rr[ra)rsrrrrgfinderrrrs rFr1zDoctestModule.collects; @; @; @; @; @W%:; @; @; @z XFF%   {##$BCC = ==>>>>?>  $33D999('))%dk22 # NN 8 E E    KK88  D} !--tyt.  s#AA('A(Nr4)r{r|r}r1rLrHrFrZrZs.ZZZZZZrHrZtype[doctest.OutputChecker]c4ddl}Gdd|j}|S)NrceZdZejdejZejdejZejdejZ d fd Z dd Z xZ S)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+) ) ) wantrBrrrr(r_ct|||rdS|tz}|tz}|t z}|s|s|sdSd d}|r"||j|}||j|}|r"||j|}||j|}|r|||}t|||S) NTFregexre.Pattern[str]txtrBr(c.tj|d|S)Nz\1\2)resub)r~rs rFremove_prefixeszX_init_checker_class..LiteralsOutputChecker.check_output..remove_prefixes~sveWc222rH)r~rrrBr(rB)r check_outputrrr_unicode_literal_re_bytes_literal_re_remove_unwanted_precision) rsr|rr allow_unicode allow_bytes allow_numberrrs rFrz?_init_checker_class..LiteralsOutputChecker.check_outputss&ww##D#{;; t'*A*C*CCM%(=(?(??K&)9););;L   \ u 3 3 3 3 E&t'?FF%od&>DD C&t'=tDD%od&.LiteralsOutputChecker._remove_unwanted_precisions11$7788E005566D5zzSYY&& FE4(( J J1'(wwz':':'(ww{';';# ww{33H!)!1AAs8}} 'X.I##veAGGII.>.>B N'S'S'SSS 0aggii&001AGGII=AEEGGfDTDVDV@WWaeegg 1QUUWWqwwyy5HIIFJrH)r|rBrrBrrr(r_)r|rBrrBr(rB) r{r|r}rcompileUNICODErrVERBOSErrrrrs@rFLiteralsOutputCheckerr{Us )bj)CRZPP&BJ'A2:NNRZ & J)  . @ @ @ @ @ @6        rHr)r OutputChecker)rrs rF_init_checker_classrRsINNNOOOOO 5OOOb ! rHdoctest.OutputCheckercHttatS)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&rrLrHrFrrs+-- ??rHc*ddl}|jdS)z+Register and return the ALLOW_UNICODE flag.rNrrregister_optionflagrs rFrrsNNN &7 & 7 77rHc*ddl}|jdS)z)Register and return the ALLOW_BYTES flag.rNrrrs rFrrsNNN &7 &} 5 55rHc*ddl}|jdS)z$Register and return the NUMBER flag.rNrrrs rFrrsNNN &7 &x 0 00rHkeyrBc ddl}t|jt|jt |jt|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 rFrrsH NNN $W%9#W%9#W%90'2S"A      rHri)scopedict[str, Any]ctS)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`. )rrLrHrFrrs  66MrH)r'rr(r)r )rNr rOrr(rP)rSr r(r_)rTrrSr rOrr(r_)r(rr) rrrrrrrr_r(r)r(r)rTrr(r)rTrr(r_)rrr(r))rr<r(r_)r(r@)r(rx)r(r)r(r)rrBr(r)r(r)_r __future__rrcollections.abcrrrr contextlibrrkrPospathlibr rrrrtypestypingr r rK_pytestr _pytest._code.coder rr _pytest._ior_pytest.compatr_pytest.configr_pytest.config.argparsingr_pytest.fixturesrr _pytest.nodesrr_pytest.outcomesrr_pytest.pathlibr_pytest.pythonr_pytest.python_apir_pytest.warning_typesrrtyping_extensionsr rrrrrrDrKr&__annotations__rGrMr^rXr\rYrorJrrrrrr#r'r]rr?rRrZrrrrrrrrLrHrFrs:::"""""" $$$$$$%%%%%%$$$$$$$$$$$$%%%%%%    ,,,,,,//////++++++&&&&&&''''''!!!!!!,,,,,,$$$$$$''''''######------!!!!!!&&&&&&!!!!!!%%%%%%//////'NNN&&&&&&#%%%+?(,  48 8888////d    AAAA9999&&&& ( ( ( ( (l ( ( (!!!!!i!!! 2222l-1 $ (CFCFCFCFCF$CFCFCFL     f>2222%%%%:[[[[[F[[[|T!T!T!T!n(888866661111    " yrH