This commit is contained in:
@@ -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",
|
||||||
|
|||||||
Reference in New Issue
Block a user