Remove has_new_line unit test changes.

This commit is contained in:
MichelleTPY 2024-12-14 00:36:33 +00:00
parent 3c8a053459
commit 29e6298d2e

View file

@ -25,7 +25,6 @@ def test_completion(prompt: str, n_predict: int, re_content: str, n_prompt: int,
assert res.body["timings"]["prompt_n"] == n_prompt
assert res.body["timings"]["predicted_n"] == n_predicted
assert res.body["truncated"] == truncated
assert res.body["has_new_line"] == False
assert match_regex(re_content, res.body["content"])
@ -48,7 +47,6 @@ def test_completion_stream(prompt: str, n_predict: int, re_content: str, n_promp
assert data["timings"]["prompt_n"] == n_prompt
assert data["timings"]["predicted_n"] == n_predicted
assert data["truncated"] == truncated
assert data["has_new_line"] == False
assert data["stop_type"] == "limit"
assert "generation_settings" in data
assert server.n_predict is not None