add extra tests for axis in RandomMotion

fepegar · Jul 6, 2024 · 1d8761c · 1d8761c
1 parent 08357f0
commit 1d8761c
Showing 1 changed file with 12 additions and 0 deletions.
diff --git a/tests/transforms/augmentation/test_random_motion.py b/tests/transforms/augmentation/test_random_motion.py
@@ -58,3 +58,15 @@ def test_wrong_image_interpolation_type(self):
  def test_wrong_image_interpolation_value(self):
  with pytest.raises(ValueError):
  RandomMotion(image_interpolation='wrong')
+
+ def test_out_of_range_axis(self):
+ with pytest.raises(ValueError):
+ RandomMotion(axes=3)
+
+ def test_out_of_range_axis_in_tuple(self):
+ with pytest.raises(ValueError):
+ RandomMotion(axes=(0, -1, 2))
+
+ def test_wrong_axes_type(self):
+ with pytest.raises(ValueError):
+ RandomMotion(axes=None)