diff --git a/tests/unit/encoders/test_bm25.py b/tests/unit/encoders/test_bm25.py index e0f6dad5b9f507ed6859c41cb6286e7eec056d39..c198715133829ff4869167b80117c40dba72d4fb 100644 --- a/tests/unit/encoders/test_bm25.py +++ b/tests/unit/encoders/test_bm25.py @@ -22,3 +22,14 @@ class TestBM25Encoder: assert all( isinstance(sublist, list) for sublist in result ), "Each item in result should be a list" + + def test_call_method_no_docs(self, bm25_encoder): + with pytest.raises(ValueError): + bm25_encoder([]) + + def test_call_method_no_word(self, bm25_encoder): + result = bm25_encoder(["doc with fake word gta5jabcxyz"]) + assert isinstance(result, list), "Result should be a list" + assert all( + isinstance(sublist, list) for sublist in result + ), "Each item in result should be a list"