Disable mem cache during API integration test (#4001)

disable mem cache during api integration test
openvinotoolkit · Oct 8, 2024 · db2eaa0 · db2eaa0
1 parent d851151
commit db2eaa0
Showing 1 changed file with 17 additions and 0 deletions.
diff --git a/tests/integration/api/conftest.py b/tests/integration/api/conftest.py
@@ -0,0 +1,17 @@
+# Copyright (C) 2024 Intel Corporation
+# SPDX-License-Identifier: Apache-2.0
+#
+from __future__ import annotations
+
+import pytest
+from otx.core.data.mem_cache import MemCacheHandlerSingleton
+
+
+@pytest.fixture(autouse=True)
+def fxt_disable_mem_cache():
+    """Disable mem cache to reduce memory usage."""
+
+    original_mem_limit = MemCacheHandlerSingleton.CPU_MEM_LIMITS_GIB
+    MemCacheHandlerSingleton.CPU_MEM_LIMITS_GIB = 99999999
+    yield
+    MemCacheHandlerSingleton.CPU_MEM_LIMITS_GIB = original_mem_limit