Update test

DiamondLightSource · Dec 16, 2024 · 05b026c · 05b026c
1 parent 097d6cb
commit 05b026c
Showing 1 changed file with 3 additions and 1 deletion.
diff --git a/tests/test_linepattern.py b/tests/test_linepattern.py
@@ -58,7 +58,9 @@ def test_select_good_peaks(self):
                                         radius=radius, sigma=sigma)
         np.testing.assert_array_equal(result, peaks)
         self.assertTrue(len(result) == 2)
-        tol, radius, sigma = 0.2, 3, 1
+        list_data = np.array([0, 1, 2, 9, 2, 1, 0, 0, 3, 10, 3, 0, 0])
+        peaks = np.asarray([3, 9])
+        tol, radius, sigma = 0.3, 3, 1
         result = lipa.select_good_peaks(list_data, peaks, tol=tol,
                                         radius=radius, sigma=sigma)
         np.testing.assert_array_equal(result, peaks)