Add more test case

openvinotoolkit · Mar 15, 2024 · 2698d1b · 2698d1b
1 parent 2f6f9f2
commit 2698d1b
Showing 1 changed file with 10 additions and 0 deletions.
diff --git a/tests/unit/engine/test_engine.py b/tests/unit/engine/test_engine.py
@@ -1,6 +1,7 @@
 # Copyright (C) 2024 Intel Corporation
 # SPDX-License-Identifier: Apache-2.0
 
+import pytest
 from otx.engine import Engine
 
 
@@ -26,3 +27,12 @@ def test_from_config_with_model_name(self, tmp_path) -> None:
         assert engine is not None
         assert engine.datamodule.config.train_subset.batch_size == 2
         assert engine.datamodule.config.test_subset.subset_name == "TESTING"
+
+        with pytest.raises(FileNotFoundError):
+            engine = Engine.from_model_name(
+                model_name="wrong_model",
+                data_root=data_root,
+                task=task_type,
+                work_dir=tmp_path,
+                **overriding,
+            )