first commit

tests/models/test_layout_detection.py

@@ -0,0 +1,51 @@
+import pytest
+
+from paddleocr import LayoutDetection
+from ..testing_utils import (
+    TEST_DATA_DIR,
+    check_simple_inference_result,
+    check_wrapper_simple_inference_param_forwarding,
+)
+from .object_detection_common import check_result_item_keys
+
+
+@pytest.fixture(scope="module")
+def layout_detection_predictor():
+    return LayoutDetection()
+
+
+@pytest.mark.parametrize(
+    "image_path",
+    [
+        TEST_DATA_DIR / "doc_with_formula.png",
+    ],
+)
+def test_predict(layout_detection_predictor, image_path):
+    result = layout_detection_predictor.predict(str(image_path))
+
+    check_simple_inference_result(result)
+    check_result_item_keys(result[0])
+
+
+@pytest.mark.parametrize(
+    "params",
+    [
+        {"img_size": 640},
+        {"threshold": 0.5},
+        {"layout_nms": True},
+        {"layout_unclip_ratio": True},
+        {"layout_merge_bboxes_mode": True},
+    ],
+)
+def test_predict_params(
+    monkeypatch,
+    layout_detection_predictor,
+    params,
+):
+    check_wrapper_simple_inference_param_forwarding(
+        monkeypatch,
+        layout_detection_predictor,
+        "paddlex_predictor",
+        "dummy_path",
+        params,
+    )