def test_append_experiment_log(self, tmpdir):
project = HidraProjectFile(os.path.join(tmpdir, 'project_file.hdf'),
HidraProjectFileMode.OVERWRITE)
group = project._project_h5[HidraConstants.RAW_DATA][
HidraConstants.SAMPLE_LOGS]
project.append_experiment_log('vx', np.array([0.0, 0.1, 0.2]))
with pytest.raises(KeyError):
group['vx'].attrs['units']
project.append_experiment_log('vy',
np.array([0.3, 0.4, 0.5]),
units='mm')
assert group['vy'].attrs['units'] == 'mm'
def test_read_log_units(self, tmpdir):
project = HidraProjectFile(os.path.join(tmpdir, 'project_file.hdf'),
HidraProjectFileMode.OVERWRITE)
with pytest.raises(AssertionError) as exception_info:
project.read_log_units('vx')
assert 'Missing sample log: vx' in str(exception_info.value)
project.append_experiment_log('vx', np.array([0.0, 0.1, 0.2]))
assert project.read_log_units('vx') == ''
project.append_experiment_log('vy',
np.array([0.3, 0.4, 0.5]),
units='mm')
assert project.read_log_units('vy') == 'mm'