test(rapidocr_onnxruntime): Add two unit testings about cuda and dml

python/tests/test_ort.py

@@ -1,6 +1,7 @@
 # -*- encoding: utf-8 -*-
 # @Author: SWHL
 # @Contact: [email protected]
+import logging
 import sys
 from pathlib import Path
 
@@ -20,6 +21,22 @@
 package_name = "rapidocr_onnxruntime"
 
 
+def test_ort_cuda_warning(caplog):
+    engine = RapidOCR(det_use_cuda=True)
+    caplog.set_level(logging.WARNING)
+
+    assert caplog.records[0].levelname == "WARNING"
+    assert "CUDAExecutionProvider" in caplog.records[0].message
+
+
+def test_ort_dml_warning(caplog):
+    engine = RapidOCR(det_use_dml=True)
+    caplog.set_level(logging.WARNING)
+
+    assert caplog.records[0].levelname == "WARNING"
+    assert "DirectML" in caplog.records[0].message
+
+
 def test_mode_one_img():
     img_path = tests_dir / "issue_170.png"
     result, _ = engine(img_path)