o ˜…›i61ã@s”ddlmZmZmZddlmZmZddlmZm Z m Z ddl Z ddl Z ddl Z ddlZddlmZiZded<ded <ed ed <iZd ed <ed ed <d ed<iZgd ¢ed<gd¢ed<gd¢ed<gd¢ed<gd¢Zdd„Zd1dd„Zdd„Zdd„ZgZgZeD]\ZZeeZeeeƒrœe eef¡q‡e eef¡q‡dd „Zd!d"„Z d#d$„Z!d%d&„Z"d'd(„Z#d)d*„Z$d+d,„Z%d-d.„Z&d/d0„Z'dS)2é)ÚxÚyÚz)ÚIS_WASMÚskip)ÚcodegenÚ make_routineÚget_code_generatorN)ÚPathzy program main include "codegen.h" integer :: result; result = 0 %(statements)s call exit(result) end program ÚF95z #include "codegen.h" #include #include int main() { int result = 0; %(statements)s return result; } ÚC89ÚC99z™ if (fabs(%(call)s)>%(threshold)s) { printf("Numerical validation failed: %(call)s=%%e threshold=%(threshold)s\n", %(call)s); result = -1; } zÁ if (abs(%(call)s)>%(threshold)s) then write(6,"('Numerical validation failed:')") write(6,"('%(call)s=',e15.5,'threshold=',e15.5)") %(call)s, %(threshold)s result = -1; end if )zcc -c codegen.c -o codegen.ozcc -c main.c -o main.oz#cc main.o codegen.o -lm -o test.exeÚcc)z$gfortran -c codegen.f90 -o codegen.oz6gfortran -ffree-line-length-none -c main.f90 -o main.oz%gfortran main.o codegen.o -o test.exeÚgfortran)zg95 -c codegen.f90 -o codegen.oz1g95 -ffree-line-length-huge -c main.f90 -o main.oz g95 main.o codegen.o -o test.exeÚg95)z!ifort -c codegen.f90 -o codegen.ozifort -c main.f90 -o main.oz"ifort main.o codegen.o -o test.exeÚifort)©r r©r r©r r©r r©r rcCsrtrdSttjdƒ%}|D]}tj||dtjd}|dkr&WdƒdSq WdƒdS1s2wYdS)z>Run a series of commands and only return True if all ran fine.FÚwT)ÚstdoutÚshellÚstderrrN)rÚopenÚosÚdevnullÚ subprocessÚcallÚSTDOUT)ÚcommandsÚnullÚcommandÚretcode©r%úc/sda-disk/www/egybert/egybert_env/lib/python3.10/site-packages/sympy/external/tests/test_codegen.pyÚtry_runts  ÿûý ÿúr'TcCs2| ¡}|tvs J‚|tvsJ‚t dd¡ ¡}|dvr tdƒ‚t d|d|¡}t  ¡}t  |¡|r>t ||ddd n t |dƒ} | j |ddd g} |D]/\} } } }d | d  d d „| Dƒ¡| f}|dkrrt|ƒ}tt|ƒƒ}|  t|||dœ¡qO|dkr†d}n| d¡rŽd}ntd|ƒ‚t|ƒ t|dd | ¡i¡t|ƒ}|r°tdgƒ}nd}|dks¾|dkrí|rí|rídd„}|dƒ|dƒ|dƒ|dƒ|dƒ|dƒ|dƒ|d ƒt  |¡t |¡ntd!|tjd"t  |¡|s Jd#|d$ |¡fƒ‚|sJd%|d$ |¡fƒ‚d&S)'a|A driver for the codegen tests. This driver assumes that a compiler ifort is present in the PATH and that ifort is (at least) a Fortran 90 compiler. The generated code is written in a temporary directory, together with a main program that validates the generated code. The test passes when the compilation and the validation run correctly. ÚSYMPY_TEST_CLEAN_TEMPÚalways)r)ÚsuccessÚneverzSSYMPY_TEST_CLEAN_TEMP must be one of the following: 'always', 'success' or 'never'.z_sympy_%s_testz%s_rT)Úto_filesz %s(%s)-(%s)ú,css|]}t|ƒVqdS©N)Ústr)Ú.0Úargr%r%r&Ú ©s€zrun_test..r )rÚ thresholdzmain.f90ÚCzmain.cz2FIXME: filename extension unknown for language: %sÚ statementsÚz ./test.exeFr*cSstj |¡r t |¡dSdSr.)rÚpathÚisfileÚremove)Úfilenamer%r%r&Ú safe_removeÈs ÿzrun_test..safe_removez codegen.f90z codegen.cz codegen.hz codegen.ozmain.oztest.exezTEST NOT REMOVED: %s)Úfilez"failed to compile %s code with: %sÚ z"failed to execute %s code from: %sN)ÚupperÚ main_templateÚnumerical_test_templaterÚgetenvÚlowerÚ ValueErrorÚtempfileÚmkdtempÚgetcwdÚchdirrr ÚwriteÚjoinÚfortranize_double_constantsr/ÚappendÚ startswithÚNotImplementedErrorr Ú write_textr'ÚrmdirÚprintÚsysr)ÚlabelÚroutinesÚnumerical_testsÚlanguager!ÚfriendlyÚcleanÚworkÚoldworkÚcode_genÚ test_stringsÚfn_nameÚargsÚexpectedr3Ú call_stringÚf_nameÚcompiledÚexecutedr;r%r%r&Úrun_testsv    ÿ   þ ÿÿ      ÿ   ÿrccsLddl‰ˆ d¡}ˆ d¡}‡fdd„}dd„}| ||¡}| ||¡}|S) z; Replaces every literal float with literal doubles rNz\d+(\.)?\d*[eE]-?\d+z\d+\.\d*(?!\d*d)csˆ dd| d¡¡S)Nz[eE]Údr)ÚsubÚgroup©Úmatchobj©Úrer%r&Úsubs_expèsz-fortranize_double_constants..subs_expcSsd| d¡S)Nz%sd0r)rfrgr%r%r&Ú subs_floatësz/fortranize_double_constants..subs_float)rjÚcompilere)Ú code_stringÚ pattern_expÚ pattern_floatrkrlr%rir&rJàs     rJcCsDtdtƒ}ddg}ztd|g|||ddWdSty!YdSw)NÚtest)rq)çð?rrçV瞯Ò<)rq)çð¿rtrsÚ is_feasibleF)rVT)rrrcÚAssertionError)rUr!ÚroutinerTr%r%r&ruôs þÿ ÿrucCódtvr tdƒdSdS)Nrz*`cc' command didn't work as expected (C89)©Úinvalid_lang_compilersrr%r%r%r&Ú test_C89_cc ó ÿr{cCrx)Nrz*`cc' command didn't work as expected (C99)ryr%r%r%r&Ú test_C99_ccr|r}cCrx)Nrz'`ifort' command didn't work as expectedryr%r%r%r&Útest_F95_ifortr|r~cCrx)Nrz*`gfortran' command didn't work as expectedryr%r%r%r&Útest_F95_gfortranr|rcCrx)Nrz%`g95' command didn't work as expectedryr%r%r%r&Ú test_F95_g95!r|r€cCs<ddg}dtttfg}tD] \}}td||||ƒqdS)N)rq)rrg@g@g5@rs)rq)rtg@çÀrrsrqÚ basic_codegen)rrrÚvalid_lang_commandsrc)rTÚ name_exprÚlangr!r%r%r&Útest_basic_codegen(sþ ÿr†cCsˆddlm}ddlm}ddlm}ddlm}m}m }ddl m }m }ddl m}ddlm} m} m} m} m} m}d ttƒfd | tƒfd | tƒfd | tƒfd | tƒfd|tƒfd|tƒfd|tƒfd| tƒfd|tƒfd|tƒfd|tƒfd|tƒfg }g}|D]\}}dD]}|| t|¡ƒ}| ||f|df¡q‡qtD]"\}}| d¡rµd|tƒfd|tƒfg}ng}td|||||ƒqŸdS)Nr©ÚN)Úln)Úlog)ÚcoshÚsinhÚtanh)ÚceilingÚfloor)Úsqrt)ÚacosÚasinÚatanÚcosÚsinÚtanÚ test_fabsÚ test_acosÚ test_asinÚ test_atanÚtest_cosÚ test_coshÚtest_logÚtest_lnÚtest_sinÚ test_sinhÚ test_sqrtÚtest_tanÚ test_tanh)çš™™™™™É?çà?çš™™™™™é?ç›+¡†›„=r4Ú test_floorÚ test_ceilÚintrinsic_math1)Úsympy.core.evalfrˆÚsympy.functionsr‰Ú&sympy.functions.elementary.exponentialrŠÚ%sympy.functions.elementary.hyperbolicr‹rŒrÚ#sympy.functions.elementary.integersrŽrÚ(sympy.functions.elementary.miscellaneousrÚ(sympy.functions.elementary.trigonometricr‘r’r“r”r•r–ÚabsrÚsubsrKrƒrLrc)rˆr‰rŠr‹rŒrrŽrrr‘r’r“r”r•r–r„rTÚnameÚexprÚxvalr^r…r!Ú name_expr_Cr%r%r&Útest_intrinsic_math1_codegen2sF                  ó þ   ÿûr¸c Cs¢ddlm}ddlm}d|ttƒfdttfg}g}|D]"\}}dD]\}}|| t|¡ t|¡ƒ}| |||f|df¡q#qtD] \} } t d||| | ƒqBdS) Nrr‡)Úatan2Ú test_atan2Útest_pow))r¤çÍÌÌÌÌÌô?)r¥çš™™™™™É¿)r¦r¦r§Úintrinsic_math2) r«rˆr±r¹rrr³rKrƒrc) rˆr¹r„rTr´rµr¶Úyvalr^r…r!r%r%r&Útest_instrinsic_math2_codegenXs    þ  þ ÿrÀc Csðddlm}ddlm}m}m}d|tƒ|tƒ|tƒd  ¡fd||||||||tttƒƒƒƒƒƒƒƒfg}g}|D](\}}dD]!\}} } ||  t|¡  t| ¡  t| ¡ƒ} |  ||| | f| df¡qDq>t D] \} } t d ||| | ƒqidS) Nrr‡)r”r•r–Útest1éÚtest2))r¤r¼g333333Ó¿)r¥r½g)r¦gÍÌÌÌÌÌ@r¦gê-™—q=Úcomplicated_codegen)r«rˆr±r”r•r–rrrÚexpandr³rKrƒrc)rˆr”r•r–r„rTr´rµr¶r¿Úzvalr^r…r!r%r%r&Útest_complicated_codegenis ".þ  þ  ÿÿrÇ)T)(Ú sympy.abcrrrÚsympy.testing.pytestrrÚsympy.utilities.codegenrrr rQrrDrÚpathlibr r?r@Úcompile_commandsÚcombinations_lang_compilerr'rcrJrurƒrzr…Úcompilerr!rKr{r}r~rr€r†r¸rÀrÇr%r%r%r&ÚsT        _   &