a ze"G@sPdZddlZddlZddlZddlZddlZddlZddlZddl Z ddl Z GdddZ Gddde Z Gddde Z Gd d d ZGd d d ZGd ddZGdddejee ZGdddejee ZGdddejee ZGdddejee ZGdddejee ZGdddejee ZGdddejZedkrLedS)zUnit tests for the memoryview Some tests are in test_bytes. Many tests that require _testbuffer.ndarray are in test_buffer. Nc@seZdZdZeddZeddZddZdd Zd d Z d d Z ddZ ddZ ddZ ddZddZddZddZddZddZd d!Zd"d#Zd$d%Zd&d'Zd(d)Zd*d+Zd,d-Zd.d/Zd0d1Zd2d3Zd4d5Zd6d7Zd8S)9AbstractMemoryTestsabcdefcCs|jSN) source_bytesselfrH/opt/bitninja-python-dojo/embedded/lib/python3.9/test/test_memoryview.py_sourceszAbstractMemoryTests._sourcecCstd|j|jgSr)filterro_typerw_typerrrr _typesszAbstractMemoryTests._typescs$||j}t|}|||dtd|dt|dtd|dtd|dtd|t fdd|t fd d|t fd d|t fd d|t fd d|t fd d|t fddd|t||dS)NraficsdS)Nrrmrr 'z=AbstractMemoryTests.check_getitem_with_type..csdS)Nrrrrr r(rcs tjSrsysmaxsizerrrr r)rcs tj Srrrrrr r*rcsdSrrrrrr r,rcsdS)Nrrrrr r-rcsdS)Narrrrr r.r) r r getrefcount_view assertEqualordassertIsInstanceint assertRaises IndexError TypeError)rtpb oldrefcountrrr check_getitem_with_types"   z+AbstractMemoryTests.check_getitem_with_typecCs|jD]}||qdSr)rr*)rr'rrr test_getitem2s z AbstractMemoryTests.test_getitemcsJ|jD]>}||j}|||tfddttDqdS)Ncsg|] }|qSrr).0irrr :rz1AbstractMemoryTests.test_iter..)rr rr listrangelen)rr'r(rrr test_iter6s   zAbstractMemoryTests.test_itercs|js|d||j}t|}||fdd}|t|d|t|d|t|tdd| t||dS)Nno read-only type to testcs |d<dS)Nrr)valuerrr setitemBsz:AbstractMemoryTests.test_setitem_readonly..setitemrA) r skipTestr rrrr$r& memoryviewr )rr(r)r5rrr test_setitem_readonly<s     z)AbstractMemoryTests.test_setitem_readonlyc s|js|d|j||j}t|}||tdd<||dddd<||dddd <||d d dd<||d d dd<||d d ddd <||dd dd<dd d d<||dfdd}|t |dd|t |dd|t |tj d|t |tj d|t |dd|t |dd|t |t ddddfd|t |dt dddfd|t |dd|t |ddt dddt ddd f}|t ||djdkr tnt }|||dd |||dd|t|t ddd|t|t dd dd|t||dS)Nno writable type to test1rs1bcdef0s0bcdefs12s012defrrrscdedefsababcfcs||<dSrr)keyr4rr'rr r5csz:AbstractMemoryTests.test_setitem_writable..setitemrrrr)rrcsab)r r7r rrrr!_check_contentsr$r%rr&sliceNotImplementedErrorformat ValueErrorr )rr(r)r5ZslicesexcrrAr test_setitem_writableJsR     z)AbstractMemoryTests.test_setitem_writablec Cs|jD]|}||j}||}|t|d=Wdn1sD0Y|t|dd=Wdq1sx0YqdS)Nr=)rr rr$r&rr'r(rrrr test_delitem~s    $ z AbstractMemoryTests.test_delitemcsXjD]L}|j}|}dfdddD}|||tqdS)Nrc3s|]}t|gVqdSr) getitem_typebytes)r,rBrrr sz3AbstractMemoryTests.test_tobytes..r)rrr tobytesjoinr r"rN)rr'rr(Zexpectedrrr test_tobytess  z AbstractMemoryTests.test_tobytescCs8|jD],}|||j}|}||tdqdS)Nr)rrr tolistr r/)rr'rlrrr test_tolists zAbstractMemoryTests.test_tolistcs|jD]}|||j|jD]p}||dk||dk||dk||dk||dk||dkq"|k|ddk|ddddk|ddk|dk|dk|dk|dkdfD]^|tfdd |tfd d |tfd d |tfd d q,qdS) Nrsabcdesabcde1rrrZabcdefcskSrrrrBrrr rrz2AbstractMemoryTests.test_compare..cskSrrrrVrr rrcskSrrrrVrr rrcskSrrrrVrr rr)rrr assertTrueZ assertFalser$r&)rr'Ztp_comprrVr test_compares,   z AbstractMemoryTests.test_comparecCs|||j}||j|j||j|j||jd||jd|t|d||j|jf||j d|S)Nr=)rrr) rr r rFitemsizendimshaper1strides suboffsets)rr'rrrr check_attributes_with_typesz.AbstractMemoryTests.check_attributes_with_typecCs.|js|d||j}||jddS)Nr3T)r r7r^r readonlyrrrrr test_attributes_readonlys  z,AbstractMemoryTests.test_attributes_readonlycCs.|js|d||j}||jddS)Nr:F)r r7r^r r_r`rrr test_attributes_writables  z,AbstractMemoryTests.test_attributes_writablecCs~|jD]r}||j}t|}||}t|}t|d}||||d|t||d}|t||qdS)Nzutf-8) rr rrrstrrCencoder )rr'r(r)rZoldviewrefcountsrrr test_getbuffers      z"AbstractMemoryTests.test_getbufferc Cs|jD]}t|tsqGddd}Gddd|}Gddd}||d}||}|}||_||_t|}d}}}t | |du|||d}|}||_ ||_t|}d}}t | |du|qdS)Nc@seZdZddZdS)z+AbstractMemoryTests.test_gc..MyViewcSst||_dSr)r8r)rbaserrr __init__sz4AbstractMemoryTests.test_gc..MyView.__init__N)__name__ __module__ __qualname__rhrrrr MyViewsrlc@s eZdZdS)z-AbstractMemoryTests.test_gc..MySourceNrirjrkrrrr MySourcesrnc@s eZdZdS)z-AbstractMemoryTests.test_gc..MyObjectNrmrrrr MyObjectsroabc) r isinstancetyperroweakrefrefgcZcollectrWx) rr'rlrnror(rrswrrrr test_gcs.        zAbstractMemoryTests.test_gcc Cs|td}|t|Wdn1s.0Y||Wdn1sZ0Y||Wdn1s0Y||dWdn1s0Y|d|d<Wdn1s0Y|t|Wdn1s 0Y||jWdn1s80Y||jWdn1sd0Y||jWdn1s0Y||j Wdn1s0Y||j Wdn1s0Y||j Wdn1s0Y|6|Wdn1s@0YWdn1s`0Y| dt || dt||||||t||j||||jdS)NZreleasedrxzreleased memory)ZassertRaisesRegexrGrNrPrSr1rFrYrZr_r[r\ZassertInrcreprr ZassertNotEqualr8r )rrr'checkrrr _check_releaseds* ,,,,,.,,,,,,@ z#AbstractMemoryTests._check_releasedc Cs|jD]}||j}||}|}|||Wdn1sD0Y|||||}||Wdq1s0YqdSr)rr rassertIsr}release)rr'r(rcmrrr test_contextmanagers   *  z'AbstractMemoryTests.test_contextmanagercCsL|jD]@}||j}||}||||||||qdSr)rr rrr}rKrrr test_release#s    z AbstractMemoryTests.test_releasecCsJ|j}|dur|d||j}||}td}|t|j|dS)Nr3sZZZZ) r r7r rioBytesIOr$r&readinto)rr'r(rr-rrr test_writable_readonly-s    z*AbstractMemoryTests.test_writable_readonlycCs|t|jidSr)r$r&rrrrr test_getbuf_fail8sz$AbstractMemoryTests.test_getbuf_failcCs|j}|dur|d||j}||}|t|td||t|td||}||tt|dS)Nr3r) r r7r rr hashrr$rGrKrrr test_hash;s    zAbstractMemoryTests.test_hashcCs>|j}|dur|d||j}||}|tt|dS)Nr:)r r7r rr$rGrrKrrr test_hash_writableMs    z&AbstractMemoryTests.test_hash_writablecs||jD]p}||j}||}g|ffdd }t||}|||~tj||d|d|qdS)Ncs|dSr)append)rxr(Lrr callback\sz2AbstractMemoryTests.test_weakref..callbackr) rr rrtrur~testsupport gc_collect)rr'r(rrrxrrr test_weakrefVs     z AbstractMemoryTests.test_weakrefc Csj|jD]^}||j}||}tt|}|tt|||tt|t|dddqdS)Nr)rr rr/reversedrSr )rr'r(rZaslistrrr test_reversedes    z!AbstractMemoryTests.test_reversedcCsl|jD]`}||j}||}|}||j|t|j|||| |qdSr) rr r toreadonlyrWr_r8r rSr)rr'r(rZmmrrr test_toreadonlyms    z#AbstractMemoryTests.test_toreadonlycCstdgd}t|}|d}|d}|dd}t|}~||dd||dd||jd||jd|d}||dd||dd||jd||jddS)NH)rrrBrr?rI)arrayr8castr rF)rrrwrr(rBd_rrr test_issue22668xs     z#AbstractMemoryTests.test_issue22668N) rirjrkrpropertyr rr*r+r2r9rIrLrRrUrXr^rarbrfryr}rrrrrrrrrrrrrr rs<  4    &     rc@s eZdZeZeZeZdZdZ dS)BaseBytesMemoryTestsr=rN) rirjrkrNr bytearrayr rMrYrFrrrr rs rc@sTeZdZdZddZddZedjZdZe dddZ e d d d Z dS) BaseArrayMemoryTestsNcCstdt|SNr-)rr/rr(rrr rrzBaseArrayMemoryTests.cCstdt|Sr)rr/rPrrrr rrr-z/XXX test should be adapted for non-byte bufferscCsdSrrrrrr rfsz#BaseArrayMemoryTests.test_getbufferz:XXX NotImplementedError: tolist() only supports byte viewscCsdSrrrrrr rUsz BaseArrayMemoryTests.test_tolist) rirjrkr r rMrrYrFunittestskiprfrUrrrr rs  rc@seZdZddZddZdS)BaseMemoryviewTestscCst|Srr8)robjrrr rszBaseMemoryviewTests._viewcCs||||dSrr rr'rcontentsrrr rCsz#BaseMemoryviewTests._check_contentsN)rirjrkrrCrrrr rsrc@s(eZdZdZddZddZddZdS) BaseMemorySliceTestsXabcdefYcCst|}|ddSNr=rrrrrrr rszBaseMemorySliceTests._viewcCs||dd||dSrrrrrr rCsz$BaseMemorySliceTests._check_contentscCsF|jD]:}t||j}t|}|dd|t||qdS)Nr=r?)rr8r rrr )rr'rr)rrr test_refss    zBaseMemorySliceTests.test_refsN)rirjrkrrrCrrrrr rsrc@s eZdZdZddZddZdS)BaseMemorySliceSliceTestsrcCst|}|ddddS)Nrr=rrrrr rszBaseMemorySliceSliceTests._viewcCs||dd||dSrrrrrr rCsz)BaseMemorySliceSliceTests._check_contentsN)rirjrkrrrCrrrr rsrc@seZdZddZdS)BytesMemoryviewTestcCsv|jD]j}||j}|t||t|d|tt|tt|||jtt|d|jtt|ddqdS)N)object)argumentT)rr rWr8r$r&)rr'obrrr test_constructors   z$BytesMemoryviewTest.test_constructorN)rirjrkrrrrr rsrc@seZdZddZdS)ArrayMemoryviewTestcCsHtdtd}t|}tdtddd}||dd<|||dS)Nr- r)rr0r8r )rrrZnew_arrr test_array_assigns  z%ArrayMemoryviewTest.test_array_assignN)rirjrkrrrrr rsrc@s eZdZdS)BytesMemorySliceTestNrmrrrr rsrc@s eZdZdS)ArrayMemorySliceTestNrmrrrr rsrc@s eZdZdS)BytesMemorySliceSliceTestNrmrrrr rsrc@s eZdZdS)ArrayMemorySliceSliceTestNrmrrrr rsrc@s,eZdZddZddZddZddZd S) OtherTestc Cstjd}t|d}|}t|d}|dd|dd<|dd|dd<||jddD]}| |p|}t||}t||dd|dd<t||dd|dd<||jdWdqj1s0YqjdS)Nctypesg333333?rr?ZBbc) rr import_modulerNZc_doubler8rr r4ZsubTest)rrZp6rrrFrrr test_ctypes_casts  zOtherTest.test_ctypes_castcCs6dd}t|}|ddd}||dddS)Nr<i@ rZ30)r8r hex)rrwZm1Zm2rrr test_memoryview_hexszOtherTest.test_memoryview_hexcCs@td}|tt|Wdn1s20YdS)Nrp)r8r$r&copyr`rrr test_copys zOtherTest.test_copyc CsVtd}ttjdD]:}|tt||Wdq1sF0YqdS)Nrpr=)r8r0pickleHIGHEST_PROTOCOLr$r&dumps)rrprotorrr test_pickles zOtherTest.test_pickleN)rirjrkrrrrrrrr rsr__main__)__doc__rZ test.supportrrrvrtrrrrrrrrrrZTestCaserrrrrrrrimainrrrr sN   '