add test for UserWarning raised

aai-institute · Feb 21, 2024 · 589eb1b · 589eb1b
1 parent 4c5eb61
commit 589eb1b
Showing 1 changed file with 3 additions and 1 deletion.
diff --git a/tests/transforms/test_transform.py b/tests/transforms/test_transform.py
@@ -23,4 +23,6 @@ def test_transform_undo(plus_one_transform, random_tensor):
 
 
 def test_transform_undo_not_bijective(abs_transform, random_tensor):
-    assert torch.allclose(abs_transform.undo(random_tensor), random_tensor)
+    with pytest.warns(UserWarning) as record:
+        assert torch.allclose(abs_transform.undo(random_tensor), random_tensor)
+    assert len(record) == 1