Skip to content

Commit 7a0d45a

Browse files
committed
Add invalid tests
1 parent 27f1167 commit 7a0d45a

1 file changed

Lines changed: 11 additions & 0 deletions

File tree

tests/integration_tests/test_hana_reranker.py

Lines changed: 11 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -59,6 +59,17 @@ def test_rerank(reranker, documents, query, top_n, return_documents, rank_fields
5959
prev_score = item[1]
6060

6161

62+
@pytest.mark.parametrize("invalid_top_n", [-1, 0, len(HanaTestConstants.TEXTS) + 1])
63+
def test_rerank_with_invalid_top_n(reranker, documents, invalid_top_n):
64+
with pytest.raises(ValueError, match="top_n must be greater than 0 and less than or equal to the number of documents"):
65+
reranker.rerank(documents, HanaTestConstants.TEXTS[0], invalid_top_n)
66+
67+
68+
def test_rerank_with_invalid_metadata_key(reranker, documents):
69+
with pytest.raises(ValueError, match="Invalid metadata key invalid-key"):
70+
reranker.rerank(documents, HanaTestConstants.TEXTS[0], rank_fields=["invalid-key"])
71+
72+
6273
def test_compress_documents(reranker, documents):
6374
documents.append(Document(page_content="abc", metadata={"start": 400, "quality": "ugly", "Owner": "Bob"}))
6475
compressed_docs = reranker.compress_documents(query=HanaTestConstants.TEXTS[0], documents=documents)

0 commit comments

Comments
 (0)