Fix OCR thread test isolation
check / local-gate (push) Failing after 17m10s

This commit is contained in:
JWeBot Contributors
2026-10-16 11:19:55 +08:00
parent 99a251c19a
commit 572159f571
+10 -2
View File
@@ -526,7 +526,11 @@ class OcrFallbackTests(unittest.TestCase):
captured.update(kwargs) captured.update(kwargs)
fake_module = types.SimpleNamespace(RapidOCR=FakeRapidOCR) fake_module = types.SimpleNamespace(RapidOCR=FakeRapidOCR)
with mock.patch.dict(sys.modules, {"rapidocr": fake_module}), mock.patch.dict( fake_numpy = types.SimpleNamespace(array=lambda value: value)
with mock.patch.dict(
sys.modules,
{"rapidocr": fake_module, "numpy": fake_numpy},
), mock.patch.dict(
MODULE.os.environ, MODULE.os.environ,
{ {
"WECHAT_OCR_ONNX_INTRA_OP_THREADS": "", "WECHAT_OCR_ONNX_INTRA_OP_THREADS": "",
@@ -551,7 +555,11 @@ class OcrFallbackTests(unittest.TestCase):
captured.update(kwargs) captured.update(kwargs)
fake_module = types.SimpleNamespace(RapidOCR=FakeRapidOCR) fake_module = types.SimpleNamespace(RapidOCR=FakeRapidOCR)
with mock.patch.dict(sys.modules, {"rapidocr": fake_module}), mock.patch.dict( fake_numpy = types.SimpleNamespace(array=lambda value: value)
with mock.patch.dict(
sys.modules,
{"rapidocr": fake_module, "numpy": fake_numpy},
), mock.patch.dict(
MODULE.os.environ, MODULE.os.environ,
{ {
"WECHAT_OCR_ONNX_INTRA_OP_THREADS": "2", "WECHAT_OCR_ONNX_INTRA_OP_THREADS": "2",