Fix test

TorchIO-project · Sep 23, 2024 · 0475b39 · 0475b39
1 parent c9024b7
commit 0475b39
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/tests/transforms/augmentation/test_random_ghosting.py b/tests/transforms/augmentation/test_random_ghosting.py
@@ -73,5 +73,5 @@ def test_out_of_range_restore(self):
             RandomGhosting(restore=-1)
 
     def test_wrong_restore_type(self):
-        with pytest.raises(TypeError):
+        with pytest.raises(ValueError):
             RandomGhosting(restore='wrong')