o 8Va2/@sddlmZmZmZddlmZddlmZmZm Z ddl Z ddl Z ddl Z ddl Z iZded<ded<eded <iZd ed<eded <d ed<iZgd ed <gded<gded<gded<gdZddZd0ddZddZddZgZgZeD]\ZZeeZeeereeefqeeefqddZd d!Zd"d#Zd$d%Zd&d'Z d(d)Z!d*d+Z"d,d-Z#d.d/Z$dS)1)xyz)skip)codegen make_routineget_code_generatorNzy 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 r r r r rr r r rcCs<ttjd}|D]}tj||dtjd}|dkrdSqdS)z>Run a series of commands and only return True if all ran fine.wT)stdoutshellstderrrF)openosdevnull subprocesscallZSTDOUT)commandsZnullZcommandZretcoderC/usr/lib/python3/dist-packages/sympy/external/tests/test_codegen.pytry_runts  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. 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).0argrrr szrun_test..r )r thresholdzmain.f90Czmain.cz2FIXME: filename extension unknown for language: %srZ statementsNz ./test.exeFr#cSstj|r t|dSdSr&)rpathisfileremove)filenamerrr 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 ValueErrortempfileZmkdtempgetcwdchdirrrwritejoinfortranize_double_constantsr'append startswithNotImplementedErrorrr!rmdirprintsysr)ZlabelZroutinesnumerical_testslanguagerfriendlyZcleanZworkZoldworkZcode_genZ test_stringsZfn_nameargsexpectedr+Z call_stringZf_namefZcompiledZexecutedr2rrr run_testsz                  rMcsLddld}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)subgroupZmatchobjrerr subs_expsz-fortranize_double_constants..subs_expcSsd|dS)Nz%sd0r)rPrQrrr subs_floatsz/fortranize_double_constants..subs_float)rScompilerO)Z code_stringZ pattern_expZ pattern_floatrTrUrrRr r@s     r@cCsDtdt}ddg}ztd|g|||ddWdSty!YdSw)Ntest)rW)?rXV瞯<)rW)rZrY is_feasibleF)rIT)rrrMAssertionError)rHrZroutinerGrrr r[s  r[cCdtvr tddSdS)Nrz*`cc' command didn't work as expected (C89)invalid_lang_compilersrrrrr test_C89_cc  r`cCr])Nrz*`cc' command didn't work as expected (C99)r^rrrr test_C99_ccrarbcCr])Nrz'`ifort' command didn't work as expectedr^rrrr test_F95_ifortrarccCr])Nrz*`gfortran' command didn't work as expectedr^rrrr test_F95_gfortranrardcCr])Nrz%`g95' command didn't work as expectedr^rrrr test_F95_g95 rarecCs<ddg}dtttfg}tD] \}}td||||qdS)N)rW)rXg@g@g5@rY)rW)rZg@rfrYrWZ basic_codegen)rrrvalid_lang_commandsrM)rG name_exprlangrrrr test_basic_codegen's rjcCsXddlm}m}m}m}m}m}m}m}m }m } m } m } m } m} m}dttfd|tfd|tfd|tfd|tfd|tfd |tfd |tfd | tfd | tfd | tfd| tfd| tfg }g}|D]\}}dD]}||t|}|||f|dfqoqitD]"\}}|drd|tfd|tfg}ng}td|||||qdS)Nr)acosasinatanceilingcoscoshfloorloglnsinsinhsqrttantanhNZ test_fabsZ test_acosZ test_asinZ test_atanZtest_cosZ test_coshZtest_logZtest_lnZtest_sinZ test_sinhZ test_sqrtZtest_tanZ test_tanh)皙??皙?+=r,Z test_floorZ test_ceilZintrinsic_math1)sympyrkrlrmrnrorprqrrrsrtrurvrwrxryabsrsubsrArgrBrM)rkrlrmrnrorprqrrrsrtrurvrwrxryrhrGnameexprxvalrKrirZ name_expr_Crrr test_intrinsic_math1_codegen1s:D                 rc Csddlm}m}d|ttfdttfg}g}|D]"\}}dD]\}}||t|t|}||||f|dfqqtD] \} } td||| | q>dS)Nr)atan2ryZ test_atan2Ztest_pow))rz?)r{皙ɿ)r|r|r}Zintrinsic_math2) r~rryrrrrArgrM) rryrhrGrrryvalrKrirrrr test_instrinsic_math2_codegenRs     rc Csddlm}m}m}m}d|t|t|tdfd||||||||tttfg}g}|D](\}}dD]!\}} } || t| t|  t| } | ||| | f| dfq@q:t D] \} } t d||| | qedS) Nr)rtrorwryZtest1Ztest2))rzrg333333ӿ)r{rg)r|g@r|g-q=Zcomplicated_codegen) r~rtrorwryrrrexpandrrArgrM)rtrorwryrhrGrrrrZzvalrKrirrrr test_complicated_codegenbs".    r)T)%Z sympy.abcrrrZsympy.testing.pytestrZsympy.utilities.codegenrrrrFrr;rr6r7Zcompile_commandsZcombinations_lang_compilerr!rMr@r[rgr_riZcompilerrrAr`rbrcrdrerjrrrrrrr sR        `   !