def test_get_pos_values_illegal(self):
with h5py.File(test_h5_file_path, mode='r') as h5_f:
usi_main = USIDataset(h5_f['/Raw_Measurement/source_main'])
with self.assertRaises(KeyError):
_ = usi_main.get_pos_values('blah')
with self.assertRaises(TypeError):
_ = usi_main.get_pos_values(np.array(5))
def test_get_pos_values(self):
with h5py.File(test_h5_file_path, mode='r') as h5_f:
usi_main = USIDataset(h5_f['/Raw_Measurement/source_main'])
expected = usi_main.h5_pos_vals[:5, 0]
actual = usi_main.get_pos_values('X')
self.assertTrue(np.allclose(expected, actual))
expected = usi_main.h5_pos_vals[0:None:5, 1]
actual = usi_main.get_pos_values('Y')
self.assertTrue(np.allclose(expected, actual))
def test_get_pos_values(self):
with h5py.File(test_h5_file_path, mode='r') as h5_f:
usi_main = USIDataset(h5_f['/Raw_Measurement/source_main'])
for dim_name in ['X', 'Y']:
expected = h5_f['/Raw_Measurement/' + dim_name][()]
actual = usi_main.get_pos_values(dim_name)
self.assertTrue(np.allclose(expected, actual))
