Skip to content

Commit 3a4a910

Browse files
committed
test(mnn): add mnn unit testings
1 parent 0ae0375 commit 3a4a910

File tree

2 files changed

+18
-6
lines changed

2 files changed

+18
-6
lines changed

python/tests/test_engine.py

Lines changed: 14 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -17,7 +17,13 @@
1717

1818
@mark.parametrize(
1919
"engine_type",
20-
[EngineType.ONNXRUNTIME, EngineType.PADDLE, EngineType.OPENVINO, EngineType.TORCH],
20+
[
21+
EngineType.ONNXRUNTIME,
22+
EngineType.PADDLE,
23+
EngineType.OPENVINO,
24+
EngineType.TORCH,
25+
EngineType.MNN,
26+
],
2127
)
2228
def test_ppocrv5_rec_mobile(engine_type):
2329
engine = RapidOCR(
@@ -36,7 +42,13 @@ def test_ppocrv5_rec_mobile(engine_type):
3642

3743
@mark.parametrize(
3844
"engine_type",
39-
[EngineType.ONNXRUNTIME, EngineType.PADDLE, EngineType.OPENVINO, EngineType.TORCH],
45+
[
46+
EngineType.ONNXRUNTIME,
47+
EngineType.PADDLE,
48+
EngineType.OPENVINO,
49+
EngineType.TORCH,
50+
EngineType.MNN,
51+
],
4052
)
4153
def test_ppocrv5_det_mobile(engine_type):
4254
engine = RapidOCR(

python/tests/test_rec_language.py

Lines changed: 4 additions & 4 deletions
Original file line numberDiff line numberDiff line change
@@ -75,7 +75,7 @@ def test_arabic_lang(engine_type):
7575

7676
@mark.parametrize(
7777
"engine_type",
78-
[EngineType.ONNXRUNTIME, EngineType.OPENVINO, EngineType.PADDLE],
78+
[EngineType.ONNXRUNTIME, EngineType.OPENVINO, EngineType.PADDLE, EngineType.MNN],
7979
)
8080
def test_el_lang(engine_type):
8181
engine = RapidOCR(
@@ -95,7 +95,7 @@ def test_el_lang(engine_type):
9595

9696
@mark.parametrize(
9797
"engine_type",
98-
[EngineType.ONNXRUNTIME, EngineType.OPENVINO, EngineType.PADDLE],
98+
[EngineType.ONNXRUNTIME, EngineType.OPENVINO, EngineType.PADDLE, EngineType.MNN],
9999
)
100100
def test_th_lang(engine_type):
101101
engine = RapidOCR(
@@ -176,7 +176,7 @@ def test_korean_lang(ocr_version, gt):
176176

177177
@mark.parametrize(
178178
"engine_type",
179-
[EngineType.ONNXRUNTIME, EngineType.OPENVINO, EngineType.PADDLE],
179+
[EngineType.ONNXRUNTIME, EngineType.OPENVINO, EngineType.PADDLE, EngineType.MNN],
180180
)
181181
def test_latin_lang(engine_type):
182182
engine = RapidOCR(
@@ -198,7 +198,7 @@ def test_latin_lang(engine_type):
198198

199199
@mark.parametrize(
200200
"engine_type",
201-
[EngineType.ONNXRUNTIME, EngineType.OPENVINO, EngineType.PADDLE],
201+
[EngineType.ONNXRUNTIME, EngineType.OPENVINO, EngineType.PADDLE, EngineType.MNN],
202202
)
203203
def test_eslav_lang(engine_type):
204204
engine = RapidOCR(

0 commit comments

Comments
 (0)