o jg1@sddlmZmZmZddlmZddlmZddlm Z m Z m Z ddl Z ddl Z ddlZddlZedZiZded<d ed <ed ed <iZd ed <ed ed <d ed<iZgded<gded<gded<gded<gdZddZd2ddZddZddZgZgZeD]\ZZeeZeeereeefqeeefqd d!Zd"d#Z d$d%Z!d&d'Z"d(d)Z#d*d+Z$d,d-Z%d.d/Z&d0d1Z'dS)3)xyz) import_module)skip)codegen make_routineget_code_generatorN pyodide_jszy 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; } C89C99z 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.execc)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.exegfortran)zg95 -c codegen.f90 -o codegen.oz1g95 -ffree-line-length-huge -c main.f90 -o main.oz g95 main.o codegen.o -o test.exeg95)z!ifort -c codegen.f90 -o codegen.ozifort -c main.f90 -o main.oz"ifort main.o codegen.o -o test.exeifort)r rr rr rr rr rcCsrtrdSttjd%}|D]}tj||dtjd}|dkr&WddSq WddS1s2wYdS)z>Run a series of commands and only return True if all ran fine.FwT)stdoutshellstderrrN)r openosdevnull subprocesscallSTDOUT)commandsnullcommandretcoder%Y/var/www/html/zoom/venv/lib/python3.10/site-packages/sympy/external/tests/test_codegen.pytry_runws   r'TcCs^|}|tvs J|tvsJtdd}|dvr tdtd|d|}t }t ||r>t ||ddd n t |d} | j |ddd g} |D]/\} } } }d | d d d | D| f}|dkrrt|}tt|}| t|||dqO|dkrd}n|drd}ntd|t|d}| t|dd| iWdn1swYt|}|rtdg}nd}|dks|dkr|r|rdd}|d|d|d|d |d|d|d!|d"t |t|ntd#|tjd$t ||sJd%|d&|f|s-Jd'|d&|fdS)(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_TEMPalways)r)successneverzSSYMPY_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|VqdSN)str).0argr%r%r& szrun_test..r )r thresholdzmain.f90Czmain.cz2FIXME: filename extension unknown for language: %sr statementsNz ./test.exeFr*cSstj|r t|dSdSr.)rpathisfileremove)filenamer%r%r& safe_removes 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: %s)upper main_templatenumerical_test_templatergetenvlower ValueErrortempfilemkdtempgetcwdchdirrr writejoinfortranize_double_constantsr/append startswithNotImplementedErrorrr'rmdirprintsysr)labelroutinesnumerical_testslanguager!friendlycleanworkoldworkcode_gen test_stringsfn_nameargsexpectedr3 call_stringf_namefcompiledexecutedr;r%r%r&run_testsz                  rccsLddld}d}fdd}dd}|||}|||}|S) z; Replaces every literal float with literal doubles rNz\d+(\.)?\d*[eE]-?\d+z\d+\.\d*(?!\d*d)csdd|dS)Nz[eE]dr)subgroupmatchobjrer%r&subs_expsz-fortranize_double_constants..subs_expcSsd|dS)Nz%sd0r)rfrgr%r%r& subs_floatsz/fortranize_double_constants..subs_float)rjcompilere) code_string pattern_exp pattern_floatrkrlr%rir&rJs     rJcCsDtdt}ddg}ztd|g|||ddWdSty!YdSw)Ntest)rq)?rrV瞯<)rq)rtrs is_feasibleF)rUT)rrrcAssertionError)rTr!routinerSr%r%r&rus  rucCdtvr tddSdS)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_gfortran r|rcCrx)Nrz%`g95' command didn't work as expectedryr%r%r%r& test_F95_g95%r|rcCs<ddg}dtttfg}tD] \}}td||||qdS)N)rq)rrg@g@g5@rs)rq)rtg@rrsrq basic_codegen)rrrvalid_lang_commandsrc)rS name_exprlangr!r%r%r&test_basic_codegen,s rcCsddlm}ddlm}ddlm}ddlm}m}m }ddl m }m }ddl m}ddlm} m} m} m} m} m}d ttfd | tfd | tfd | tfd | tfd|tfd|tfd|tfd| tfd|tfd|tfd|tfd|tfg }g}|D]\}}dD]}||t|}|||f|dfqqtD]"\}}|drd|tfd|tfg}ng}td|||||qdS)NrN)ln)log)coshsinhtanh)ceilingfloor)sqrt)acosasinatancossintan test_fabs test_acos test_asin test_atantest_cos test_coshtest_logtest_lntest_sin test_sinh test_sqrttest_tan test_tanh)皙??皙?+=r4 test_floor test_ceilintrinsic_math1)sympy.core.evalfrsympy.functionsr&sympy.functions.elementary.exponentialr%sympy.functions.elementary.hyperbolicrrr#sympy.functions.elementary.integersrr(sympy.functions.elementary.miscellaneousr(sympy.functions.elementary.trigonometricrrrrrrabsrsubsrKrrLrc)rrrrrrrrrrrrrrrrrSnameexprxvalr]rr! name_expr_Cr%r%r&test_intrinsic_math1_codegen6sF                      rc Csddlm}ddlm}d|ttfdttfg}g}|D]"\}}dD]\}}||t|t|}||||f|dfq#qtD] \} } t d||| | qBdS) Nrr)atan2 test_atan2test_pow))r?)r皙ɿ)rrrintrinsic_math2) rrrrrrrrKrrc) rrrrSrrryvalr]rr!r%r%r&test_instrinsic_math2_codegen\s       rc Csddlm}ddlm}m}m}d|t|t|td fd||||||||tttfg}g}|D](\}}dD]!\}} } || t| t|  t| } | ||| | f| dfqDq>t D] \} } t d ||| | qidS) Nrr)rrrtest1test2))rrg333333ӿ)rrg)rg@rg-q=complicated_codegen)rrrrrrrrrexpandrrKrrc)rrrrrrSrrrrzvalr]rr!r%r%r&test_complicated_codegenms ".    r)T)( sympy.abcrrrsympy.externalrsympy.testing.pytestrsympy.utilities.codegenrrr rPrrDrr r?r@compile_commandscombinations_lang_compilerr'rcrJrurrzrcompilerr!rKr{r}r~rrrrrrr%r%r%r&sV         `   &