From ed9da13292852ffc1f6caad0707c5afaebd88c2e Mon Sep 17 00:00:00 2001
From: SWHL <liekkaskono@163.com>
Date: Sat, 11 Feb 2023 16:44:34 +0800
Subject: [PATCH] Optim unit tests

---
 python/README.md                              |  14 +++---
 python/demo.py                                |   3 +-
 python/test_images/single_line_text.jpg       | Bin 5442 -> 0 bytes
 python/tests/base_module.py                   |   1 +
 python/tests/test_all_ort.py                  |  40 ++++++++++++++++
 python/tests/test_all_vino.py                 |  41 +++++++++++++++++
 python/tests/test_cls.py                      |  41 +++++------------
 python/tests/test_det.py                      |  43 +++++-------------
 .../test_files}/ch_en_num.jpg                 | Bin
 python/tests/test_rec.py                      |  38 +++++-----------
 10 files changed, 124 insertions(+), 97 deletions(-)
 delete mode 100644 python/test_images/single_line_text.jpg
 create mode 100644 python/tests/test_all_ort.py
 create mode 100644 python/tests/test_all_vino.py
 rename python/{test_images => tests/test_files}/ch_en_num.jpg (100%)

diff --git a/python/README.md b/python/README.md
index ebe5a99..c0ab6d4 100755
--- a/python/README.md
+++ b/python/README.md
@@ -61,7 +61,7 @@
 
     rapid_ocr = RapidOCR()
 
-    img = cv2.imread('test_images/ch_en_num.jpg')
+    img = cv2.imread('tests/test_files/ch_en_num.jpg')
 
     result = rapid_ocr(img)
     print(result)
@@ -111,12 +111,10 @@
         │        ├── ch_ppocr_mobile_v2.0_cls_infer.onnx
         │        └── ch_PP-OCRv3_rec_infer.onnx
         ├── requirements.txt
