U Zʹj1@sddlmZmZmZddlmZddlmZddlm Z m Z m Z ddl Z ddl Z ddlZddlZedZiZded<d ed <ed ed <iZd ed <ed ed <d ed<iZdddged<dddged<dddged<dddged<ddd d!d"gZd#d$Zd>d&d'Zd(d)Zd*d+ZgZgZeD]<\ZZeeZeeerLeeefneeefq d,d-Zd.d/Z d0d1Z!d2d3Z"d4d5Z#d6d7Z$d8d9Z%d:d;Z&d #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.execcz$gfortran -c codegen.f90 -o codegen.oz6gfortran -ffree-line-length-none -c main.f90 -o main.oz%gfortran main.o codegen.o -o test.exegfortranzg95 -c codegen.f90 -o codegen.oz1g95 -ffree-line-length-huge -c main.f90 -o main.oz g95 main.o codegen.o -o test.exeg95z!ifort -c codegen.f90 -o codegen.ozifort -c main.f90 -o main.oz"ifort main.o codegen.o -o test.exeifortr rr rr rr rr rc CsZtrdSttjd<}|D]0}tj||dtjd}|dkrW5QRdSqW5QRXdS)z>Run a series of commands and only return True if all ran fine.FwT)stdoutshellstderrr)r openosdevnull subprocesscallSTDOUT)commandsnullcommandretcoder%`/var/www/html/bwave/python/venv/lib/python3.8/site-packages/sympy/external/tests/test_codegen.pytry_runws r'Tc CsR|}|tkst|tks ttdd}|dkr@tdt d|d|}t }t ||r|t ||ddd nt |d} | j|ddd g} |D]^\} } } }d | d d d | D| f}|dkrt|}tt|}| t|||dq|dkrd}n|dr d}n td|t|d"}|t|dd| iW5QRXt|}|rxtdg}nd}|dks|dkr|r|rdd}|d|d|d|d|d|d|d |d!t |t|ntd"|tjd#t ||s2td$|d%|f|sNtd&|d%|fd'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. ZSYMPY_TEST_CLEAN_TEMPalways)r(successneverzSSYMPY_TEST_CLEAN_TEMP must be one of the following: 'always', 'success' or 'never'.z_sympy_%s_testz%s_rT)Zto_filesz %s(%s)-(%s),css|]}t|VqdSN)str).0argr%r%r& szrun_test..r )r thresholdzmain.f90Czmain.cz2FIXME: filename extension unknown for language: %srZ statementsz ./test.exeFr)cSstj|rt|dSr,)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: %sN)upper main_templateAssertionErrornumerical_test_templatergetenvlower ValueErrortempfilemkdtempgetcwdchdirrr writejoinfortranize_double_constantsr-append startswithNotImplementedErrorrr'rmdirprintsysr)labelZroutinesnumerical_testslanguager!friendlycleanZworkZoldworkZcode_genZ test_stringsfn_nameargsexpectedr1Z call_stringf_namefcompiledZexecutedr8r%r%r&run_tests                  rZcsLddld}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)subgroupZmatchobjrer%r&subs_expsz-fortranize_double_constants..subs_expcSsd|dS)Nz%sd0r)r]r^r%r%r& subs_floatsz/fortranize_double_constants..subs_float)r`compiler\)Z code_stringZ pattern_expZ pattern_floatrarbr%r_r&rHs     rHcCsJtdt}ddg}ztd|g|||ddWdStk rDYdSXdS)Ntest)rd)?reV瞯<)rd)rgrf is_feasibleF)rRT)rrrZr=)rQr!ZroutinerPr%r%r&rhs rhcCsdtkrtddS)Nrz*`cc' command didn't work as expected (C89)invalid_lang_compilersrr%r%r%r& test_C89_ccsrkcCsdtkrtddS)Nrz*`cc' command didn't work as expected (C99)rir%r%r%r& test_C99_ccsrlcCsdtkrtddS)Nrz'`ifort' command didn't work as expectedrir%r%r%r&test_F95_ifortsrmcCsdtkrtddS)Nrz*`gfortran' command didn't work as expectedrir%r%r%r&test_F95_gfortran srncCsdtkrtddS)Nrz%`g95' command didn't work as expectedrir%r%r%r& test_F95_g95%srocCs<ddg}dtttfg}tD]\}}td||||qdS)N)rd)reg@g@g5@rf)rd)rgg@rprfrdZ basic_codegen)rrrvalid_lang_commandsrZ)rP name_exprlangr!r%r%r&test_basic_codegen,s  rtcCsddlm}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]H\}}|drpd|tfd|tfg}ng}td|||||qBdS)NrN)ln)log)coshsinhtanh)ceilingfloor)sqrt)acosasinatancossintanZ test_fabsZ test_acosZ test_asinZ test_atanZtest_cosZ test_coshZtest_logZtest_lnZtest_sinZ test_sinhZ test_sqrtZtest_tanZ test_tanh)皙??皙?+=r2Z test_floorZ test_ceilZintrinsic_math1)sympy.core.evalfrvZsympy.functionsrwZ&sympy.functions.elementary.exponentialrxZ%sympy.functions.elementary.hyperbolicryrzr{Z#sympy.functions.elementary.integersr|r}Z(sympy.functions.elementary.miscellaneousr~(sympy.functions.elementary.trigonometricrrrrrrabsrsubsrIrqrJrZ)rvrwrxryrzr{r|r}r~rrrrrrrrrPnameexprxvalrVrsr!Z name_expr_Cr%r%r&test_intrinsic_math1_codegen6sF                      rc Csddlm}ddlm}d|ttfdttfg}g}|D]D\}}dD]6\}}||t|t|}||||f|dfqFq:tD]\} } t d||| | qdS) Nrru)atan2Z test_atan2Ztest_pow))r?)r皙ɿ)rrrZintrinsic_math2) rrvrrrrrrIrqrZ) rvrrrrPrrryvalrVrsr!r%r%r&test_instrinsic_math2_codegen\s       rc Csddlm}ddlm}m}m}d|t|t|td fd||||||||tttfg}g}|D]P\}}dD]B\}} } || t| t|  t| } | ||| | f| dfqq|t D]\} } t d ||| | qdS) Nrru)rrrZtest1Ztest2))rrg333333ӿ)rrg)rg@rg-q=Zcomplicated_codegen)rrvrrrrrrrexpandrrIrqrZ)rvrrrrrrPrrrrZzvalrVrsr!r%r%r&test_complicated_codegenms$ ".   r)T)(Z sympy.abcrrrZsympy.externalrZsympy.testing.pytestrZsympy.utilities.codegenrrr rNrrBrr r<r>Zcompile_commandsZcombinations_lang_compilerr'rZrHrhrqrjrscompilerr!rIrkrlrmrnrortrrrr%r%r%r&sv        `   &