K iddlZddlZddlmZmZddlmZmZmZm Z m Z m Z m Z m Z mZmZddlmZmZmZmZmZmZmZmZmZmZmZmZmZmZmZm Z m!Z!m"Z"m#Z#m$Z$m%Z%ddl&m'Z'ddl(m)Z)ddl*m+Z+ddl,m-Z-dd l.m/Z/m0Z0m1Z1dd l2m3Z3dd l4m5Z5m6Z6e+d Z7e+d Z8dZ9e3dZ:e6e3dZ;e3dZdZ?dZ@dZAdZBdZCe3dZDy)N)Symbolsymbols) AssignmentPrint DeclarationFunctionDefinitionReturnreal FunctionCallVariableElementinteger) allocatableArrayConstructorisigndsigncmplxkind literal_dpProgramModuleuse Subroutine dimensionassumed_extent ImpliedDoLoop intent_outsizeDoSubroutineCallsum_arraybind_C)render_as_module) unchanged) import_module)fcode) has_fortrancompile_run_stringscompile_link_import_strings) may_xfail)skipXFAILcythonnumpycXtdd}t|}t|ddk(sJy)NxTr free source_formatzsize(x))rrr')r1sxs e/mnt/ssd/data/python-lab/Trading/venv/lib/python3.12/site-packages/sympy/codegen/tests/test_fnodes.py test_sizer8s,sA aB 6 *i 77 7cts tdtdd}tt |dzt |z dzg}t |dgd }ttd |g|}t|gd td t|gd fd gd\\}}}d|vsJ|dk(sJ|dtjk(sJy)NNo fortran compiler found.aTr2?:indimintentrmsmod_rmsrms.f90)main.f90zprogram myprog use mod_rms, only: rms real*8, dimension(4), parameter :: x = [4, 2, 2, 2] print "(f7.5)", dsqrt(7d0) - rms(x) end program cleanz0.00000 exit_status) r(r,rr r!rr"rr r$r)osEX_OK)r<bodyarrfdstdoutstderrinfos r7test_size_assumed_shaperTs = )*sA DAJtAw&+ , -D uT *C D%# 5BbT9%0 $bT956  2 VVd    R<<  "(( ** *r9cts tdtdd\}}t|dz|ddd}t d|d g}t |d gt g }td |jt||t|gg}t|d d}td|fgd\\}}}djD]} | |vrJ|dk(sJ|dtjk(sJy)Nr;za iTrr=ir?)rBattrsidlprogir3standardr5rGrHz-28 -27 -1 1 27 28rJrK)r(r,rrrr"rras_Declarationrrr'r)splitrLrM) r<iidlacprogfsrcrQrRrSnumstrs r7test_ImpliedDoLooprf8s = )* 5$ 'DAq 1aQ *C 3R. )B acU;-0A 9 1b qc  D F ;D0:t2D1ETRVVd&,,.  R<<  "(( ** *r9c^tdd}tj|d}t|}t ||dzg}t d||g}t s tdtdt|d fgd \\}}}d |vsJd |vsJ|dk(sJ|dtjk(sJy)Nr1Tr2*foor;rGZr]rH4243rJrK) rr deducedrrrr(r,r)r'rLrM)r1vxdeclprntrcrQrRrSs r7 test_ProgramrsOssA   !R B r?D !QqS?D 54, 'D = )*0:uTTV?W2X1YaefVVd 6>> 6>> R<<  "(( ** *r9c tdd}tj|}ttd|gt |dzg}t dg|g}tddg}tdtddg td |gg}ts td td t|d fdt|d fgd\\}}}d|vsJtd|vsJ|dk(sJy)Nr1Tr2sqrr=mod_sqgE@foobaronlyz"Square of 42 = "r;z mod_sq.f90rkrlrGrHrmirJ)rr rorr r rr rrrr(r,r)r'str) r1v_xsqrvsq_callprg_sqrQrRrSs r7 test_Moduler`ssA   1 C D%#1 ?B Hb2$ 'F53%(G X HE7# "G,- F = )*0 uVb12 U6B/02VVd 6>> u:   R< $K LC   1 C 3 C71qc?JqM!Q$$67!SG S3%CC3Q  " C sA   Ail^ 4D se ,C 5 CseDsTF# tDz4 !  D = )*0 %b)* %r*+2VVd#1a[ )3q!t8 )C ) s3x="  '' ' %1vcr{f$$$% R<< *s&G3c~tdd}ttd|sJttd|dddk(sJy) Nr1TrVri_r3r\z isign(1, x))rr%rr'r1s r7 test_isignrs=sD!A UAq !! ! q!r @M QQ Qr9ctd}ttd|sJttt d|dddk(sJy)Nr1rirr3r\z dsign(1d0, x))rr%rr'rrs r7 test_dsignrs@s A UAq !! ! z!}a(2V LP_ __ _r9c@td}ttd|sJy)Nr1ri)rr%rrs r7 test_cmplxrss A UAq !! !r9c>td}tt|sJy)Nr1)rr%rrs r7 test_kindrss A T1  r9c:ttdddk(sJy)Nrr3r40d0)r'rr9r7test_literal_dprs Af 5 >> >r9c lts tdts tdts tdt dd}t dd}t t |d z|z d zg}t||gd }ttd ||g|td g}t|gd}tj5}td|fddjddzfg|\}}t!|j#tjgddz dksJ dddy#1swYyxYw)Nr;zCython not found.zNumPy not found.r<Tr2srVr=r>r@rArDrrErFz_rms.pyxz#cython: language_level={} 3zscdef extern double rms(double*, int*) def py_rms(double[::1] x): cdef int s = x.size return rms(&x[0], &s) ) build_dir)@g@rrgҩ*@g+=)r(r,r.nprr r!r"rr r#r$tempfileTemporaryDirectoryr*formatabspy_rms) r<rrNrOrPf_modfolderrrSs r7 test_bind_Crs, = )*  !   sAsD!A DAJqL2% & 'D s4 (C D%#q4u  OB bT9 -E  $ $ & L&/   .55c:.. 01  T3::bhh'789FBCeKKK L L Ls A D**D3)ErLrsympy.core.symbolrrsympy.codegen.astrrrrr r r r r rsympy.codegen.fnodesrrrrrrrrrrrrrrrrrr r!r"r#sympy.codegen.futilsr$sympy.core.exprr%sympy.externalr&sympy.printing.codeprinterr'sympy.utilities._compilationr(r)r*!sympy.utilities._compilation.utilr+sympy.testing.pytestr,r-r.rr8rTrfrsrrrrrrrrrr9r7rs / 2%(,ff7, x 78  + +0 + +*  +  +   * & &RR ` "  ? L Lr9