test for invalid model

microsoft · Apr 22, 2023 · d71a252 · d71a252
1 parent e8b001f
commit d71a252
Showing 1 changed file with 6 additions and 0 deletions.
diff --git a/tests/trainers/test_regression.py b/tests/trainers/test_regression.py
@@ -286,6 +286,12 @@ def test_trainer(
         except MisconfigurationException:
             pass
 
+    def test_invalid_model(self, model_kwargs: dict[str, Any]) -> None:
+        model_kwargs["model"] = "invalid_model"
+        match = "Model type 'invalid_model' is not valid."
+        with pytest.raises(ValueError, match=match):
+            PixelwiseRegressionTask(**model_kwargs)
+
     @pytest.fixture
     def model_kwargs(self) -> dict[str, Any]:
         return {