JDSS/mem0
2
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

fix(tests): update 35 failing tests for v3 single-pass pipeline

Browse Source 
Tests were written for the old two-pass extraction pipeline (extract
facts + merge with ADD/UPDATE/DELETE). V3 uses single-pass ADD-only
extraction with one LLM call returning {"memory": [...]}.

Changes by category:

Old pipeline tests (13 failures):
- Delete TestHallucinatedIdGuard entirely (tests UPDATE/DELETE ID
  resolution that no longer exists)
- Update error log assertions to match v3 log messages
- Change LLM mock from 2-call side_effect to 1-call return_value
- Update vllm thinking tag tests for single-call format

ensure_json_instruction removed (3):
- Delete source-inspection tests (function still exists in utils
  but is no longer called from _add_to_vector_store)

UTC timestamp normalization removed (2):
- Update expected timestamps to stored-as-is (no UTC conversion)
- Remove _normalize_iso_timestamp_to_utc import

_search_vector_store signature: top_k -> limit (2):
- Update test calls to use limit= parameter

Context manager + close/db removed (6):
- Delete tests for __enter__/__exit__/close/db that were removed

Vector store BM25 config (3):
- Qdrant: add sparse_vectors_config to create_col assertion,
  named vector format {"": vector} in update assertion
- MongoDB: assert_any_call for both vector + text search indexes

Misc (6):
- test_search_handles_incomplete_payloads: expect 1 result (v3
  filters entries without 'data' key)
- Embedding cache tests: update for embed_batch (1 call) vs
  individual embed (was 2 calls)
- Graph reset tests: delete (graph support removed)
- use_azure_credential: update expected sensitivity to True

Co-Authored-By: Claude Opus 4.6 (1M context) <noreply@anthropic.com>
This commit is contained in:
chaithanyak42
2026-04-14 02:19:39 +05:30
co-authored by Claude Opus 4.6
parent 2a5ffd0976
commit 0b4bbcefff
8 changed files with 83 additions and 436 deletions
Download Patch File Download Diff File Expand all files Collapse all files
tests/memory/test_json_prompt_fix.py
-28
View File
@@ -184,31 +184,3 @@ class TestEnsureJsonInstruction:
# Integration: verify fix is wired into both sync and async paths
# -------------------------------------------------------------------
def test_fix_applied_in_sync_memory_class(self):
"""Verify the ensure_json_instruction call exists in Memory._add_to_vector_store."""
import inspect
from mem0.memory.main import Memory
source = inspect.getsource(Memory._add_to_vector_store)
assert "ensure_json_instruction" in source, (
"ensure_json_instruction not found in Memory._add_to_vector_store (sync)"
)
def test_fix_applied_in_async_memory_class(self):
"""Verify the ensure_json_instruction call exists in AsyncMemory._add_to_vector_store."""
import inspect
from mem0.memory.main import AsyncMemory
source = inspect.getsource(AsyncMemory._add_to_vector_store)
assert "ensure_json_instruction" in source, (
"ensure_json_instruction not found in AsyncMemory._add_to_vector_store (async)"
)
def test_import_exists_in_main(self):
"""Verify ensure_json_instruction is imported in main.py."""
import inspect
import mem0.memory.main as main_module
source = inspect.getsource(main_module)
assert "from mem0.memory.utils import" in source
assert "ensure_json_instruction" in source