jfbUdZddlmZddlZddlmZddlZddlZddlZddl m Z ddl Z ddl Z ddl Z ddlZddlmZddlmZddlmZdd lmZdd lmZdd lmZdd lmZddlZdd lmZddlmZddlmZddlmZddlmZddl 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.ddl/m0Z0ddl1m2Z2ddl3m4Z4ddl5m6Z6er ddl7Z7ddl8m9Z9d Z:d!Z;d"Ze:e;efZ?da@daAd%eBd&<djd+ZCdkd,ZDdld2ZEdmd5ZFdnd8ZGdmd9ZHGd:d;eZIGd<d=eJZKdod?ZL dpdqdIZMGdJdKe+ZNdrdMZOdsdNZPdtdOZQGdPdQe2ZRdudTZSdvdWZTedwdYZUGdZd[e2ZVdxd]ZWdyd_ZXdzd`ZYdzdaZZdzdbZ[d{deZ\e'dfgd|diZ]dS)}z4Discover and run doctests in modules and test files.) annotationsN)contextmanager)Path)Any)Callable) Generator)Iterable)Pattern)Sequence) 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)r0collectz--doctest-modules store_trueFzRun doctests in all .py modulesdoctestmodules)actionr0helpdestz--doctest-reportr%z9Choose another output format for diffs on doctest failure doctestreport)r/r0r6choicesr7z--doctest-globappendpatz2Doctests file matching pattern, default: test*.txt doctestglob)r5r0metavarr6r7z--doctest-ignore-import-errorsz Ignore doctest collection errorsdoctest_ignore_import_errorsz--doctest-continue-on-failurezz_is_doctest..s-88$z$%%888888rI)rVsession isinitpath getoptionrX)rUrTrPglobss ` rGr]r]sh {&&&6>+D+DT+J+J&t   ] + + < }E 8888%888 8 88rIc|jdkS)Nz __main__.py)rbrSs rGrZrZs 9 %%rIceZdZd dZd dZd S) ReprFailDoctestreprlocation_lines0Sequence[tuple[ReprFileLocation, Sequence[str]]]r)r*c||_dSrK)rq)selfrqs rG__init__zReprFailDoctest.__init__s#5rItwrc~|jD]4\}}|D]}||||5dSrK)rqline toterminal)rtrv reprlocationlinesrxs rGryzReprFailDoctest.toterminalsZ#'#: ( ( L%    # #B ' ' ' ' ( (rIN)rqrrr)r*)rvrr)r*)__name__ __module__ __qualname__ruryrMrIrGrprps<5555 ((((((rIrpc eZdZdfd ZxZS)MultipleDoctestFailuresfailures Sequence[doctest.DocTestFailure]r)r*cVt||_dSrK)superrur)rtr __class__s rGruz MultipleDoctestFailures.__init__s$   rI)rrr)r*)r|r}r~ru __classcell__rs@rGrrs=!!!!!!!!!!rIrtype[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)rrur)rtrrrrrs rGruz8_init_runner_class..PytestDoctestRunner.__init__s2 GG  Wg;  W W W':D $ $ $rItestdoctest.DocTestexampledoctest.ExamplegotrCchj|||}|jr||dS|rK)DocTestFailurerr:)rtoutrrrfailuredoctests rGreport_failurez>_init_runner_class..PytestDoctestRunner.report_failuresC-g,T7C@@G'  7##### rIexc_info>tuple[type[BaseException], BaseException, types.TracebackType]ct|dtr|dt|dtjrt jdj|||}|jr||dS|)NzQuitting debugger) isinstancerbdbBdbQuitr exitUnexpectedExceptionrr:)rtrrrrrrs rGreport_unexpected_exceptionzK_init_runner_class..PytestDoctestRunner.report_unexpected_exceptions(1+'788 "qk!(1+s{33 3 12221g1$JJG'  7##### rINNrT) rrrrrrrr`r)r*)rrrrrrCr)r*)rrrrrrr)r*)r|r}r~__doc__rurrr)rrs@rGPytestDoctestRunnerrs  59#' (,  ; ; ; ; ; ; ;                rIr)r DebugRunner)rrs @rG_init_runner_classrsLNNN-------g1---^ rITrrrrrrrdoctest.DocTestRunnercRttat||||S)Nrrrr)rLrrs rG _get_runnerrs:)++  /    rIcpeZdZdfd Zedfd ZddZddZddZddZ dfd Z ddZ xZ S) DoctestItemrbrCrPDoctestTextfile | DoctestModulerunnerrdtestrr)r*c t||||_||_d|_|jj}||dd}||_|j |_ | dS)N)nodefunccls) rrurrobjrj_fixturemanagergetfixtureinfo _fixtureinfo names_closure fixturenames _initrequest)rtrbrPrrfm fixtureinfors rGruzDoctestItem.__init__s v&&&   \ )''T$'GG ''5 rIr!cNt||||S)zThe public named constructor.)rbrPrr)rr\)rrPrbrrrs rGr\zDoctestItem.from_parents'ww""VFRW"XXXrIc@i|_t|d|_dS)NT) _ispytest)funcargsr_requestrts rGrzDoctestItem._initrequests!+- "44888 rIc|jt|jj}|jdD] \}}|||< |jj|dS)N) getfixturedoctest_namespace)r _fillfixturesdictgetfixturevalueitemsrrmupdate)rtrmrbvalues rGsetupzDoctestItem.setup s ##%%% =>>>=889LMMSSUU  KD%E$KK &&&&&rIct|j|g}|j|j||rt |dS)N)r)_check_all_skippedr$_disable_output_capturing_for_darwinrrunr)rtrs rGruntestzDoctestItem.runtest'se4:&&& 1133313  111  4)(33 3 4 4rIcTtjdkrdS|jjd}|rm|d|\}}tj |tj |dSdS)zFDisable output capturing. Otherwise, stdout is lost to doctest (#985).DarwinNcapturemanagerT)in_) platformsystemrU pluginmanager getpluginsuspend_global_captureread_global_capturesysstdoutwritestderr)rtcapmanrerrs rGrz0DoctestItem._disable_output_capturing_for_darwin1s ?   ( ( F*445EFF  "  ) )d ) 3 3 31133HC J  S ! ! ! J  S ! ! ! ! !  " "rIexcinfoExceptionInfo[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)Nrrr8Fc:g|]\}}d|jzdz|fzS)z%03d %sr)lineno)rgixrs rG z,DoctestItem.repr_failure..cs?=CaIT[1!4a 88rI z?EXAMPLE LOCATION UNKNOWN, not showing all tests of that examplez>>>z???  z... zUNEXPECTED EXCEPTION: c8g|]}|dS)r)strip)rgrs rGrz,DoctestItem.repr_failure..ws/&'AGGDMMrI)#rrrrrrrr repr_failurerrfilenamerr/r|r _get_checker_get_report_choicerUrl docstring splitlines enumeratemaxsourcer:output_differencersplitr from_exc_infor tracebackformat_exceptionrp)rtrrrrqrrrrmessagerzr report_choicer{indentrx inner_excinforrs @rGrzDoctestItem.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@FFFFFFFFrIrdict[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 rG_get_flag_lookuprsaNNN &=%;$9! ' ? 1-//)++!!    rIcp|d}t}d}|D] }|||z}|S)Nr,r)getinir)rUoptionflags_strflag_lookup_tableflag_accflags rGget_optionflagsr#sKmm$9::O(**H,,%d++ OrIcb|d}|r|drd}|S)Nr?usepdbF)getvalue)rUrs rG_get_continue_on_failurer's? &0M N N( ??8 $ $ ("'  rIceZdZdZddZdS)r^Nr)Iterable[DoctestItem]c#Kddl}|jd}|j|}t |j}|jj}ddi}t|j}td|tt|j}|j } | ||||d} | j r't|| j|| VdSdS)Nrr1r|__main__Frrrrrbrr)rrUrrT read_textrCrbr#rrr' DocTestParser get_doctestexamplesrr\) rtrencodingtextrrbrmrrr(rs rGr2zDoctestTextfile.collects;%%&899y""8,,ty>>y~Z(%dk22 # NN 8 E E    '&((!!$tXqAA = ))49V4*       rIr)r))r|r}r~rr2rMrIrGr^r^s- CrIr^rrcvddltfd|jD}|rtddSdS)zVRaise pytest.skip() if all examples in the given DocTest have the SKIP option set.rNc3XK|]$}|jjdV%dS)FN)optionsgetSKIP)rgrrs rGriz%_check_all_skipped..s5PPQaimmGL%88PPPPPPrIz!all tests skipped by +SKIP option)rallr1r)r all_skippedrs @rGrrsYNNNPPPP$-PPPPPK2 01111122rIrobjectc(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 rG _is_mockedr?s! SMtTT rIGenerator[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)rc | |turtS|fdS#t$r'}tjd|ddtd}~wwxYw)NrBc8t|p SrK)r?)r_stoprs rGzF_patch_unwrap_mock_aware.._mock_aware_unwrap..sjoo6TtrIzGot 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` @rG_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 @rG_patch_unwrap_mock_awarerRst.KJN"(GN% $$$$$s 8AceZdZddZdS)r[r)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)rtr source_linesrs rGrazBDoctestModule.collect..MockAwareDocTestFinder._find_linenosm"#x008%c6377sM222%nS11!77//$rI)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) rttestsrrbmodulerbrmseenrs rGrez;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)rtrgr<rs rGrmzBDoctestModule.collect..MockAwareDocTestFinder._from_module)s= "&)*CDD-!'!77//???rIr ) r|r}r~r version_infopy_ver_info_minormicrois_find_lineno_brokenrarermrrs@rGMockAwareDocTestFinderrVs  # 0! 4 !G+Q%0OS5E5Ka5OQ%0OS5E5Ka5O " % (')) ')) @ @ @ @ @ @ @ @ @*)))rIrrr>zunable to import module Fr,r-)r DocTestFinderrr CollectErrorrUr&rrTrjrparsefactoriesr#rrr'findr|r1rr\rb)rtrrrrgfinderrrrs rGr2zDoctestModule.collects; @; @; @; @; @W%:; @; @; @z XFF%   {##$BCC = ==>>>>?>  $33D999('))%dk22 # NN 8 E E    KK88  D} !--tyt.  s#AA('A(Nr4)r|r}r~r2rMrIrGr[r[s.ZZZZZZrIr[type[doctest.OutputChecker]cBddl}ddlGfdd|j}|S)NrceZdZdjZdjZdjZd 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+) ) ) wantrCrrrr)r`ct|||rdS|tz}|tz}|t z}|s|s|sdSd fd }|r"||j|}||j|}|r"||j|}||j|}|r|||}t|||S) NTFregex Pattern[str]txtrCr)c2|d|S)Nz\1\2)sub)r~rres rGremove_prefixeszX_init_checker_class..LiteralsOutputChecker.check_output..remove_prefixessvveWc222rI)r~rrrCr)rC)r check_outputrrr_unicode_literal_re_bytes_literal_re_remove_unwanted_precision) rtr|rr allow_unicode allow_bytes allow_numberrrrs rGrz?_init_checker_class..LiteralsOutputChecker.check_outputts0ww##D#{;; t'*A*C*CCM%(=(?(??K&)9););;L   \ u 3 3 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 1QUUWWqwwyy5HIIFJrI)r|rCrrCrrr)r`)r|rCrrCr)rC) r|r}r~compileUNICODErrVERBOSErrrr)rrs@rGLiteralsOutputCheckerr{Vs !jj)CRZPPJJ'A2:NNZZ & J)  . @ @ @ @ @ @ @6        rIr)rr OutputChecker)rrrs @rG_init_checker_classrRs_NNN IIIOOOOOOO 5OOOb ! rIdoctest.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'rrMrIrGrrs+-- ??rIc*ddl}|jdS)z+Register and return the ALLOW_UNICODE flag.rNrrregister_optionflagrs rGrrsNNN &7 & 7 77rIc*ddl}|jdS)z)Register and return the ALLOW_BYTES flag.rNrrrs rGrrsNNN &7 &} 5 55rIc*ddl}|jdS)z$Register and return the NUMBER flag.rNrrrs rGrrsNNN &7 &x 0 00rIkeyrCc 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 rGrrsH NNN $W%9#W%9#W%90'2S"A      rIrj)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`. )rrMrIrGrrs  66MrI)r(rr)r*r )rOrrPrr)rQ)rTrr)r`)rUrrTrrPrr)r`)r)rr) rrrrrrrr`r)r)r)r)rUrr)r)rUrr)r`)rrr)r*)rr<r)r`)r)r@)r)rx)r)r)r)r)rrCr)r)r)r)^r __future__rr contextlibrrkrPospathlibrrrrtypestypingrrrr r r r rK_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 rtyping_extensionsr!rrrrrrErLr'__annotations__rHrNr_rYr]rZrprJrrrrrr#r'r^rr?rRr[rrrrrrrrMrIrGrs:::"""""" %%%%%%    ,,,,,,//////++++++&&&&&&''''''!!!!!!,,,,,,$$$$$$''''''######------!!!!!!&&&&&&!!!!!!%%%%%%//////'NNN&&&&&&#%%%+?(,  48 8888////d    AAAA9999&&&& ( ( ( ( (l ( ( (!!!!!i!!! 2222l-1 $ (CFCFCFCFCF$CFCFCFL     f>2222%%%%:[[[[[F[[[|U!U!U!U!p(888866661111    " yrI