server : add more test cases (#10569)
* server : add split model test * add test speculative * add invalid cases
This commit is contained in:
parent
3a8e9af402
commit
b782e5c7d4
6 changed files with 186 additions and 1 deletions
|
@ -36,3 +36,20 @@ def test_rerank():
|
|||
assert most_relevant["relevance_score"] > least_relevant["relevance_score"]
|
||||
assert most_relevant["index"] == 2
|
||||
assert least_relevant["index"] == 3
|
||||
|
||||
|
||||
@pytest.mark.parametrize("documents", [
|
||||
[],
|
||||
None,
|
||||
123,
|
||||
[1, 2, 3],
|
||||
])
|
||||
def test_invalid_rerank_req(documents):
|
||||
global server
|
||||
server.start()
|
||||
res = server.make_request("POST", "/rerank", data={
|
||||
"query": "Machine learning is",
|
||||
"documents": documents,
|
||||
})
|
||||
assert res.status_code == 400
|
||||
assert "error" in res.body
|
||||
|
|
Loading…
Add table
Add a link
Reference in a new issue