X-Ph%VdZddlmZddlZddlZddlZddlZddlZddlm Z ddl m Z ddl m Z ddlmZddlmZmZdd lmZdd lmZdd lmZmZdd lmZdd lmZddlmZddl m!Z!m"Z"ddl#m$Z$ddl%m&Z&ddl'm(Z(ddl)m*Z*ddl+m,Z,ddl-m.Z.ddl/m0Z0ej12e0dZ3ej12e0dZ4GddeZ5dId Z6ej7dJd%Z8dKd'Z9 dLdMd-Z: dLdNd/Z;dOd1Z< dPdQd7Z=dRd8Z>dSd:Z?dTd<Z@dUd>ZAdVdAZBdWdEZCdXdFZDdYdHZEdS)ZzHelpers for writing tests) annotationsN)Iterator)Callable)build) CompileError) ExpressionMypyFile)Options) test_temp_dir)DataDrivenTestCase DataSuite)assert_string_arrays_equal)Type)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_prefixS/var/www/html/test/jupyter/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..2sLtX>>r$r#)r&r(s``r%builtins_wrapperr1+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)r2r/r(_default_builtinsbuiltinss r%use_custom_builtinsr?5s>  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) rr5r(r6r8r r9r:r;)r&r2r/r(r<r=r>s r%r.r.Js>  a 7  D ! !] 2 2$  E 3 7<< }== x000 DNNN (r$ input_linesrcompiler_optionsCompilerOptions | None list[FuncIR]c8t||djS)Nr)build_ir_for_single_file2 functions)rBrCs r%build_ir_for_single_filerI_s %[2B C CA F PPr$9tuple[ModuleIR, MypyFile, dict[Expression, Type], Mapper]cNd|}|ptd}t}d|_d|_d|_d|_|jpd|_d|_d|_ d|_ ddi|j d<tj dd|}tj |g|t }|jrt!|jt#|}t%ddi}t'|jdg|j|j|||}|jr!t!|t3|d } | jD]} t9| |j| jj} | J| | |j|fS) N  ) capi_versionT)rNmypyc__main__main)sourcesoptions alt_lib_pathr)r8rr show_tracebackhide_error_codesuse_builtins_fixturesstrict_optionalpython_version export_types preserve_astsallow_empty_bodiesper_module_optionsr BuildSourcer errorsrrrrrgraphtypes num_errors new_messageslistvaluesrHrfullnametree) rBrC program_textrVsourceresultrbmappermodulesmodulefnrjs r%rGrGes99[))L(O?+O+O+OiiG!G#G$(G!"G-<FGG G!%G.5t_Gz*  vz< @ @F[&7 W W WF }*6=))) G__F Z& ' 'F j !"FL&,HXZ`G26..00111 '.."" # #A &F!!R    < ( -D     4v --r$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]rLw)file)old_cwdr5r(r8ryopenread splitlinesnamecountindexlen startswithprint) r/rr 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 ))rrconfig getoptionrrryline)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.*$]+)\(ru)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---)striprr)rrrs r%remove_comment_linesrsl A  7799   % % aggii.B.B5.I.I   HHQKKKK Hr$rc|}t|D]\}}td|dz|fzdS)Nz%-4d %sru)r} 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)rrr)rrcfilectexts r%show_crsx M++! + +LE5 """" # # # #E * * * * + Gr$dirdeltaintctj|D]_\}}}|D]V}tj||}tj|j|z}tj|||fW`dS)N)times)r5walkr(r8statst_mtimeutime)rrdirpathr< filenamesr~r( 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)rrmrrword_size_tokenn replace_strs r%replace_word_sizers F    +&& B;;"566l002215OOBCC())Ama/00K MM$,, DD E E E E MM$     Mr$r~cdd|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 StripAssertrM) strip_assertsrPz_python([3-9]+)_([0-9]+)(_|\b)ru_py_PythonFz,Invalid _py* suffix (should be _pythonX_Y): )rrrsearchrrrPr\)r~rVrs r%%infer_ir_build_options_from_test_namers"4.t4 2tMT,APVWWWG 3T::AL #AGGAJJQWWQZZA!(!5 $)t++KKTKKKKK Nr$)r&r'r(r)r*r')r2r)r/r r*r3)r&r'r2r)r/r r*r@r-)rBrrCrDr*rE)rBrrCrDr*rJ)r/r rrrr*r@)NN) r/r rrrr)rrrrr*r@)rrr*r)rrr*r)rr)r*r@)rr)r*r@)rrr*r@)rr)rrr*r@)rrr*r)r~r)r*rD)F__doc__ __future__r contextlibr5os.pathrr9collections.abcrtypingrmypyr mypy.errorsr mypy.nodesrr mypy.optionsr mypy.test.configr mypy.test.datar r mypy.test.helpersr mypy.typesrmypyc.analysis.ircheckr mypyc.commonrr mypyc.errorsrmypyc.ir.func_irrmypyc.ir.module_irrmypyc.irbuild.mainrmypyc.irbuild.mapperr mypyc.optionsrmypyc.test.configrr(r8ICODE_GEN_BUILTINS TESTUTIL_PATHrr1contextmanagerr?r.rIrGrrrrrrrrrrr#r$r%rs""""""  $$$$$$$$$$$$++++++++ ******88888888888888777777::::::::######''''''''''''''''''))))))......W\\"24DEE -/EFF #####Y### ????     (,HLQQQQQHL'.'.'.'.'.T@"&"& $     )))) 22229999     r$