bg HddlZddlZddlZddlmZmZGddZdS)N)assert_array_equalIS_PYPYc.eZdZejeddZdZdZ ejeddZ ej de j e je je je je je je je je je je je je jgdZd Zd Zd Zej d ed dZ dZ!dZ"dZ#dZ$dZ%dZ&dS) TestDLPackzPyPy can't get refcounts.)reasonctjd}|}tj|dksJ~tj|dksJdSN)nparange __dlpack__sys getrefcountselfxys o/builddir/build/BUILD/cloudlinux-venv-1.0.7/venv/lib64/python3.11/site-packages/numpy/core/tests/test_dlpack.pytest_dunder_dlpack_refcountz&TestDLPack.test_dunder_dlpack_refcount s[ IaLL LLNNq!!Q&&&& q!!Q&&&&&&ctjd}|dtjt 5|dddddS#1swxYwYdS)Nr )stream)r rrpytestraises RuntimeErrorrrs rtest_dunder_dlpack_streamz$TestDLPack.test_dunder_dlpack_streams IaLL D !!! ]< ( ( # # LLL " " " # # # # # # # # # # # # # # # # # #sA((A,/A,c*tjdtjfdtjfg}tjd|}|d}t jt5tj|ddddS#1swxYwYdS)Nintchar)r dtype) r r%int32int8zerosrr BufferError from_dlpack)rdtrzs r%test_strides_not_multiple_of_itemsizez0TestDLPack.test_strides_not_multiple_of_itemsizes Xrx(627*;< = = HT $ $ $ eH ]; ' '   N1                     s&BB B ctjd}tj|}tj|dksJ~tj|dksJdSr )r rr*rrrs rtest_from_dlpack_refcountz$TestDLPack.test_from_dlpack_refcount s] IaLL N1  q!!Q&&&& q!!Q&&&&&&rr%ctjd|}tj|}|j|jksJt ||dSNr )r rastyper*r%r)rr%rrs rtest_dtype_passthroughz!TestDLPack.test_dtype_passthrough(sY IaLL   & & N1  w!'!!!!1a     rctjtjd}tjt 5tj|ddddS#1swxYwYdS)Nz 2021-05-27)r asarray datetime64rrr)r*rs rtest_invalid_dtypezTestDLPack.test_invalid_dtype6s Jr}\22 3 3 ]; ' '   N1                     sA""A&)A&c tjd}tjd|}t jt 5tj|ddddS#1swxYwYdS)Nz=i8r r$)r r% newbyteorderrrrr)r*)rr+rs rtest_invalid_byte_swappingz%TestDLPack.test_invalid_byte_swapping<s Xe__ ) ) + + Iar " " " ]; ' '   N1                     sA88A<?A<cDtjdd}|d}t|tj||dddf}t|tj||dddf}t|tj||d}t|tj|tj|}t|tj|dS)N)r r rr)r rreshaperr*diagonalcopy)rry1y2y3y4y5s rtest_non_contiguouszTestDLPack.test_non_contiguousCs IbMM ! !& ) ) qT2r~b11222 qqq!tW2r~b11222 q!!!tW2r~b11222 qT2r~b11222 [^^ " "2r~b1122222rndim!cd|z}tj|tj}|tj|jksJdS)N)rr$)r r(float64r*shape)rrFrJrs rtest_higher_dimszTestDLPack.test_higher_dimsUsEt  HU"* - - -q))///////rctjd}|dksJtj|}|dksJ|ddd}|dksJdS)Nr )rrr )r r__dlpack_device__r*)rrrr,s rtest_dlpack_devicezTestDLPack.test_dlpack_device\s IaLL""$$.... N1  ""$$.... cccF""$$......rc`tjd}|}tr1)r rrr)rr_s rdlpack_deleter_exceptionz#TestDLPack.dlpack_deleter_exceptionds" IaLL LLNNrctjt5|ddddS#1swxYwYdS)N)rrrrQ)rs r test_dlpack_destructor_exceptionz+TestDLPack.test_dlpack_destructor_exceptionis ]< ( ( , ,  ) ) + + + , , , , , , , , , , , , , , , , , ,s<AActjd}d|j_t jt 5|ddddS#1swxYwYdS)Nr F)r rflags writeablerrr)rrs r test_readonlyzTestDLPack.test_readonlyms IaLL! ]; ' '   LLNNN                  sAA #A cvtjd}tj|}t||dS)Ng?)r arrayr*rrs r test_ndim0zTestDLPack.test_ndim0ss4 HSMM N1  1a     rc tjdddtjdtjd}tj|}t ||dS)Nf8) r r)Pir$F)r%rJstridesbufferorder)r ndarrayonesuint8r*rrs rtest_size1dims_arraysz TestDLPack.test_size1dims_arraysxs\ JTZ gd"(;;;3 H H H N1  1a     rN)'__name__ __module__ __qualname__rmarkskipifrrr r-r/ parametrizer bool_r'int16r&int64rguint16uint32uint64float16float32rI complex64 complex128r3r7r:rErangerKrNrQrSrWrZrhrrrrs [(CDD''ED'### [(CDD''ED' [W  28RX ")RY  BJ  bm '!!! 333$ [VUU2YY//000/0 /// ,,, !!! !!!!!rr)rrnumpyr numpy.testingrrrrzrrr}sw 55555555t!t!t!t!t!t!t!t!t!t!r