6 f$2dZddlmZddlZddlZddlZddlZddlZddlm Z m Z ddl m Z ddl mZddlmZddlmZdd lmZmZdd lmZdd lmZdd lmZmZdd lmZddlm Z ddl!m"Z"ddl#m$Z$ddl%m&Z&ddl'm(Z(ddl)m*Z*ej+,e*dZ-ej+,e*dZ.GddeZ/dFdZ0ej1dGd"Z2dHd$Z3 dIdJd*Z4 dIdKd,Z5dLd.Z6 dMdNd4Z7dOd5Z8dPd7Z9dQd9Z:dRd;Z;dSd>ZdVdEZ?dS)WzHelpers for writing tests) annotationsN)CallableIterator)build) CompileError)Options) test_temp_dir)DataDrivenTestCase DataSuite)assert_string_arrays_equal)assert_func_ir_valid)IS_32_BIT_PLATFORM PLATFORM_SIZE)Errors)FuncIR)ModuleIR)build_ir)Mapper)CompilerOptions)test_data_prefixzfixtures/ir.pyzfixtures/testutil.pyc"eZdZUgZded<eZdS)MypycDataSuite list[str]filesN)__name__ __module__ __qualname__r__annotations__r data_prefixI/var/www/equiseq/venv/lib/python3.11/site-packages/mypyc/test/testutil.pyrr"s&E"KKKr!rfunc$Callable[[DataDrivenTestCase], None]pathstrreturncfdS)zDecorate a function that implements a data-driven test case to copy an alternative builtins module implementation in place before performing the test case. Clean up after executing the test case. c&t|SN) perform_test)testcaser#r%s r"z"builtins_wrapper../sLtX>>r!r )r#r%s``r"builtins_wrapperr.(s ? > > > >>r! builtins_pathr,r Iterator[None]c#K|jD],\}}tj|dkrd}nZ-tjtjt d}tj||d} dV|rtj |dSdS#|rtj |wwxYw)Nz builtins.pyiFT) rosr%basenameabspathjoinr shutilcopyfileremove)r/r,r%_default_builtinsbuiltinss r"use_custom_builtinsr<2s>  a 7  D ! !^ 3 3$  E 4 7??27<< ~#N#NOO x000   Ih        Ih     s B//CNonec,|jD],\}}tj|dkrd}n=-tjt d}t j||d}|||rtj|dSdS)Nz builtins.pyFT) rr2r%r3r5r r6r7r8)r#r/r,r%r9r:r;s r"r+r+Gs>  a 7  D ! !] 2 2$  E 3 7<< }== x000 DNNN (r! input_linesrcompiler_optionsCompilerOptions | None list[FuncIR]c,t||jSr*)build_ir_for_single_file2 functions)r?r@s r"build_ir_for_single_filerF\s %[2B C C MMr!rc d|}|ptd}t}d|_d|_d|_d|_|jpd|_d|_d|_ d|_ ddi|j d<tj dd|}tj |g|t }|jrt!|jt#|}t%|jdg|j|jt-ddi||}|jr!t!|t3|d }|jD]} t9| |S) N ) capi_versionT)rJmypyc__main__main)sourcesoptions alt_lib_pathr)r5rrshow_tracebackhide_error_codesuse_builtins_fixturesstrict_optionalpython_version export_types preserve_astsallow_empty_bodiesper_module_optionsr BuildSourcer errorsrrrrgraphtypesr num_errors new_messageslistvaluesrEr ) r?r@ program_textrRsourceresultr^modulesmodulefns r"rDrDbs99[))L(O?+O+O+OiiG!G#G$(G!"G-<FGG G!%G.5t_Gz*  vz< @ @F[&7 W W WF }*6=))) G__F j !"   D!"" G26..00111 '.."" # #A &F!!R    Mr!outputc^|j Jdtj|j|j}t |5}|}dddn #1swxYwYd|jd}| |dkrdS| |}|dz}|t|krN|| ds3|dz }|t|kr|| d3|||}| d} ||| dzd<|dgz|||<d|} t |d 5}t| | ddddS#1swxYwYdS) Nztest was not properly set upz[case ] z[out]rHw)file)old_cwdr2r%r5rropenread splitlinesnamecountindexlen startswithprint) r,rk testcase_pathf data_lines test_slug start_idxstop_idxtest out_startdatas r"update_testcase_outputrs6   ' ')G ' ' 'GLL!18=AAM m  +VVXX((** +++++++++++++++ *)))I ""a''  ++I2~H S__ $ $Z-A-L-LX-V-V $A  S__ $ $Z-A-L-LX-V-V $ i( )D 7##I"DQ%)RD[Jy!" 99Z D mS ! !Q ds$ 'A>>BBF""F&)F&actualmessageexpectedlist[str] | None formattedc d}||n|j}||kr+|jddrt||t |||d|jd|jddS)NTz --update-dataFz (z, line ))rkconfig getoptionrr rrline)r,rrrr__tracebackhide__expected_outputs r"assert_test_outputrs"*"6hhHOO&  X_%>%>PU%V%V x000G!U!Ux}!U!UX]!U!U!Ur!cg}|D]A}tjd|}|r(||dB|S)Nzdef ([_a-zA-Z0-9.*$]+)\(rn)rematchappendgroup)rressms r"get_func_namesrsR C ## H0! 4 4  # JJqwwqzz " " " Jr!acg}|D]f}|dr(|dsQ||g|S)zrReturn a copy of array with comments removed. Lines starting with '--' (but not with '---') are removed. z--z---)stripr{r)rrrs r"remove_comment_linesrsl A  7799   % % aggii.B.B5.I.I   HHQKKKK Hr!rc|}t|D]\}}td|dz|fzdS)Nz%-4d %srn)rv enumerater|)rlinesirs r"print_with_line_numbersrsR LLNNEU##))4 i1q5$-'(((())r!textc6td|zdzdzdS)Nz====================  z====================)r|)rs r"headingrs% .4 # % 011111r!cfileslist[list[tuple[str, str]]]ctd|D],}|D]'\}}td|dt|(-tddS)Nz Generated Cz== z ==zEnd C)rr|r)rrcfilectexts r"show_crsx M++! + +LE5 """" # # # #E * * * * + Gr!dirdeltaintctj|D]_\}}}|D]V}tj||}tj|j|z}tj|||fW`dS)N)times)r2walkr%r5statst_mtimeutime)rrdirpathr9 filenamesrwr% new_mtimes r"fudge_dir_mtimesrs!#99I 9 9D7<<..D .6I HT)Y!7 8 8 8 8 8 999r!chg}|D]}|d}|dkrz||dd}t|dd}tt|z}|||||||S)z2Replace WORDSIZE with platform specific word sizes WORD_SIZENr )findsplitrr&rrreplace)rrgrryword_size_tokenn replace_strs r"replace_word_sizers F    +&& B;;"566l002215OOBCC())Ama/00K MM$,, DD E E E E MM$     Mr!rwcdd|vr trdSd|vr tsdStd|vd}tjd|}|rVt |dt |d f|_|j|_nd |vsd |vr Jd ||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_8* (or for any Python version): Use Python 3.8+ C API features (default: lowest supported version) *StripAssert*: Don't generate code for assert statements _64bitN_32bit StripAssertrI) strip_assertsrLz_python([3-9]+)_([0-9]+)(_|\b)rn_py_PythonFz,Invalid _py* suffix (should be _pythonX_Y): )rrrsearchrrrLrX)rwrRrs r"%infer_ir_build_options_from_test_namers"4.t4 2tMT,APVWWWG 3T::AL #AGGAJJQWWQZZA!(!5 $)t++KKTKKKKu Nr!)r#r$r%r&r'r$)r/r&r,r r'r0)r#r$r/r&r,r r'r=r*)r?rr@rAr'rB)r?rr@rAr'r)r,r rkrr'r=)NN) r,r rrrr&rrrrr'r=)rrr'r)rrr'r)rr&r'r=)rr&r'r=)rrr'r=)rr&rrr'r=)rrr'r)rwr&r'rA)@__doc__ __future__r contextlibr2os.pathrr6typingrrmypyr mypy.errorsr mypy.optionsrmypy.test.configr mypy.test.datar r mypy.test.helpersr mypyc.analysis.ircheckr mypyc.commonrr mypyc.errorsrmypyc.ir.func_irrmypyc.ir.module_irrmypyc.irbuild.mainrmypyc.irbuild.mapperr mypyc.optionsrmypyc.test.configrr%r5ICODE_GEN_BUILTINS TESTUTIL_PATHrr.contextmanagerr<r+rFrDrrrrrrrrrrr r!r"rsL""""""  %%%%%%%%$$$$$$ ******88888888888888777777::::::::######''''''''''''''''''))))))......W\\"24DEE -/EFF #####Y### ????     (,HLNNNNNHL)))))X@"&"& $     )))) 22229999     r!