From 8d201d5163270b703a0fc7481c431d8c0bb12662 Mon Sep 17 00:00:00 2001
From: James Briggs <35938317+jamescalam@users.noreply.github.com>
Date: Sat, 2 Dec 2023 14:37:00 -0800
Subject: [PATCH] add coverage for bm25 fit

---
 tests/unit/encoders/test_bm25.py | 9 ++++++++-
 1 file changed, 8 insertions(+), 1 deletion(-)

diff --git a/tests/unit/encoders/test_bm25.py b/tests/unit/encoders/test_bm25.py
index 3a94602e..9822d29b 100644
--- a/tests/unit/encoders/test_bm25.py
+++ b/tests/unit/encoders/test_bm25.py
@@ -10,7 +10,14 @@ def bm25_encoder():
 
 class TestBM25Encoder:
     def test_initialization(self, bm25_encoder):
-        bm25_encoder = BM25Encoder()
+        assert len(bm25_encoder.idx_mapping) != 0
+
+    def test_fit(self, bm25_encoder):
+        bm25_encoder.fit([
+            "some docs",
+            "and more docs",
+            "and even more docs"
+        ])
         assert len(bm25_encoder.idx_mapping) != 0
 
     def test_call_method(self, bm25_encoder):