o µ2śh6ć@s“dZddlZddlmZddlZddlmZmZGdd„deƒZGdd „d eƒZ Gd d „d eƒZ Gd d „d eƒZ Gdd„deƒZ Gdd„deƒZ Gdd„deƒZGdd„deƒZdS)a Dataset slicing test module. Tests all supported slicing operations, including read/write and broadcasting operations. Does not test type conversion except for corner cases overlapping with slicing; for example, when selecting specific fields of a compound type. éNé)ŚTestCase)ŚFileŚMultiBlockSlicec@seZdZdd„Zdd„ZdS)Ś BaseSlicingcCst| ”dƒ|_dS)NŚw)rŚmktempŚf©Śself©r śR/var/www/html/chatgem/venv/lib/python3.10/site-packages/h5py/tests/test_slicing.pyŚsetUpszBaseSlicing.setUpcCó|jr |j ”dSdS©N©r Ścloser r r r ŚtearDownó’zBaseSlicing.tearDownN)Ś__name__Ś __module__Ś __qualname__rrr r r r rs rc@ó8eZdZdZdd„Zdd„Zdd„Zdd „Zd d „Zd S) ŚTestSingleElementzM Feature: Retrieving a single element works with NumPy semantics cCs,|jjdddd}|d}| |tj”dS)z; Single-element selection with [index] yields array scalar Śx©rŚi1©ŚdtyperN©r Ścreate_datasetŚassertIsInstanceŚnpŚint8©r ŚdsetŚoutr r r Śtest_single_index)óz#TestSingleElement.test_single_indexcCó:|jjdddd}|d}| |tj”| |jd”dS)z3 Single-element selection with [()] yields ndarray rrrrr N©r r r!r"ŚndarrayŚ assertEqualŚshaper$r r r Śtest_single_null/óz"TestSingleElement.test_single_nullcCr))z* Slicing with [...] yields scalar ndarray rr r ©r-r.Nr*r$r r r Śtest_scalar_index6r/z#TestSingleElement.test_scalar_indexcCs,|jjdddd}|d}| |tj”dS)z' Slicing with [()] yields array scalar rr rr0Nrr$r r r Śtest_scalar_null=r(z"TestSingleElement.test_scalar_nullcCsXt ddg”}tjd|d}|jjdd|d}| |d|d”| |dtj”dS) z6 Compound scalar is numpy.void, not tuple (issue 135) )ŚaŚi4)ŚbŚf8)érŚfoo©ŚdatarN)r"rŚonesr r r,r!Śvoid)r ŚdtŚvr%r r r Ś test_compoundCs zTestSingleElement.test_compoundN) rrrŚ__doc__r'r.r1r2r?r r r r r#s rc@r) ŚTestObjectIndexzH Feature: numpy.object_ subtypes map to real Python objects cCs:|jjddtjd}|jj|d<| t|dƒtj”dS)z@ Indexing a reference dataset returns a h5py.Reference instance rrrrN©r r Śh5pyŚ ref_dtypeŚrefr,ŚtypeŚ Reference©r r%r r r Śtest_referenceQó zTestObjectIndex.test_referencecCsN|j dd”}|jd}|jjddtjd}||d<| t|dƒtj”dS) zL Indexing a region reference dataset returns a h5py.RegionReference r)é rK.ŚyrrrN)r r Ś regionrefrCŚregionref_dtyper,rFŚRegionReference)r Śdset1ŚregrefŚdset2r r r Ś test_regrefWs  zTestObjectIndex.test_regrefcCs\t ddtjfg”}|jjdd|d}d|jdjf|d<|d}| t|d ƒtj ”d S) z> Compound types of which a reference is an element work right )r3Śir5rrré*ś/rrN) r"rrCrDr r rEr,rFrG©r r=r%r&r r r Śtest_reference_field`s z$TestObjectIndex.test_reference_fieldcCs:|jjddtjd}|jj|d<| t|dƒtj”dS)z: Indexing returns a real Python object on scalar datasets rr rNrBrHr r r Ś test_scalarjrJzTestObjectIndex.test_scalarcCs:|jjddtjddd}d|d<| t|dƒt”dS) zJ Indexing a byte string dataset returns a real python byte string rrŚascii)Śencodingrs Hello there!rN)r r rCŚ string_dtyper,rFŚbytesrHr r r Ś test_bytestrpszTestObjectIndex.test_bytestrN) rrrr@rIrSrXrYr^r r r r rAKs  rAc@s0eZdZdZdd„Zdd„Zdd„Zdd „Zd S) ŚTestSimpleSlicingzQ Feature: Simple NumPy-style slices (start:stop:step) are supported. cCs4t| ”dƒ|_t d”|_|jjd|jd|_dS)NrrKrr9)rrr r"ŚarangeŚarrr r%r r r r r}s zTestSimpleSlicing.setUpcCrrrr r r r r‚rzTestSimpleSlicing.tearDowncCs$| |jdd…|jdd…”dS)z0 Negative stop indexes work as they do in NumPy ééž’’’N)ŚassertArrayEqualr%rar r r r Śtest_negative_stop†s$z$TestSimpleSlicing.test_negative_stopcCst|j dd”}t d”}|dd…df|dd…df<| t”||dd…df<WdƒdS1s3wYdS)z0Assigning to a 1D slice of a 2D dataset Śx2)rKrb)rKrNrr)r r r"ŚzerosŚ assertRaisesŚ TypeError)r r%rr r r Ś test_writeŠs   "’zTestSimpleSlicing.test_writeN)rrrr@rrrerjr r r r r_ws  r_c@r) ŚTestArraySlicingz< Feature: Array types are handled appropriately cCsĄt d”}|jjdd|d}| |jd”| |j|”|d}| |jt d””| |jd”|d}| |jt d””| |jd ”|d d d …}| |jt d””| |jd ”d S)z; Read arrays tack array dimensions onto end of shape tuple ś(3,)f8r©rKr.r6)rKér©rnrbé)rnrnN)r"rr r r,r-rWr r r Ś test_readšs zTestArraySlicing.test_readcCsTt d”}|jjdd|d}| t” d|d<WdƒdS1s#wYdS)z@ Array fill from constant is not supported (issue 211). ś(3,)irrmrrU.N)r"rr r rhri)r r=r%r r r Śtest_write_broadcast°s   "’z%TestArraySlicing.test_write_broadcastcCsRt d”}|jjdd|d}t gd¢”}||d<|d}| t ||k””dS)zA Write a single element to the array Issue 211. rlrrmr)rrbg@r7N)r"rr r ŚarrayŚ assertTrueŚall)r r=r%r:r&r r r Śtest_write_elementŗs z#TestArraySlicing.test_write_elementcCsœt d”}tjd|d}tjd|d}|jjdd|d}||dddd …f<| |dddd …f|”||d d d …d d…f<| |d d d …d d…f|”dS)z Write slices to array type rr)rbr)r7ér)rKé é rrbr7rnrrxérzN)r"rr;r r rd)r r=Śdata1Śdata2r%r r r Śtest_write_slicesČs "z"TestArraySlicing.test_write_slicescCsHt d”}|jjdd|d}|d}||d<| t |d|k””dS)zO Read the contents of an array and write them back Issue 211. rlrrmr.N)r"rr r rurvrWr r r Śtest_roundtripŲs zTestArraySlicing.test_roundtripN) rrrr@rqrsrwr~rr r r r rk”s  rkc@s(eZdZdZdd„Zdd„Zdd„ZdS) ŚTestZeroLengthSlicingz. Slices resulting in empty arrays cCsŲtgd¢ƒD]c\}}|jjd||tdt|ƒd}| |j|”|d}| |tj ”| |j|”|dd…}| |tj ”| |j|”t|ƒdkri|dd…dd…f}| |tj ”| |jdd…d ”qdS) z] Slice a dataset with a zero in its shape vector along the zero-length dimension )©r)rrn)rrbrśx%dr©rŚmaxshape.Nrrb)rr© Ś enumerater r ŚintŚlenr,r-r!r"r+©r rTr-r%r&r r r Ś test_slice_zero_length_dimensionģs   €ōz6TestZeroLengthSlicing.test_slice_zero_length_dimensioncCs|tgd¢ƒD]5\}}|jjd||tdt|ƒd}| |j|”|dd…}| |tj ”| |jd|dd…”qdS)z_ Slice a dataset with a zero in its shape vector along a non-zero-length dimension ))rnr)rrbr)rbrrr‚rrƒNrrr…r‰r r r Śtest_slice_other_dimensionżs  ūz0TestZeroLengthSlicing.test_slice_other_dimensioncCs‚tgd¢ƒD]8\}}|jjd|t |t”dt|ƒd}| |j|”|dd…}|  |tj ”| |jd|dd…”qdS)z5 Get a slice of length zero from a non-empty dataset )ro)rbrb)rbrrxr‚r)r:r„rrN) r†r r r"rgr‡rˆr,r-r!r+r‰r r r Śtest_slice_of_length_zeros& ūz/TestZeroLengthSlicing.test_slice_of_length_zeroN)rrrr@rŠr‹rŒr r r r r€ęs  r€c@sTeZdZdZe gd¢”ZejdedZdd„Z dd„Z d d „Z d d „Z d d„Z dS)ŚTestFieldNamesz* Field names for read & write ))r3r )r5rT)ŚcŚf4©édrcCs0t |”|jjdd|jd|_|j|jd<dS)Nrrr.)rrr r r=r%r:r r r r rs zTestFieldNames.setUpcCs| |jd|jd”dS)z! Test read with field selections r3N)rdr%r:r r r r rqszTestFieldNames.test_readcCsN| |jd|jd”d|jd<|j ”}d|d<| |jd|d”dS)z, Unicode field names for for read and write r3rUN)rdr%r:Ścopy)r r:r r r Śtest_unicode_names"s   z!TestFieldNames.test_unicode_namescCsŗ|j ”}|dd9<||jd<| t |jd|k””|dd9<||jd<| t |jd|k””|dd9<|dd9<||jd<| t |jd|k””d S) z" Test write with field selections r3rb.r5r7rnrŽ)r3rŽN©r:r’r%rur"rv©r r}r r r rj*s    zTestFieldNames.test_writecCs:|j ”}d|d<d|jd<| t |jd|k””dS)z4 Test write with non-compound source (single-field) gš?r5.Nr”r•r r r Śtest_write_noncompound8s  z%TestFieldNames.test_write_noncompoundN)rrrr@r"rr=r;r:rrqr“rjr–r r r r rs rcsŒeZdZ‡fdd„Zdd„Zdd„Zdd„Zd d „Zd d „Zd d„Z dd„Z dd„Z dd„Z dd„Z dd„Zdd„Zdd„Zdd„Z‡ZS)ŚTestMultiBlockSlicecs.tƒ ”t d”|_|jjd|jd|_dS)NrKrr9)Śsuperrr"r`rar r r%r ©Ś __class__r r rBs  zTestMultiBlockSlice.setUpcCs2tƒ}| | d”d”tj |j||j”dS)NrK©rrrKr©rr,Śindicesr"ŚtestingŚassert_array_equalr%ra©r Śmbslicer r r Ś test_defaultGsz TestMultiBlockSlice.test_defaultcCs<tddddd}| | d”d”tj |j||j”dS)NrrKr©ŚstartŚcountŚstrideŚblockr›rœr r r r Śtest_default_explicitNsz)TestMultiBlockSlice.test_default_explicitcCó>tdd}| | d”d”tj |j|t gd¢””dS)Nr7)r¤rK)r7rr{r)r7rxr{érpry©rr,rr"ržrŸr%rtr r r r Ś test_startTó "zTestMultiBlockSlice.test_startcCr©)NrŖ©r„rK)rrrŖr)rrrbrnr7rxr{r«r r r r Ś test_countZs ’zTestMultiBlockSlice.test_countcCsDtdd}| t”| d”WdƒdS1swYdS)Nrzr®rK©rrhŚ ValueErrorrr r r r Ś!test_count_more_than_length_errorbs   "’z5TestMultiBlockSlice.test_count_more_than_length_errorcCr©)Nrb)r¦rK)rrbrxr)rrbr7r{rpr«r r r r Ś test_stridegr­zTestMultiBlockSlice.test_stridecCsB| t”tddd d”WdƒdS1swYdS)Nr©r¦r§rK)rhr±rrr r r r Śtest_stride_zero_errorms "žz*TestMultiBlockSlice.test_stride_zero_errorcCs8tddd}| | d”d”tj |j||j”dS)Nrbr“rK)rrbrxrbrœr r r r Śtest_stride_block_equalrs z+TestMultiBlockSlice.test_stride_block_equalcCsp| t” tddWdƒn1swY| t”tdddWdƒdS1s1wYdS)Nrn)r§rbr“)rhr±rr r r r Ś!test_block_more_than_stride_errorxs  ’ "’z5TestMultiBlockSlice.test_block_more_than_stride_errorcCs@tddd}| | d”d”tj |j|t gd¢””dS)Nrnrbr“rK)rrnrnrb)rrrnr7r{rŖr«r r r r Śtest_stride_more_than_blocks "z/TestMultiBlockSlice.test_stride_more_than_blockcCsJtddddd}| t”| d”WdƒdS1swYdS)Nrbrxr7r£rKr°r r r r Ś test_block_overruns_extent_error…s  "’z4TestMultiBlockSlice.test_block_overruns_extent_errorcCsDtddddd}| | d”d”tj |j|t gd¢””dS) Nrrbrxr7r£rK)rrxrbr7)rrbrnr7r{rŖrpryr«r r r r Śtest_fully_described‹s ’z(TestMultiBlockSlice.test_fully_describedcCsBtdddd}| | d”d”tj |j|t gd¢””dS)Nrrnrb©r¤r¦r§rK)rrnrnrb)rrbr7rxrŖrpr«r r r r Śtest_count_calculated“s"z)TestMultiBlockSlice.test_count_calculatedcCsHtdddd}| t”| d”WdƒdS1swYdS)Nrpr7rnr»rKr°r r r r Ś test_zero_count_calculated_erroršs  "’z4TestMultiBlockSlice.test_zero_count_calculated_error)rrrrr¢rØr¬rÆr²r³rµr¶r·rør¹rŗr¼r½Ś __classcell__r r r™r r—@s  r—)r@Śnumpyr"ŚcommonrrCrrrrrAr_rkr€rr—r r r r Śs   (,R*0