mtfyun/rapidocr
1
0 Fork 0
mirror of https://github.com/RapidAI/RapidOCR.git synced 2026-11-11 05:20:49 +08:00
Code Issues Packages Projects Releases Wiki Activity

test: update unit testing when adjusting the format of returning word box

Browse Source
This commit is contained in:
SWHL committed 2025-06-18 08:22:56 +08:00
1 parent ea04c30d40
commit 0b2d7f0d24
3 files changed
+11 -39
Download Patch File Download Diff File Expand all files Collapse all files

No files matched your search

python/tests/test_main.py
+5 -35
View File
@@ -406,44 +406,14 @@ def test_input_three_ndim_one_channel(engine):
),
(
"text_vertical_words.png",
(
"已",
"取",
"之",
"時",
"不",
"參",
"一",
"人",
"見",
"而",
"是",
"非",
"不",
"得",
"問",
"之",
"人",
"要",
"取",
"之",
"有",
"是",
"是",
"非",
"非",
"之",
"士",
"師",
"也",
),
("已", "取", "之", "時", "不", "參", "一", "人", "見", "而"),
),
],
)
def test_cn_word_ocr(engine, img_name: str, words: List[str]):
img_path = tests_dir / img_name
result = engine(img_path, return_word_box=True)
txts, _, _ = list(zip(*result.word_results))
txts, _, _ = list(zip(*result.word_results[0]))
assert txts == words
@@ -454,12 +424,12 @@ def test_cn_word_ocr(engine, img_name: str, words: List[str]):
def test_en_word_ocr(engine, img_name: str, words: str):
img_path = tests_dir / img_name
result = engine(img_path, return_word_box=True)
txts, _, _ = list(zip(*result.word_results))
txts, _, _ = list(zip(*result.word_results[0]))
assert txts[0] == words
def test_en_return_single_char_box(engine):
img_path = tests_dir / "en.jpg"
result = engine(img_path, return_word_box=True, return_single_char_box=True)
txts, _, _ = list(zip(*result.word_results))
assert txts[:3] == ("3", "M", "o")
txts, _, _ = list(zip(*result.word_results[0]))
assert txts[:3] == ("3",)