EngineX-Hygon/sglang
5
0
Fork 0
Code Issues Pull Requests Actions 7 Projects Releases Wiki Activity

misc: simplify test (#964)

Browse Source
This commit is contained in:
yichuan~
2024-08-07 16:23:27 +08:00
committed by GitHub
parent 58a0970853
commit 5f6fa04a3f
1 changed files with 6 additions and 16 deletions
Download Patch File Download Diff File Expand all files Collapse all files

22
test/srt/test_openai_server.py
View File

@@ -313,22 +313,12 @@ class TestOpenAIServer(unittest.TestCase):
result_file_id = batch_job.output_file_id result_file_id = batch_job.output_file_id
file_response = client.files.content(result_file_id) file_response = client.files.content(result_file_id)
result_content = file_response.read() result_content = file_response.read().decode("utf-8") # Decode bytes to string
results = [
if mode == "completion": json.loads(line)
result_file_name = "batch_job_complete_results.jsonl" for line in result_content.split("\n")
else: if line.strip() != ""
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
assert len(results) == len(content) assert len(results) == len(content)
def test_completion(self): def test_completion(self):