jj:UddlmZddlZddlmZddlmZddlmZddlmZddlZddl Z ddl m Z ddl Z ddl mZddlZddlZdd lmZdd lmZdd lmZdd lmZdd lmZddlmZddlmZddlmZddlmZddlmZddlmZddlmZddlm Z ddlm!Z!ddlm"Z"ddlm#Z#ddlm$Z$ddl%Z%ddl&Z&ddl'm(Z(ddl'm)Z)ddl'm*Z*ddl'm+Z+ddl,m-Z-ddl.m/Z/dd l.m0Z0dd!l1m2Z2dd"l3m4Z4dd#l5m6Z6dd$l5m7Z7ejpd%krdd&l9m:Z:e d'Z;eeZMejd;<Gd?d@eMZNejd;<GdAdBeNZOejd;<GdCdDeNZPejd;<GdEdFeMZQGdGdHeQZRejd;<GdIdJeMZSejd;<GdKdLeMZTejd;<GdMdNeMZUejd;<GdOdPeMZVejd;<GdQdReMZWd[dSZXdTZYee%jjdUZ\e\jdVk(r e\jZ\ee&jjZ_d\dWZ` d]dXZay)^) annotationsN)Callable)Iterable)Mapping)Sequence)StringIO)Path) extract_tb)format_exception)format_exception_only) FrameSummary)CodeType) FrameType) TracebackType)Any)ClassVar)Final)final)Generic)Literal)overload) SupportsIndex) TypeAlias)TypeVar) findsource getrawcode)getstatementrange_ast)Source)TerminalWriter) safeformatsaferepr) get_real_func)check_ispytest) absolutepath) bestrelpath )BaseExceptionGroup)longshortlinenonativevalueauto.ceZdZdZdZd dZeddZdZdZ e ddZ e ddZ e dd Z e dd Zdd Zddd Zy)Codez#Wrapper around Python code objects.rawc||_yNr5)selfobjs T/opt/eval-harness/bench_task/venv/lib/python3.12/site-packages/_pytest/_code/code.py__init__z Code.__init__?s c$|t|Sr8r)clsr:s r; from_functionzCode.from_functionBs:c?##r=c4|j|jk(Sr8r5)r9others r;__eq__z Code.__eq__Fsxx599$$r=Nc4|jjdz SN)r6co_firstlinenor9s r; firstlinenozCode.firstlinenoLsxx&&**r=c.|jjSr8)r6co_namerHs r;namez Code.namePsxxr=ctj|j}|sy t|}|j s t d|S#t $r|cYSwxYw)zoReturn a path object pointing to source code, or an ``str`` in case of ``OSError`` / non-existing file.zpath check failed.)inspectgetfiler6r&existsOSError)r9filenameps r;pathz Code.pathTs[??488, X&A88:233H O s'A AAc6t|j\}}|S)zJReturn a _pytest._code.Source object for the full source file of the code.)rr6)r9full_s r; fullsourcezCode.fullsourcefsTXX&a r=c,t|jS)zGReturn a _pytest._code.Source object for the code object's source only.)rr6rHs r;sourcez Code.sourcelsdhhr=ctj|j\}}}|s"t|d|jjSt |}||j |||j |t|S)zReturn a tuple with the argument names for the code object. If 'var' is set True also return the names of the variable and keyword arguments when present. N)rOgetargsr6tuple co_argcountlistappend)r9varargsvarargsvarkwresults r;r]z Code.getargsqst 'txx8gu4 4 456 6d   MM' "   MM% V}r=)r:rreturnNone)r:objectrgr4rgintrgstrrg Path | str)rg Source | NonergrF)rbboolrgztuple[str, ...])__name__ __module__ __qualname____doc__ __slots__r< classmethodr@rC__hash__propertyrIrLrUrYr[r]r=r;r4r4:s-I$$%H ++  "  r=r4ceZdZdZdZd dZeddZeddZeddZ eddZ eddZ d Z dd Z ddd Zy )FramezgWrapper around a Python frame holding f_locals and f_globals in which expressions can be evaluated.r5c||_yr8r5)r9frames r;r<zFrame.__init__s r=c4|jjdz SrE)r6f_linenorHs r;linenoz Frame.linenosxx  1$$r=c.|jjSr8)r6 f_globalsrHs r;rzFrame.f_globalssxx!!!r=c.|jjSr8)r6f_localsrHs r;rzFrame.f_localssxx   r=c@t|jjSr8)r4r6f_coderHs r;codez Frame.codesDHHOO$$r=c|jj tdS|jjj|jS)zStatement this frame is at.rN)rrYr getstatementrrHs r; statementzFrame.statements= 99   '": yy##00==r=c |jj}|j|t||j|S)zEvaluate 'code' in the frame. 'vars' are optional additional local variables. Returns the result of the evaluation. )rcopyupdateevalr)r9rvarsrs r;rz Frame.evals5==%%'D$..(33r=ct|S)zCReturn a 'safe' (non-recursive, one-line) string repr for 'object'.r")r9ris r;reprz Frame.reprs r=cg}|jj|D]#} |j||j|f%|S#t$rY3wxYw)zReturn a list of tuples (name, value) for all arguments. If 'var' is set True, also include the variable and keyword arguments when present. )rr]rarKeyError)r9rbretvalargs r;r]z Frame.getargssa 99$$S) C  sDMM#$678     s A AAN)rrrgrhrjrgzdict[str, Any])rgr4rq)ririrgrmrr)rbrs)rtrurvrwrxr<r{rrrrrrrr]r|r=r;r~r~s.I%%""!!%%>> 4  r=r~ceZdZdZdZ d ddZ ddZeddZddZ e jdkredd Z edd Z edd Znedd Z edd Z eddZed dZeddZd!dZed"dZed#dZed$dZddZ d d%dZeeZd&dZd!dZed!dZy)'TracebackEntryzA single entry in a Traceback. _rawentry _repr_styleNc ||_||_yr8r)r9rawentry repr_styles r;r<zTracebackEntry.__init__s !)",r=c.t|j|Sr8)rr)r9rs r;with_repr_stylezTracebackEntry.with_repr_stylesdnnj99r=c4|jjdz SrE)r tb_linenorHs r;rzTracebackEntry.linenos~~''!++r=c:t|jd}|dS)NrF)limitr)r r)r9 stack_summarys r;get_python_framesummaryz&TracebackEntry.get_python_framesummarys#4>>; Qr=r(cyr8r|rHs r;end_lineno_relativez"TracebackEntry.end_lineno_relativer=cyr8r|rHs r;colnozTracebackEntry.colnorr=cyr8r|rHs r; end_colnozTracebackEntry.end_colnorr=c|j}|jy|jdz |jjjz SrE)r end_linenorrrI)r9 frame_summarys r;rz"TracebackEntry.end_lineno_relativesC 88:M''/ ++a/$**//2M2MM Mr=c6|jjS)z>Starting byte offset of the expression in the traceback entry.)rrrHs r;rzTracebackEntry.colnos//177 7r=c6|jjS)z)rrrUrrHs r;__repr__zTracebackEntry.__repr__s/!$**//"6"6!7qq8IKKr=c|jjj}|J|j|jS)z6_pytest._code.Source object for the current statement.)rrrYrr)r9r[s r;rzTracebackEntry.statement s8++!!!""4;;//r=cB|jjjS)zPath to the source code.)rrrUrHs r;rUzTracebackEntry.pathszz###r=c.|jjS)zLocals of underlying frame.)rrrHs r;localszTracebackEntry.localsszz"""r=cB|jjjSr8)rrrIrHs r;getfirstlinesourcez!TracebackEntry.getfirstlinesourceszz***r=cp|jjj}|ydx}}|4|jjj}||j |d}|j } t |j||\}}}|||||< |||S#t$r|jdz}YwxYw)zReturn failing source code.N)astnoderF) rrrYrUgetrrr SyntaxError)r9astcacher[keyrstartrXends r; getsourcezTracebackEntry.getsources ++ >g  **//&&C",,sD1'') (3 VWOGQ 8#7 ' eC   "++/C "s/BB54B5cd}|jj|jjfD] } |d}n|rt |r||S|S#t$rY.wxYw)aReturn True if the current frame has a var __tracebackhide__ resolving to True. If __tracebackhide__ is a callable, it gets called with the ExceptionInfo instance and can decide whether to hide the traceback. Mostly for internal use. F__tracebackhide__)rrr Exceptioncallable)r9excinfotbh maybe_ns_dcts r;ishiddenzTracebackEntry.ishidden:svMR!ZZ00$**2F2FG L  "#67  8C=w<    sA AAc  |jjj} t|jj }d|jd|jdzd|d|d S#t $rt$rd}Y.fps--0&(..<z Traceback.cut..s$FSV$Fs) osfspathrrrUrmrr parentsrrIrr) r9rUrrI excludepathpath_ excludepath_xrcodepaths r;cutz Traceback.cutzs  "))D/*2t +8N  *A77<>%'"5c":; ;7&s+ +r=c^ttrfd}n}tt||S)aReturn a Traceback instance with certain items removed. If the filter is an `ExceptionInfo`, removes all the ``TracebackEntry``s which are hidden (see ishidden() above). Otherwise, the filter is a function that gets a single argument, a ``TracebackEntry`` instance, and should return True when the item should be added to the ``Traceback``, False when not. c(|j Sr8)r)r excinfo_or_fns r;z"Traceback.filter..sqzz-88r=)r ExceptionInforfilter)r9rfns ` r;rzTraceback.filters+ m] 38BBD)**r=ci}t|D]\}}|jjjt |jjj |j f}|j|g}t|jj}|r|D] }||k(s |ccS|j|y)z~Return the index of the frame/TracebackEntry where recursion originates if appropriate, None if no recursion occurred.N) enumeraterrrUidr6r setdefaultdictrra)r9cacheientryrvalueslocotherlocs r;recursionindexzTraceback.recursionindexsCE!$ HAu ++""''EKK,<,<,@,@)A5<<OC%%c2.Fu{{++,C &!H3 ! MM#  r=)rz(TracebackType | Iterable[TracebackEntry]rgrh)NNNN) rUzos.PathLike[str] | str | NonerrrIrrzos.PathLike[str] | Nonergr)rrrgr)rrrgr)rzSupportsIndex | slicergzTracebackEntry | Traceback)rz?ExceptionInfo[BaseException] | Callable[[TracebackEntry], bool]rgrr) rtrurvrwr<rrrrr __classcell__rs@r;rrfs[! 4! !&/3!"&/3 "+"" " - "  "HDD 77, +V+  +(r=rTcT t|dg}|st |tr |j}n t|}dj|g|S#t$rQttjj dddd}tj dkrt ||rg}nYwxYw)N __notes__z urllib.error HTTPErrorr|)r) r) getattrrrmodulesrrrr+messagermjoin)excinclude_subexception_msgnotesrrs r;stringify_exceptionr"s ["- $ 38J(K++c( 99     CKKOONDA;PRS   g %*S)*DE   s A AB'&B'E)bound covariantceZdZUdZdZded<ded<ded<d ed < d#d d  d$dZe d% d&dZe d% d'dZ ed%d(dZ ed)dZ d*dZ e d+dZe d,dZe d-dZe d.dZe d/dZej&d0dZd.dZd1d2dZd3dZd4dZ d5 d6dZd7dZ d8 d9d Zd d d! d:d"Zy );rzJWraps sys.exc_info() objects and offers help for navigating the traceback.zAssertionError('assert r_assert_start_repr'tuple[type[E], E, TracebackType] | None_excinform _striptextTraceback | None _tracebackNF _ispytestcDt|||_||_||_yr8)r%r)r*r,)r9r striptext tracebackr.s r;r<zExceptionInfo.__init__s! y! ##r=c||jsJdt|||jf}|j||S)aEReturn an ExceptionInfo for an existing exception. The exception must have a non-``None`` ``__traceback__`` attribute, otherwise this function fails with an assertion error. This means that the exception must have been raised, or added a traceback with the :py:meth:`~BaseException.with_traceback()` method. :param exprinfo: A text string helping to determine if we should strip ``AssertionError`` from the output. Defaults to the exception message/``__str__()``. .. versionadded:: 7.4 zTExceptions passed to ExcInfo.from_exception(...) must have a non-None __traceback__.) __traceback__type from_exc_info)r? exceptionexprinfoexc_infos r;from_exceptionzExceptionInfo.from_exception sJ.&&  3 &OY 0G0GH  844r=cd}|Rt|dtr?t|ddd}|t|d}|r|j |j rd}|||dS)z@Like :func:`from_exception`, but using old-style exc_info tuple.rNNrFmsgzAssertionError: Tr-)rAssertionErrorrr# startswithr')r?r8r7r*s r;r5zExceptionInfo.from_exc_info'sl   8A; Gx{E48H#HQK0H//0F0FG/ 8Z488r=ctj}|dJd|dJd|dJd|d|d|df}tj||S)a=Return an ExceptionInfo matching the current traceback. .. warning:: Experimental API :param exprinfo: A text string helping to determine if we should strip ``AssertionError`` from the output. Defaults to the exception message/``__str__()``. rzno current exceptionrF)rr8rr5)r?r7tupr8s r; from_currentzExceptionInfo.from_current8s|lln1v!9#99!1v!9#99!1v!9#99!FCFCF+**8X>>r=c|ddS)z!Return an unfilled ExceptionInfo.NTr-r|)r?s r; for_laterzExceptionInfo.for_laterLs44((r=c8|jJd||_y)z< rFz tblen=r)r)rrtr#lenr1rHs r;rzExceptionInfo.__repr__sS == >4>>**+1XdmmA6F-G,HPSTXTbTbPcOddeffr=c dfd |rSt|jtr9|jx}%|dt|jjdSt |j}dj |}|j}|r3|j|jr|t|jd}|S)aReturn the exception as a string. This is usually a single line ": ", but may also include additional lines for the exception notes, and detailed information for SyntaxError's. :param tryshort: If true, and the exception is an AssertionError, strip 'AssertionError: ' from the beginning. ct|jdk7ryt|jdx}tr|S|S)NrFr)rQ exceptionsrr+)ege_get_single_subexcs r;rWz1ExceptionInfo.exconly.._get_single_subexcsB2==!Q&r}}Q//!1CD)!,,Hr=Nz [single exception in ]rN)rU!BaseExceptionGroup[BaseException]rgzBaseException | None) rr1r+r4rtr rrstripr=r*rQ)r9tryshortsubexclinestextrWs @r;exconlyzExceptionInfo.exconlys 1  !  4::'9:-djj99FZ5d4::6F6O6O5PPQR R%djj1wwu~{{} t/C023 r=c.t|j|S)zReturn True if the exception is an instance of exc. Consider using ``isinstance(excinfo.value, exc)`` instead. )rr1)r9rs r; errisinstancezExceptionInfo.errisinstances $**c**r=cJtdt|j dz dD]|}|j|}|j|r$|jj j j|j}}|jd}t||dz|cSy)NrFTr[) rangerQr1rrrr6 co_filenamerr_ReprFileLocation)r9rrrUrr_s r; _getreprcrashzExceptionInfo._getreprcrashsrC//!3R8 CANN1%E>>$'${{//33??f,,,5'fqj'BB  C r=c $|dk(rhttt|j|j|j r|j dj nd|jSt||||||||} | j|S)a.Return str()able representation of this exception info. The formatting parameters are ineffective if ``style="native"``, since in this case the native formatting is used. :param bool showlocals: Show locals per traceback entry. :param str style: long|short|line|no|native|value traceback style. :param bool abspath: If paths should be changed to absolute or left unchanged. :param tbfilter: A filter for traceback entries. * If false, don't hide any entries. * If true, hide internal entries and entries that contain a local variable ``__tracebackhide__ = True``. * If a callable, delegates the filtering to the callable. :param bool funcargs: Show function arguments per traceback entry. :param bool truncate_locals: Whether to show a size-limited `repr()` of locals, or a full pretty-printing. :param bool truncate_args: Whether to show a size-limited truncated `repr()` of function arguments, or a full pretty-printing. :param bool chain: If chained exceptions should be shown. .. versionchanged:: 3.9 Added the ``chain`` parameter. r0rN reprtraceback reprcrash) showlocalsstyleabspathtbfilterfuncargstruncate_locals truncate_argschain) ReprExceptionInfoReprTracebackNativer r4r1r1rrhExceptionInfoFormatter repr_excinfo) r9rmrnrorprqrrrsrtfmts r;getreprzExceptionInfo.getreprsf H $1$  7;~~q)334,,.  %!+'  %%r=cd}t|j}d|d|}||k(r|dz }tj||sJ|y)zCheck whether the regular expression `regexp` matches the string representation of the exception using :func:`python:re.search`. If it matches `True` is returned, otherwise an `AssertionError` is raised. Tz/Regex pattern did not match. Expected regex: z Actual message: z* Did you mean to `re.escape()` the regex?)r"r1research)r9regexprr1r;s r;matchzExceptionInfo.matchsf !#DJJ/!!' +!!&  + U? @ @Cyy',,'r=c |||kDry|jD]f}t|tr|j|||||dzry|||k7r5t||sB|"t |}t j ||sfyy)zFReturn `True` if a `BaseExceptionGroup` contains a matching exception.FrFT)rTrr+_group_containsr"r|r})r9 exc_groupexpected_exceptionr target_depth current_depthrr1s r;rzExceptionInfo._group_containss  $=<+G'' C#12''+UL-RSBS (} /Lc#56 +C0yy.  r=)rdepthcd}t|jtsJ|d}| |dk\sJ||j|j|||S)aCheck whether a captured exception group contains a matching exception. :param Type[BaseException] | Tuple[Type[BaseException]] expected_exception: The expected exception type, or a tuple if one of multiple possible exception types are expected. :param str | re.Pattern[str] | None match: If specified, a string containing a regular expression, or a regular expression object, that is tested against the string representation of the exception and its `PEP-678 ` `__notes__` using :func:`re.search`. To match a literal string that may contain :ref:`special characters `, the pattern can first be escaped with :func:`re.escape`. :param Optional[int] depth: If `None`, will search for a matching exception at any nesting depth. If >= 1, will only match an exception if it's at the specified depth (depth = 1 being the exceptions contained within the topmost exception group). .. versionadded:: 8.0 .. warning:: This helper makes it easy to check for the presence of specific exceptions, but it is very bad for checking that the group does *not* contain *any other exceptions*. You should instead consider using :class:`pytest.RaisesGroup` z=Captured exception is not an instance of `BaseExceptionGroup`z!`depth` must be >= 1 if specifiedrF)rr1r+r)r9rrrr;s r;group_containszExceptionInfo.group_contains9s\HN$**&89>3>91 5A:33.##DJJ0BE5QQr=)rNN) rr(r0rmr1r+r.rsrgrhr8)r6r#r7 str | NonergExceptionInfo[E])r8 tuple[type[E], E, TracebackType]r7rrgr)r7rrgExceptionInfo[BaseException])rgr)r8rrgrh)rgztype[E])rgr#)rgrrl)rgr)r1rrgrhrr)r[rsrgrm)rEXCEPTION_OR_MORErgrs)rgReprFileLocation | None)Fr,FTFTTT)rmrsrnTracebackStylerorsrpz:bool | Callable[[ExceptionInfo[BaseException]], Traceback]rqrsrrrsrsrsrtrsrgz&ReprExceptionInfo | ExceptionChainRepr)r~zstr | re.Pattern[str]rgz Literal[True]rE) rrYrrrstr | re.Pattern[str] | Nonerrrrkrgrs)rrrrrrrgrs)rtrurvrwr'__annotations__r<ryr9r5rArCrFr{r4r1rrKr1setterrr_rarhrzrrrr|r=r;rrsU#<<55O  &* $  $8 $ $$ $  $  $ $ 5  5  5 55: $9299  99 ??&))!       ""   g #J+ ! &OS $"I&I&I& I& M I&  I&I&I&I& 0I&V0$( 4.,  !    D/3 (R-(R, (R  (R  (Rr=rrTracebackFilterceZdZUdZdZded<dZded<dZded <d Zd ed <d Z ded<d Z ded<dZ ded<d Z ded<d Z ded<d Zded<ej eddZded<d%dZd&dZd'dZ d( d)dZ d*dZ d+ d,dZd-dZ d. d/d Zd0d!Zd1d"Z d2d#Zd3d$Zy)4rwzHelper object to format ExceptionInfo's and individual exception parts into TerminalRepr's. See :func:`ExceptionInfo.getrepr` for parameters. rr flow_markerr# fail_markerFrsrmr,rrnTrorrprqrrrsrt)default_factoryinitrzdict[str | Path, ast.AST]rc" t|jt|dz }dt|t|j z zS#t$rt$r, t|d}n#t$rt$rYYywxYwYewxYw)NrFrcr)rmrrQrrr)r9r[ss r; _getindentz!ExceptionInfoFormatter._getindents F''F a89ACFS_,--!    r O$     s5&AB#A21B2BBBB Bc`|j|j}||j}|Sr8)rrdeindent)r9rr[s r;_getentrysourcez&ExceptionInfoFormatter._getentrysources+/  __&F r=c|jrig}|jjdD]=\}}|jr t |}n t |d}|j ||f?t |Sy)NT)rb)maxsize)rqrr]rsr#ra ReprFuncArgs)r9rrcargnameargvaluestr_reprs r; repr_argsz ExceptionInfoFormatter.repr_argsst ==D%*[[%8%8T%8%B 1!%%'1H'$?H Wh/0  1  % %r=Nc g}||dkr|t|z }||t|jk\s|dkr td}d}d} |r~|j| |j|j z|j |j |j||j|j ||||n|jd|D]} |j| | z|j|jdz|j|z|j |j |j||j||||||j|dzdD]} |j| | z|8|rdn|j|} |j |j|| d |S) z,Return formatted and marked up source lines.Nrr )raw_liner.rrrr rFrT)indentmarkall) rQr]rrastripextendget_highlight_arrows_for_line raw_linesrr get_exconly) r9r[ line_indexrr-end_line_indexrrr] space_prefixr.rs r; get_sourcez!ExceptionInfoFormatter.get_sources  *q. #f+ %J >Z3v||+<< QE]FJ  LL Z(@(F(F(HH I LL22#--j9j1779%-' 3  [j1 2 \D01 2 LL))E1FLL4LL M LL22#--j9j1%-' 3  Z!^%56 2 \D01 2  QT__V%z<10 = )rmsortrarrr#r! ReprLocals)r9rr]rkeysrLr1rs r; repr_localsz"ExceptionInfoFormatter.repr_localss ??E$*;CSVs]C;D; IIK =t >)LL!<= ++#+E?#-e#4LLD:S !;< =$e$ $+BJ>>*-D*41A7KK))%,,7JUHj+uM M g  S/55d;<UD$e< < T--ga-@AUD$e<