a aK6@sdZddlZddlmZmZddlZddlmZmZm Z ddlm Z m Z GdddeZ Gd d d e Z Gd d d e ZGd ddeZGddde ZGddde ZGddde ZGddde 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)utTestCase)h5sh5th5d)FileMultiBlockSlicec@seZdZddZddZdS) BaseSlicingcCst|d|_dS)Nw)rmktempfselfr6lib/python3.9/site-packages/h5py/tests/test_slicing.pysetUpszBaseSlicing.setUpcCs|jr|jdSNr closerrrrtearDown szBaseSlicing.tearDownN)__name__ __module__ __qualname__rrrrrrr sr c@s8eZdZdZddZddZddZdd Zd d Zd S) TestSingleElementzM Feature: Retrieving a single element works with NumPy semantics cCs,|jjdddd}|d}||tjdS)z; Single-element selection with [index] yields array scalar xri1dtyperNr create_datasetassertIsInstancenpZint8rdsetoutrrrtest_single_index*sz#TestSingleElement.test_single_indexcCs:|jjdddd}|d}||tj||jddS)z3 Single-element selection with [()] yields ndarray rrrrrNr r!r"r#ndarray assertEqualshaper$rrrtest_single_null0sz"TestSingleElement.test_single_nullcCs:|jjdddd}|d}||tj||jddS)z* Slicing with [...] yields scalar ndarray rrr r+r.Nr(r$rrrtest_scalar_index7sz#TestSingleElement.test_scalar_indexcCs,|jjdddd}|d}||tjdS)z' Slicing with [()] yields array scalar rrrr-Nr r$rrrtest_scalar_null>sz"TestSingleElement.test_scalar_nullcCsXtddg}tjd|d}|jjdd|d}||d|d||dtjdS) z6 Compound scalar is numpy.void, not tuple (issue 135) )aZi4)bf8)rZfoodatarN)r#ronesr r!r*r"Zvoid)rdtvr%rrr test_compoundDs zTestSingleElement.test_compoundN) rrr__doc__r'r,r.r/r9rrrrr$s rc@s8eZdZdZddZddZddZdd Zd d Zd S) TestObjectIndexzH Feature: numpy.object_ subtypes map to real Python objects cCs:|jjddtjd}|jj|d<|t|dtjdS)z@ Indexing a reference dataset returns a h5py.Reference instance rrrrNr r!h5py ref_dtyperefr*type Referencerr%rrrtest_referenceRs zTestObjectIndex.test_referencecCsN|jdd}|jd}|jjddtjd}||d<|t|dtjdS) zL Indexing a region reference dataset returns a h5py.RegionReference r) rD.yrrrN)r r!Z regionrefr=Zregionref_dtyper*r@ZRegionReference)rZdset1ZregrefZdset2rrr test_regrefXs  zTestObjectIndex.test_regrefcCs\tddtjfg}|jjdd|d}d|jdjf|d<|d}|t|d tj d S) z> Compound types of which a reference is an element work right )r0ir1rrr*/rrN) r#rr=r>r r!r?r*r@rArr7r%r&rrrtest_reference_fieldas z$TestObjectIndex.test_reference_fieldcCs:|jjddtjd}|jj|d<|t|dtjdS)z: Indexing returns a real Python object on scalar datasets rrrNr<rBrrr test_scalarks zTestObjectIndex.test_scalarcCs:|jjddtjddd}d|d<|t|dtdS) zJ Indexing a byte string dataset returns a real python byte string rrascii)encodingrs Hello there!rN)r r!r=Z string_dtyper*r@bytesrBrrr test_bytestrqszTestObjectIndex.test_bytestrN) rrrr:rCrFrKrLrPrrrrr;Ls   r;c@s0eZdZdZddZddZddZdd Zd S) TestSimpleSlicingzQ Feature: Simple NumPy-style slices (start:stop:step) are supported. cCs4t|d|_td|_|jjd|jd|_dS)Nr rDrr4)rr r r#arangearrr!r%rrrrr~s zTestSimpleSlicing.setUpcCs|jr|jdSrrrrrrrszTestSimpleSlicing.tearDowncCs$||jdd|jdddS)z0 Negative stop indexes work as they do in NumPy N)assertArrayEqualr%rSrrrrtest_negative_stopsz$TestSimpleSlicing.test_negative_stopcCsr|jdd}td}|dddf|dddf<|t ||dddf<Wdn1sd0YdS)z0Assigning to a 1D slice of a 2D dataset Zx2)rDrT)rDrNrr)r r!r#zeros assertRaises TypeError)rr%rrrr test_writes   zTestSimpleSlicing.test_writeN)rrrr:rrrWr[rrrrrQxs rQc@s8eZdZdZddZddZddZdd Zd d Zd S) TestArraySlicingz< Feature: Array types are handled appropriately cCstd}|jjdd|d}||jd||j||d}||jtd||jd|d}||jtd||jd |d d d }||jtd||jd d S)z; Read arrays tack array dimensions onto end of shape tuple (3,)f8rrDr.r2)rDrr_rT)r_r_N)r#rr r!r*r+rJrrr test_reads zTestArraySlicing.test_readcCsRtd}|jjdd|d}|td|d<Wdn1sD0YdS)z@ Array fill from constant is not supported (issue 211). (3,)irr^rrH.N)r#rr r!rYrZ)rr7r%rrrtest_write_broadcasts  z%TestArraySlicing.test_write_broadcastcCsRtd}|jjdd|d}tgd}||d<|d}|t||kdS)zA Write a single element to the array Issue 211. r]rr^r)rrTg@r3N)r#rr r!array assertTrueall)rr7r%r5r&rrrtest_write_elements  z#TestArraySlicing.test_write_elementcCstd}tjd|d}tjd|d}|jjdd|d}||dddd f<||dddd f|||d d d d df<||d d d d df|dS)z Write slices to array type rc)rTr)r3r)rD rrTr3r_rrirkN)r#rr6r r!rV)rr7Zdata1data2r%rrrtest_write_slicess z"TestArraySlicing.test_write_slicescCsHtd}|jjdd|d}|d}||d<|t|d|kdS)zO Read the contents of an array and write them back Issue 211. r]rr^r.N)r#rr r!rfrgrJrrrtest_roundtrips  zTestArraySlicing.test_roundtripN) rrrr:rbrdrhrnrorrrrr\s  r\c@s(eZdZdZddZddZddZdS) TestZeroLengthSlicingz. Slices resulting in empty arrays cCstgdD]\}}|jjd||tdt|d}||j||d}||tj ||j||dd}||tj ||j|t|dkr |ddddf}||tj ||jddd q dS) z] Slice a dataset with a zero in its shape vector along the zero-length dimension )r)rr_)rrTrx%drrmaxshape.NrrT)rr enumerater r!intlenr*r+r"r#r)rrGr+r%r&rrr test_slice_zero_length_dimensions   z6TestZeroLengthSlicing.test_slice_zero_length_dimensioncCs|tgdD]j\}}|jjd||tdt|d}||j||dd}||tj ||jd|ddq dS)z_ Slice a dataset with a zero in its shape vector along a non-zero-length dimension ))r_r)rrTr)rTrrrrrrsNrrruryrrrtest_slice_other_dimensions   z0TestZeroLengthSlicing.test_slice_other_dimensioncCstgdD]p\}}|jjd|t|tdt|d}||j||dd}| |tj ||jd|ddq dS)z5 Get a slice of length zero from a non-empty dataset )r`)rTrT)rTrrirrr)r5rtrrqN) rvr r!r#rXrwrxr*r+r"r)ryrrrtest_slice_of_length_zeros & z/TestZeroLengthSlicing.test_slice_of_length_zeroN)rrrr:rzr{r|rrrrrps rpc@sTeZdZdZegdZejdedZddZ ddZ d d Z d d Z d dZ dS)TestFieldNamesz* Field names for read & write ))r0r )r1rG)cZf4drcCs0t||jjdd|jd|_|j|jd<dS)Nrrr.)r rr r!r7r%r5rrrrrs zTestFieldNames.setUpcCs||jd|jddS)z! Test read with field selections r0N)rVr%r5rrrrrbszTestFieldNames.test_readcCsN||jd|jdd|jd<|j}d|d<||jd|ddS)z, Unicode field names for for read and write r0rHN)rVr%r5copy)rr5rrrtest_unicode_names#s   z!TestFieldNames.test_unicode_namescCs|j}|dd9<||jd<|t|jd|k|dd9<||jd<|t|jd|k|dd9<|dd9<||jd<|t|jd|kd S) z" Test write with field selections r0rT.r1r3r_r~)r0r~Nr5rr%rfr#rgrrmrrrr[+s    zTestFieldNames.test_writecCs:|j}d|d<d|jd<|t|jd|kdS)z4 Test write with non-compound source (single-field) g?r1.Nrrrrrtest_write_noncompound9s  z%TestFieldNames.test_write_noncompoundN)rrrr:r#rr7r6r5rrbrr[rrrrrr}sr}cseZdZfddZddZddZddZd d Zd d Zd dZ ddZ ddZ ddZ ddZ ddZddZddZddZZS)TestMultiBlockSlicecs2tt|td|_|jjd|jd|_dS)NrDrr4) superrrr#rRrSr r!r%r __class__rrrCs zTestMultiBlockSlice.setUpcCs2t}||ddtj|j||jdS)NrDrrrDrr r*indicesr#testingassert_array_equalr%rSrZmbslicerrr test_defaultHsz TestMultiBlockSlice.test_defaultcCs<tddddd}||ddtj|j||jdS)NrrDrstartcountstrideblockrrrrrrtest_default_explicitOsz)TestMultiBlockSlice.test_default_explicitcCs>tdd}||ddtj|j|tgddS)Nr3)rrD)r3rrlr)r3rirlrarjr r*rr#rrr%rerrrr test_startUs zTestMultiBlockSlice.test_startcCs>tdd}||ddtj|j|tgddS)NrrrD)rrrr)rrrTr_r3rirlrrrrr test_count[s  zTestMultiBlockSlice.test_countcCsBtdd}|t|dWdn1s40YdS)NrkrrDr rY ValueErrorrrrrr!test_count_more_than_length_errorcs  z5TestMultiBlockSlice.test_count_more_than_length_errorcCs>tdd}||ddtj|j|tgddS)NrT)rrD)rrTrir)rrTr3rlrarrrrr test_stridehs zTestMultiBlockSlice.test_stridecCs@|t"tddddWdn1s20YdS)NrrrrD)rYrr rrrrrtest_stride_zero_errorns z*TestMultiBlockSlice.test_stride_zero_errorcCs8tddd}||ddtj|j||jdS)NrTrrD)rrTrirTrrrrrtest_stride_block_equalss z+TestMultiBlockSlice.test_stride_block_equalcCsn|ttddWdn1s*0Y|ttdddWdn1s`0YdS)Nr_)rrTr)rYrr rrrr!test_block_more_than_stride_errorys ( z5TestMultiBlockSlice.test_block_more_than_stride_errorcCs@tddd}||ddtj|j|tgddS)Nr_rTrrD)rr_r_rT)rrr_r3rlrrrrrrtest_stride_more_than_blocks z/TestMultiBlockSlice.test_stride_more_than_blockcCsHtddddd}|t|dWdn1s:0YdS)NrTrir3rrDrrrrr test_block_overruns_extent_errors z4TestMultiBlockSlice.test_block_overruns_extent_errorcCsDtddddd}||ddtj|j|tgddS) NrrTrir3rrD)rrirTr3)rrTr_r3rlrrarjrrrrrtest_fully_describeds z(TestMultiBlockSlice.test_fully_describedcCsBtdddd}||ddtj|j|tgddS)Nrr_rTrrrrD)rr_r_rT)rrTr3rirrarrrrrtest_count_calculatedsz)TestMultiBlockSlice.test_count_calculatedcCsFtdddd}|t|dWdn1s80YdS)Nrar3r_rrDrrrrr test_zero_count_calculated_errors z4TestMultiBlockSlice.test_zero_count_calculated_error)rrrrrrrrrrrrrrrrrr __classcell__rrrrrAs r)r:Znumpyr#commonrrr=rrrrr r rr;rQr\rpr}rrrrr s  (,R*0