use np.allclose

fortyninemaps · Mar 28, 2017 · baf6fef · baf6fef
1 parent f039da6
commit baf6fef
Showing 1 changed file with 2 additions and 6 deletions.
diff --git a/tests/aai_tests.py b/tests/aai_tests.py
@@ -10,18 +10,14 @@ def test_read_aai_corner(self):
         pe = peaks(n=49)
         control = karta.RegularGrid((0.0, 0.0, 30.0, 30.0, 0.0, 0.0), values=pe)
         grid = karta.read_aai(os.path.join(TESTDATA, "peaks49_corner.asc"))
-        print(grid[::-1][:8,:8])
-        print(control[:8,:8])
-        self.assertTrue(np.all(grid[::-1] == control[:,:]))
+        self.assertTrue(np.allclose(grid[::-1], control[:,:]))
         return
 
     def test_read_aai_center(self):
         pe = peaks(n=49)
         control = karta.RegularGrid((0.0, 0.0, 30.0, 30.0, 0.0, 0.0), values=pe)
         grid = karta.read_aai(os.path.join(TESTDATA, "peaks49_center.asc"))
-        print(grid[::-1][:8,:8])
-        print(control[:8,:8])
-        self.assertTrue(np.all(grid[::-1] == control[:,:]))
+        self.assertTrue(np.allclose(grid[::-1], control[:,:]))
         return
 
     def test_write_aai(self):