dev(narugo): add test case for model not found

deepghs · Sep 17, 2024 · 5280b44 · 5280b44
1 parent 10514ed
commit 5280b44
Showing 1 changed file with 4 additions and 0 deletions.
diff --git a/test/detect/test_head.py b/test/detect/test_head.py
@@ -33,3 +33,7 @@ def test_detect_heads(self):
 
     def test_detect_heads_none(self):
         assert detect_heads(get_testfile('png_full.png')) == []
+
+    def test_detect_heads_not_found(self):
+        with pytest.raises(ValueError):
+            _ = detect_heads(get_testfile('genshin_post.png'), model_name='not_found')