Skip to main content
Glama
test_text_image_unwarping.py1.04 kB
import pytest from paddleocr import TextImageUnwarping from ..testing_utils import ( TEST_DATA_DIR, check_simple_inference_result, check_wrapper_simple_inference_param_forwarding, ) @pytest.fixture(scope="module") def text_image_unwarping_predictor(): return TextImageUnwarping() @pytest.mark.parametrize( "image_path", [ TEST_DATA_DIR / "book.jpg", ], ) def test_predict(text_image_unwarping_predictor, image_path): result = text_image_unwarping_predictor.predict(str(image_path)) check_simple_inference_result(result) assert result[0].keys() == { "input_path", "page_index", "input_img", "doctr_img", } @pytest.mark.parametrize( "params", [ {}, ], ) def test_predict_params( monkeypatch, text_image_unwarping_predictor, params, ): check_wrapper_simple_inference_param_forwarding( monkeypatch, text_image_unwarping_predictor, "paddlex_predictor", "dummy_path", params, )

Latest Blog Posts

MCP directory API

We provide all the information about MCP servers via our MCP API.

curl -X GET 'https://glama.ai/api/mcp/v1/servers/PaddlePaddle/PaddleOCR'

If you have feedback or need assistance with the MCP directory API, please join our Discord server