Fixing unit test warnings

the-lay · Feb 13, 2022 · 6965e0d · 6965e0d
1 parent 62871d9
commit 6965e0d
Show file tree

Hide file tree

Showing 2 changed files with 4 additions and 3 deletions.
diff --git a/tests/test_merger.py b/tests/test_merger.py
@@ -24,9 +24,9 @@ def test_init(self):
 
         # Check save_visits
         merger3 = Merger(tiler=tiler, save_visits=False)
-        self.assert_(merger3.data_visits is None)
+        self.assertTrue(merger3.data_visits is None)
         merger3 = Merger(tiler=tiler, save_visits=True)
-        self.assert_(merger3.data_visits is not None)
+        self.assertTrue(merger3.data_visits is not None)
 
         # Check data and weights dtypes
         merger4 = Merger(

diff --git a/tests/test_tiler.py b/tests/test_tiler.py
@@ -36,7 +36,8 @@ def test_init(self):
             )
 
         # test tile shape broadcasting
-        tiler = Tiler(data_shape=(300, 300), tile_shape=(10,))
+        with self.assertWarns(UserWarning):
+            tiler = Tiler(data_shape=(300, 300), tile_shape=(10,))
         assert np.allclose(tiler.tile_shape, (1, 10))
 
     def test_repr(self):