o Tj\@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)Nsrcarray_from_pyobj)r__file__resolveparent)testrootri/home/dinkstrade/pdmp-scanner/venv/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.ctest_array_from_pyobj_ext) module_name)wraprr build_meson)rrrr setup_modules rcCst|d}t|S)N)r array_attrs flags2names)arrflagsrrr flags_info'srcCs0g}dD]}t|tt|d@r||q|S)N) CONTIGUOUSFORTRANOWNDATA ENSURECOPY ENSUREARRAYALIGNED NOTSWAPPED WRITEABLEWRITEBACKIFCOPY UPDATEIFCOPYBEHAVED BEHAVED_ROCARRAYFARRAYr)absgetattrrappend)rinfoflagnamerrrr,s  rc@s@eZdZgfddZddZddZddZd d Zd d Zd S)IntentcCsP|dd|_d}|D]}|dkr|tjO}q |ttd|O}q ||_dS)Nroptional F2PY_INTENT_) intent_listr F2PY_OPTIONALr+upperr)selfr2rirrr__init__Ds  zIntent.__init__cCs&|}|dkr d}||j|gS)Nin_in)lower __class__r2)r5namerrr __getattr__NszIntent.__getattr__cCsdd|jS)Nz intent(%s),)joinr2r5rrr__str__TszIntent.__str__cCs d|jS)Nz Intent(%r)r2r@rrr__repr__W zIntent.__repr__cstfdd|DS)Nc3s|]}|jvVqdSNrB).0r<r@rr [sz#Intent.is_intent..)allr5namesrr@r is_intentZszIntent.is_intentcGst|jt|ko |j|SrE)lenr2rKrIrrris_intent_exact]szIntent.is_intent_exactN) __name__ __module__ __qualname__r7r=rArCrKrMrrrrr/Cs   r/)BOOLBYTEUBYTESHORTUSHORTINTUINTLONGULONGLONGLONG ULONGLONGFLOATDOUBLECFLOATSTRING1STRING5rrQrRrSrTrUrVrWrXrYrZr[r\)rWr\r]r]r^r_r`win32)Darwinarm) LONGDOUBLECDOUBLE CLONGDOUBLE)rYr\r]rfrf)r^rgrhrhrgc@sPeZdZiZddZddZddZddZd d Zd d Z d dZ ddZ dS)TypecCst|tjr"|}d}tD]\}}t|ts!|j|jur!|}nq|j|d}|dur1|St |}| |||j|<|SrE) isinstancenpdtyperitemstype _type_cachegetr4object__new___init)clsr<dtype0nr6objrrrrrs   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) NrrrSTRINGrrSNPY_)r4NAMErr NPY_STRINGtype_numelsizerkrl startswithintr+itemsizernnumchar dtypechar)r5r<r-rrrrss$      z Type._initc Cs4d|jd|jd|jd|jd|jd|j S)NzType(z )|type_num=z, dtype=z, type=z , elsize=z , dtypechar=)r{r}rlrnr~rr@rrrrCsz Type.__repr__csfddtjDS)Ncg|]}|qSrr;rF_mr@rr z#Type.cast_types..) _cast_dictr{r@rr@r cast_typesszType.cast_typescsfddtDS)Ncrrrrr@rrrrz"Type.all_types..) _type_namesr@rr@r all_typesszType.all_typescCs:t|jj}g}tD]}t|j|kr|t|q |SrErr{ alignmentrr,rir5bitstypesr<rrr smaller_types zType.smaller_typescCsFt|jj}g}tD]}||jkrq t|j|kr |t|q |SrErrrrr equal_typess  zType.equal_typescCs:t|jj}g}tD]}t|j|kr|t|q |SrErrrrr larger_typesrzType.larger_typesN) rNrOrProrrrsrCrrrrrrrrrris ric@s4eZdZddZddZddZddZd d Zd S) Arrayc Cs,d|jd|jd|jd|jd|j S)NzArray(z, z)|arr=)rndimsintentrwrr@rrrrCs 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)NrrrrrcacherlCF)orderr#writerrcopy)#rnrrrdeepcopyobj_copyrwrcallr}r~rrrjrkndarrayrarr_attrrLrK F2PY_INTENT_Crpyarr pyarr_attrreprarrayreshaperrlsetflagstobytesr arr_equalrihas_shared_memory)r5typrrrwrrrr7s    *  zArray.__init__cCs|j|jkrdS||kS)NF)shaperH)r5arr1arr2rrrrVs  zArray.arr_equalcCs t|jSrE)strrr@rrrrA[rDz Array.__str__cCs@|j|jurdSt|jtjsdSt|j}|d|jdkS)z6Check that created array shares data with input array.TFr)rwrrjrkrrrr)r5obj_attrrrrr^s  zArray.has_shared_memoryN)rNrOrPrCr7rrArrrrrrs W rc@seZdZddZdS) TestIntentcCstttjjdks JtjjdsJtjjdrJtjjdds&Jtjjdds0Jtjdr8JdS)Nzintent(in,out)rr9)rrr8outrrKrMr@rrr test_in_outis zTestIntent.test_in_outN)rNrOrPrrrrrrhs 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)autousescopeparamscs"tjj_fddj_dS)Ncsttj|||SrE)rriparam)r5rrrwrequestrrwsz-TestSharedMemory.setup_type..)rirrtrnr)r5rrrr setup_typetszTestSharedMemory.setup_typecCs.|jjdr|jj}d|d|gSddgS)Nrx12rrrnr{rr~r5r~rrrnum2seqzszTestSharedMemory.num2seqcCsR|jjdr!|jj}d|d|d|gd|d|d|ggSgdgd gS) Nrxrr3456)rrr)rarrrrrrrnum23seqs zTestSharedMemory.num23seqcCs$|dgtj|j}|rJdS)Nr)rrr8rrr5arrrtest_in_from_2seqsz"TestSharedMemory.test_in_from_2seqcCsx|jD]4}tj|j|jd}|t|jgtj|}|j |jj kr3| s2Jt |jj|jfq| r9JqdSNr) rnrrkrrrlrLrr8r~rrr5trwrrrrtest_in_from_2casttypes z'TestSharedMemory.test_in_from_2casttyperwrorrrinp2seq23seqcCsdt|d|}tj||jj|d}|j|dkd||j|dkr%tjj p'tj|}| s0JdS)z5Test if intent(in) array can be passed without copiesrrlrrrrN) r+rkrrnrlrrrr8rr)r5rrrseqrwrrrrtest_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)rkrrrnrlrLrinoutrr8 TypeErrorrr SystemError)r5rwrmsgrrrtest_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)rkrrrnrlrLrr8rr ValueErrorrrr)r5rwrrrrrrtest_f_inout_23seqs& z#TestSharedMemory.test_f_inout_23seqcCsPtj|j|jjd}t|jt|jdf}||tjjj |}| s&JdSNrr) rkrrrnrlrLrr8rrrr5rwrrrrrtest_c_inout_23seqsz#TestSharedMemory.test_c_inout_23seqcCsL|jD]}tj|j|jd}|t|jgtjj |}| r#JqdSr) rnrrkrrrlrLrr8rrrrrrtest_in_copy_from_2casttypes z,TestSharedMemory.test_in_copy_from_2casttypecCs6|t|jt|jdgtj|j}|rJdSNr)rrLrrr8rrrrrtest_c_in_from_23seqs z%TestSharedMemory.test_c_in_from_23seqcCsV|jD]#}tj|j|jd}|t|jt|jdgtj|}| r(JqdSr) rnrrkrrrlrLrr8rrrrrtest_in_from_23casttypesz(TestSharedMemory.test_in_from_23casttypecCst|jD]2}tj|j|jdd}|t|jt|jdgtj|}|j |jj kr1| s0Jq| r7JqdSNrrr) rnrrkrrrlrLrr8r~rrrrrtest_f_in_from_23casttypesz*TestSharedMemory.test_f_in_from_23casttypecCst|jD]2}tj|j|jd}|t|jt|jdgtjj |}|j |jj kr1| s0Jq| r7JqdSr) rnrrkrrrlrLrr8rr~rrrrrtest_c_in_from_23casttypesz*TestSharedMemory.test_c_in_from_23casttypecCsZ|jD]%}tj|j|jdd}|t|jt|jdgtjj |}| r*JqdSr) rnrrkrrrlrLrr8rrrrrrtest_f_copy_in_from_23casttypesz/TestSharedMemory.test_f_copy_in_from_23casttypecCsZ|jD]%}tj|j|jd}|t|jt|jdgtjj j |}| r*JqdSr) rnrrkrrrlrLrr8rrrrrrrtest_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)rnrr~rkrrrlrLrr8rrrrrrrr)r5rrwrrrrrrtest_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)Nrxrrrz1intent(cache) should have failed on smaller array)rnrr{r~rk issubdtyperlintegerrriinfomaxrrLrr8rrrrr)r5ris_intrwrrrrr$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) rrrhiderrrrrrr5rrrrrrtest_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)rrrrrrrkzerosrnrlrrrrrrrrrr 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) rrr0rrrrkrrnrlrr)r5rrrrrtest_optional_noneds    z#TestSharedMemory.test_optional_nonecCs@|j}t|f}||tj|}|jj|ksJ|rJdSrE)rrLrrr0rrrrrrrtest_optional_from_2seqvs  z(TestSharedMemory.test_optional_from_2seqcCsx|j}t|t|df}||tj|}|jj|ksJ|r#J||tjj|}|jj|ks4J|r:JdSr) rrLrrr0rrrrrrrrtest_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) rkrrrnrlrrrinplacerrrrrr 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 ) rnrrkrrrlrrrr rr)r5rrwrrrrrtest_inplace_from_casttypes& ,$z+TestSharedMemory.test_inplace_from_casttypeN)"rNrOrPpytestfixturerrpropertyrrrrmark parametrizerrrrrrrrrrrrrrrrrr r rrrrrrrs@        rr)#sysrplatformrpathlibrnumpyrknumpy._core._type_aliasesr _c_names_dictrrdictrlrrrrr/rrrintpr clongdoublersystem processorextendrirrrrrrrsf         " Qo