tests : improve "tokens" type check

Co-authored-by: Xuan Son Nguyen <thichthat@gmail.com>
This commit is contained in:
Georgi Gerganov 2024-12-18 10:02:01 +02:00
parent 8bcfc5551e
commit 5bf29af841
No known key found for this signature in database
GPG key ID: 449E073F9DC10735

View file

@ -29,7 +29,8 @@ def test_completion(prompt: str, n_predict: int, re_content: str, n_prompt: int,
assert type(res.body["has_new_line"]) == bool
assert match_regex(re_content, res.body["content"])
if return_tokens:
assert res.body["tokens"] != []
assert len(res.body["tokens"]) > 0
assert all(type(tok) == int for tok in res.body["tokens"])
else:
assert res.body["tokens"] == []
@ -61,7 +62,8 @@ def test_completion_stream(prompt: str, n_predict: int, re_content: str, n_promp
assert data["generation_settings"]["seed"] == server.seed
assert match_regex(re_content, content)
else:
assert data["tokens"] != []
assert len(res.body["tokens"]) > 0
assert all(type(tok) == int for tok in res.body["tokens"])
content += data["content"]