do not group nxevent_data by pulse

scipp · Feb 4, 2025 · 4ccc8fe · 4ccc8fe
1 parent 12dec8b
commit 4ccc8fe
Show file tree

Hide file tree

Showing 2 changed files with 6 additions and 5 deletions.
diff --git 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
@@ -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}