Test exact equality, fix random seed

cta-observatory · Mar 20, 2020 · f824053 · f824053
1 parent c26b534
commit f824053
Showing 1 changed file with 3 additions and 6 deletions.
diff --git a/ctapipe/image/tests/test_hillas.py b/ctapipe/image/tests/test_hillas.py
@@ -211,6 +211,8 @@ def test_straight_line_width_0():
     trans = np.zeros(len(long))
     pix_id = np.arange(len(long))
 
+    np.random.seed(0)
+
     for dx in (-1, 0, 1):
         for dy in (-1, 0, 1):
             for psi in np.linspace(0, np.pi, 20):
@@ -231,9 +233,4 @@ def test_straight_line_width_0():
 
                 img = np.random.poisson(5, size=len(long))
                 result = hillas_parameters(geom, img)
-                if np.isnan(result.width):
-                    print('img', img)
-                    cov = np.cov(x, y, aweights=img)
-                    print('cov', repr(cov))
-                    print('eig', repr(np.linalg.eigh(cov)[0]))
-                assert np.isclose(result.width.value, 0)
+                assert result.width.value == 0