laywerrobot/lib/python3.6/site-packages/h5py/tests/old/test_dimension_scales.py

207 lines
7.6 KiB
Python
Raw Normal View History

2020-08-27 21:55:39 +02:00
# 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.
from __future__ import absolute_import
import sys
import numpy as np
from ..common import ut, TestCase
from h5py.highlevel import File, Group, Dataset
import h5py
class BaseDataset(TestCase):
"""
data is a 3-dimensional dataset with dimensions [z, y, x]
The z dimension is labeled. It does not have any attached scales.
The y dimension is not labeled. It has one attached scale.
The x dimension is labeled. It has two attached scales.
data2 is a 3-dimensional dataset with no associated dimension scales.
"""
def setUp(self):
self.f = File(self.mktemp(), 'w')
self.f['data'] = np.ones((4, 3, 2), 'f')
self.f['data2'] = np.ones((4, 3, 2), 'f')
self.f['x1'] = np.ones((2), 'f')
h5py.h5ds.set_scale(self.f['x1'].id)
h5py.h5ds.attach_scale(self.f['data'].id, self.f['x1'].id, 2)
self.f['x2'] = np.ones((2), 'f')
h5py.h5ds.set_scale(self.f['x2'].id, b'x2 name')
h5py.h5ds.attach_scale(self.f['data'].id, self.f['x2'].id, 2)
self.f['y1'] = np.ones((3), 'f')
h5py.h5ds.set_scale(self.f['y1'].id, b'y1 name')
h5py.h5ds.attach_scale(self.f['data'].id, self.f['y1'].id, 1)
self.f['z1'] = np.ones((4), 'f')
h5py.h5ds.set_label(self.f['data'].id, 0, b'z')
h5py.h5ds.set_label(self.f['data'].id, 2, b'x')
def tearDown(self):
if self.f:
self.f.close()
class TestH5DSBindings(BaseDataset):
"""
Feature: Datasets can be created from existing data
"""
def test_create_dimensionscale(self):
""" Create a dimension scale from existing dataset """
self.assertTrue(h5py.h5ds.is_scale(self.f['x1'].id))
self.assertEqual(h5py.h5ds.get_scale_name(self.f['x1'].id), b'')
self.assertEqual(self.f['x1'].attrs['CLASS'], b"DIMENSION_SCALE")
self.assertEqual(h5py.h5ds.get_scale_name(self.f['x2'].id), b'x2 name')
def test_attach_dimensionscale(self):
self.assertTrue(
h5py.h5ds.is_attached(self.f['data'].id, self.f['x1'].id, 2)
)
self.assertFalse(
h5py.h5ds.is_attached(self.f['data'].id, self.f['x1'].id, 1))
self.assertEqual(h5py.h5ds.get_num_scales(self.f['data'].id, 0), 0)
self.assertEqual(h5py.h5ds.get_num_scales(self.f['data'].id, 1), 1)
self.assertEqual(h5py.h5ds.get_num_scales(self.f['data'].id, 2), 2)
def test_detach_dimensionscale(self):
self.assertTrue(
h5py.h5ds.is_attached(self.f['data'].id, self.f['x1'].id, 2)
)
h5py.h5ds.detach_scale(self.f['data'].id, self.f['x1'].id, 2)
self.assertFalse(
h5py.h5ds.is_attached(self.f['data'].id, self.f['x1'].id, 2)
)
self.assertEqual(h5py.h5ds.get_num_scales(self.f['data'].id, 2), 1)
# TODO: update condition once the bug is fixed upstream
@ut.skipUnless(
h5py.version.hdf5_version_tuple > (2, 0, 0),
"Reading non-existent label segfaults"
)
def test_label_dimensionscale(self):
self.assertEqual(h5py.h5ds.get_label(self.f['data'].id, 0), b'z')
self.assertEqual(h5py.h5ds.get_label(self.f['data'].id, 1), b'')
self.assertEqual(h5py.h5ds.get_label(self.f['data'].id, 2), b'x')
def test_iter_dimensionscales(self):
def func(dsid):
res = h5py.h5ds.get_scale_name(dsid)
if res == b'x2 name':
return dsid
res = h5py.h5ds.iterate(self.f['data'].id, 2, func, 0)
self.assertEqual(h5py.h5ds.get_scale_name(res), b'x2 name')
class TestDimensionManager(BaseDataset):
def test_create_scale(self):
# test recreating or renaming an existing scale:
self.f['data'].dims.create_scale(self.f['x1'], b'foobar')
self.assertEqual(self.f['data'].dims[2]['foobar'], self.f['x1'])
# test creating entirely new scale:
self.f['data'].dims.create_scale(self.f['data2'], b'foobaz')
self.f['data'].dims[2].attach_scale(self.f['data2'])
self.assertEqual(self.f['data'].dims[2]['foobaz'], self.f['data2'])
def test_get_dimension(self):
with self.assertRaises(IndexError):
self.f['data'].dims[3]
def test_len(self):
self.assertEqual(len(self.f['data'].dims), 3)
self.assertEqual(len(self.f['data2'].dims), 3)
def test_iter(self):
dims = self.f['data'].dims
self.assertEqual(
[d for d in dims],
[dims[0], dims[1], dims[2]]
)
class TestDimensionsHighLevel(BaseDataset):
def test_len(self):
self.assertEqual(len(self.f['data'].dims[0]), 0)
self.assertEqual(len(self.f['data'].dims[1]), 1)
self.assertEqual(len(self.f['data'].dims[2]), 2)
self.assertEqual(len(self.f['data2'].dims[0]), 0)
self.assertEqual(len(self.f['data2'].dims[1]), 0)
self.assertEqual(len(self.f['data2'].dims[2]), 0)
def test_get_label(self):
self.assertEqual(self.f['data'].dims[2].label, 'x')
self.assertEqual(self.f['data'].dims[1].label, '')
self.assertEqual(self.f['data'].dims[0].label, 'z')
self.assertEqual(self.f['data2'].dims[2].label, '')
self.assertEqual(self.f['data2'].dims[1].label, '')
self.assertEqual(self.f['data2'].dims[0].label, '')
def test_set_label(self):
self.f['data'].dims[0].label = 'foo'
self.assertEqual(self.f['data'].dims[2].label, 'x')
self.assertEqual(self.f['data'].dims[1].label, '')
self.assertEqual(self.f['data'].dims[0].label, 'foo')
def test_detach_scale(self):
self.f['data'].dims[2].detach_scale(self.f['x1'])
self.assertEqual(len(self.f['data'].dims[2]), 1)
self.assertEqual(self.f['data'].dims[2][0], self.f['x2'])
self.f['data'].dims[2].detach_scale(self.f['x2'])
self.assertEqual(len(self.f['data'].dims[2]), 0)
def test_attach_scale(self):
self.f['x3'] = self.f['x2'][...]
self.f['data'].dims[2].attach_scale(self.f['x3'])
self.assertEqual(len(self.f['data'].dims[2]), 3)
self.assertEqual(self.f['data'].dims[2][2], self.f['x3'])
def test_get_dimension_scale(self):
self.assertEqual(self.f['data'].dims[2][0], self.f['x1'])
with self.assertRaises(RuntimeError):
self.f['data2'].dims[2][0], self.f['x2']
self.assertEqual(self.f['data'].dims[2][''], self.f['x1'])
self.assertEqual(self.f['data'].dims[2]['x2 name'], self.f['x2'])
def test_get_items(self):
self.assertEqual(
self.f['data'].dims[2].items(),
[('', self.f['x1']), ('x2 name', self.f['x2'])]
)
def test_get_keys(self):
self.assertEqual(self.f['data'].dims[2].keys(), ['', 'x2 name'])
def test_get_values(self):
self.assertEqual(
self.f['data'].dims[2].values(),
[self.f['x1'], self.f['x2']]
)
def test_iter(self):
self.assertEqual([i for i in self.f['data'].dims[2]], ['', 'x2 name'])
def test_repr(self):
self.assertEqual(repr(self.f['data'].dims[2])[1:16], '"x" dimension 2')
def test_attributes(self):
self.f["data2"].attrs["DIMENSION_LIST"] = self.f["data"].attrs[
"DIMENSION_LIST"]
self.assertEqual(len(self.f['data2'].dims[0]), 0)
self.assertEqual(len(self.f['data2'].dims[1]), 1)
self.assertEqual(len(self.f['data2'].dims[2]), 2)