U ZʹjB@sddlmZddlmZmZmZmZmZmZddl m Z ddl m Z ddl mZddlmZddlmZmZddlmZmZmZmZmZmZmZmZmZmZmZdd l m!Z!dd l"m#Z#dd dZ$ddZ%ddZ&ddZ'ddZ(ddZ)ddZ*ddZ+ddZ,dd Z-d!d"Z.d#d$Z/d%d&Z0d'd(Z1d)d*Z2d+d,Z3d-d.Z4d/d0Z5d1d2Z6d3d4Z7d5d6Z8d7d8Z9d9d:Z:d;d<Z;d=d>ZdCdDZ?dEdFZ@dGdHZAdIdJZBdKdLZCdMdNZDdOdPZEdQdRZFdSdTZGdUdVZHdWdXZIdYdZZJd[d\ZKd]d^ZLd_d`ZMdadbZNdcddZOdedfZPdgdhZQdidjZRdkdlZSdmdnZTdodpZUdqdrZVdsdtZWdudvZXdwdxZYdydzZZd{d|Z[d}d~Z\ddZ]ddZ^dS))StringIO)symbolsEqpiCatalanLambdaDummyEquality)Symbol)erf)Integral)Matrix MatrixSymbol) codegen make_routineCCodeGen C89CodeGen C99CodeGen InputArgument CodeGenErrorFCodeGenCodeGenArgumentListErrorOutputArgument InOutArgument)raises)implemented_functionfileFcCs*t}|||||||}||S)a5Wrapper for dump_fn. dump_fn writes its results to a stream object and this wrapper returns the contents of that stream as a string. This auxiliary function is used by many tests below. The header and the empty lines are not generated to facilitate the testing of the output. )rgetvalueclose)Zdump_fnZroutinesprefixheaderemptyoutputsourcer%a/var/www/html/bwave/python/venv/lib/python3.8/site-packages/sympy/utilities/tests/test_codegen.py get_strings r'cstd\ttfddttfddtdtgd}dd|jDgkstd d|jDttttgksttdtgd}d d|jDt ttgkstd d l m }m }t |d dg\}}tddd}|d|}tdt|||||||gd}dd|jD|j|j|gks^ttddfddftdtgd}dd|jDgkstdS)Nza x y zcstdgdSNtestargument_sequencerr%)exprxzr%r&&sz-test_Routine_argument_order..cstdtgdSr()rrr%ar-r.yr/r%r&r0(sr)r*cSsg|] }|jqSr%name.0argr%r%r& +sz/test_Routine_argument_order..cSsg|] }t|qSr%typer6r%r%r&r9,scSsg|] }t|qSr%r:r6r%r%r&r9/sr IndexedBaseIdxABmTintegericSsg|] }|jqSr%r4r6r%r%r&r97scSsg|] }|jqSr%r4r6r%r%r&r9;s)rrrrr argumentsAssertionErrorrrr sympy.tensorr=r>maplabelr )rr=r>r?r@rArDr%r1r&test_Routine_argument_order#s6      "$ rNcCs"t}t|jg}|dkstdS)Nz$#include "file.h" #include rr'dump_crIcode_genr$r%r%r&test_empty_c_code>s rScCsBt}t|jgdd}|dddks*t|dddks>tdS)NTr!RzR/****************************************************************************** *a* * * * See http://www.sympy.org/ for more information. * * * * This file is part of 'project' * ******************************************************************************/ #include "file.h" #include rOrQr%r%r&test_empty_c_code_with_commentDs  rWcCs"t}t|jg}|dkstdS)Nz9#ifndef PROJECT__FILE__H #define PROJECT__FILE__H #endif )rr'dump_hrIrQr%r%r&test_empty_c_headerVs rYcCsLtd\}}}|||}td|}t}t|j|g}d}||ksHtdS)Nx,y,zr)#include "file.h" #include double test(double x, double y, double z) { double test_result; test_result = z*(x + y); return test_result; } rrrr'rPrIr.r3r/r-routinerRr$expectedr%r%r&test_simple_c_code\s   r`cCsLtd\}}}|||}td|}t}t|j|g}d}||ksHtdS)Nzif, typedef, whiler)z#include "file.h" #include double test(double if_, double typedef_, double while_) { double test_result; test_result = while_*(if_ + typedef_); return test_result; } )rrrr'rPrIr]r%r%r&test_c_code_reserved_wordsns   racCs@tdtt}t}t|j|g}dtd}||ks double test() { double test_result; double const Catalan = %s; test_result = pow(M_PI, Catalan); return test_result; } )rrrrr'rPevalfrIr^rRr$r_r%r%r&test_numbersymbol_c_codes recCsRtd\}}}||}td||||gd}t}t|j|g}d}||ksNtdS)NrZr)r*z#include "file.h" #include double test(double z, double x, double y) { double test_result; test_result = x + y; return test_result; } r\r]r%r%r&test_c_code_argument_orders rfcCsLtd\}}}|||}td|}t}t|j|g}d}||ksHtdS)NrZr)d#ifndef PROJECT__FILE__H #define PROJECT__FILE__H double test(double x, double y, double z); #endif )rrrr'rXrIr]r%r%r&test_simple_c_headers  rhcCsHtd\}}}|||}ddg}td|fddddd}||ksDtdS) NrZ)file.cr[)file.hrgr)CrFr!r"rrrI)r.r3r/r-r_resultr%r%r&test_simple_c_codegens rocsRtd\}}}|||}|||}td||gtttfdddS)NrZr)cstjgSNr'rXr%rRr^r%r&r0z)test_multiple_results_c..)rrrrrr.r3r/expr1expr2r%rrr&test_multiple_results_cs  rwcCsttdddS)NcSs tdgSNr)r,r%r%r%r&r0rsz#test_no_results_c..r ValueErrorr%r%r%r&test_no_results_csr{cCsxddlm}ddlm}ddlm}m}m}ddlm }m }ddl m }ddl m}m} m} m} m} m} td}d ||fd ||fd | |fd | |fd ||fd| |fd||fd||fd||fd||fd| |fd||fd||fd| |fd||fg}t|ddddd}|dddks2t|dddksHt|dddks^t|ddd ksttdS)!NrAbslogcoshsinhtanh)ceilingfloorsqrtacosasinatancossintanr.Z test_fabs test_acos test_asin test_atanZ test_ceiltest_cos test_coshZ test_floortest_logtest_lntest_sin test_sinh test_sqrttest_tan test_tanhC89rFrlrirEa#include "file.h" #include double test_fabs(double x) { double test_fabs_result; test_fabs_result = fabs(x); return test_fabs_result; } double test_acos(double x) { double test_acos_result; test_acos_result = acos(x); return test_acos_result; } double test_asin(double x) { double test_asin_result; test_asin_result = asin(x); return test_asin_result; } double test_atan(double x) { double test_atan_result; test_atan_result = atan(x); return test_atan_result; } double test_ceil(double x) { double test_ceil_result; test_ceil_result = ceil(x); return test_ceil_result; } double test_cos(double x) { double test_cos_result; test_cos_result = cos(x); return test_cos_result; } double test_cosh(double x) { double test_cosh_result; test_cosh_result = cosh(x); return test_cosh_result; } double test_floor(double x) { double test_floor_result; test_floor_result = floor(x); return test_floor_result; } double test_log(double x) { double test_log_result; test_log_result = log(x); return test_log_result; } double test_ln(double x) { double test_ln_result; test_ln_result = log(x); return test_ln_result; } double test_sin(double x) { double test_sin_result; test_sin_result = sin(x); return test_sin_result; } double test_sinh(double x) { double test_sinh_result; test_sinh_result = sinh(x); return test_sinh_result; } double test_sqrt(double x) { double test_sqrt_result; test_sqrt_result = sqrt(x); return test_sqrt_result; } double test_tan(double x) { double test_tan_result; test_tan_result = tan(x); return test_tan_result; } double test_tanh(double x) { double test_tanh_result; test_tanh_result = tanh(x); return test_tanh_result; } rja#ifndef PROJECT__FILE__H #define PROJECT__FILE__H double test_fabs(double x); double test_acos(double x); double test_asin(double x); double test_atan(double x); double test_ceil(double x); double test_cos(double x); double test_cosh(double x); double test_floor(double x); double test_log(double x); double test_ln(double x); double test_sin(double x); double test_sinh(double x); double test_sqrt(double x); double test_tan(double x); double test_tanh(double x); #endif )$sympy.functions.elementary.complexesr}&sympy.functions.elementary.exponentialr%sympy.functions.elementary.hyperbolicrrrZ#sympy.functions.elementary.integersrr(sympy.functions.elementary.miscellaneousr(sympy.functions.elementary.trigonometricrrrrrrrrrI)r}rrrrrrrrrrrrrr. name_exprrnr%r%r&test_ansi_math1_codegens@                      rcCsddlm}td\}}d|||fd||fg}t|ddddd }|ddd ksXt|dd d kslt|d dd kst|d d dkstdS)Nratan2x,y test_atan2test_powrrFrlrirEa*#include "file.h" #include double test_atan2(double x, double y) { double test_atan2_result; test_atan2_result = atan2(x, y); return test_atan2_result; } double test_pow(double x, double y) { double test_pow_result; test_pow_result = pow(x, y); return test_pow_result; } rjz#ifndef PROJECT__FILE__H #define PROJECT__FILE__H double test_atan2(double x, double y); double test_pow(double x, double y); #endif rrrrrI)rr.r3rrnr%r%r&test_ansi_math2_codegens      rc Csddlm}m}m}td\}}}d||||||dfd|||||||||||fg}t|ddd d d }|ddd kst|dd d kst|d ddkst|d d dkstdS)NrrrrrZtest1test2rrFrlrirEa-#include "file.h" #include double test1(double x, double y, double z) { double test1_result; test1_result = pow(sin(x), 7) + 7*pow(sin(x), 6)*cos(y) + 7*pow(sin(x), 6)*tan(z) + 21*pow(sin(x), 5)*pow(cos(y), 2) + 42*pow(sin(x), 5)*cos(y)*tan(z) + 21*pow(sin(x), 5)*pow(tan(z), 2) + 35*pow(sin(x), 4)*pow(cos(y), 3) + 105*pow(sin(x), 4)*pow(cos(y), 2)*tan(z) + 105*pow(sin(x), 4)*cos(y)*pow(tan(z), 2) + 35*pow(sin(x), 4)*pow(tan(z), 3) + 35*pow(sin(x), 3)*pow(cos(y), 4) + 140*pow(sin(x), 3)*pow(cos(y), 3)*tan(z) + 210*pow(sin(x), 3)*pow(cos(y), 2)*pow(tan(z), 2) + 140*pow(sin(x), 3)*cos(y)*pow(tan(z), 3) + 35*pow(sin(x), 3)*pow(tan(z), 4) + 21*pow(sin(x), 2)*pow(cos(y), 5) + 105*pow(sin(x), 2)*pow(cos(y), 4)*tan(z) + 210*pow(sin(x), 2)*pow(cos(y), 3)*pow(tan(z), 2) + 210*pow(sin(x), 2)*pow(cos(y), 2)*pow(tan(z), 3) + 105*pow(sin(x), 2)*cos(y)*pow(tan(z), 4) + 21*pow(sin(x), 2)*pow(tan(z), 5) + 7*sin(x)*pow(cos(y), 6) + 42*sin(x)*pow(cos(y), 5)*tan(z) + 105*sin(x)*pow(cos(y), 4)*pow(tan(z), 2) + 140*sin(x)*pow(cos(y), 3)*pow(tan(z), 3) + 105*sin(x)*pow(cos(y), 2)*pow(tan(z), 4) + 42*sin(x)*cos(y)*pow(tan(z), 5) + 7*sin(x)*pow(tan(z), 6) + pow(cos(y), 7) + 7*pow(cos(y), 6)*tan(z) + 21*pow(cos(y), 5)*pow(tan(z), 2) + 35*pow(cos(y), 4)*pow(tan(z), 3) + 35*pow(cos(y), 3)*pow(tan(z), 4) + 21*pow(cos(y), 2)*pow(tan(z), 5) + 7*cos(y)*pow(tan(z), 6) + pow(tan(z), 7); return test1_result; } double test2(double x, double y, double z) { double test2_result; test2_result = cos(cos(cos(cos(cos(cos(cos(cos(x + y + z)))))))); return test2_result; } rjz#ifndef PROJECT__FILE__H #define PROJECT__FILE__H double test1(double x, double y, double z); double test2(double x, double y, double z); #endif rrrrrexpandrrI)rrrr.r3r/rrnr%r%r&test_complicated_codegen,s". 1 rcCs@ddlm}m}ddlm}|ddd\}}|d}|d}|d }|d |}|d |} td t||||| f|| fd dddd\\} } \} } | dkstd}| |dd||| iks | |dd| ||iks | |dd||| iks | |dd| ||iks t| dks.t| dks void matrix_vector(double *A, int m, int n, double *x, double *y) { for (int i=0; isympy.core.symbolrrrrIr=r>rnrAr?r.r3rDrf1codef2 interfacer_r%r%r& test_loops_cps:     rc Csddlm}m}tddtd\}}|d}|d}|||}d|jj|jd }td t||||}t }t } t | j |g} | |kst ttt |j |gW5QRXdS) Nrr<i mTrCclsr.r3z#include "file.h" #include void test_dummies(int m_%(mno)i, double *x, double *y) { for (int i_%(ino)i=0; i_%(ino)irrrL dummy_indexrrrrr'rPrIrNotImplementedError) r=r>rDrAr.r3r_rMZc89c99rr%r%r&test_dummy_loops_cs    rcCsbddlm}m}ddlm}|ddd\}}}}|d||fd}|d }|d } |d ||d f} |d |} tdt| | || | f|| fddddd\\} } \}}| dkstd|ddd}| |dd| || iksB| |dd| | |iksB| |dd| || iksB| |dd| | |iksBt|dksPt|dks^tdS)Nrr<rn m o pTrBr?shaper.r3rDrrrrFrlria##include "file.h" #include void matrix_vector(double *A, int m, int n, int o, int p, double *x, double *y) { for (int i=o; i<%(upperi)s; i++){ y[i] = 0; } for (int i=o; i<%(upperi)s; i++){ for (int j=0; jrrrAopr?r.r3rDrrrrrr_r%r%r&test_partial_loops_cs<     rc Csddlm}ddlm}m}td\}}}td||||||g}t}|j|gdddd}|ddd kstt d } |dd | kst dS) Nrr rrrZfoor)Frltest.cz#include "test.h" #include double foo(double x, double *y) { (*y) = sin(x); double foo_result; foo_result = cos(x); return foo_result; } rE sympy.core.relationalr rrrrrrwriterI r rrr.r3r/rMcrnr_r%r%r&test_output_arg_cs  rc Csddlm}ddlm}m}td\}}}td||||||g}t}|j|gdddd}|ddd kstt d } |dd | kst dS) Nrr rz if, while, zrr)Frlrz#include "test.h" #include double foo(double if_, double *while_) { (*while_) = sin(if_); double foo_result; foo_result = cos(if_); return foo_result; } rErrr%r%r& test_output_arg_c_reserved_wordss  rc Cstddd}tddd}t|}t|}||}tddd}|d|}d |jd _d |jd _t|j|gdd }d }||kst dS) Nr?rGbrEr)T)projectcserout)r z#include "test.h" #include void c(double *A, double *b, double *out) { out[0] = A[0]*b[0] + A[1]*b[1] + A[2]*b[2]; out[1] = A[3]*b[0] + A[4]*b[1] + A[5]*b[2]; out[2] = A[6]*b[0] + A[7]*b[1] + A[8]*b[2]; } ) rrrr^rHZ result_var_namer'rPrI) ZA_symZb_symr?rrZcgenrMrr_r%r%r&test_multidim_c_argument_cses       rc Cstd\}}}td\}}tddd}t|tdd|gg}t||||}t|d|}d|||gf} d} t| d dd d ||||||fd } | d d} | | kstdS) NrZB,Cr?rErGrFr)z#include "test.h" #include void test(double x, double *C, double z, double y, double *A, double *B) { (*C) = z*(x + y); A[0] = 1; A[1] = 2; A[2] = x; (*B) = 2*x; } rFr!r"r+rrrr rrrI) r.r3r/r@rkr?rurvexpr3rr_rnr$r%r%r& test_ccode_results_named_ordereds    rcCstddd}tddd}tddd}tddd}dt||d ddft||dddft||ddd fgf}t|d dd d d }|d d}d}||kstdS)Nr?rrGr@rErkDr)rrFrFrla#include "test.h" #include void test(double *A, double *B, double *C, double *D) { B[0] = A[0]; B[1] = A[1]; B[2] = A[2]; C[0] = A[3]; C[1] = A[4]; C[2] = A[5]; D[0] = A[2]; D[1] = A[5]; D[2] = A[8]; D[3] = A[11]; D[4] = A[14]; } rr rrIr?r@rkrrrnr$r_r%r%r&test_ccode_matrixsymbol_slice8s     rc Cstd\}}}}tddd}dt|t||g||||g||||gggf}tdd}t||ddd }|d d}d } || kstdS) Nza b c derGrEr)T)rF)rRr!r"rz#include "test.h" #include void test(double a, double b, double c, double d, double *e) { const double x0 = a*b; const double x1 = c*d; e[0] = x0; e[1] = x0 + x1; e[2] = x0*x1; } )rrr rrrrI) r2rrdrr generatorrnr$r_r%r%r&test_ccode_cseUs <   rcCsJtddd}d}t}t||dd|fd}|dd}d}||ksFtdS) Nr.rFrE)r)g?F)rRr!r"r+rz#include "test.h" #include double test(double *x) { double test_result; test_result = 1.0; return test_result; } )rrrrI)r.rrrnr$r_r%r%r&test_ccode_unused_array_argis   rcCsptddd}tddd}tddd}d|d |df}t|d d d |||fd }|d d}d }||ksltdS)NXrGrEYr/TrBZtestBugrrkF)languager!r"r+z#include "testBug.h" #include double testBug(double *X, double *Y, int z) { double testBug_result; testBug_result = X[0] + X[1]; return testBug_result; } )rrrrI)rrr/rrnr$r_r%r%r& test_ccode_unused_array_arg_func|s     rcCs"t}t|jg}|dkstdSNrr'dump_f95rIrQr%r%r&test_empty_f_codes rcCsBt}t|jgdd}|dddks*t|dddks>tdS)NTrTrUzR!****************************************************************************** !*rVa* !* * !* See http://www.sympy.org/ for more information. * !* * !* This file is part of 'project' * !****************************************************************************** rrQr%r%r&test_empty_f_code_with_headers  rcCs"t}t|jg}|dkstdSr)rr'rXrIrQr%r%r&test_empty_f_headers rcCsLtd\}}}|||}td|}t}t|j|g}d}||ksHtdS)NrZr)REAL*8 function test(x, y, z) implicit none REAL*8, intent(in) :: x REAL*8, intent(in) :: y REAL*8, intent(in) :: z test = z*(x + y) end function rrrr'rrIr]r%r%r&test_simple_f_codes   rcCsJtdtt}t}t|j|g}dtdtdf}||ksFtdS)Nr)zREAL*8 function test() implicit none REAL*8, parameter :: Catalan = %sd0 REAL*8, parameter :: pi = %sd0 test = pi**Catalan end function rb)rrrrr'rrcrIrdr%r%r&test_numbersymbol_f_codesrcCsNtd}tdt|td|}t}t|j|g}d}||ksJt|dS)Nr.r)zhREAL*8 function test(x) implicit none REAL*8, intent(in) :: x test = erf(x) + erf(2.0d0*x) end function )rrr rr'rrI)r.r^rRr$r_r%r%r&test_erf_f_codesrcCsRtd\}}}||}td||||gd}t}t|j|g}d}||ksNtdS)NrZr)r*zREAL*8 function test(z, x, y) implicit none REAL*8, intent(in) :: z REAL*8, intent(in) :: x REAL*8, intent(in) :: y test = x + y end function rr]r%r%r&test_f_code_argument_orders rcCsLtd\}}}|||}td|}t}t|j|g}d}||ksHtdS)NrZr)interface REAL*8 function test(x, y, z) implicit none REAL*8, intent(in) :: x REAL*8, intent(in) :: y REAL*8, intent(in) :: z end function end interface )rrrr'rXrIr]r%r%r&test_simple_f_headers   rcCsHtd\}}}|||}td|fddddd}ddg}||ksDtdS) NrZr)F95rFrl)file.f90r)rjrrm)r.r3r/r-rnr_r%r%r&test_simple_f_codegens r csRtd\}}}|||}|||}td||gtttfdddS)NrZr)cstjgSrprqr%rrr%r&r0"rsz)test_multiple_results_f..)rrrrrrtr%rrr&test_multiple_results_fs  r cCsttdddS)NcSs tdgSrxr,r%r%r%r&r0&rsz#test_no_results_f..ryr%r%r%r&test_no_results_f%sr cCs\ddlm}ddlm}ddlm}m}m}ddlm }ddl m }m }m }m} m} m} td} d|| fd || fd || fd || fd | | fd || fd|| fd|| fd| | fd|| fd|| fd| | fd|| fg } t| ddddd}|dddkstd}|dd|ks(t|dddks>td}|dd|ksXtdS)Nrr|r~rrrr.Ztest_absrrrrrrrrrrrrrrFrlr aREAL*8 function test_abs(x) implicit none REAL*8, intent(in) :: x test_abs = abs(x) end function REAL*8 function test_acos(x) implicit none REAL*8, intent(in) :: x test_acos = acos(x) end function REAL*8 function test_asin(x) implicit none REAL*8, intent(in) :: x test_asin = asin(x) end function REAL*8 function test_atan(x) implicit none REAL*8, intent(in) :: x test_atan = atan(x) end function REAL*8 function test_cos(x) implicit none REAL*8, intent(in) :: x test_cos = cos(x) end function REAL*8 function test_cosh(x) implicit none REAL*8, intent(in) :: x test_cosh = cosh(x) end function REAL*8 function test_log(x) implicit none REAL*8, intent(in) :: x test_log = log(x) end function REAL*8 function test_ln(x) implicit none REAL*8, intent(in) :: x test_ln = log(x) end function REAL*8 function test_sin(x) implicit none REAL*8, intent(in) :: x test_sin = sin(x) end function REAL*8 function test_sinh(x) implicit none REAL*8, intent(in) :: x test_sinh = sinh(x) end function REAL*8 function test_sqrt(x) implicit none REAL*8, intent(in) :: x test_sqrt = sqrt(x) end function REAL*8 function test_tan(x) implicit none REAL*8, intent(in) :: x test_tan = tan(x) end function REAL*8 function test_tanh(x) implicit none REAL*8, intent(in) :: x test_tanh = tanh(x) end function rErjaAinterface REAL*8 function test_abs(x) implicit none REAL*8, intent(in) :: x end function end interface interface REAL*8 function test_acos(x) implicit none REAL*8, intent(in) :: x end function end interface interface REAL*8 function test_asin(x) implicit none REAL*8, intent(in) :: x end function end interface interface REAL*8 function test_atan(x) implicit none REAL*8, intent(in) :: x end function end interface interface REAL*8 function test_cos(x) implicit none REAL*8, intent(in) :: x end function end interface interface REAL*8 function test_cosh(x) implicit none REAL*8, intent(in) :: x end function end interface interface REAL*8 function test_log(x) implicit none REAL*8, intent(in) :: x end function end interface interface REAL*8 function test_ln(x) implicit none REAL*8, intent(in) :: x end function end interface interface REAL*8 function test_sin(x) implicit none REAL*8, intent(in) :: x end function end interface interface REAL*8 function test_sinh(x) implicit none REAL*8, intent(in) :: x end function end interface interface REAL*8 function test_sqrt(x) implicit none REAL*8, intent(in) :: x end function end interface interface REAL*8 function test_tan(x) implicit none REAL*8, intent(in) :: x end function end interface interface REAL*8 function test_tanh(x) implicit none REAL*8, intent(in) :: x end function end interface )rr}rrrrrrrrrrrrrrrrrrI)r}rrrrrrrrrrrr.rrnr_r%r%r&test_intrinsic_math_codegen)s:                 CPr cCsddlm}td\}}d|||fd||fg}t|ddddd }|ddd ksXtd }|dd |kspt|d dd kstd}|d d |kstdS)NrrrrrrrFrlr zREAL*8 function test_atan2(x, y) implicit none REAL*8, intent(in) :: x REAL*8, intent(in) :: y test_atan2 = atan2(x, y) end function REAL*8 function test_pow(x, y) implicit none REAL*8, intent(in) :: x REAL*8, intent(in) :: y test_pow = x**y end function rErjainterface REAL*8 function test_atan2(x, y) implicit none REAL*8, intent(in) :: x REAL*8, intent(in) :: y end function end interface interface REAL*8 function test_pow(x, y) implicit none REAL*8, intent(in) :: x REAL*8, intent(in) :: y end function end interface r)rr.r3rrnr_r%r%r&test_intrinsic_math2_codegens    rc Csddlm}m}m}td\}}}d||||||dfd|||||||||||fg}t|ddd d d }|ddd kstd }|dd |kst|d ddkstd}|d d |kstdS)NrrrZrrrrrFrlr a{REAL*8 function test1(x, y, z) implicit none REAL*8, intent(in) :: x REAL*8, intent(in) :: y REAL*8, intent(in) :: z test1 = sin(x)**7 + 7*sin(x)**6*cos(y) + 7*sin(x)**6*tan(z) + 21*sin(x) & **5*cos(y)**2 + 42*sin(x)**5*cos(y)*tan(z) + 21*sin(x)**5*tan(z) & **2 + 35*sin(x)**4*cos(y)**3 + 105*sin(x)**4*cos(y)**2*tan(z) + & 105*sin(x)**4*cos(y)*tan(z)**2 + 35*sin(x)**4*tan(z)**3 + 35*sin( & x)**3*cos(y)**4 + 140*sin(x)**3*cos(y)**3*tan(z) + 210*sin(x)**3* & cos(y)**2*tan(z)**2 + 140*sin(x)**3*cos(y)*tan(z)**3 + 35*sin(x) & **3*tan(z)**4 + 21*sin(x)**2*cos(y)**5 + 105*sin(x)**2*cos(y)**4* & tan(z) + 210*sin(x)**2*cos(y)**3*tan(z)**2 + 210*sin(x)**2*cos(y) & **2*tan(z)**3 + 105*sin(x)**2*cos(y)*tan(z)**4 + 21*sin(x)**2*tan & (z)**5 + 7*sin(x)*cos(y)**6 + 42*sin(x)*cos(y)**5*tan(z) + 105* & sin(x)*cos(y)**4*tan(z)**2 + 140*sin(x)*cos(y)**3*tan(z)**3 + 105 & *sin(x)*cos(y)**2*tan(z)**4 + 42*sin(x)*cos(y)*tan(z)**5 + 7*sin( & x)*tan(z)**6 + cos(y)**7 + 7*cos(y)**6*tan(z) + 21*cos(y)**5*tan( & z)**2 + 35*cos(y)**4*tan(z)**3 + 35*cos(y)**3*tan(z)**4 + 21*cos( & y)**2*tan(z)**5 + 7*cos(y)*tan(z)**6 + tan(z)**7 end function REAL*8 function test2(x, y, z) implicit none REAL*8, intent(in) :: x REAL*8, intent(in) :: y REAL*8, intent(in) :: z test2 = cos(cos(cos(cos(cos(cos(cos(cos(x + y + z)))))))) end function rErja4interface REAL*8 function test1(x, y, z) implicit none REAL*8, intent(in) :: x REAL*8, intent(in) :: y REAL*8, intent(in) :: z end function end interface interface REAL*8 function test2(x, y, z) implicit none REAL*8, intent(in) :: x REAL*8, intent(in) :: y REAL*8, intent(in) :: z end function end interface r) rrrr.r3r/rrnr_r%r%r&test_complicated_codegen_f95 s".rcCsddlm}m}ddlm}|ddd\}}t|d\}}}|d|}|d |} td t||||| f|| fd d d d d\\} } \} } | dkstd}| |ddiks| |ddikst| dkst| dkstdS)Nrr<rzn,mTrBAxyrDrrrrFrlr a~subroutine matrix_vector(A, m, n, x, y) implicit none INTEGER*4, intent(in) :: m INTEGER*4, intent(in) :: n REAL*8, intent(in), dimension(1:m, 1:n) :: A REAL*8, intent(in), dimension(1:n) :: x REAL*8, intent(out), dimension(1:m) :: y INTEGER*4 :: i INTEGER*4 :: j do i = 1, m y(i) = 0 end do do i = 1, m do j = 1, n y(i) = %(rhs)s + y(i) end do end do end subroutine r A(i, j)*x(j) x(j)*A(i, j)rjainterface subroutine matrix_vector(A, m, n, x, y) implicit none INTEGER*4, intent(in) :: m INTEGER*4, intent(in) :: n REAL*8, intent(in), dimension(1:m, 1:n) :: A REAL*8, intent(in), dimension(1:n) :: x REAL*8, intent(out), dimension(1:m) :: y end subroutine end interface ) rJr=r>rrrKrrrIrr%r%r& test_loopsGs.      rc Csddlm}m}tddtd\}}|d}|d}|||}d|jj|jd }td t||||}t }t |j |g} | |kst dS) Nrr<rTrr.r3a>subroutine test_dummies(m_%(mcount)i, x, y) implicit none INTEGER*4, intent(in) :: m_%(mcount)i REAL*8, intent(in), dimension(1:m_%(mcount)i) :: x REAL*8, intent(out), dimension(1:m_%(mcount)i) :: y INTEGER*4 :: i_%(icount)i do i_%(icount)i = 1, m_%(mcount)i y(i_%(icount)i) = x(i_%(icount)i) end do end subroutine )ZicountZmcountr) rJr=r>rrrLrrrrr'rrI) r=r>rDrAr.r3r_rMrrr%r%r&test_dummy_loops_f95zs   rcCsddlm}m}ddlm}|ddd\}}}}|d\}}} ||||||||f}|||||}|| |||} tdt| | ||fd d d d d \\} } \} } | d kstd}| |ddiks| |ddikst| dkst| dkstdS)Nrr<rzi,j,n,mTrBzA,x,yrrrFrlr aasubroutine matrix_vector(A, m, n, x, y) implicit none INTEGER*4, intent(in) :: m INTEGER*4, intent(in) :: n REAL*8, intent(in), dimension(1:m, 1:n) :: A REAL*8, intent(in), dimension(1:n) :: x REAL*8, intent(inout), dimension(1:m) :: y INTEGER*4 :: i INTEGER*4 :: j do i = 1, m do j = 1, n y(i) = %(rhs)s + y(i) end do end do end subroutine rrrrjainterface subroutine matrix_vector(A, m, n, x, y) implicit none INTEGER*4, intent(in) :: m INTEGER*4, intent(in) :: n REAL*8, intent(in), dimension(1:m, 1:n) :: A REAL*8, intent(in), dimension(1:n) :: x REAL*8, intent(inout), dimension(1:m) :: y end subroutine end interface r)r=r>rrDrrrAr?r.r3rrrrr_r%r%r&test_loops_InOuts0   rcCsddlm}m}ddlm}|ddd\}}}}|d||fd}|d }|d } |d ||d f} |d |} tdt| | || | f|| fddddd\\} } \}}ddt|dtd|t|d|d}| |ddiks| |ddikstdS)Nrr<rrTrBr?rr.r3rDrrrrrFrlasubroutine matrix_vector(A, m, n, o, p, x, y) implicit none INTEGER*4, intent(in) :: m INTEGER*4, intent(in) :: n INTEGER*4, intent(in) :: o INTEGER*4, intent(in) :: p REAL*8, intent(in), dimension(1:m, 1:p) :: A REAL*8, intent(in), dimension(1:n) :: x REAL*8, intent(out), dimension(1:%(iup-ilow)s) :: y INTEGER*4 :: i INTEGER*4 :: j do i = %(ilow)s, %(iup)s y(i) = 0 end do do i = %(ilow)s, %(iup)s do j = 1, n y(i) = %(rhs)s + y(i) end do end do end subroutine rrrE)rZiupZilowziup-ilowrrr) rJr=r>rrrrstrrIrr%r%r&test_partial_loops_fs2     rc Csddlm}ddlm}m}td\}}}td||||||g}t}|j|gdddd}|ddd kstt |dd d kst dS) Nrr rrZrr)Frlztest.f90rEz~REAL*8 function foo(x, y) implicit none REAL*8, intent(in) :: x REAL*8, intent(out) :: y y = sin(x) foo = cos(x) end function ) rr rrrrrrrrI) r rrr.r3r/rMrrnr%r%r&test_output_arg_fs  rcCsddlm}m}ddlm}|ddd\}}t|d\}}}|d|}t} td t|||d } t d t ||| ||} t | j | g} d } d }| | |ks| | |dddkst dS)Nrr<rrTrBrrDfuncrEZ test_inlinezsubroutine test_inline(m, x, y) implicit none INTEGER*4, intent(in) :: m REAL*8, intent(in), dimension(1:m) :: x REAL*8, intent(out), dimension(1:m) :: y INTEGER*4 :: i do i = 1, m y(i) = %s*%s end do end subroutine )zx(i)z (x(i) + 1)r)rJr=r>rrrKrrrrrr'rrI)r=r>rrrAr?r.r3rDrrr^rr_argsr%r%r&test_inline_functions    rcCsPtd}d}|D] }||7}qtd|}t}t|j|g}d}||ksLtdS)Nzx:20rr)a REAL*8 function test(x0, x1, x10, x11, x12, x13, x14, x15, x16, x17, x18, & x19, x2, x3, x4, x5, x6, x7, x8, x9) implicit none REAL*8, intent(in) :: x0 REAL*8, intent(in) :: x1 REAL*8, intent(in) :: x10 REAL*8, intent(in) :: x11 REAL*8, intent(in) :: x12 REAL*8, intent(in) :: x13 REAL*8, intent(in) :: x14 REAL*8, intent(in) :: x15 REAL*8, intent(in) :: x16 REAL*8, intent(in) :: x17 REAL*8, intent(in) :: x18 REAL*8, intent(in) :: x19 REAL*8, intent(in) :: x2 REAL*8, intent(in) :: x3 REAL*8, intent(in) :: x4 REAL*8, intent(in) :: x5 REAL*8, intent(in) :: x6 REAL*8, intent(in) :: x7 REAL*8, intent(in) :: x8 REAL*8, intent(in) :: x9 test = x0 + x1 + x10 + x11 + x12 + x13 + x14 + x15 + x16 + x17 + x18 + & x19 + x2 + x3 + x4 + x5 + x6 + x7 + x8 + x9 end function r)r.r-symr^rRr$r_r%r%r&test_f_code_call_signature_wrap s  rcs$td\ttfdddS)Nzx,XcstdfddS)Nr)f95r )rr%rr.r%r&r0Jrsz!test_check_case..)rrrr%r%rr&test_check_caseHs r c Csltd}tddd}ztd||fddWn:tk rf}z|jddrVtd W5d}~XYnXdS) Nr.T)Z my_assumptionr)rr rzFortran ignores case.z$This exception should not be raised!)rrrr startswithrI)x1Zx2rr%r%r&test_check_case_false_positiveMs r#cCstd\}}dd|fg}t|dddd}t|ddddd}|dd|ddksXtd||f}t|dddd}t|ddddd}|dd|ddkstdt||g||||ggf}t|d ddd}t|d dddd}|dd|ddkstdS) NrrrFrFrlrrEr)rrrIr)r.r3rrnZ expresultr%r%r& test_c_fortran_omit_routine_name[s   r$c Cstd\}}}||}t||g|dgg}d||ff}t|ddddd}|dd}d }|d } | d } | d} |d | i}||kstdS) NrZr)rFrlrrEa2REAL*8 function test(x, y, z, out_%(hash)s) implicit none REAL*8, intent(in) :: x REAL*8, intent(in) :: y REAL*8, intent(in) :: z REAL*8, intent(out), dimension(1:2, 1:2) :: out_%(hash)s out_%(hash)s(1, 1) = x out_%(hash)s(2, 1) = z out_%(hash)s(1, 2) = y out_%(hash)s(2, 2) = 16 test = x + y end function r_hash)rrr splitlinessplitrI) r.r3r/e1e2rrnr$r_r2rrr%r%r&test_fcode_matrix_outputms     r,c Cstd\}}}td\}}tddd}t|tdd|gg}t||||}t|d|}d|||gf} t| ddd d ||||||fd } | d d} d } | | kstdS) NrZrr?rErGrFr)rFrra!subroutine test(x, z, y, C, A, B) implicit none REAL*8, intent(in) :: x REAL*8, intent(in) :: z REAL*8, intent(in) :: y REAL*8, intent(out) :: C REAL*8, intent(out) :: B REAL*8, intent(out), dimension(1:1, 1:3) :: A C = z*(x + y) A(1, 1) = 1 A(1, 2) = 2 A(1, 3) = x B = 2*x end subroutine r) r.r3r/r@rkr?rurvrrrnr$r_r%r%r& test_fcode_results_named_ordereds    r-cCstddd}tddd}tddd}tddd}dt||d ddft||dddft||dddfgf}t|d dd d d }|d d}d }||kstdS)Nr?rFrGr@rErkrr)rrFrlasubroutine test(A, B, C, D) implicit none REAL*8, intent(in), dimension(1:2, 1:3) :: A REAL*8, intent(out), dimension(1:1, 1:3) :: B REAL*8, intent(out), dimension(1:1, 1:3) :: C REAL*8, intent(out), dimension(1:2, 1:1) :: D B(1, 1) = A(1, 1) B(1, 2) = A(1, 2) B(1, 3) = A(1, 3) C(1, 1) = A(2, 1) C(1, 2) = A(2, 2) C(1, 3) = A(2, 3) D(1, 1) = A(1, 3) D(2, 1) = A(2, 3) end subroutine rrr%r%r&test_fcode_matrixsymbol_slices     r.cCs|tddd}d|dddff}t|ddddd}|d d}d }|d}|d }|d}|d |i}||ksxtdS) Nr?rFrGr)rErFrlrzsubroutine test(A, out_%(hash)s) implicit none REAL*8, intent(in), dimension(1:2, 1:3) :: A REAL*8, intent(out), dimension(1:2, 1:1) :: out_%(hash)s out_%(hash)s(1, 1) = A(1, 2) out_%(hash)s(2, 1) = A(2, 2) end subroutine r&r')rrr(r)rI)r?rrnr$r_r2rrr%r%r&&test_fcode_matrixsymbol_slice_autonames     r/cCstd\}}}}td||fddd|fd}|dd}d}||ksHtd }td|||fd dd||fd}|dd}||kstdS) Nzx y z tfrF)r!r" global_varsrrEzPREAL*8 function f(x) implicit none REAL*8, intent(in) :: x f = x*y end function z#include "f.h" #include double f(double x, double y) { double f_result; f_result = x*y + z; return f_result; } rkrm)r.r3r/trnr$r_r%r%r&test_global_varss    r3c Csddlm}ddlm}|dddiid}td\}}|||}t|d gd }d }td |fd d |d}|dd} | |kstt|d}|j d d}td |fd d |d}|dd} | |kstdS)NrC99CodePrinter)expZuser_functionsr6Zfastexp)settingszx yz#include "fastexp.h")printerpreprocessor_statementsz#include "expr.h" #include "fastexp.h" double expr(double x, double y) { double expr_result; expr_result = fastexp(x + y); return expr_result; } r-F)r!r"rRrE)r8z#include "expr.h" #include #include "fastexp.h" double expr(double x, double y) { double expr_result; expr_result = fastexp(x + y); return expr_result; } ) sympy.printing.cr5rr6rrrrIr9append) r5r6r8r.r3r-genr_rnr$r%r%r&test_custom_codegens(           r=cCs^ddlm}Gddd|}td}|d}ddg}td |fd d d d |d }||ksZtdS)Nrr4c@seZdZddZdS)z*test_c_with_printer..CustomPrintercSsd||j||jS)Nzfastpow({}, {}))formatZ_printbaser6)selfr-r%r%r& _print_Pow+s z5test_c_with_printer..CustomPrinter._print_PowN)__name__ __module__ __qualname__rAr%r%r%r& CustomPrinter*srEr.rG)riz#include "file.h" #include double test(double x) { double test_result; test_result = fastpow(x, 3); return test_result; } )rjzP#ifndef PROJECT__FILE__H #define PROJECT__FILE__H double test(double x); #endif r)rkrF)r!r"r8)r:r5rrrI)r5rEr.r-r_rnr%r%r&test_c_with_printer's rFcCsddl}d|jj_tddd}tddd}td||fddddd }|dd }d }||ks`ttd}tddd}td||fddddd }|dd }d }||kstd|jj_dS) NrTr.)realr3r)rFrlrEzsREAL*8 function test(x, y) implicit none REAL*8, intent(in) :: x REAL*8, intent(in) :: y test = x + y end function z{COMPLEX*16 function test(x, y) implicit none COMPLEX*16, intent(in) :: x REAL*8, intent(in) :: y test = x + y end function )sympy.utilities.codegenZ utilitiesrZCOMPLEX_ALLOWEDr rI)Zsympyr.r3rnr$r_r%r%r&test_fcode_complexDs"        rIN)rFF)_iorZ sympy.corerrrrrrrr rr Z'sympy.functions.special.error_functionsr Zsympy.integrals.integralsr Zsympy.matricesrrrHrrrrrrrrrrrZsympy.testing.pytestrZsympy.utilities.lambdifyrr'rNrSrWrYr`rarerfrhrorwr{rrrrrrrrrrrrrrrrrrrrrrr r r r rrrrrrrrrr r#r$r,r-r.r/r3r=rFrIr%r%r%r&s      4    <D*, 3.>31/(.