Spaces:
Running
Running
File size: 4,093 Bytes
122d3ff |
1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28 29 30 31 32 33 34 35 36 37 38 39 40 41 42 43 44 45 46 47 48 49 50 51 52 53 54 55 56 57 58 59 60 61 62 63 64 65 66 67 68 69 70 71 72 73 74 75 76 77 78 79 80 81 82 83 84 85 86 87 88 89 90 91 92 93 94 95 96 97 98 99 100 101 102 103 104 105 106 107 108 109 110 111 112 113 114 115 116 117 118 119 |
import numpy as np
import h5py
from .common import ut, TestCase
class TestDatasetSwmrRead(TestCase):
""" Testing SWMR functions when reading a dataset.
Skip this test if the HDF5 library does not have the SWMR features.
"""
def setUp(self):
TestCase.setUp(self)
self.data = np.arange(13).astype('f')
self.dset = self.f.create_dataset('data', chunks=(13,), maxshape=(None,), data=self.data)
fname = self.f.filename
self.f.close()
self.f = h5py.File(fname, 'r', swmr=True)
self.dset = self.f['data']
def test_initial_swmr_mode_on(self):
""" Verify that the file is initially in SWMR mode"""
self.assertTrue(self.f.swmr_mode)
def test_read_data(self):
self.assertArrayEqual(self.dset, self.data)
def test_refresh(self):
self.dset.refresh()
def test_force_swmr_mode_on_raises(self):
""" Verify when reading a file cannot be forcibly switched to swmr mode.
When reading with SWMR the file must be opened with swmr=True."""
with self.assertRaises(Exception):
self.f.swmr_mode = True
self.assertTrue(self.f.swmr_mode)
def test_force_swmr_mode_off_raises(self):
""" Switching SWMR write mode off is only possible by closing the file.
Attempts to forcibly switch off the SWMR mode should raise a ValueError.
"""
with self.assertRaises(ValueError):
self.f.swmr_mode = False
self.assertTrue(self.f.swmr_mode)
class TestDatasetSwmrWrite(TestCase):
""" Testing SWMR functions when reading a dataset.
Skip this test if the HDF5 library does not have the SWMR features.
"""
def setUp(self):
""" First setup a file with a small chunked and empty dataset.
No data written yet.
"""
# Note that when creating the file, the swmr=True is not required for
# write, but libver='latest' is required.
self.f = h5py.File(self.mktemp(), 'w', libver='latest')
self.data = np.arange(4).astype('f')
self.dset = self.f.create_dataset('data', shape=(0,), dtype=self.data.dtype, chunks=(2,), maxshape=(None,))
def test_initial_swmr_mode_off(self):
""" Verify that the file is not initially in SWMR mode"""
self.assertFalse(self.f.swmr_mode)
def test_switch_swmr_mode_on(self):
""" Switch to SWMR mode and verify """
self.f.swmr_mode = True
self.assertTrue(self.f.swmr_mode)
def test_switch_swmr_mode_off_raises(self):
""" Switching SWMR write mode off is only possible by closing the file.
Attempts to forcibly switch off the SWMR mode should raise a ValueError.
"""
self.f.swmr_mode = True
self.assertTrue(self.f.swmr_mode)
with self.assertRaises(ValueError):
self.f.swmr_mode = False
self.assertTrue(self.f.swmr_mode)
def test_extend_dset(self):
""" Extend and flush a SWMR dataset
"""
self.f.swmr_mode = True
self.assertTrue(self.f.swmr_mode)
self.dset.resize( self.data.shape )
self.dset[:] = self.data
self.dset.flush()
# Refresh and read back data for assertion
self.dset.refresh()
self.assertArrayEqual(self.dset, self.data)
def test_extend_dset_multiple(self):
self.f.swmr_mode = True
self.assertTrue(self.f.swmr_mode)
self.dset.resize( (4,) )
self.dset[0:] = self.data
self.dset.flush()
# Refresh and read back 1st data block for assertion
self.dset.refresh()
self.assertArrayEqual(self.dset, self.data)
self.dset.resize( (8,) )
self.dset[4:] = self.data
self.dset.flush()
# Refresh and read back 1st data block for assertion
self.dset.refresh()
self.assertArrayEqual(self.dset[0:4], self.data)
self.assertArrayEqual(self.dset[4:8], self.data)
|