ó \c@@sdZddlmZddlZddlZddlmZmZddl Z ddl m Z m Z m Z ddl m Z defd „ƒYZd efd „ƒYZd efd „ƒYZdefd„ƒYZdefd„ƒYZdefd„ƒYZdefd„ƒYZdS(s 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. i(tabsolute_importNi(tuttTestCase(th5sth5tth5d(tFilet BaseSlicingcB@seZd„Zd„ZRS(cC@st|jƒdƒ|_dS(Ntw(Rtmktemptf(tself((s:lib/python2.7/site-packages/h5py/tests/old/test_slicing.pytsetUp!scC@s|jr|jjƒndS(N(R tclose(R ((s:lib/python2.7/site-packages/h5py/tests/old/test_slicing.pyttearDown$s (t__name__t __module__R R(((s:lib/python2.7/site-packages/h5py/tests/old/test_slicing.pyRs tTestSingleElementcB@s;eZdZd„Zd„Zd„Zd„Zd„ZRS(sM Feature: Retrieving a single element works with NumPy semantics cC@s<|jjddddƒ}|d}|j|tjƒdS(s; Single-element selection with [index] yields array scalar txitdtypeti1iN(i(R tcreate_datasettassertIsInstancetnptint8(R tdsettout((s:lib/python2.7/site-packages/h5py/tests/old/test_slicing.pyttest_single_index.s cC@sO|jjddddƒ}|d}|j|tjƒ|j|jdƒdS( s3 Single-element selection with [()] yields ndarray RiRRN(i((i(R RRRtndarrayt assertEqualtshape(R RR((s:lib/python2.7/site-packages/h5py/tests/old/test_slicing.pyttest_single_null4s cC@sR|jjdddddƒ}|d}|j|tjƒ|j|jdƒdS( s* Slicing with [...] yields scalar ndarray RRRR .N(((R RRRRRR(R RR((s:lib/python2.7/site-packages/h5py/tests/old/test_slicing.pyttest_scalar_index;s cC@s?|jjdddddƒ}|d}|j|tjƒdS(s' Slicing with [()] yields array scalar RRRRN(((R RRRR(R RR((s:lib/python2.7/site-packages/h5py/tests/old/test_slicing.pyttest_scalar_nullBs cC@sxtjd d gƒ}tjd d|ƒ}|jjddd|ƒ}|j|d |d ƒ|j|d tjƒd S(s6 Compound scalar is numpy.void, not tuple (issue 135) tati4tbtf8iRtfootdataiN(R"R#(R$R%(i(i(RRtonesR RRRtvoid(R tdttvR((s:lib/python2.7/site-packages/h5py/tests/old/test_slicing.pyt test_compoundHs (RRt__doc__RRR R!R,(((s:lib/python2.7/site-packages/h5py/tests/old/test_slicing.pyR(s     tTestObjectIndexcB@s;eZdZd„Zd„Zd„Zd„Zd„ZRS(sH Feature: numpy.object_ subtypes map to real Python objects cC@s[|jjdddtjdtjƒƒ}|jj|d<|jt|dƒtjƒdS(s@ Indexing a reference dataset returns a h5py.Reference instance RiRtrefiN(i(R Rth5pyt special_dtypet ReferenceR/Rttype(R R((s:lib/python2.7/site-packages/h5py/tests/old/test_slicing.pyttest_referenceVs*cC@sw|jjdd ƒ}|jd}|jjdd dtjdtjƒƒ}||d<|jt|dƒtjƒd S( sL Indexing a region reference dataset returns a h5py.RegionReference Ri .tyiRR/iN(i i (i(R Rt regionrefR0R1tRegionReferenceRR3(R tdset1tregreftdset2((s:lib/python2.7/site-packages/h5py/tests/old/test_slicing.pyt test_regref\s  * cC@stjdtjƒ}tjd d|fgƒ}|jjdd d|ƒ}d|jd jf|d <|d }|jt |dƒtjƒd S(s> Compound types of which a reference is an element work right R/R"tiR$RiRi*t/iN(R"R<(i( R0R1R2RRR RR/RR3(R treftypeR*RR((s:lib/python2.7/site-packages/h5py/tests/old/test_slicing.pyttest_reference_fieldes  cC@s[|jjdddtjdtjƒƒ}|jj|d<|jt|dƒtjƒdS(s: Indexing returns a real Python object on scalar datasets RRR/N((((R RR0R1R2R/RR3(R R((s:lib/python2.7/site-packages/h5py/tests/old/test_slicing.pyt test_scalarps*cC@sO|jjdddtjdtƒƒ}d|d<|jt|dƒtƒdS( sJ Indexing a byte string dataset returns a real python byte string RiRtvlens Hello there!iN(i(R RR0R1tbytesRR3(R R((s:lib/python2.7/site-packages/h5py/tests/old/test_slicing.pyt test_bytestrvs' (RRR-R4R;R?R@RC(((s:lib/python2.7/site-packages/h5py/tests/old/test_slicing.pyR.Ps   tTestSimpleSlicingcB@s2eZdZd„Zd„Zd„Zd„ZRS(sQ Feature: Simple NumPy-style slices (start:stop:step) are supported. cC@sLt|jƒdƒ|_tjdƒ|_|jjdd|jƒ|_dS(NRi RR'(RR R RtarangetarrRR(R ((s:lib/python2.7/site-packages/h5py/tests/old/test_slicing.pyR ƒscC@s|jr|jjƒndS(N(R R (R ((s:lib/python2.7/site-packages/h5py/tests/old/test_slicing.pyRˆs cC@s(|j|jdd!|jdd!ƒdS(s0 Negative stop indexes work as they do in NumPy iiþÿÿÿN(tassertArrayEqualRRF(R ((s:lib/python2.7/site-packages/h5py/tests/old/test_slicing.pyttest_negative_stopŒscC@sz|jjddƒ}tjdƒ}|dd…df|dd…df<|jtƒ||dd…dfs    (-R*