fix: test_api bug

Author: CaralHsi

tests/api/test_product_router.py

@@ -5,50 +5,22 @@
 input request formats and return properly formatted responses.
 """
 
-# Mock sklearn before importing any memos modules to avoid import errors
-import importlib.util
-import sys
-
-from unittest.mock import MagicMock, Mock, patch
+from unittest.mock import Mock, patch
 
 import pytest
 
 from fastapi.testclient import TestClient
 
 # Patch the MOS_PRODUCT_INSTANCE directly after import
+# Patch MOS_PRODUCT_INSTANCE and MOSProduct so we can test the FastAPI router
+# without initializing the full MemOS product stack.
 import memos.api.routers.product_router as pr_module
 
 
-# Create a proper mock module with __spec__
-sklearn_mock = MagicMock()
-sklearn_mock.__spec__ = importlib.util.spec_from_loader("sklearn", None)
-sys.modules["sklearn"] = sklearn_mock
-
-sklearn_fe_mock = MagicMock()
-sklearn_fe_mock.__spec__ = importlib.util.spec_from_loader("sklearn.feature_extraction", None)
-sys.modules["sklearn.feature_extraction"] = sklearn_fe_mock
-
-sklearn_fet_mock = MagicMock()
-sklearn_fet_mock.__spec__ = importlib.util.spec_from_loader("sklearn.feature_extraction.text", None)
-sklearn_fet_mock.TfidfVectorizer = MagicMock()
-sys.modules["sklearn.feature_extraction.text"] = sklearn_fet_mock
-
-# Mock sklearn.metrics as well
-sklearn_metrics_mock = MagicMock()
-sklearn_metrics_mock.__spec__ = importlib.util.spec_from_loader("sklearn.metrics", None)
-sklearn_metrics_mock.roc_curve = MagicMock()
-sys.modules["sklearn.metrics"] = sklearn_metrics_mock
-
-
-# Create mock instance
 _mock_mos_instance = Mock()
-
 pr_module.MOS_PRODUCT_INSTANCE = _mock_mos_instance
 pr_module.get_mos_product_instance = lambda: _mock_mos_instance
-
-# Mock MOSProduct class before importing to prevent initialization
 with patch("memos.mem_os.product.MOSProduct", return_value=_mock_mos_instance):
-    # Import after patching
     from memos.api import product_api
 
 

tests/api/test_server_router.py

@@ -5,11 +5,7 @@
 input request formats and return properly formatted responses.
 """
 
-# Mock sklearn before importing any memos modules to avoid import errors
-import importlib.util
-import sys
-
-from unittest.mock import MagicMock, Mock, patch
+from unittest.mock import Mock, patch
 
 import pytest
 
@@ -25,25 +21,8 @@
 )
 
 
-# Create a proper mock module with __spec__
-sklearn_mock = MagicMock()
-sklearn_mock.__spec__ = importlib.util.spec_from_loader("sklearn", None)
-sys.modules["sklearn"] = sklearn_mock
-
-sklearn_fe_mock = MagicMock()
-sklearn_fe_mock.__spec__ = importlib.util.spec_from_loader("sklearn.feature_extraction", None)
-sys.modules["sklearn.feature_extraction"] = sklearn_fe_mock
-
-sklearn_metrics_mock = MagicMock()
-sklearn_metrics_mock.__spec__ = importlib.util.spec_from_loader("sklearn.metrics", None)
-sys.modules["sklearn.metrics"] = sklearn_metrics_mock
-
-sklearn_fet_mock = MagicMock()
-sklearn_fet_mock.__spec__ = importlib.util.spec_from_loader("sklearn.feature_extraction.text", None)
-sklearn_fet_mock.TfidfVectorizer = MagicMock()
-sys.modules["sklearn.feature_extraction.text"] = sklearn_fet_mock
-
-
+# Patch init_server so we can import server_api without starting the full MemOS stack,
+# and keep sklearn and other core dependencies untouched for other tests.
 @pytest.fixture(scope="module")
 def mock_init_server():
     """Mock init_server before importing server_api."""