From 73614fb52dd29dcd13d1c8e88a38cb3015e49827 Mon Sep 17 00:00:00 2001 From: Timmy Date: Thu, 19 Mar 2026 14:20:26 -0400 Subject: [PATCH] fix: remove numpy dependency from test_memory_embeddings Use MagicMock to simulate .tolist() instead of importing numpy, which is not in the test dependencies. Fixes broken unit test on main. --- tests/unit/test_memory_embeddings.py | 6 +++--- 1 file changed, 3 insertions(+), 3 deletions(-) diff --git a/tests/unit/test_memory_embeddings.py b/tests/unit/test_memory_embeddings.py index ac7ea1ae..6d3e6bb5 100644 --- a/tests/unit/test_memory_embeddings.py +++ b/tests/unit/test_memory_embeddings.py @@ -112,10 +112,10 @@ class TestEmbedText: assert len(vec) == 128 # hash fallback dimension def test_uses_model_when_available(self): - import numpy as np - + mock_encoding = MagicMock() + mock_encoding.tolist.return_value = [0.1, 0.2, 0.3] mock_model = MagicMock() - mock_model.encode.return_value = np.array([0.1, 0.2, 0.3]) + mock_model.encode.return_value = mock_encoding emb.EMBEDDING_MODEL = mock_model result = embed_text("test")