test

Author: jokasimr

tests/reflectometry/normalization_tests.py

@@ -0,0 +1,43 @@
+import numpy as np
+import pytest
+import scipp as sc
+
+from ess.reflectometry.normalization import reduce_sample_over_zw
+
+
+@pytest.fixture
+def sample(request):
+    n = 50
+    da = sc.DataArray(
+        data=sc.ones(dims=('events',), shape=(n,)),
+        coords={
+            'wavelength': sc.linspace('events', 1, 5, n),
+            'wire': sc.array(dims=('events',), values=np.random.randint(0, 5, n)),
+            'stripe': sc.array(dims=('events',), values=np.random.randint(0, 10, n)),
+        },
+    )
+    return da.group('wire', 'stripe')
+
+
+@pytest.fixture
+def reference(request):
+    n = 50
+    da = sc.DataArray(
+        data=sc.ones(dims=('events',), shape=(n,)),
+        coords={
+            'wavelength': sc.linspace('events', 1, 5, n),
+            'wire': sc.array(dims=('events',), values=np.random.randint(0, 5, n)),
+            'stripe': sc.array(dims=('events',), values=np.random.randint(0, 10, n)),
+        },
+    )
+    return da.group('wire').bin(wavelength=2).bins.sum()
+
+
+def test_reduce_sample_over_zw_when_data_not_dimensionless(sample, reference):
+    sample = sample.copy(deep=True)
+    sample.bins.unit = '1/s'
+    reduce_sample_over_zw(
+        sample,
+        reference,
+        reference.coords['wavelength'],
+    )