From ac8db937608eb9599c0a44aea4d65b16cc189f0d Mon Sep 17 00:00:00 2001
From: jamescalam <james.briggs@hotmail.com>
Date: Fri, 29 Nov 2024 14:05:29 +0100
Subject: [PATCH] fix: tfidf tests
---
tests/unit/encoders/test_tfidf.py | 4 ++--
tests/unit/test_router.py | 1 +
2 files changed, 3 insertions(+), 2 deletions(-)
diff --git a/tests/unit/encoders/test_tfidf.py b/tests/unit/encoders/test_tfidf.py
index 7664433d..5052a035 100644
--- a/tests/unit/encoders/test_tfidf.py
+++ b/tests/unit/encoders/test_tfidf.py
@@ -37,7 +37,7 @@ class TestTfidfEncoder:
result = tfidf_encoder(["test"])
assert isinstance(result, list), "Result should be a list"
assert all(
- isinstance(sublist, list) for sublist in result
+ isinstance(sparse_emb.embedding, np.ndarray) for sparse_emb in result
), "Each item in result should be a list"
def test_call_method_no_docs_tfidf_encoder(self, tfidf_encoder):
@@ -55,7 +55,7 @@ class TestTfidfEncoder:
result = tfidf_encoder(["doc with fake word gta5jabcxyz"])
assert isinstance(result, list), "Result should be a list"
assert all(
- isinstance(sublist, list) for sublist in result
+ isinstance(sparse_emb.embedding, np.ndarray) for sparse_emb in result
), "Each item in result should be a list"
def test_call_method_with_uninitialized_model(self, tfidf_encoder):
diff --git a/tests/unit/test_router.py b/tests/unit/test_router.py
index ef36e0ab..a20c68e1 100644
--- a/tests/unit/test_router.py
+++ b/tests/unit/test_router.py
@@ -19,6 +19,7 @@ from platform import python_version
PINECONE_SLEEP = 6
+
def mock_encoder_call(utterances):
# Define a mapping of utterances to return values
mock_responses = {
--
GitLab