From 5f6fa04a3f3f8c2451835d6e99ddaf5b9d48c8f8 Mon Sep 17 00:00:00 2001 From: yichuan~ <73766326+yichuan520030910320@users.noreply.github.com> Date: Wed, 7 Aug 2024 16:23:27 +0800 Subject: [PATCH] misc: simplify test (#964) --- test/srt/test_openai_server.py | 22 ++++++---------------- 1 file changed, 6 insertions(+), 16 deletions(-) diff --git a/test/srt/test_openai_server.py b/test/srt/test_openai_server.py index db1a3c027..b0df6738f 100644 --- a/test/srt/test_openai_server.py +++ b/test/srt/test_openai_server.py @@ -313,22 +313,12 @@ class TestOpenAIServer(unittest.TestCase): result_file_id = batch_job.output_file_id file_response = client.files.content(result_file_id) - result_content = file_response.read() - - if mode == "completion": - result_file_name = "batch_job_complete_results.jsonl" - else: - result_file_name = "batch_job_chat_results.jsonl" - with open(result_file_name, "wb") as file: - file.write(result_content) - results = [] - with open(result_file_name, "r", encoding="utf-8") as file: - for line in file: - json_object = json.loads(line.strip()) - results.append(json_object) - for delete_fid in [uploaded_file.id, result_file_id]: - del_pesponse = client.files.delete(delete_fid) - assert del_pesponse.deleted + result_content = file_response.read().decode("utf-8") # Decode bytes to string + results = [ + json.loads(line) + for line in result_content.split("\n") + if line.strip() != "" + ] assert len(results) == len(content) def test_completion(self):