mirror of
https://github.com/PaddlePaddle/FastDeploy.git
synced 2026-04-23 00:17:25 +08:00
tests: fix cache_transfer_manager threading and init mocks (#6502)
tests: fix cache_transfer_manager threading and init mocks
This commit is contained in:
@@ -110,6 +110,13 @@ class TestCacheTransferManager(unittest.TestCase):
|
||||
patcher2.start()
|
||||
self.addCleanup(patcher2.stop)
|
||||
|
||||
# --------------------------
|
||||
# mock threading.Thread
|
||||
# --------------------------
|
||||
patcher_thread = patch("fastdeploy.cache_manager.cache_transfer_manager.threading.Thread")
|
||||
patcher_thread.start()
|
||||
self.addCleanup(patcher_thread.stop)
|
||||
|
||||
# --------------------------
|
||||
# mock _init_cpu_cache 和 _init_gpu_cache
|
||||
# --------------------------
|
||||
|
||||
Reference in New Issue
Block a user