fix: Prevent Mock object issues in graph memory tests (#3627)

Co-authored-by: parshvadaftari <daftariparshva@gmail.com>
This commit is contained in:
Ronak Bhalgami
2025-10-22 03:53:12 +05:30
committed by GitHub
co-authored by parshvadaftari
parent 4a30745592
commit eb2f8a3738
3 changed files with 25 additions and 8 deletions
+19 -4
View File
@@ -7,11 +7,25 @@ from mem0.memory.kuzu_memory import MemoryGraph
class TestKuzu:
"""Test that Kuzu memory works correctly"""
# Create distinct embeddings that won't match with threshold=0.7
# Each embedding is mostly zeros with ones in different positions to ensure low similarity
alice_emb = np.zeros(384)
alice_emb[0:96] = 1.0
bob_emb = np.zeros(384)
bob_emb[96:192] = 1.0
charlie_emb = np.zeros(384)
charlie_emb[192:288] = 1.0
dave_emb = np.zeros(384)
dave_emb[288:384] = 1.0
embeddings = {
"alice": np.random.uniform(0.0, 0.9, 384).tolist(),
"bob": np.random.uniform(0.0, 0.9, 384).tolist(),
"charlie": np.random.uniform(0.0, 0.9, 384).tolist(),
"dave": np.random.uniform(0.0, 0.9, 384).tolist(),
"alice": alice_emb.tolist(),
"bob": bob_emb.tolist(),
"charlie": charlie_emb.tolist(),
"dave": dave_emb.tolist(),
}
@pytest.fixture
@@ -26,6 +40,7 @@ class TestKuzu:
# Mock graph store config
config.graph_store.config.db = ":memory:"
config.graph_store.threshold = 0.7
# Mock LLM config
config.llm.provider = "mock_llm"