o g’×iN7ć@sødZddlZddlmZmZddlZddlmZmZGdd„deƒZ Gdd „d e ƒZ Gd d „d e ƒZ Gd d „d eƒZ Gdd„de ƒZ Gdd„de ƒZGdd„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Ś make_name)ŚFileŚMultiBlockSlicec@seZdZdd„Zdd„ZdS)Ś BaseSlicingcCst| ”dƒ|_dS)NŚw)rŚmktempŚf©Śself©r śU/home/ubuntu/SoloSpeech/.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 rrs rc@ó8eZdZdZdd„Zdd„Zdd„Zdd „Zd d „Zd S) ŚTestSingleElementzM Feature: Retrieving a single element works with NumPy semantics cCs.|jjtƒddd}|d}| |tj”dS)z; Single-element selection with [index] yields array scalar ©rŚi1©ŚdtyperN©r Ścreate_datasetrŚassertIsInstanceŚnpŚint8©r ŚdsetŚoutr r rŚtest_single_index)óz#TestSingleElement.test_single_indexcCó<|jjtƒddd}|d}| |tj”| |jd”dS)z3 Single-element selection with [()] yields ndarray rrrr N©r r rr!r"ŚndarrayŚ assertEqualŚshaper$r r rŚtest_single_null/óz"TestSingleElement.test_single_nullcCr))z* Slicing with [...] yields scalar ndarray r r ©r-r.Nr*r$r r rŚtest_scalar_index6r/z#TestSingleElement.test_scalar_indexcCs.|jjtƒddd}|d}| |tj”dS)z' Slicing with [()] yields array scalar r rr0Nrr$r r rŚtest_scalar_null=r(z"TestSingleElement.test_scalar_nullcCsZt ddg”}tjd|d}|jjtƒd|d}| |d|d”| |dtj”dS)z6 Compound scalar is numpy.void, not tuple (issue 135) )ŚaŚi4)ŚbŚf8)ér©ŚdatarN) r"rŚonesr r rr,r!Śvoid)r ŚdtŚvr%r r rŚ test_compoundCs zTestSingleElement.test_compoundN) rrrŚ__doc__r'r.r1r2r>r r r rr#s rc@r) ŚTestObjectIndexzH Feature: numpy.object_ subtypes map to real Python objects cCs<|jjtƒdtjd}|jj|d<| t|dƒtj”dS)z@ Indexing a reference dataset returns a h5py.Reference instance rrrN© r r rŚh5pyŚ ref_dtypeŚrefr,ŚtypeŚ Reference©r r%r r rŚtest_referenceQó zTestObjectIndex.test_referencecCsX|j tdƒdd”}|jd}|jjtdƒdtjd}||d<| t|dƒtj”d S) zL Indexing a region reference dataset returns a h5py.RegionReference Śx)é rKŚf4.ŚyrrrN) r r rŚ regionrefrBŚregionref_dtyper,rEŚRegionReference)r Śdset1ŚregrefŚdset2r r rŚ test_regrefWs  zTestObjectIndex.test_regrefcCs^t ddtjfg”}|jjtƒd|d}d|jdjf|d<|d}| t |dƒtj ”d S) z> Compound types of which a reference is an element work right )r3Śir5rré*ś/rrN) r"rrBrCr r rrDr,rErF©r r<r%r&r r rŚtest_reference_field`s z$TestObjectIndex.test_reference_fieldcCs<|jjtƒdtjd}|jj|d<| t|dƒtj”dS)z: Indexing returns a real Python object on scalar datasets r rNrArGr r rŚ test_scalarjrIzTestObjectIndex.test_scalarcCs<|jjtƒdtjddd}d|d<| t|dƒt”dS)zJ Indexing a byte string dataset returns a real python byte string rŚascii)Śencodingrs Hello there!rN)r r rrBŚ string_dtyper,rEŚbytesrGr r rŚ test_bytestrpszTestObjectIndex.test_bytestrN) rrrr?rHrTrYrZr_r r r rr@Ks  r@c@s0eZdZdZdd„Zdd„Zdd„Zdd „Zd S) ŚTestSimpleSlicingzQ Feature: Simple NumPy-style slices (start:stop:step) are supported. cCs4t| ”dƒ|_t d”|_|jjd|jd|_dS)NrrKrJr8)rr r r"ŚarangeŚarrr r%r r r rr}s zTestSimpleSlicing.setUpcCrrrr r r rr‚rzTestSimpleSlicing.tearDowncCs$| |jdd…|jdd…”dS)z0 Negative stop indexes work as they do in NumPy ééž’’’N)ŚassertArrayEqualr%rbr r r rŚtest_negative_stop†s$z$TestSimpleSlicing.test_negative_stopcCsx|j tƒdd”}t d”}|dd…df|dd…df<| t”||dd…df<WdƒdS1s5wYdS)z0Assigning to a 1D slice of a 2D dataset )rKrcrL)rKrNrr)r r rr"ŚzerosŚ assertRaisesŚ TypeError)r r%rJr r rŚ test_writeŠs   "’zTestSimpleSlicing.test_writeN)rrrr?rrrfrjr r r rr`ws  r`c@r) ŚTestArraySlicingz< Feature: Array types are handled appropriately cCsĀt d”}|jjtƒd|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,)f8©rKr.r6)rKér©rnrcé)rnrnN)r"rr r rr,r-rXr r rŚ test_readšs zTestArraySlicing.test_readcCsVt d”}|jjtƒd|d}| t” d|d<WdƒdS1s$wYdS)z@ Array fill from constant is not supported (issue 211). ś(3,)irmrrV.N)r"rr r rrhri)r r<r%r r rŚtest_write_broadcast°s   "’z%TestArraySlicing.test_write_broadcastcCsTt d”}|jjtƒd|d}t gd¢”}||d<|d}| t ||k””dS)zA Write a single element to the array Issue 211. rlrmr)rrcg@r7N)r"rr r rŚarrayŚ assertTrueŚall)r r<r%r9r&r r rŚtest_write_elementŗs z#TestArraySlicing.test_write_elementcCsžt d”}tjd|d}tjd|d}|jjtƒd|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)rcr)r7é)rKé é rrcr7rnrrxérzN)r"rr:r r rre)r r<Śdata1Śdata2r%r r rŚtest_write_slicesČs "z"TestArraySlicing.test_write_slicescCsJt d”}|jjtƒd|d}|d}||d<| t |d|k””dS)zO Read the contents of an array and write them back Issue 211. rlrmr.N)r"rr r rrurvrXr r rŚtest_roundtripŲs zTestArraySlicing.test_roundtripN) rrrr?rqrsrwr~rr r r rrk”s  rkc@s(eZdZdZdd„Zdd„Zdd„ZdS) ŚTestZeroLengthSlicingz. Slices resulting in empty arrays cCsŽtgd¢ƒD]f\}}|jjtd|›ƒ|tdt|ƒd}| |j|”|d}| |t j ”| |j|”|dd…}| |t j ”| |j|”t|ƒdkrl|dd…dd…f}| |t j ”| |jdd…d ”qdS) z] Slice a dataset with a zero in its shape vector along the zero-length dimension )©r)rrn)rrcrrJr©rŚmaxshape.Nrrc)rr© Ś enumerater r rŚintŚlenr,r-r!r"r+©r rUr-r%r&r r rŚ test_slice_zero_length_dimensionģs&  €ōz6TestZeroLengthSlicing.test_slice_zero_length_dimensioncCs‚tgd¢ƒD]8\}}|jjtd|›ƒ|tdt|ƒd}| |j|”|dd…}| |t j ”| |jd|dd…”qdS)z_ Slice a dataset with a zero in its shape vector along a non-zero-length dimension ))rnr)rrcr)rcrrrJrr‚Nrrr„rˆr r rŚtest_slice_other_dimensionżs& ūz0TestZeroLengthSlicing.test_slice_other_dimensioncCsˆtgd¢ƒD];\}}|jjtd|›ƒt |t”dt|ƒd}| |j |”|dd…}|  |tj ”| |j d|dd…”qdS)z5 Get a slice of length zero from a non-empty dataset )ro)rcrc)rcrrxrJr)r9rƒrrN) r…r r rr"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 rr€ę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 )r5rU)ŚcrL©édrcCs0t |”|jjdd|jd|_|j|jd<dS)NrJrŽr.)rrr r r<r%r9r r r rrs zTestFieldNames.setUpcCs| |jd|jd”dS)z! Test read with field selections r3N)rer%r9r r r rrqszTestFieldNames.test_readcCs\| |jd|jd”|j ”}|jjtƒ|d}d|d<d|d<| |d|d”dS)z, Unicode field names for for read and write r3r8rVN)rer%r9Ścopyr r r©r r9r%r r rŚtest_unicode_names"s  z!TestFieldNames.test_unicode_namescCsĄ|j ”}|jjtƒ|d}|dd9<||d<| t |d|k””|dd9<||d<| t |d|k””|dd9<|dd9<||d <| t |d|k””d S) z" Test write with field selections r8r3rc.r5r7rnr)r3rN©r9rr r rrur"rvr‘r r rrj,s zTestFieldNames.test_writecCsH|j ”}|jjtƒ|d}d|d<d|d<| t |d|k””dS)z4 Test write with non-compound source (single-field) r8gš?r5.Nr“r‘r r rŚtest_write_noncompound<s z%TestFieldNames.test_write_noncompoundN)rrrr?r"rr<r:r9rrqr’rjr”r r r rrŒ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)NrKrJr8)Śsuperrr"rarbr r r%r ©Ś __class__r rrHs  zTestMultiBlockSlice.setUpcCs2tƒ}| | d”d”tj |j||j”dS)NrK©rrrKr©rr,Śindicesr"ŚtestingŚassert_array_equalr%rb©r Śmbslicer r rŚ test_defaultMsz TestMultiBlockSlice.test_defaultcCs<tddddd}| | d”d”tj |j||j”dS)NrrKr©ŚstartŚcountŚstrideŚblockr™ršržr r rŚtest_default_explicitTsz)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_startZó "zTestMultiBlockSlice.test_startcCr§)NrØ©r£rK)rrrØr)rrrcrnr7rxr{r©ržr r rŚ test_count`s ’zTestMultiBlockSlice.test_countcCsDtdd}| t”| d”WdƒdS1swYdS)Nrzr¬rK©rrhŚ ValueErrorr›ržr r rŚ!test_count_more_than_length_errorhs   "’z5TestMultiBlockSlice.test_count_more_than_length_errorcCr§)Nrc)r¤rK)rrcrxr)rrcr7r{rpr©ržr r rŚ test_stridemr«zTestMultiBlockSlice.test_stridecCsB| t”tddd d”WdƒdS1swYdS)Nr©r¤r„rK)rhrÆrr›r r r rŚtest_stride_zero_errorss "žz*TestMultiBlockSlice.test_stride_zero_errorcCs8tddd}| | d”d”tj |j||j”dS)Nrcr²rK)rrcrxrcršržr r rŚtest_stride_block_equalxs z+TestMultiBlockSlice.test_stride_block_equalcCsp| t” tddWdƒn1swY| t”tdddWdƒdS1s1wYdS)Nrn)r„rcr²)rhrÆrr r r rŚ!test_block_more_than_stride_error~s  ’ "’z5TestMultiBlockSlice.test_block_more_than_stride_errorcCs@tddd}| | d”d”tj |j|t gd¢””dS)Nrnrcr²rK)rrnrnrc)rrrnr7r{rØr©ržr r rŚtest_stride_more_than_block…s "z/TestMultiBlockSlice.test_stride_more_than_blockcCsJtddddd}| t”| d”WdƒdS1swYdS)Nrcrxr7r”rKr®ržr r rŚ test_block_overruns_extent_error‹s  "’z4TestMultiBlockSlice.test_block_overruns_extent_errorcCsDtddddd}| | d”d”tj |j|t gd¢””dS) Nrrcrxr7r”rK)rrxrcr7)rrcrnr7r{rØrpryr©ržr r rŚtest_fully_described‘s ’z(TestMultiBlockSlice.test_fully_describedcCsBtdddd}| | d”d”tj |j|t gd¢””dS)Nrrnrc©r¢r¤r„rK)rrnrnrc)rrcr7rxrØ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—rr•Fs  r•)r?Śnumpyr"ŚcommonrrrBrrrrr@r`rkr€rŒr•r r r rŚs   (,R*6