o 2gt@s@UdZddlmZddlZddlZddlZddlZddlZddlZddl Z ddl m Z ddl m Z ddlmZddlmZmZmZmZmZmZmZddlmZddlZdd lmZdd lmZm Z m!Z!ej"#eZ$d d gZ%d e&d<GdddeZ'GdddeZ(ee'e(fZ)de&d<deddZ*dfddZ+Gdddej,Z-dgd!d"Z.e Gd#d$d$Z/dhd(d)Z0did,d-Z1did.d/Z2djd1d2Z3dkd6d7Z4dld9d:Z5dld;d<Z6dmd?d@Z7ej8dAdBdndEdFZ9dodKdLZ:e;dMZZ?GdXdYdYeZ@GdZdOdOejAZBdqd\d]ZCdrd`daZDdrdbdcZEGdddQdQZFdS)szGUtilities for processing .test files containing test case descriptions.) annotationsN)abstractmethod) dataclass)Path)AnyFinalIterator NamedTupleNoReturnPatternUnion) TypeAlias)defaults)PREFIXtest_data_prefix test_temp_dirgdblldbrSUPPORTED_DEBUGGERSc@s&eZdZUded<ded<ded<dS) UpdateFilestrmodulecontent target_pathN__name__ __module__ __qualname____annotations__rrF/home/garg/my-data/venv/lib/python3.10/site-packages/mypy/test/data.pyr   rc@seZdZUded<ded<dS) DeleteFilerrpathNrrrrr r"'s  r" _TypeAlias FileOperationfilenamerreturncCs8tj|\}}|d}|ddkr|d|S)N/__init__.)osr#splitextsplitpopjoin)r&_partsrrr _file_arg_to_module/s    r3caseDataDrivenTestCaseNonec+ s4tjj}jj}jjrtjj}nt j}jj }g}g}g}i}i} i} i} g} i} g}dNfdd }|d}| d|d d D]%dNfd d }j d vrj d us^Jtdj}||j }j dkry| tj j dvr| ||fqGj dkr| |t|tjfqGj dkr| ||fqGj dkrj d usJ|tjjj }t|dd}| ||d|fWd n1swYqGj dkrj d usJ|tjjj }t|dd}| ||d|fWd n 1swYqGj dkrZj d us+J|tjjj }t|dd}| ||d|fWd n 1sTwYqGtdj rj dkrjd n tj tdd }|dks|Jj d urtn ddj dD}|| |<qGtd j rЈj d!krd n tj td!d }|dksJj d urtn d"dj dD}|| |<qGtd#j r j d$krd n tj td$d }|dksJj d urgn d%d&j dD}|| |<qGj d'krRj d usJtd(j }|d ur,|d)j t|d*}|d*kr?|d+||||d }| |t|qGtd,j rCj d urcg}nj d}d-}|D]}| d.r|d/d0} | d1vr|d2|d0d }!zt!d3d4|!d5D}"Wnt"y||!d6Ynw| d7kr|"t#j$kr||d8t#j$t%j&|"k}qm| d9kr|"t#j$kr||d:t#j$d t|"krd;ksn|d<|!t%j&d t|"|"k}qm|rBd=d&jD}#tjj'd>krj(rd?d&|#D}#j d@ks'j dAkr*|#}ntj td@d }|d ks._case_failr__main__Ncs4jjd}tjjd|d|dddS)Nr8r9Fr:)r?r<r=r>)r7 item_abs_liner4itemrr _item_failYs$z#parse_test_case.._item_fail>r>fixtureoutfile outfile-re rI>r>rIrKrJbuiltinsutf8encodingz builtins.pyitypingz typing.pyi _typeshedz _typeshed.pyiz stale[0-9]*$stalecSh|]}|qSrstrip.0trrr z"parse_test_case..,zrechecked[0-9]*$ recheckedcSrTrrUrWrrr rZr[ztargets[0-9]*$targetscSsg|]}|qSrrUrWrrr r[z#parse_test_case..deletez(.*)\.([0-9]+)$zInvalid delete section rDzCan't delete during step z out[0-9]*$Tversion >==>=z5Only >= and == version checks are currently supportedcss|]}t|VqdSN)int)rXxrrr sz"parse_test_case..r+z is not a valid python versionrez. always true since minimum runtime version is rdz/ always false since minimum runtime version is zFOnly minor or patch version checks are currently supported with "==": cSg|]}t|qSr)expand_variablesrXr?rrr r_r[\cSrkr) fix_win_pathrmrrr r_r[outout1F triggered zInvalid section header [z ] in case z*Required output section not found in case zStale modules after pass z& must be a subset of rechecked modulesmainzDuplicated filename z$. Did you include it multiple times?)r7rr'r ):parse_test_datadatanamesuite base_path native_sepr,r#r0 posixpathrequired_out_sectionappendidargrlr3recompilerstripSdirnamer>openreadmatchrglensetr.group setdefaultadd startswithtuple ValueErrorrPYTHON3_VERSIONsys version_infosepnormalize_outputkeysissubset expand_errorsinputoutputoutput_inline_startoutput2r? last_linefiles output_filesexpected_stale_modulesexpected_rechecked_modules deleted_pathsrrexpected_fine_grained_targets test_modules)+r4 test_itemsrzr0out_section_missingrrrrr stale_modulesrechecked_modulesrrr^rrA first_itemrHcontentsr#mpathfsrc_pathpassnummodules reprocessedmnumfullargs version_checkr compare_op version_strra tmp_output section_strrr file_path seen_filesr>r1rrFr parse_test_case7sD            &( &( &&                         rcseZdZUdZded<ded<ded<ded<d ed <d Zd Zd ed<ded<ded<ded<d ed<ded<ded<ded<ded<ded<d?fd*d+ Zd@d,d-Zd@d.d/Z d@d0d1Z dAd3d4Z 5dBdCd:d;Z dDd=d>Z ZS)Er5zNHolds parsed data-driven test cases, and handles directory setup and teardown.DataSuiteCollectorparent list[str]rrrgrzdict[int, list[str]]rrtrzlist[tuple[str, str]]rrzdict[int, set[str]]rrrboolrrz$list[tuple[str, str | Pattern[str]]]rrrrry DataSuiter>rrx writescache only_whenplatform str | Noneskipxfailrwr?r'r6c s|t||||_||_||_||_||_|dkrtjdks(|dkr*tjdkr*d} | |_ | |_ | |_ | |_ d|_ d|_dS)Nwindowswin32posixT)superr*ryr>rrrrrrrrwr?old_cwdtmpdir) selfrryr>rxrrrrrrrwr? __class__rr r*s zDataDrivenTestCase.__init__c Cs|jrtn |jr|tjj|t}|dusJd|}|z| |WdSt yz|j dd}|ry|j dusFJtj|tj|j j}td|jd|tj|sq|jsiJtj|j|}t|j j|w)NzShould not happen--save-failures-tozCopying data from test z to )rr<r add_markermark getparentrobjsetuprun_case Exceptionconfig getoptionrr,r#r0basenamerxprintisabsrshutilcopytree)rrrysave_dir target_dirrrr runtest5s,     zDataDrivenTestCase.runtestc s|t|dt|_tjdd|_t|jjt t i|j D]]\}}t d|}|rVt|d}|dks;Jt dd|}t|}t|||}|g|q"tj|}tj|d d t|d d d  } | |Wdn1szwYq"|jD]\}} |dksJ| D]}t|}|gt||qqrtnd} fddtd| dD|_dS)Nr@z mypy-test-)prefixz .*\.([0-9]+)$rCrDz \.[0-9]+$rtT)exist_okwrNrOcsg|]}|gqSr)get)rXrstepsrr r_psz,DataDrivenTestCase.setup..) rr,getcwdrtempfileTemporaryDirectoryrchdirrxmkdirrrrrrgrsubmodule_from_pathrrr~r#rmakedirsrwriteritemsr"maxranger) rr#rrrrr operationdirrpathsmax_steprrr rNs:         "zDataDrivenTestCase.setupcCsR|jdur t|j|jdur!z|jWn ty Ynwd|_d|_dSrf)rr,rrcleanupOSErrorrrrr teardownrs     zDataDrivenTestCase.teardowntuple[str, int, str]cCs|j|j|jfSrf)r>r?rxrrrr reportinfo}zDataDrivenTestCase.reportinfoNexcinfo#pytest.ExceptionInfo[BaseException]style Any | NonecCslt|jtr |}nt|jtjjr|jjs|}n |j ||_ |j dd}d|j d|j d|S)Nshort)rzdata: r8z: ) isinstancevalue SystemExitexconlyr<r=rr;r_traceback_filter tracebackgetreprr>r?)rrrexcreprrrr repr_failures    zDataDrivenTestCase.repr_failurelist[list[FileOperation]]cCs|jS)axReturn a list of descriptions of file operations for each incremental step. The first list item corresponds to the first incremental step, the second for the second step, etc. Each operation can either be a file modification/creation (UpdateFile) or deletion (DeleteFile). Defaults to having two steps if there aern't any operations. rrrrr find_stepss zDataDrivenTestCase.find_steps)rrryrr>rrxrrrrrrrrrrrrrrwrr?rgr'r6r'r6)r'rrf)rrrrr'r)r'r)rrr__doc__rr>r?r*rrrrr r __classcell__rrrr r5s6  !  $  r#cCs<tdd|}td|}|d=d|}tdd|}|S)Nz\.pyi?$rtz[/\\]rr+z \.__init__$)rrr.r0)r#r2rrrr rs   rc@sHeZdZUdZded<ded<ded<ded <ded <edd d ZdS)TestItemzVParsed test caseitem. An item is of the form [id arg] .. data .. rrrrrrwrgr?end_liner'cCs|j|jt|jSrf)rr?rrwrrrr trimmed_newlinesszTestItem.trimmed_newlinesN)r'rg)rrrrrpropertyrrrrr rs rraw_datarxlist[TestItem]c Csdd|dg|d}g}g}d}d}d}d}|t|kr||} ||drm| drm|rHt|}t|}|t||||d||}| dd }d}d |vrj|| d dd}|d| d }g}n4||d r|||ddn!||d s|||n||d r|||dd|d7}|t|ks|rt|}t|}|t||||d|d|S)z>Parse a list of lines that represent a sequence of test items.rtz[case ]rLNr[rCr)rsz\[z--z----rD) r.rrVrendswithcollapse_line_continuation strip_listr~rindex) rrxlinesretrwrrii0srrr rvsD    rvlrcCsNg}|D] }|tdd|q|r%|ddkr%||r%|ddks|S)zReturn a stripped copy of l. Strip whitespace at the end of all lines, and strip all empty lines from the end of the array. z\s+$rtr))r~rrr/)r$rr#rrr rsrc CsXg}d}|D]#}tdd|}|r|dtdd|7<n|||d}q|S)NFz\\$rtr)z^ +rn)rrr~r)r$r%contr#ssrrr rs  rr#cCs |dtS)Nz)replaceroot_dir)r#rrr rl s rlrrfnamc Cstt|D]s}||dddD]e}td|}|rx|ddkr)d}n|ddkr3d}n |dd kr\d+):)? (?P.*)$EerrorNnoteWwarningcolmessagez\##r8r9) rrr.rsearchrVrr(r~) rrr*r!possible_err_commentrseverityr1r2rrr rs*   &*rr?cCsP|ttdd}td|}|s|S|\}}}d|dd|p%d|S)z`Changes Windows paths to Linux paths in error messages. E.g. foo\bar.py -> foo/bar.py. rnr(z^([\S/]+):(\d+:)?(\s+.*)z{}:{}{}rt)r(r)rrgroupsformat)r?rr&linenor2rrr ro,s  rocCsLtd|}|s |Sd|d|d|ddd||ddS)zChanges filename paths to Linux paths in Cobertura output files. E.g. filename="pkg\subpkg\a.py" -> filename="pkg/subpkg/a.py". z&[a-zA-Z_0-9]+)(?P-writescache)?(?P-only_when_cache|-only_when_nocache)?(?P-skip_path_normalization)?(-(?Pposix|windows))?(?P-skip)?(?P-xfail)?rDataFileCollectorryrr>Iterator[DataDrivenTestCase]c csBt|dd }|}Wdn1swYtjd|tjtjBd}t|}t|dd}t }|D]e} t|}t | } | sMt d| | d } | |vr`t d | |j|tj|||t| |jt| d | d | d t| dt| d| d ||d V||dd7}|| hq9dS)zIterate over raw test cases in file, at collection time, ignoring sub items. The collection phase is slow, so any heavy processing should be deferred to after uninteresting tests are filtered (when using -k PATTERN switch). zutf-8rONz^\[case ([^]+)]+)\][ \t]*$\n)flagsrLrCzInvalid testcase id rxz1Found a duplicate test name "{}" in {} on line {}rrrrrskip_path_normalization) rryr>rxrrrrrrrwr?)rrrr.DOTALL MULTILINEiternextrDr_case_name_pattern fullmatch RuntimeErrorrr8rxr5rUadd_test_name_suffixtest_name_suffixrupdate) rryr>rrwcases cases_iterline_no test_namescase_idrrxrrr split_test_casessJ        rjc@seZdZdddZdS)rr'Iterator[DataFileCollector]ccsH|j}tj|jsJd|jd|jD] }tj||dVqdS)zNCalled by pytest on each of the object returned from pytest_pycollect_makeitemzTest data prefix (z) not set correctlyrRN)rr,r#isdir data_prefixrrWrU)rry data_filerrr collects  zDataSuiteCollector.collectN)r'rk)rrrrorrrr rsrc@s&eZdZUded<ded<ded<dS) DataFileFixrgr9 end_linenorrNrrrrr rpr!rpcsreZdZUdZded<ded<edfd d Zdd d Zdfdd Zdfdd Z dddZ dddZ Z S)rWzvRepresents a single `.test` data driven test file. More context: https://github.com/python/mypy/issues/11662 rrzlist[DataFileFix]_fixesrxrr'cs"tj||d}t|tsJ|S)N)rx)rrUrrW)clsrrxrPrrr rUszDataFileCollector.from_parentrXccs0t||jjtj|jjj|jdEdHdS)N)rryr>)rjrrr,r#r0rmrxrrrr ros zDataFileCollector.collectr6cstg|_dSrf)rrrrrrrr rs  zDataFileCollector.setupcst|dSrf)rr _apply_fixesrrrr rs  zDataFileCollector.teardownfixrpcCs|j|dSrf)rrr~)rrurrr enqueue_fixrzDataFileCollector.enqueue_fixcCsn|jsdSt|jjj|j}|d}t|jddD]}|j ||j d|j d<q| d |dS)NrLT)reverserC)rrrrrrmrx read_textr.sortedrr9rq write_textr0)r data_pathrrurrr rtszDataFileCollector._apply_fixes)rrrxrr'rW)r'rXr)rurpr'r6) rrrrr classmethodrUrorrrvrtrrrrr rWs   suffixcCs<td|}|r|d}|dt| ||S||S)Nz-[-A-Za-z0-9]+$r)rr4rr)rxr}r magic_suffixrrr rbs  rbtestcasercCsd|jvp d|jvS)N incremental)rxlowerr>)rrrr is_incrementalsrcCsDtdd|jDr dS|jD]\}}tj|drdSqdS)Ncss|] }td|VqdS)z # flags[2-9]:N)rrrmrrr risz#has_stable_flags..Fz mypy.ini.T)anyrrr,r#rr)rr&rrrr has_stable_flagssrc@sBeZdZUded<eZeZdZdZ dZ dddZ e dd d Z d S)rrrFrtr'r6cCsdS)zSetup fixtures (ad-hoc)Nrrrrr r2szDataSuite.setuprr5cCstrf)NotImplementedError)rrrrr r5szDataSuite.run_caseNr)rr5r'r6)rrrrrrzrrmr}r{rcrrrrrrr r!s  )r&rr'r)r4r5r'r6)r#rr'r)rrrxrr'r)r$rr'r)r#rr'r)rrrrr*rr'r6)r?rr'r)r=rr'r6)rrLr'r6)rPrrxrrrQr'r)rrWryrr>rr'rX)rxrr}rr'r)rr5r'r)Gr __future__rr,os.pathr|rrrrabcr dataclassesrpathlibrrQrrrr r r r typing_extensionsr r$r<r>rmypy.test.configrrrr#normpathr)rrrr"r%r3rItemr5rrrvrrrlrror<rJhookimplrOrVrr_rjClassrrp CollectorrWrbrrrrrrr sh    $     = ,   -         / /