Commit 5280b44d authored by narugo1992's avatar narugo1992
Browse files

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

parent 10514ed2

test/detect/test_head.py

+4 −0
Original line number Diff line number Diff line
@@ -33,3 +33,7 @@ class TestDetectHead: 


    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')