Use array float comparison for tuples

bjheinen · May 13, 2024 · c0c0313 · c0c0313
1 parent 78df69f
commit c0c0313
Showing 1 changed file with 5 additions and 5 deletions.
diff --git a/tests/core/test_data_utils.py b/tests/core/test_data_utils.py
@@ -85,11 +85,11 @@ def test_find_integration_limits(self):
         limits_c = data_utils.find_integration_limits(*rdf_data.T, rho=0.05, peak_search_limit=20, search_method='first')
         limits_d = data_utils.find_integration_limits(*rdf_data.T, rho=0.05, peak_search_limit=20, search_method='prominent')
         limits_e = data_utils.find_integration_limits(*rdf_data.T, peak_search_limit=20, search_method='prominent')
-        self.assertEqual(limits_a, rdf_peaks)
-        self.assertEqual(limits_b, rdf_peaks)
-        self.assertEqual(limits_c, rdf_peaks)
-        self.assertEqual(limits_d, rdf_peaks)
-        self.assertEqual(limits_e, rdf_peaks)
+        self.assertFloatArrayEqual(limits_a, rdf_peaks)
+        self.assertFloatArrayEqual(limits_b, rdf_peaks)
+        self.assertFloatArrayEqual(limits_c, rdf_peaks)
+        self.assertFloatArrayEqual(limits_d, rdf_peaks)
+        self.assertFloatArrayEqual(limits_e, rdf_peaks)
 
 
 class TestRebinData(unittest.TestCase, CustomAssertions):