fixed some test functions.

LSST-strong-lensing · Nov 22, 2024 · 691cb02 · 691cb02
1 parent 50949d8
commit 691cb02
Showing 1 changed file with 1 addition and 3 deletions.
diff --git a/tests/test_lens.py b/tests/test_lens.py
@@ -580,9 +580,7 @@ def test_point_source_arrival_time_multi(self):
         point_source_arival_time2 = self.lens_class2.point_source_arrival_times()
         point_source_arival_time3 = self.lens_class3.point_source_arrival_times()
         # Test multisource point source arival time.
-        assert np.all(point_source_arival_time1[0]) == np.all(
-            point_source_arival_time3[0]
-        )
+        assert point_source_arival_time1[0] == point_source_arival_time3[0]
         assert np.all(point_source_arival_time2[0]) == np.all(
             point_source_arival_time3[1]
         )