-        ├── resources
-        │     └── fonts
-        │        └── FZYTK.TTF
-        └── test_images
-            ├── ch_en_num.jpg
-            └── single_line_text.jpg
+        └── resources
+              └── fonts
+                 └── FZYTK.TTF
+
         ```
 
 3. 安装运行环境
@@ -178,7 +176,7 @@
 
     - `min_height`是用来过滤只有一行文本的图像（如下图），这类图像不会进入文本检测模块，直接进入后续过程。
 
-      ![](./test_images/single_line_text.jpg)
+      ![](https://github.com/RapidAI/RapidOCR/releases/download/v1.1.0/single_line_text.jpg)
 
 - `Det`部分
     |    参数名称      | 取值范围   | 默认值   |                       作用                       |
diff --git a/python/demo.py b/python/demo.py
index 4830577..035b037 100644
--- a/python/demo.py
+++ b/python/demo.py
@@ -90,8 +90,9 @@ def visualize(image_path, result, font_path="resources/fonts/FZYTK.TTF"):
 if __name__ == '__main__':
     rapid_ocr = RapidOCR()
 
-    image_path = 'test_images/ch_en_num.jpg'
+    image_path = 'tests/test_files/ch_en_num.jpg'
     img = cv2.imread(image_path)
+    img = np.zeros([640, 640, 3])
     result, elapse_list = rapid_ocr(img)
     print(result)
     print(elapse_list)
diff --git a/python/test_images/single_line_text.jpg b/python/test_images/single_line_text.jpg
deleted file mode 100644
index f98cc39fc2581cbc5d15eb0dbdd83066b186325b..0000000000000000000000000000000000000000
GIT binary patch
literal 0
HcmV?d00001

literal 5442
zcmbW4cQhPMyT^B<_Yg!_y%S0FP7r1F602^A5@D?xJ&BSKb)zRl^v<rjRu5U7)rsB-
zs|7(2T)+3d=iYPfU-#bcoO%A3`JS2Q^URz%XRfENmjJi5HNYAGAP@im-Za4VEI<uF
zN=!^bOhigTLPADHN=`{nMM*(Hd5`W64LvI}I~yxA2*knjfS-fwJ~s#?AT4<Rk*I`(
z1UtX1qKufr191tle-;6fk&#hSP%=?bF^O@4IK}?Yx^4r|k^=aFEd)R=z%5!J0WI*l
z1HgWhClT;p0{o|dw+IM{h)GDv$SH0N=-Yr>KmvkWgakxHgoHQlH#hSDLRum^PEi%&
zJ4Vk*xIF2_LKAXGxmBy$8H`7M@rc`bg^`gn-etPS%*)3wASfguDJ3l<E2s8ET|-j~
ztZnkt6k=v>VQKH+_`>O>vy1mDA74NJfWUX}!y`UKMnxwkC8wl*`ka=Yo0nfuSX5k6
zT3u6%uEW$fG=9f+bar+3^!AO8jZaMen3|qhURhmR-}t$?g+Dkv`h9$I`seKYA1)w(
z;NMs`@!w$ogNyct>lPs)0U^mhT)<m?H=Tf%kcd;1m`=rr<hkb^F0oKjdewxSs&+DN
zapPYMc3vaoj64#{ya)fF{fq2>2Nw4KBKt3}|K^$jP!a%d4v&BqpbXf&^5GtysK4EG
z%ivaFG!YLm!;2jqxZR<J1+rt6#m@Q3^M;=r?cEHsf_$E;G{vPkt?r{mv*AR6KQe|^
zXlP+<oO1BgPcZfq`}`lCRGOF1Id-7!GJr!38e31R)W*|DhvxL*mr^1M0w1}kbpPhm
z)LsKh_iL|AQ*f@6AM_c%B3%4U=zzL3hb0`s_8a#$gOQ?pt1;d+fKoAo?9wXfDv?<@
zlgU%w&y`#a6-<9JCz_QXPN$x0*x%u)vgofY@sXv$IQN{Mlo|es?%Tl2+hRP19{BR&
z@)O+JdfOcV2m7WhwDL0>5sP1vOu`}7cBK;i8KRx`E@Nb97+mQ^+D<`jW&n5SJT#42
zv0gy!)r-~Ew<%sD_r?+fS#^yBdq~`rEh~J4a^v&(C+iuY6ubQ{uN7DBro{Kq_dd`L
z11j|6)uD}J5|>It1{38w`>qame|2qnil<KP-CkqlW|wlIM+<p0O0zlB|D6}?!>>_-
zCN7#DCMl8}p@KT((zkSov)sAj^v_a-Jv3gIX9UFUi2un{gq4~^N41sc^>wR}@3!OQ
zz~*&C9&Ef#>0`Q3VVyV>^7jFYP0wM<8IUQqj%Hxq05jS6a;(!@BDL`mNo7NF){vw9
zz!$jyW<*5N+HpQsXg!#@j)PZ2b3juDrQl-ODk)qdSw_Jk^KcEVqL`4-YHc6Cs+=PG
zROfLx-~3=0_G$DX9{&SYeu82c>7pj>d5$ltI#Ei=xo`hccI`LA&u_Dl=V|o@cPze4
zeCQWpi&EOZpK&O?dJjaW=s7;wIE0#A+^p^`?GA7b_FSiDw7XxCx}VUij5RE@E}6n-
z*Dfs80rpM$;N$eFMK_O{jHLF_)4FOyHy*8ckb6Q{?4_Q!B{il9_W;G+5bbPvuA!Be
zqSGUVl8QY_uUi<v@r<p@*~rn<A}do~uY`YhO3V&17~^^C^W5q=+i|B}Ih2NZra6$f
zu*3Q2Vd{5YJB6OX<xgG2fbBUma6bM3f1tfWhFlAK5n*09+W-Q19wy4&4m>R}Hc&ld
zHj^}}>XIA7WE^@BJ)#|FUiW!pVlZ)x8QDA5fkCB)V9&(oB4ZEPARJ!QB#`dP<NWzJ
zZpOiyz#3b&n+q=VNLt>E*Lj77itSKD>cXO?sU~B$_M6pvKecJT_P<#<`|o=38}G4)
zRYN()Ibg*P0_X<)y_X}P(D#O5`ziA1g<XZ@L@^#s!#->dNlxhkcRY4EPX7tSS!n<t
zoyVINO0Fy>98*_QIl2*zHJfnl>mYRPT@keRuj&?MWI89%8yp<yo?2YcgkP3p%<eLd
z$R61W@mjySFQ%-GW~K+`yP?-RwscD8U5xR*ca8d$Kb@$RifM5@rUhxSQ+80sFSMxO
z0+qfNPHzpfSGgaqWB-mRWFNE>MO_MZ`Q0h%E%OtpE#~HiE6wg)1KvIc|7Oq*l6A97
zpxfHD_|PvTRAutpR&&ducNwQ%lXVBLwna~%{AXRdu%iqU^!3GNflV^pM#V6If;${Q
z#4r^1V2>8vtW#DePlRE=(5}5&tmcUn7c~Vm*^BIX0C19w-Nqg|CfX-Qx+4-lo0~ln
zG;75f_zB}l08WXI%gNqyyySo16z_EP)$tmz;_@-};d%_Vcr()>_i!RXJg!6cJ>|)k
z%H;8%k-{(7qL=r2TR(<^i-PEj*sLlBKL43jab=t!v^`Cfj(zJ&<Zfkk*6tLA5f15_
zX5t<<qblAG+$$;=VyFwPg|_vA7G2}QBCLroy`dt>iL(~%F0tBb%w0uOl)>7?g#!&M
zS$|o+n9<h9?k6dIFS9~n>*I+WUHndcN4NUwJ8`3Joh{@w!8>R+q^BpG0TVwRP{Hxe
z6ZAYE`0ZeRJaJ(iJYVJ`N5B<+f#oZi)5YhPJ@unF2~NpMFEnrRS`>jZai{xO4u8Hn
zP~9WCgek2)MaP&jo0Lb%ejJFKqe`M))(=E1EnP7T`6Fy-nD!~ttJRAmr=N;k<Y-cK
z_b^e{uQGgu^S|dG6{&czH&Vd}<mR^NZB|-%e0KhmD`#-D8na*-=Fw!ZC*LaG+i5d4
z46xdlc6T}2Cx>p<HTu*Zx%~RFl{XN>y9IAT`0k2?B1m)(pdEMmDI8@wVYnfdj05cy
zje;}tCJBcyn{{{XesR??FnME|J)oMPheU?|P*U^hf&26xp||t-JI16w4J{=zw>COW
zYl!+r5pN3?rl~e=(t~kG$kZ|vbMoU3;AfKQE-AQIjO;cB?Hi;nZul5Aau8J;WS7MD
zwK<W~xI#3(1_iy+fo?@iJR)aIv(9RJg(U7Kq(<8?7HS(KV%3E9h4K`)!pvtN9|P0q
zZ}l)a5M<CP{t6cQc}$Vpm|=<dN_d2$FPy~7jByTosq})vyl5I&@#vk#j8o(r`+WRu
z?Gv1vPNac{Y1M|0%@=md%vUjSQVT^;cW?Tf0qQt(w5TyPlMlNQgj^zo?&mL5FxK0O
z!fb4J*G(jI;C-s>Nh)u|Zbu2nPuW3OHl3J<koz0`Nc3G8OKH2jT^TGe9G9AKKUHei
z(|=8q(3VcRs!e1wPnLh!6pP-ya<f7!QM~E14kPl&ES)=wF`!wP_RqD_p)%bq_^2PX
z=cbU7P?WHqT*Ep?neg|6OBQ$pZSnefxv+oYF)K&iypS5T!%$j?iYl?}ru^hyPL2z1
z1$?ZiY>KHxSiQ^^<}LYL1@X>muD_UoC{z0*SXDQU6~HeO=Y;Nm9+Tk|M^t;B#_2`O
z4`w=(zWI>7WmsKSV(ma1K_Ac44+Hs`lpmEYZ<wDDD>Zp}o(-#Tgm@^zOiN-K{bY~<
z6}8`pA0#%qp3=^AyVJ~cRF&3Es(dZLcxlK%lBH%M@=8>Ef@~*)9p~%^`YvK9L94S1
zwLbl1?zPTH`1e{wkH#tjk$)93nH=kDn{f@$=6=;_jB_!%xFgzYYtUkE!`;5h%5qr-
zE9kif)M=iZsaj_<ZB+Zjz%^D*^}mhT)uqDJ_%`==MI+3tJybK7ASrw^qDA_&sUPkZ
z<0;%%LU5~$bRzF_92RdapQ5;BbSaP5oT6KT%4hKl0PxbdLvxA~I?+VbModbazi;+2
z2!!}r<|c1LTY)2vS>E=>J9!0bbwVa2KHk}8)aW~bbICjN#N{6vtNt;AEG5MrY?*(B
zB2chcX~hp9GuGm*en*D<f`yg6H`?>42L)Yc)PI3HdD%v7s1`F{i|`lQhUg9{7`<cN
zGc|78SHNmL*PRiZv-eA*gHWwz+HxVTGA5nGyn71X$1CCBFNFRp8Yi=`p(|Nq|5B2s
zUv(vfq@&s=gS52Z`<`tClNBm?Tf`FV)4uXyvDbjSBAdQK@WQ%EemO33ox{EmmhuBF
zJ?d77#-w9)!u*b&Ezk8!3_d1tQ2WNT{~MJOzf`r-aH&6LoO<@`J5&j9*L{XJq+j3~
z;1{nf_Zma(b0@QMg5~?E_rjv(d}rYz2#H&Xs1Ha9d<0q!kyq)WVFE;NyYi2_AtBo)
z2>Og3hcMQVFOI??(!oK5YZnQl^}ALY?+ojiT@w~uEMs(NQ}_0yrs<!j%sRI{CMA3x
zz;gs8{#s<Sl@w{;dh~mVdJ09KX7`)?Y%RE1eE*5v7@sLaGtL4IIgpy;k~`)M;i|uI
zfXeloOEFenw(3EVyZMKqqKu@{)G{*^?BAQrXW9K%N?!U_&q@`K77oc>f$fJ51Wf@~
z5gD2JV-UR*UF5@$?@nTLXMZm~yaqfsM@r5k7cj+BpWJKko1>~}9VP1;C{a9t+d0w|
z>l{q9S5xviPuirq8&1(1ez;2FW=;^-`T2?~2@Ao9&RFXO0K-=Bc&iVQm3C<|MhWd_
zsw1#ZL+))`s_`T(-nmvd;LfWcR%!jlhz&DRw-@jo59nHO{3-s87Alp;nfu-<>$bG<
zu;aJ!>~Rqv!JR#hs2pfj9%_;!lL^G0?7eFuuI&J0%XhQGG>rXKs+uDv<`frKum}Ds
z6wM?(S%WO3<j!uLyQUiU%YSF8745tDn{u_1*`#R;?s_9kZa_U(TX!^6zA$HPrCX!A
zmr$(h(MDv16%_I^uyMhQtO4FFwm8-_Ajcr0>4KBSWNpOt7jbUq*8ttt`r3Eu(IbD?
z5a7M&Ual~+x5FP$VzQ{8&$4>Zu%6u)^*&ql@_y1Ak1t^%^}q6^Cy(%VXCPS)@g0Yq
zjxcg+)kI1Z$w2Tl`@>0e%FJ+=s1aQ1Y2}2E)kl))sZP&qHBcQHX`;^L5v?{bvB|TZ
zdU_kWmnpI|hFrz?_}fPri&3i>K9WUgk1pzs{FTkuQ3^dtPVC|ffbT+#*=?7Tspig)
zhO8k!BP6zCScw<NM(jR46xNVaZ<#)!DcpHWpId&%Z)%E?=}^@A?-hHdncV`%WVOCW
z4cEtBLrB3fa$-Qc&Tsd{()wbbg_z<a7?+;3wQl9Xlh4<H3W0CmOSXsOQc+mDU+3*Y
zJ?Di;L1Sv3CRb2kXl!(A|Dd-8vT!?G4HLO0dt4J6teuH-vyB?X|Cw))rL-dp-3qlD
zd#1??b6fZN5H;(Q66<iet2%qo5t21*CNZ-<n6r%amW~z4P1mv8g;#^Z%=Q8=0)LnW
zJB#d<6gB}*U`_#xnH3lp%HX!oR}45!`ylx8htV%sJ&`tyKjX3`9a%fD!S$Rutpxh*
z=#soZ;IQDzro>Xw7^<uqX08B*DO9Pm%|5D3RU{siy+UuocoH%yBs+UyJ^@8mQx79}
zFiEe3PN~+GO)YfTSQLRMTqxjRvh<VF`BlCho%Z$+X48IK+}e8El7|wko(b+@t?R;`
z)t51!<Z+=s(q)rNG9lh*K_um|EY_N`HVa+l(F>uG{MP^#_7thZ(I19&Qm66d33SwB
zc)8n}YLKnMv{>Pb{{Ab{^wx5C>hxD>i63Nud)*5%LrY>VZeSV7FYe2`d^770xN2-9
zsg35-`m$V;qQ`Ebq^;7%oi`Cqo!(GQ&jK}fo@oA*Np*yJY$d7IS)|OC^+d-fsdF5%
zNpcv>?dvF){3KEIT@w>)AltWapDXUv-aN#@l9fdEzLAgJsUSiV==tu?4+j6rx*FSq
zH=%aw$@1#C+Qkh7ET8_=%(BWJqg)mlFA)PmucIXA@Ct%{oCk`XU;0Jg?dUw{MR9+j
zYK{%doG+_)Ab)Epi=t3#L|3;QQ5G-%a1;4$kt5sFWtAyhaiP|#vQnC1Wt9ZUHAY9^
zEnhJgBv1MghI?WK+ttP&DP@NQldr8bXIYIWVWr{hUtKQt^UYYCp0pe0DR*eym$yY6
zkJNpwaQX2N24||Q-8?PFkIGyF3SH#hSQQmm)c?}Ubsn)0pp7(dHqq$NMg-QsZTjeS
z^SeocZ(ReZof#Z`zbyr%LRZ5*M%E_$Qot)>LM)_cVVwl0l6=2mX^J3<$ob&ye5a3{
z2Fu2MIS%q(8bOo4XA-)hGQL$IlTs(A(%VucC#LoFcgd>;ino(ZIeu1gOD3^&yw-(}
z*<=b^nS!nR{aJPie-l(1A%ZUDeyEjpy3~z60~crpaM=rSL42)4JQD_q^%cvw`mlQA
ze^uQ%Mpt!6k-94R=qaA~N?G#93IGkyg<Ovb-t+X)n{vHrZjNi=6Ie;S+7OQ1)0^-9
z{7j70km2dWn@x_9lJTK^8)fb8ROQd0wMO65dd<fL_&0@AJA5Q_qtUUvrgPgb!gC#k
zRQQw98>@5B#y~<RrQb%+>@C@!IJF)<2qafeh~m6>l^b1Dl^*y`Kv9T<(|?FBS|~w`
ziTZJoE02-7xMZ;eC%|O#Q;)p=O0-~Hu2IM>#5j`8Q=LBg$wQ`l4A8ur;khW<NPT_h
zr$nyjM(WOuVzA;fo&b75SCJ>&<d5Wx)Jb`gSk|9ooQhmD&RS~J8D891Z{swO6swN+
zJ;{y!(4h6&C;->HQlMoCihcb(T2Nni<~F1ukY~W<?939y#nqVcmfpx*Ybye~5#nX|
zx+pzlR_wvC&1AHYUVtge(MDFN&!EqxTz85*-}WF(Zbbc6_p(AC^1|4YNhKpTG$h+Y
z5ZA2nNai><nko<7erV`mn_hIE&@q%n6G?A^*zh)kIGXv*_VW0au6uc_GmP9BENxj@
zvQK~emquHljPqr;Cg)<(u*r=2eH}JMf!y4g^douNuzLq1LB?{#Za1D%@%brd%GU8T
zx8Fv(Aj2hO>`J)8QtXI2fsq;Q2zc-jS_rNy#^*UcqRuCiA9LVR%|{Q^|71KLk4?I%
z*^At`bz|^L6Q4M&jGfhJsWf4%=WMz_86x)k+ry@N<L8EE2{?LFOinUalp4oUp&lX8
RPY|;}$tGOVA^3XwKLEqaT1@}|

diff --git a/python/tests/base_module.py b/python/tests/base_module.py
index b6c7e18..13bd0bd 100644
--- a/python/tests/base_module.py
+++ b/python/tests/base_module.py
@@ -13,6 +13,7 @@ class BaseModule():
         self.package_name = package_name
         self.root_dir = Path(__file__).resolve().parent.parent
         self.package_dir = self.root_dir / self.package_name
+        self.tests_dir = self.root_dir / 'tests'
 
         sys.path.append(str(self.root_dir))
         sys.path.append(str(self.package_dir))
diff --git a/python/tests/test_all_ort.py b/python/tests/test_all_ort.py
new file mode 100644
index 0000000..1a77900
--- /dev/null
+++ b/python/tests/test_all_ort.py
@@ -0,0 +1,40 @@
+# -*- encoding: utf-8 -*-
+# @Author: SWHL
+# @Contact: liekkaskono@163.com
+import sys
+from pathlib import Path
+
+import numpy as np
+import cv2
+import pytest
+
+
+root_dir = Path(__file__).resolve().parent.parent
+sys.path.append(str(root_dir))
+
+from rapidocr_onnxruntime import RapidOCR
+
+rapid_ocr = RapidOCR()
+tests_dir = root_dir / 'test_files'
+
+
+def test_normal():
+    image_path = tests_dir / 'ch_en_num.jpg'
+    img = cv2.imread(str(image_path))
+    result, _ = rapid_ocr(img)
+    assert result[0][1] == '正品促销'
+    assert len(result) == 17
+
+
+def test_empty():
+    img = None
+    with pytest.raises(AttributeError) as exc_info:
+        rapid_ocr(img)
+        raise AttributeError
+    assert exc_info.type is AttributeError
+
+
+def test_zeros():
+    img = np.zeros([640, 640, 3])
+    result, _ = rapid_ocr(img)
+    assert result is None
diff --git a/python/tests/test_all_vino.py b/python/tests/test_all_vino.py
new file mode 100644
index 0000000..fe1cca0
--- /dev/null
+++ b/python/tests/test_all_vino.py
@@ -0,0 +1,41 @@
+# -*- encoding: utf-8 -*-
+# @Author: SWHL
+# @Contact: liekkaskono@163.com
+import sys
+from pathlib import Path
+
+import numpy as np
+import cv2
+import pytest
+
+
+root_dir = Path(__file__).resolve().parent.parent
+sys.path.append(str(root_dir))
+
+from rapidocr_openvino import RapidOCR
+
+rapid_ocr = RapidOCR()
+
+tests_dir = root_dir / 'test_files'
+
+
+def test_normal():
+    image_path = tests_dir / 'ch_en_num.jpg'
+    img = cv2.imread(str(image_path))
+    result, _ = rapid_ocr(img)
+    assert result[0][1] == '正品促销'
+    assert len(result) == 17
+
+
+def test_empty():
+    img = None
+    with pytest.raises(AttributeError) as exc_info:
+        rapid_ocr(img)
+        raise AttributeError
+    assert exc_info.type is AttributeError
+
+
+def test_zeros():
+    img = np.zeros([640, 640, 3])
+    result, _ = rapid_ocr(img)
+    assert result is None
diff --git a/python/tests/test_cls.py b/python/tests/test_cls.py
index ed4cc6c..c9ec655 100644
--- a/python/tests/test_cls.py
+++ b/python/tests/test_cls.py
@@ -1,15 +1,20 @@
 # -*- encoding: utf-8 -*-
 # @Author: SWHL
 # @Contact: liekkaskono@163.com
-from base_module import BaseModule
 import cv2
+import pytest
+from base_module import BaseModule
 
 
-class TestONNXRuntime():
+@pytest.mark.parametrize(
+    'package_name',
+    ['rapidocr_onnxruntime', 'rapidocr_openvino']
+)
+def test_cls(package_name: str):
     module_name = 'ch_ppocr_v2_cls'
     class_name = 'TextClassifier'
 
-    base = BaseModule(package_name='rapidocr_onnxruntime')
+    base = BaseModule(package_name=package_name)
 
     TextClassifier = base.init_module(module_name, class_name)
 
@@ -19,29 +24,7 @@ class TestONNXRuntime():
 
     text_cls = TextClassifier(config)
 
-    def test_cls(self):
-        img_path = 'test_files/text_cls.jpg'
-        img = cv2.imread(img_path)
-        result = self.text_cls([img])
-        assert result[1][0][0] == '180'
-
-
-class TestOpenVINO():
-    module_name = 'ch_ppocr_v2_cls'
-    class_name = 'TextClassifier'
-
-    base = BaseModule(package_name='rapidocr_openvino')
-
-    TextClassifier = base.init_module(module_name, class_name)
-
-    yaml_path = base.package_dir / module_name / 'config.yaml'
-    config = base.read_yaml(str(yaml_path))
-    config['model_path'] = str(base.package_dir / config['model_path'])
-
-    text_cls = TextClassifier(config)
-
-    def test_cls(self):
-        img_path = 'test_files/text_cls.jpg'
-        img = cv2.imread(img_path)
-        result = self.text_cls([img])
-        assert result[1][0][0] == '180'
+    img_path = base.tests_dir / 'test_files' / 'text_cls.jpg'
+    img = cv2.imread(str(img_path))
+    result = text_cls([img])
+    assert result[1][0][0] == '180'
diff --git a/python/tests/test_det.py b/python/tests/test_det.py
index 1e03a06..c8e9ec5 100644
--- a/python/tests/test_det.py
+++ b/python/tests/test_det.py
@@ -2,14 +2,19 @@
 # @Author: SWHL
 # @Contact: liekkaskono@163.com
 import cv2
+import pytest
 from base_module import BaseModule
 
 
-class TestONNXRuntime():
+@pytest.mark.parametrize(
+    'package_name',
+    ['rapidocr_onnxruntime', 'rapidocr_openvino']
+)
+def test_det(package_name):
     module_name = 'ch_ppocr_v3_det'
     class_name = 'TextDetector'
 
-    base = BaseModule(package_name='rapidocr_onnxruntime')
+    base = BaseModule(package_name)
     TextDetector = base.init_module(module_name, class_name)
 
     yaml_path = base.package_dir / module_name / 'config.yaml'
@@ -17,33 +22,7 @@ class TestONNXRuntime():
     config['model_path'] = str(base.package_dir / config['model_path'])
 
     text_det = TextDetector(config)
-
-    def test_det(self):
-        img_path = 'test_files/text_det.jpg'
-
-        img = cv2.imread(img_path)
-
-        dt_boxes, elapse = self.text_det(img)
-        assert dt_boxes.shape == (18, 4, 2)
-
-
-class TestOpenVINO():
-    module_name = 'ch_ppocr_v3_det'
-    class_name = 'TextDetector'
-
-    base = BaseModule(package_name='rapidocr_openvino')
-    TextDetector = base.init_module(module_name, class_name)
-
-    yaml_path = base.package_dir / module_name / 'config.yaml'
-    config = base.read_yaml(str(yaml_path))
-    config['model_path'] = str(base.package_dir / config['model_path'])
-
-    text_det = TextDetector(config)
-
-    def test_det(self):
-        img_path = 'test_files/text_det.jpg'
-
-        img = cv2.imread(img_path)
-
-        dt_boxes, elapse = self.text_det(img)
-        assert dt_boxes.shape == (18, 4, 2)
+    img_path = base.tests_dir / 'test_files' / 'text_det.jpg'
+    img = cv2.imread(str(img_path))
+    dt_boxes, elapse = text_det(img)
+    assert dt_boxes.shape == (18, 4, 2)
diff --git a/python/test_images/ch_en_num.jpg b/python/tests/test_files/ch_en_num.jpg
similarity index 100%
rename from python/test_images/ch_en_num.jpg
rename to python/tests/test_files/ch_en_num.jpg
diff --git a/python/tests/test_rec.py b/python/tests/test_rec.py
index 34b4041..98b79d0 100644
--- a/python/tests/test_rec.py
+++ b/python/tests/test_rec.py
@@ -2,14 +2,19 @@
 # @Author: SWHL
 # @Contact: liekkaskono@163.com
 import cv2
+import pytest
 from base_module import BaseModule
 
 
-class TestONNXRuntime():
+@pytest.mark.parametrize(
+    'package_name',
+    ['rapidocr_onnxruntime', 'rapidocr_openvino']
+)
+def test_det(package_name):
     module_name = 'ch_ppocr_v3_rec'
     class_name = 'TextRecognizer'
 
-    base = BaseModule(package_name='rapidocr_onnxruntime')
+    base = BaseModule(package_name)
     TextRecognizer = base.init_module(module_name, class_name)
 
     yaml_path = base.package_dir / module_name / 'config.yaml'
@@ -18,28 +23,7 @@ class TestONNXRuntime():
 
     text_rec = TextRecognizer(config)
 
-    def test_rec(self):
-        img_path = 'test_files/text_rec.jpg'
-        img = cv2.imread(img_path)
-        rec_res, elapse = self.text_rec([img])
-        assert rec_res[0][0] == '韩国小馆'
-
-
-class TestOpenVINO():
-    module_name = 'ch_ppocr_v3_rec'
-    class_name = 'TextRecognizer'
-
-    base = BaseModule(package_name='rapidocr_openvino')
-    TextRecognizer = base.init_module(module_name, class_name)
-
-    yaml_path = base.package_dir / module_name / 'config.yaml'
-    config = base.read_yaml(str(yaml_path))
-    config['model_path'] = str(base.package_dir / config['model_path'])
-
-    text_rec = TextRecognizer(config)
-
-    def test_rec(self):
-        img_path = 'test_files/text_rec.jpg'
-        img = cv2.imread(img_path)
-        rec_res, elapse = self.text_rec([img])
-        assert rec_res[0][0] == '韩国小馆'
+    img_path = base.tests_dir / 'test_files' / 'text_rec.jpg'
+    img = cv2.imread(str(img_path))
+    rec_res, elapse = text_rec([img])
+    assert rec_res[0][0] == '韩国小馆'