fix: resolve import path issues in moved test files

- Fix remaining src.contextforge_memory imports in test files
- Add contextforge_memory import to fix module reload issues
- Update all import paths to use contextforge_memory instead of src.contextforge_memory
- Resolves CI failures with KeyError: 'contextforge_memory' and ModuleNotFoundError

Author: AlmostBald-TRADING

tests/test_simple_imports.py

@@ -14,10 +14,10 @@ class TestSimpleImports:
 
     def test_main_module_import(self):
         """Test that the main module can be imported."""
-        import src.contextforge_memory
+        import contextforge_memory
 
-        assert hasattr(src.contextforge_memory, "__version__")
-        assert hasattr(src.contextforge_memory, "InMemoryCosineIndex")
+        assert hasattr(contextforge_memory, "__version__")
+        assert hasattr(contextforge_memory, "InMemoryCosineIndex")
 
     def test_embeddings_module_imports(self):
         """Test that embeddings modules can be imported."""
@@ -53,7 +53,7 @@ def test_vector_index_module_imports(self):
 
     def test_import_with_reload(self):
         """Test that modules can be reloaded without issues."""
-        import src.contextforge_memory.embeddings.base as base_module
+        import contextforge_memory.embeddings.base as base_module
 
         # Reload the module
         importlib.reload(base_module)

tests/test_strict_provider_detailed.py

@@ -14,6 +14,7 @@
 
 import pytest
 
+import contextforge_memory  # noqa: F401  # Required for module reloading
 from contextforge_memory.embeddings.openai import OpenAIEmbeddingsProvider
 from contextforge_memory.embeddings.sentencetransformers import (
     SentenceTransformersProvider,

tests/test_threadpool_config.py

@@ -18,14 +18,14 @@ class TestThreadPoolConfiguration:
 
     def test_default_pool_size(self):
         """Test that default pool size is 4 when no env var is set."""
-        from src.contextforge_memory.main import ThreadPoolConfig
+        from contextforge_memory.main import ThreadPoolConfig
 
         config = ThreadPoolConfig()
         assert config.pool_size == 4
 
     def test_explicit_pool_size(self):
         """Test that explicit pool sizes are accepted."""
-        from src.contextforge_memory.main import ThreadPoolConfig
+        from contextforge_memory.main import ThreadPoolConfig
 
         test_cases = [1, 4, 8, 16, 32]
 
@@ -35,7 +35,7 @@ def test_explicit_pool_size(self):
 
     def test_from_env_with_valid_values(self):
         """Test that from_env() works with valid environment values."""
-        from src.contextforge_memory.main import ThreadPoolConfig
+        from contextforge_memory.main import ThreadPoolConfig
 
         test_cases = [1, 4, 8, 16, 32]
 
@@ -46,7 +46,7 @@ def test_from_env_with_valid_values(self):
 
     def test_from_env_with_invalid_values(self):
         """Test that from_env() handles invalid environment values gracefully."""
-        from src.contextforge_memory.main import ThreadPoolConfig
+        from contextforge_memory.main import ThreadPoolConfig
 
         invalid_values = ["invalid", "-1", "0", "not_a_number"]
 
@@ -58,15 +58,15 @@ def test_from_env_with_invalid_values(self):
 
     def test_from_env_with_missing_var(self):
         """Test that from_env() uses default when env var is missing."""
-        from src.contextforge_memory.main import ThreadPoolConfig
+        from contextforge_memory.main import ThreadPoolConfig
 
         with patch.dict(os.environ, {}, clear=True):
             config = ThreadPoolConfig.from_env()
             assert config.pool_size == 4
 
     def test_pool_creation(self):
         """Test that create_pool() returns a working ThreadPoolExecutor."""
-        from src.contextforge_memory.main import ThreadPoolConfig
+        from contextforge_memory.main import ThreadPoolConfig
 
         config = ThreadPoolConfig(pool_size=2)
         pool = config.create_pool()
@@ -86,7 +86,7 @@ def simple_task(x):
 
     def test_pool_with_concurrent_tasks(self):
         """Test that the pool can handle concurrent tasks efficiently."""
-        from src.contextforge_memory.main import ThreadPoolConfig
+        from contextforge_memory.main import ThreadPoolConfig
 
         config = ThreadPoolConfig(pool_size=3)
         pool = config.create_pool()
@@ -150,7 +150,7 @@ def test_pool_with_real_app_integration(self):
 
     def test_environment_parsing_edge_cases(self):
         """Test edge cases in environment variable parsing."""
-        from src.contextforge_memory.main import ThreadPoolConfig
+        from contextforge_memory.main import ThreadPoolConfig
 
         # Test with very large numbers
         with patch.dict(os.environ, {"CONTEXTFORGE_IO_POOL_SIZE": "1000"}):