143 lines
4.7 KiB
Python
143 lines
4.7 KiB
Python
# This file is part of h5py, a Python interface to the HDF5 library.
|
|
#
|
|
# http://www.h5py.org
|
|
#
|
|
# Copyright 2008-2013 Andrew Collette and contributors
|
|
#
|
|
# License: Standard 3-clause BSD; see "license.txt" for full license terms
|
|
# and contributor agreement.
|
|
|
|
"""
|
|
Tests for the (internal) selections module
|
|
"""
|
|
|
|
import numpy as np
|
|
import h5py
|
|
import h5py._hl.selections as sel
|
|
import h5py._hl.selections2 as sel2
|
|
|
|
from .common import TestCase, ut
|
|
|
|
class BaseSelection(TestCase):
|
|
def setUp(self):
|
|
self.f = h5py.File(self.mktemp(), 'w')
|
|
self.dsid = self.f.create_dataset('x', ()).id
|
|
|
|
def tearDown(self):
|
|
if self.f:
|
|
self.f.close()
|
|
|
|
class TestTypeGeneration(BaseSelection):
|
|
|
|
"""
|
|
Internal feature: Determine output types from dataset dtype and fields.
|
|
"""
|
|
|
|
def test_simple(self):
|
|
""" Non-compound types are handled appropriately """
|
|
dt = np.dtype('i')
|
|
out, format = sel2.read_dtypes(dt, ())
|
|
self.assertEqual(out, format)
|
|
self.assertEqual(out, np.dtype('i'))
|
|
|
|
def test_simple_fieldexc(self):
|
|
""" Field names for non-field types raises ValueError """
|
|
dt = np.dtype('i')
|
|
with self.assertRaises(ValueError):
|
|
out, format = sel2.read_dtypes(dt, ('a',))
|
|
|
|
def test_compound_simple(self):
|
|
""" Compound types with elemental subtypes """
|
|
dt = np.dtype( [('a','i'), ('b','f'), ('c','|S10')] )
|
|
|
|
# Implicit selection of all fields -> all fields
|
|
out, format = sel2.read_dtypes(dt, ())
|
|
self.assertEqual(out, format)
|
|
self.assertEqual(out, dt)
|
|
|
|
# Explicit selection of fields -> requested fields
|
|
out, format = sel2.read_dtypes(dt, ('a','b'))
|
|
self.assertEqual(out, format)
|
|
self.assertEqual(out, np.dtype( [('a','i'), ('b','f')] ))
|
|
|
|
# Explicit selection of exactly one field -> no fields
|
|
out, format = sel2.read_dtypes(dt, ('a',))
|
|
self.assertEqual(out, np.dtype('i'))
|
|
self.assertEqual(format, np.dtype( [('a','i')] ))
|
|
|
|
# Field does not apear in named typed
|
|
with self.assertRaises(ValueError):
|
|
out, format = sel2.read_dtypes(dt, ('j', 'k'))
|
|
|
|
class TestScalarSliceRules(BaseSelection):
|
|
|
|
"""
|
|
Internal feature: selections rules for scalar datasets
|
|
"""
|
|
|
|
def test_args(self):
|
|
""" Permissible arguments for scalar slicing """
|
|
shape, selection = sel2.read_selections_scalar(self.dsid, ())
|
|
self.assertEqual(shape, None)
|
|
self.assertEqual(selection.get_select_npoints(), 1)
|
|
|
|
shape, selection = sel2.read_selections_scalar(self.dsid, (Ellipsis,))
|
|
self.assertEqual(shape, ())
|
|
self.assertEqual(selection.get_select_npoints(), 1)
|
|
|
|
with self.assertRaises(ValueError):
|
|
shape, selection = sel2.read_selections_scalar(self.dsid, (1,))
|
|
|
|
dsid = self.f.create_dataset('y', (1,)).id
|
|
with self.assertRaises(RuntimeError):
|
|
shape, selection = sel2.read_selections_scalar(dsid, (1,))
|
|
|
|
class TestSelection(BaseSelection):
|
|
|
|
""" High-level routes to generate a selection
|
|
"""
|
|
|
|
def test_selection(self):
|
|
dset = self.f.create_dataset('dset', (100,100))
|
|
regref = dset.regionref[0:100, 0:100]
|
|
|
|
# args is list, return a FancySelection
|
|
st = sel.select((10,), list([1,2,3]), dset)
|
|
self.assertIsInstance(st, sel.FancySelection)
|
|
|
|
# args[0] is tuple, return a FancySelection
|
|
st = sel.select((10,), ((1, 2, 3),), dset)
|
|
self.assertIsInstance(st, sel.FancySelection)
|
|
|
|
# args is a Boolean mask, return a PointSelection
|
|
st1 = sel.select((5,), np.array([True,False,False,False,True]), dset)
|
|
self.assertIsInstance(st1, sel.PointSelection)
|
|
|
|
# args is int, return a SimpleSelection
|
|
st2 = sel.select((10,), 1, dset)
|
|
self.assertIsInstance(st2, sel.SimpleSelection)
|
|
|
|
# args is str, should be rejected
|
|
with self.assertRaises(TypeError):
|
|
sel.select((100,), "foo", dset)
|
|
|
|
# args is RegionReference, return a Selection instance
|
|
st3 = sel.select((100,100), regref, dset)
|
|
self.assertIsInstance(st3, sel.Selection)
|
|
|
|
# args is RegionReference, but dataset is None
|
|
with self.assertRaises(TypeError):
|
|
sel.select((100,), regref, None)
|
|
|
|
# args is RegionReference, but its shape doesn't match dataset shape
|
|
with self.assertRaises(TypeError):
|
|
sel.select((100,), regref, dset)
|
|
|
|
# args is a single Selection instance, return the arg
|
|
st4 = sel.select((100,100), st3, dset)
|
|
self.assertEqual(st4,st3)
|
|
|
|
# args is a single Selection instance, but args shape doesn't match Shape
|
|
with self.assertRaises(TypeError):
|
|
sel.select((100,), st3, dset)
|