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))