update benchmark

inisis · Nov 1, 2024 · 5301e97 · 5301e97
1 parent 474d5c0
commit 5301e97
Showing 1 changed file with 10 additions and 0 deletions.
diff --git a/tests/test_benchmark.py b/tests/test_benchmark.py
@@ -104,6 +104,16 @@ def test_uiex(self, request):
         filename = f"{MODELZOO_PATH}/{name}/{name}.onnx"
         self.run_model_test(name, filename)
 
+    def test_en_number_mobile_v2_0_rec_infer(self, request):
+        name = request.node.originalname[len("test_") :]
+        filename = f"{MODELZOO_PATH}/{name}/{name}.onnx"
+        self.run_model_test(name, filename)
+
+    def test_paddleocr(self, request):
+        name = request.node.originalname[len("test_") :]
+        filename = f"{MODELZOO_PATH}/{name}/{name}.onnx"
+        self.run_model_test(name, filename)
+
 
 if __name__ == "__main__":
     pytest.main(