pPj*dZddlmZddlZddlZddlZddlZddlZddlm Z m Z ddl m Z ddl mZddlmZddlmZmZdd lmZdd lmZdd lmZmZdd lmZdd lmZddlm Z ddl!m"Z"m#Z#ddl$m%Z%ddl&m'Z'ddl(m)Z)ddl*m+Z+ddl,m-Z-ddl.m/Z/ddl0m1Z1ejdjge1dZ4ejdjge1dZ5GddeZ6 d+dZ7ejpd,dZ9 d-dZ: d. d/dZ; d. d0dZd4d"Z?d5d#Z@d6d$ZAd7d%ZBd8d&ZCd9d'ZDd:d(ZEd;d)ZFdz"builtins_wrapper..2sLtX>r%r$)r,r-s``r&builtins_wrapperr/+s  ?>r%c#K|jD]+\}}tjj|dk(s)d}nZtjj tjj t d}tj||d} d|rtjyy#|rtjwwxYww)Nz builtins.pyiFT) rosr-basenameabspathjoinr shutilcopyfileremove) builtins_pathr+r-_default_builtinsbuiltinss r&use_custom_builtinsr<5s>> a 77  D !^ 3$   77??277<< ~#NO x0   IIh    IIh  s#5CACB4C4CCc0|jD]+\}}tjj|dk(s)d}n=tjj t d}t j||d}|||rtjyy)Nz builtins.pyFT) rr1r-r2r4r r5r6r7)r,r8r+r-r9r:r;s r&r*r*Js>> a 77  D !] 2$   77<< }= x0 N (r%c4t||djS)Nr)build_ir_for_single_file2 functions) input_linescompiler_optionss r&build_ir_for_single_filerC_s %[2B CA F P PPr%cdj|}tjd|tj}|xs t d}|r0|j dj }t|d\}}n t}d |_ d |_ d |_ d |_ |jxsd|_ d |_d |_d |_d |_d |_d |_d d i|j(d <t+j,d d |}t+j*|g|t. }|j0j2j5|j6rt9|j6t;|} t=d di} t?|j@d g|jB|jD| || } | jFrt9| jItK| jMd} | jND] } tQ| |jB| jRjT}|J| ||jD| fS)N z# flags: (.*)$)flags ) capi_versionF)require_targetsTmypyc__main__main)sourcesoptions alt_lib_pathr)+r4research MULTILINErgroupsplitrr show_tracebackhide_error_codesuse_builtins_fixturesstrict_optionalpython_version export_types preserve_astsallow_empty_bodies strict_bytesdisable_bytearray_promotiondisable_memoryview_promotionper_module_optionsr BuildSourcer manager metastorecloseerrorsrrrrrgraphtypes num_errors new_messageslistvaluesr@rfullnametree)rArB program_textrF flag_listr9rQsourceresultrhmappermodulesmodulefnrps r&r?r?es99[)L II& BLL IE(P?+P KKN((* $YF 7)!G#G$(G!"G-<<GGG G!%GG*.G'+/G(.5t_Gz*   vz< @F[[&7 WF NN""$ }}6==)) G_F Z& 'F j !"FLL&,,HXZ`G6..011 '.." #A &F!R ! << ( - -D    4v --r%c|jJdtjj|j|j}t |5}|j j}dddd|jd}j|dk7ry|j|}|dz}|t|kr<||jds(|dz }|t|kr||jds(|||}|jd} ||| dzd|dgz|||dj|} t |d 5}t| | dddy#1swYxYw#1swYyxYw) Nztest was not properly set upz[case ]rK z[out]rEw)file)old_cwdr1r-r4r~openread splitlinesnamecountindexlen startswithprint) r+output testcase_pathf data_lines test_slug start_idxstop_idxtest out_startdatas r&update_testcase_outputrsq    'G)GG 'GGLL!1!18==AM m +VVX((* + q)I "a'  +I2~H S_ $Z-A-L-LX-VA  S_ $Z-A-L-LX-V i )D 7#I"DQ%)RD[Jy" 99Z D mS !Q d)++(sEE&E#&E/c d}||n |j}||k7r(|jjddr t||t |||d|j d|j dy)NTz --update-dataFz (z, line ))rconfig getoptionrrr~line)r+actualmessageexpected formatted__tracebackhide__expected_outputs r&assert_test_outputrsk"*"6hHOOO& X__%>%>PU%Vx0G9Bx}}oWX]]OST!Ur%cg}|D];}tjd|}|s|j|jd=|S)Nzdef ([_a-zA-Z0-9.*$]+)\(rK)rSmatchappendrV)rressms r&get_func_namesrsE C # HH0! 4 JJqwwqz "# Jr%cg}|D]R}|jjdr |jjdsB|j|T|S)zrReturn a copy of array with comments removed. Lines starting with '--' (but not with '---') are removed. z--z---)striprr)arrs r&remove_comment_linesrsT A  779   %aggi.B.B5.I  HHQK  Hr%cp|j}t|D]\}}td|dz|fzy)Nz%-4d %srK)r enumerater)rlinesirs r&print_with_line_numbersrs: LLNEU#)4 i1q5$-'()r%c,td|zdzdzy)Nz====================  z====================)r)texts r&headingrs .4 # % 01r%ctd|D]&}|D]\}}td|dt|!(tdy)Nz Generated Cz== z ==zEnd C)rrr)cfilesrVcfilectexts r&show_crsO M+! +LE5 Cwc" # #E * ++ Gr%c(tj|D]z\}}}|D]o}|jdrtjj ||}tj |j |z}tj|||fq|y)N)z.soz.pydz.oz.obj)times)r1walkendswithr-r4statst_mtimeutime)dirdeltadirpathr9 filenamesrr- new_mtimes r&fudge_dir_mtimesrs "$9I 9D}}:;77<<.D ..6I HHT)Y!7 8  99r%cg}|D]}|jd}|dk7rX||djd}t|dd}tt|z}|j |j ||q|j ||S)z2Replace WORDSIZE with platform specific word sizes WORD_SIZENrrI)findrWintstrrrreplace)rrtrrword_size_tokenn replace_strs r&replace_word_sizers F   +& B;"56l00215OOBC()Ama/0K MM$,, D E MM$    Mr%cd|vrtryd|vrtsytd|vdd}tjd|}|r\t |j d t |j d f}|dk\s Jd |||_|j |_nd |vsd |vr Jd|t|drd|_ |S)aLook for magic substrings in test case name to set compiler options. Return None if the test case should be skipped (always pass). Supported naming conventions: *_64bit*: Run test case only on 64-bit platforms *_32bit*: Run test caseonly on 32-bit platforms *_python3_10* (or for any Python version): Use Python 3.10+ C API features (default: lowest supported version) *StripAssert*: Don't generate code for assert statements _64bitN_32bit StripAssertrGT) strip_assertsrJstrict_traceback_checksz_python([0-9]+)_([0-9]+)(_|\b)rKzUnsupported _python* suffix: _py_Pythonz,Invalid _py* suffix (should be _pythonX_Y): experimental) rrrSrTrrVrJr\has_test_name_tagexperimental_features)rrQrversions r&%infer_ir_build_options_from_test_namers"4.4 2#t+'[_G 3T:Aqwwqz?C O4'!I%B4&#II!&!(!5!5 $)t+KDTFKKu~.(,% Nr%c`tjdtj|d|duS)zCheck if a test case name contains a tag token like ``_experimental``. A tag matches if it appears as a full underscore-delimited token: ``foo_tag_bar`` or ``foo_tag``. z(?:^|_)z(?:_|$)N)rSrTescape)rtags r&rr5s, 99 #/w7 >d JJr%)r,$Callable[[DataDrivenTestCase], None]r-rreturnr)r8rr+r rzIterator[None])r,rr8rr+r rNoner))rArrBCompilerOptions | Nonerz list[FuncIR])rArrBrrz9tuple[ModuleIR, MypyFile, dict[Expression, Type], Mapper])r+r rrrr)NN) r+r rrrrrlist[str] | Nonerrrr)rrrr)rrrr)rrrr)rrrr)rzlist[list[tuple[str, str]]]rr)rrrrrr)rrrr)rrrr)rrrrrbool)H__doc__ __future__r contextlibr1os.pathrSr5collections.abcrrmypyr mypy.errorsr mypy.mainr mypy.nodesr r mypy.optionsr mypy.test.configr mypy.test.datar rmypy.test.helpersr mypy.typesrmypyc.analysis.ircheckr mypyc.commonrr mypyc.errorsrmypyc.ir.func_irrmypyc.ir.module_irrmypyc.irbuild.mainrmypyc.irbuild.mapperr mypyc.optionsrmypyc.test.configrr-r4ICODE_GEN_BUILTINS TESTUTIL_PATHrr/contextmanagerr<r*rCr?rrrrrrrrrrrr$r%r&rs"  .$%+ *887:#''').WW\\"24DE -/EF #Y# ? .?69?)?   ( .?BN` ,HLQQ.DQQHL1.1..D1.>1.h@"&"&        $  ) 2 9  #LKr%