Skip to content

Commit 334ea5d

Browse files
committed
add gloden result
Signed-off-by: MengqingCao <cmq0113@163.com>
1 parent 682963b commit 334ea5d

File tree

1 file changed

+10
-3
lines changed

1 file changed

+10
-3
lines changed

tests/multicard/test_torchair_graph_mode.py

Lines changed: 10 additions & 3 deletions
Original file line numberDiff line numberDiff line change
@@ -62,7 +62,14 @@ def test_e2e_deepseekv3_with_torchair(monkeypatch: pytest.MonkeyPatch):
6262
) as vllm_model:
6363
vllm_output = vllm_model.generate_greedy(example_prompts,
6464
max_tokens)
65+
golden_results = [
66+
'Hello, my name is feasibility伸 spazio debtor添',
67+
'The president of the United States is begg"""\n杭州风和 bestimm',
68+
'The capital of France is frequentlyশามalinkAllowed',
69+
'The future of AI is deleting俯احت怎么样了حراف',
70+
]
6571

66-
for output in vllm_output:
67-
generated_text = output[1]
68-
print(f"Generated text: {generated_text!r}")
72+
assert len(golden_results) == len(vllm_output)
73+
for i in range(len(vllm_output)):
74+
assert golden_results[i] == vllm_output[i][1]
75+
print(f"Generated text: {vllm_output[i][1]!r}")

0 commit comments

Comments
 (0)