From 55cd79da566b0d05c6c27b01cce02c7c047f3f91 Mon Sep 17 00:00:00 2001
From: James Briggs <35938317+jamescalam@users.noreply.github.com>
Date: Sat, 2 Dec 2023 14:22:00 -0800
Subject: [PATCH] fix bm25 test
---
tests/unit/encoders/test_bm25.py | 4 ++--
1 file changed, 2 insertions(+), 2 deletions(-)
diff --git a/tests/unit/encoders/test_bm25.py b/tests/unit/encoders/test_bm25.py
index 086a80bd..3a94602e 100644
--- a/tests/unit/encoders/test_bm25.py
+++ b/tests/unit/encoders/test_bm25.py
@@ -9,11 +9,11 @@ def bm25_encoder():
class TestBM25Encoder:
- def test_initialization(self):
+ def test_initialization(self, bm25_encoder):
bm25_encoder = BM25Encoder()
assert len(bm25_encoder.idx_mapping) != 0
- def test_call_method(self):
+ def test_call_method(self, bm25_encoder):
result = bm25_encoder(["test"])
assert isinstance(result, list), "Result should be a list"
assert all(
--
GitLab