add unittest

src/yomitoku/export/export_csv.py

@@ -41,6 +41,8 @@ def save_figure(
     out_path,
     figure_dir="figures",
 ):
+    assert img is not None, "img is required for saving figures"
+
     for i, figure in enumerate(figures):
         x1, y1, x2, y2 = map(int, figure.box)
         figure_img = img[y1:y2, x1:x2, :]

src/yomitoku/export/export_html.py

@@ -110,6 +110,8 @@ def figure_to_html(
     figure_dir="figures",
     width=200,
 ):
+    assert img is not None, "img is required for saving figures"
+
     elements = []
     for i, figure in enumerate(figures):
         x1, y1, x2, y2 = map(int, figure.box)

src/yomitoku/export/export_json.py

@@ -21,6 +21,8 @@ def save_figure(
     out_path,
     figure_dir="figures",
 ):
+    assert img is not None, "img is required for saving figures"
+
     for i, figure in enumerate(figures):
         x1, y1, x2, y2 = map(int, figure.box)
         figure_img = img[y1:y2, x1:x2, :]

src/yomitoku/export/export_markdown.py

@@ -75,6 +75,8 @@ def figure_to_md(
     width=200,
     figure_dir="figures",
 ):
+    assert img is not None, "img is required for saving figures"
+
     elements = []
     for i, figure in enumerate(figures):
         x1, y1, x2, y2 = map(int, figure.box)

tests/test_cli.py

@@ -210,6 +210,7 @@ def test_validate_encoding():
         validate_encoding("utf-9")
 
     assert validate_encoding("utf-8")
+    assert validate_encoding("utf-8-sig")
     assert validate_encoding("shift-jis")
     assert validate_encoding("euc-jp")
     assert validate_encoding("cp932")