Fix tests expecting AssertionErrors

pytroll · Nov 17, 2023 · 14331e4 · 14331e4
1 parent f07a571
commit 14331e4
Showing 1 changed file with 7 additions and 6 deletions.
diff --git a/pyresample/test/test_kd_tree.py b/pyresample/test/test_kd_tree.py
@@ -21,6 +21,7 @@
 from unittest import mock
 
 import numpy as np
+import pytest
 
 from pyresample import geometry, kd_tree, utils
 from pyresample.test.utils import catch_warnings
@@ -927,8 +928,8 @@ def test_nearest_area_2d_to_area_1n(self):
         for val in ninfo[:3]:
             # vii, ia, voi
             self.assertIsInstance(val, da.Array)
-        self.assertRaises(AssertionError,
-                          resampler.get_sample_from_neighbour_info, data)
+        with pytest.raises(ValueError):
+            resampler.get_sample_from_neighbour_info(data)
 
         # rename data dimensions to match the expected area dimensions
         data = data.rename({'my_dim_y': 'y', 'my_dim_x': 'x'})
@@ -954,8 +955,8 @@ def test_nearest_area_2d_to_area_1n_no_roi(self):
         for val in ninfo[:3]:
             # vii, ia, voi
             self.assertIsInstance(val, da.Array)
-        self.assertRaises(AssertionError,
-                          resampler.get_sample_from_neighbour_info, data)
+        with pytest.raises(ValueError):
+            resampler.get_sample_from_neighbour_info(data)
 
         # rename data dimensions to match the expected area dimensions
         data = data.rename({'my_dim_y': 'y', 'my_dim_x': 'x'})
@@ -997,8 +998,8 @@ def test_nearest_area_2d_to_area_1n_3d_data(self):
         for val in ninfo[:3]:
             # vii, ia, voi
             self.assertIsInstance(val, da.Array)
-        self.assertRaises(AssertionError,
-                          resampler.get_sample_from_neighbour_info, data)
+        with pytest.raises(ValueError):
+            resampler.get_sample_from_neighbour_info(data)
 
         # rename data dimensions to match the expected area dimensions
         data = data.rename({'my_dim_y': 'y', 'my_dim_x': 'x'})