o ZZh\@sddlZddlZddlZddlZddlmZddlZddlm Z ddl m Z da ed6dedie Z dd Zd d Zd d ZddZGdddZeZgdZddgiZeddged<eddged<dged<dged<edddged<edddged<edddged<edddged<eddged<eddged<eddged<eddged<edddged<edgded <edd!ged!<d"ged"<d#ged#<dged<ejjd$ksejjd%kr2ejd&kr2eefd'kr2egd(edgd)ed*<ed*gd+ed,<ed d!d-ged-<Gd.d/d/ZGd0d1d1Z Gd2d3d3Z!Gd4d5d5Z"dS)7N)Path) c_names_dict)util CHARACTERccCsttjd}|dS)NsrcZarray_from_pyobj)r__file__resolveparent)Ztestrootr U/var/www/auris/lib/python3.10/site-packages/numpy/f2py/tests/test_array_from_pyobj.py get_testdirsrcCs*tdurtdg}tj|ddadSdS)z6 Build the required testing extension module Nz wrapmodule.cZtest_array_from_pyobj_ext) module_name)wraprrZ build_meson)rr r r setup_modules rcCst|d}t|S)N)r array_attrs flags2names)arrflagsr r r flags_info'srcCs0g}dD]}t|tt|d@r||q|S)N) CONTIGUOUSFORTRANOWNDATAZ ENSURECOPYZ ENSUREARRAYZALIGNEDZ NOTSWAPPED WRITEABLEZWRITEBACKIFCOPYZ UPDATEIFCOPYZBEHAVEDZ BEHAVED_ROZCARRAYZFARRAYr)absgetattrrappend)rinfoZflagnamer r r r,s  rc@s@eZdZgfddZddZddZddZd d Zd d Zd S)IntentcCsP|dd|_d}|D]}|dkr|tjO}q |ttd|O}q ||_dS)NroptionalZ F2PY_INTENT_) intent_listrZ F2PY_OPTIONALrupperr)selfr"rir r r __init__Ds  zIntent.__init__cCs&|}|dkr d}||j|gS)Nin_in)lower __class__r")r$namer r r __getattr__NszIntent.__getattr__cCsdd|jS)Nz intent(%s),)joinr"r$r r r __str__TszIntent.__str__cCs d|jS)Nz Intent(%r)r"r/r r r __repr__W zIntent.__repr__cstfdd|DS)Nc3s|]}|jvVqdSNr1).0r+r/r r [sz#Intent.is_intent..)allr$namesr r/r is_intentZszIntent.is_intentcGst|jt|ko |j|Sr4)lenr"r:r8r r r is_intent_exact]szIntent.is_intent_exactN) __name__ __module__ __qualname__r&r,r0r2r:r<r r r r r Cs   r )BOOLBYTEUBYTESHORTUSHORTINTUINTLONGULONGLONGLONG ULONGLONGFLOATDOUBLECFLOATSTRING1STRING5rr@rArBrCrDrErFrGrHrIrJrK)rFrKrLrLrMrNrOwin32)Darwinarm) LONGDOUBLECDOUBLE CLONGDOUBLE)rHrKrLrUrU)rMrVrWrWrVc@sPeZdZiZddZddZddZddZd d Zd d Z d dZ ddZ dS)TypecCst|tjr"|}d}tD]\}}t|ts!|j|jur!|}nq|j|d}|dur1|St |}| |||j|<|Sr4) isinstancenpdtyperitemstype _type_cachegetr#object__new___init)clsr+Zdtype0nr%objr r r ras   z Type.__new__cCs||_|jdkrt|j}tj|_d|_td|_nC|j drGt|jdd}tj|_t |jddp9d|_td|j|_nt|j}t td|j|_|j |_t|j |_|j|jkshJ|j |_ |j|_dS) NrrrSTRINGrrSZNPY_)r#NAMErrZ NPY_STRINGtype_numelsizerZr[ startswithintritemsizer]numchar dtypechar)r$r+rr r r rbs$      z Type._initc Cs4d|jd|jd|jd|jd|jd|j S)NzType(z )|type_num=z, dtype=z, type=z , elsize=z , dtypechar=)rhrir[r]rjrpr/r r r r2sz Type.__repr__csfddtjDS)Ncg|]}|qSr r*r5_mr/r r z#Type.cast_types..) _cast_dictrhr/r r/r cast_typesszType.cast_typescsfddtDS)Ncrqr rrrsr/r r rurvz"Type.all_types..) _type_namesr/r r/r all_typesszType.all_typescCs:t|jj}g}tD]}t|j|kr|t|q |Sr4rrh alignmentryrrXr$bitstypesr+r r r smaller_types zType.smaller_typescCsFt|jj}g}tD]}||jkrq t|j|kr |t|q |Sr4r{r}r r r equal_typess  zType.equal_typescCs:t|jj}g}tD]}t|j|kr|t|q |Sr4r{r}r r r larger_typesrzType.larger_typesN) r=r>r?r^rarbr2rxrzrrrr r r r rXs rXc@s4eZdZddZddZddZddZd d Zd S) Arrayc Cs,d|jd|jd|jd|jd|j S)NzArray(z, z)|arr=)r]dimsintentrerr/r r r r2s zArray.__repr__cCsF||_||_||_t||_||_t|j |j ||j ||_ t |j tjs(Jt|j |_t|dkr|jdr^|j tj@sCJ|j j drKJ|j j dsSJ|jdtj@r]Jn"|j tj@rfJ|j j dsnJ|j j drvJ|jdtj@sJ|durd|_d|_dS|drt |tjsJtt|t|j||_n!tjtj||jdj||jdrdpd d |_|jj|ksJ|jj|j j d d |jj d sJ||ft|j|_t|dkr3|jdr|jj drJ|jj ds J|jdtj@rJn|jj dsJ|jj dr(J|jdtj@s3J|jd|jdks@J|jd|jdksMJ|jddkrv|jd|jdksvJt|jd|jd|j |jf|jddd|jdddksJt|jd|jdf|jd|jdksJt|jd|jdtd|jd|jdt|jd|f|dr|jdd|jj ksJn|jdd|jj ksJ| |j|j sJt |jtjr|j t!|jj kr|ds|jddkr|"s!JdSdSdSdSdS)Nrrrrrcacher[CF)orderrwriterrcopy)#r]rrrdeepcopyZobj_copyrercallrirjrrrYrZndarrayrarr_attrr;r:Z F2PY_INTENT_CrZpyarrZ pyarr_attrreprarrayZreshaperpr[setflagstobytesr arr_equalrXhas_shared_memory)r$typrrrer r r r&s    *  zArray.__init__cCs|j|jkrdS||kS)NF)shaper7)r$Zarr1Zarr2r r r rVs  zArray.arr_equalcCs t|jSr4)strrr/r r r r0[r3z Array.__str__cCs@|j|jurdSt|jtjsdSt|j}|d|jdkS)z6Check that created array shares data with input array.TFr)rerrYrZrrrr)r$Zobj_attrr r r r^s  zArray.has_shared_memoryN)r=r>r?r2r&rr0rr r r r rs W rc@seZdZddZdS) TestIntentcCstttjjdks JtjjdsJtjjdrJtjjdds&Jtjjdds0Jtjdr8JdS)Nzintent(in,out)rr()rrr'outrr:r<r/r r r test_in_outis zTestIntent.test_in_outN)r=r>r?rr r r r rhs rc@s"eZdZejddedddZeddZedd Z d d Z d d Z ej dddgej dddgej dddgddZddZddZddZdd Zd!d"Zd#d$Zd%d&Zd'd(Zd)d*Zd+d,Zd-d.Zd/d0Zd1d2Zd3d4Zd5d6Zd7d8Zd9d:Zd;d<Z d=d>Z!d?S)@TestSharedMemoryTclass)Zautousescopeparamscs"tjj_fddj_dS)Ncsttj|||Sr4)rrXparam)r$rrrerequestr r wsz-TestSharedMemory.setup_type..)rXrrcr]r)r$rr rr setup_typetszTestSharedMemory.setup_typecCs.|jjdr|jj}d|d|gSddgS)Nrf12rrr]rhrkrjr$rjr r r num2seqzszTestSharedMemory.num2seqcCsR|jjdr!|jj}d|d|d|gd|d|d|ggSgdgd gS) Nrfrr3456)rrr)rPrrrrr r r num23seqs zTestSharedMemory.num23seqcCs$|dgtj|j}|rJdS)Nr)rrr'rrr$ar r r test_in_from_2seqsz"TestSharedMemory.test_in_from_2seqcCsx|jD]4}tj|j|jd}|t|jgtj|}|j |jj kr3| s2Jt |jj|jfq| r9JqdSNr) r]rxrZrrr[r;rr'rjrrr$trerr r r test_in_from_2casttypes z'TestSharedMemory.test_in_from_2casttyperwrorrrinpZ2seqZ23seqcCsdt|d|}tj||jj|d}|j|dkd||j|dkr%tjj p'tj|}| s0JdS)z5Test if intent(in) array can be passed without copiesrnr[rrrrN) rrZrr]r[rrrr'rr)r$rrrseqrerr r r test_in_nocopyszTestSharedMemory.test_in_nocopyc Cstj|j|jjd}|t|jgtj|}|sJz|dgtj j|j}Wt dt yG}zt | ds<WYd}~dSd}~ww)Nrrz6failed to initialize intent(inout|inplace|cache) arrayz,intent(inout) should have failed on sequence)rZrrr]r[r;rinoutrr' TypeErrorrrk SystemError)r$rermsgr r r test_inout_2seqs z TestSharedMemory.test_inout_2seqc Cstj|j|jjdd}t|jt|jdf}||tjj|}| s&Jtj|j|jjdd}t|jt|jdf}z||tjj|}Wt dt yf}zt | ds[WYd}~dSd}~ww)Nrrrrz(failed to initialize intent(inout) arrayz2intent(inout) should have failed on improper array)rZrrr]r[r;rr'rr ValueErrorrrkr)r$rerrrr r r test_f_inout_23seqs& z#TestSharedMemory.test_f_inout_23seqcCsPtj|j|jjd}t|jt|jdf}||tjjj |}| s&JdSNrr) rZrrr]r[r;rr'rrrr$rerrr r r test_c_inout_23seqsz#TestSharedMemory.test_c_inout_23seqcCsL|jD]}tj|j|jd}|t|jgtjj |}| r#JqdSr) r]rxrZrrr[r;rr'rrrr r r test_in_copy_from_2casttypes z,TestSharedMemory.test_in_copy_from_2casttypecCs6|t|jt|jdgtj|j}|rJdSNr)rr;rrr'rrr r r test_c_in_from_23seqs z%TestSharedMemory.test_c_in_from_23seqcCsV|jD]#}tj|j|jd}|t|jt|jdgtj|}| r(JqdSr) r]rxrZrrr[r;rr'rrr r r test_in_from_23casttypesz(TestSharedMemory.test_in_from_23casttypecCst|jD]2}tj|j|jdd}|t|jt|jdgtj|}|j |jj kr1| s0Jq| r7JqdSNrrr) r]rxrZrrr[r;rr'rjrrr r r test_f_in_from_23casttypesz*TestSharedMemory.test_f_in_from_23casttypecCst|jD]2}tj|j|jd}|t|jt|jdgtjj |}|j |jj kr1| s0Jq| r7JqdSr) r]rxrZrrr[r;rr'rrjrrr r r test_c_in_from_23casttypesz*TestSharedMemory.test_c_in_from_23casttypecCsZ|jD]%}tj|j|jdd}|t|jt|jdgtjj |}| r*JqdSr) r]rxrZrrr[r;rr'rrrr r r test_f_copy_in_from_23casttypesz/TestSharedMemory.test_f_copy_in_from_23casttypecCsZ|jD]%}tj|j|jd}|t|jt|jdgtjj j |}| r*JqdSr) r]rxrZrrr[r;rr'rrrrr r r test_c_copy_in_from_23casttypesz/TestSharedMemory.test_c_copy_in_from_23casttypec Cs6|jD]}|j|jjkrqtj|j|jd}t|jf}||tj j j |}| s.J||tj j |}| s=Jtj|j|jdd}||tj j j |}| sWJ||tj j |}| skJt |jz||tj j |ddd}Wtdty}zt|dsWYd}~qd}~wwdS)Nrrr(failed to initialize intent(cache) arrayz8intent(cache) should have failed on multisegmented array)r]rzrjrZrrr[r;rr'rrrrrrrkr)r$rrerrrr r r test_in_cache_from_2casttypes8     z-TestSharedMemory.test_in_cache_from_2casttypec Cs|jD]`}|jdkr q|j|jjkrqt|jtj}|r.t|j dt |jj kr.qtj |j |jd}t |j f}z| |tjj|Wtdtye}zt|ds[WYd}~qd}~wwdS)Nrfrrrz1intent(cache) should have failed on smaller array)r]rzrhrjrZZ issubdtyper[integerrlrZiinfomaxrr;rr'rrrrkr)r$rZis_intrerrr r r $test_in_cache_from_2casttype_failures0    z5TestSharedMemory.test_in_cache_from_2casttype_failurec Csd}||tjjd}|jj|ksJd}||tjjd}|jj|ks&Jd}z||tjjd}Wt dtyQ}zt|dsFWYd}~dSd}~ww)Nrrrrr2failed to create intent(cache|hide)|optional arrayz8intent(cache) should have failed on undefined dimensions) rrrhiderrrrrkrr$rrrr r r test_cache_hidden3s(z"TestSharedMemory.test_cache_hiddenc Cs\d}||tjd}|jj|ksJ||jtj||jj ds"Jd}||tjd}|jj|ks4J||jtj||jj dsDJ|jj drP|jj drRJd}||tj jd}|jj|kseJ||jtj||jj dsuJ|jj ds|jj dsJd}z ||tjd}Wtdt y}zt |dsWYd}~dSd}~ww) Nrrrrrrrz7intent(hide) should have failed on undefined dimensions)rrrrrrrZzerosr]r[rrrrrkrrr r r test_hiddenGs8   zTestSharedMemory.test_hiddencCs d}||tjd}|jj|ksJ||jtj||jj ds"Jd}||tjd}|jj|ks4J||jtj||jj dsDJ|jj drP|jj drRJd}||tj jd}|jj|kseJ||jtj||jj dsuJ|jj ds|jj dsJdS)Nrrrrr) rrr!rrrrZrr]r[rr)r$rrr r r test_optional_noneds    z#TestSharedMemory.test_optional_nonecCs@|j}t|f}||tj|}|jj|ksJ|rJdSr4)rr;rrr!rrrrr r r test_optional_from_2seqvs  z(TestSharedMemory.test_optional_from_2seqcCsx|j}t|t|df}||tj|}|jj|ksJ|r#J||tjj|}|jj|ks4J|r:JdSr) rr;rrr!rrrrrr r r test_optional_from_23seq}s z)TestSharedMemory.test_optional_from_23seqcCstj|j|jjd}|jds|jdsJ|j}||tj|}|dd|j ddks7Jt ||j fd|j dd<|dd|j ddkrYtjd|jjdks\JJ|j |uscJ|jdsjJ|jdrqJdSNrrrrr6) rZrrr]r[rrrinplacerrrr r r test_inplaces,<zTestSharedMemory.test_inplacecCs:|jD]}||jur qtj|j|jd}|jj|jksJ|jj|jjus)J|jds3|jds5J|j}||tj |}|dd|j ddksVJt ||j fd|j dd<|dd|j ddkrxtjd|jjdks{JJ|j |usJ|jdsJ|jdrJ|jj|jjusJqdSr) r]rxrZrrr[rrrrrr)r$rrerrr r r test_inplace_from_casttypes& ,$z+TestSharedMemory.test_inplace_from_casttypeN)"r=r>r?pytestZfixtureryrpropertyrrrrmarkZ parametrizerrrrrrrrrrrrrrrrrrrrr r r r rrs@        rr )#sysrplatformrpathlibrnumpyrZZnumpy._core._type_aliasesrZ _c_names_dictrrdictr[rrrrr rryrwZintprmZ clongdoubler|system processorextendrXrrrr r r r sf         " Qo