diff --git a/src/tof/result.py b/src/tof/result.py index 16745dc..8137ce6 100644 --- a/src/tof/result.py +++ b/src/tof/result.py @@ -427,8 +427,8 @@ def to_nxevent_data(self, key: Optional[str] = None) -> sc.DataArray: ) % period.to(unit=dt.unit) out = ( event_data.drop_coords(['tof', 'speed', 'time', 'wavelength']) - .group('distance', 'event_time_zero') - .rename_dims(event_time_zero='pulse') - ).rename_dims(distance='detector_number') + .group('distance') + .rename_dims(distance='detector_number') + ) out.coords['Ltotal'] = out.coords.pop('distance') return out diff --git a/tests/model_test.py b/tests/model_test.py index 0d551eb..d43ad8c 100644 --- a/tests/model_test.py +++ b/tests/model_test.py @@ -421,7 +421,8 @@ def test_to_nxevent_data(): for key, npulses in zip(('monitor', 'detector'), (1, 2)): nxevent_data = res.to_nxevent_data(key) assert sc.identical(res['monitor'].data.sum().data, nxevent_data.sum().data) - assert nxevent_data.sizes['pulse'] == npulses + grouped = nxevent_data.group('event_time_zero') + assert grouped.sizes['event_time_zero'] == npulses assert nxevent_data.bins.concat().value.coords[ 'event_time_offset' ].min() >= sc.scalar(0.0, unit='us') @@ -431,4 +432,4 @@ def test_to_nxevent_data(): # Test when we include all detectors at once nxevent_data = res.to_nxevent_data() - assert nxevent_data.sizes == {'detector_number': 2, 'pulse': 2} + assert nxevent_data.sizes == {'detector_number': 2}