fix tests adopted for earlier commit with extended bin number

OpenCOMPES · May 8, 2023 · 6450199 · 6450199
1 parent 31e2af1
commit 6450199
Show file tree

Hide file tree

Showing 2 changed files with 2 additions and 2 deletions.
diff --git a/tests/calibrator/test_energy.py b/tests/calibrator/test_energy.py
@@ -43,7 +43,7 @@ def test_bin_data_and_read_biases_from_files():
     )
     ec.bin_data(data_files=files)
     assert ec.traces.ndim == 2
-    assert ec.traces.shape == (2, 1001)
+    assert ec.traces.shape == (2, 1000)
     assert ec.tof.ndim == 1
     assert ec.biases.ndim == 1
     assert len(ec.biases) == 2

diff --git a/tests/calibrator/test_momentum.py b/tests/calibrator/test_momentum.py
@@ -41,7 +41,7 @@ def test_bin_data_and_slice_image():
         width=10,
         apply=True,
     )
-    assert sed_processor.mc.slice.shape == (513, 513)
+    assert sed_processor.mc.slice.shape == (512, 512)
 
 
 def test_feature_extract():