dev(narugo): more unittests

deepghs · Sep 8, 2024 · 6218299 · 6218299
1 parent 89fef6e
commit 6218299
Show file tree

Hide file tree

Showing 2 changed files with 6 additions and 2 deletions.
diff --git a/test/sd/test_nai.py b/test/sd/test_nai.py
@@ -206,5 +206,9 @@ def test_save_image_with_naimeta_both_no_with_title(self, nai3_clear_file, nai3_
                 )
             assert get_naimeta_from_image('image.png') is None
 
-    def test_image_error_with_wrong_format(self):
-        assert get_naimeta_from_image(get_testfile('118519492_p0.png')) is None
+    @pytest.mark.parametrize(['file'], [
+        ('118519492_p0.png',),
+        ('118438300_p1.png',),
+    ])
+    def test_image_error_with_wrong_format(self, file):
+        assert get_naimeta_from_image(get_testfile(file)) is None
diff --git a/test/testfile/118438300_p1.png b/test/testfile/118438300_p1.png