def test_get_spec_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_spec_values('blah')
with self.assertRaises(TypeError):
_ = usi_main.get_spec_values(np.array(5))
def test_get_spec_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_spec_vals[0, ::2]
actual = usi_main.get_spec_values('Bias')
self.assertTrue(np.allclose(expected, actual))
expected = usi_main.h5_spec_vals[1, 0:None:7]
actual = usi_main.get_spec_values('Cycle')
self.assertTrue(np.allclose(expected, actual))
def test_get_spec_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 ['Bias', 'Cycle']:
expected = h5_f['/Raw_Measurement/' + dim_name][()]
actual = usi_main.get_spec_values(dim_name)
self.assertTrue(np.allclose(expected, actual))