File tree Expand file tree Collapse file tree 1 file changed +10
-3
lines changed Expand file tree Collapse file tree 1 file changed +10
-3
lines changed Original file line number Diff line number Diff line change @@ -62,7 +62,14 @@ def test_e2e_deepseekv3_with_torchair(monkeypatch: pytest.MonkeyPatch):
62
62
) as vllm_model :
63
63
vllm_output = vllm_model .generate_greedy (example_prompts ,
64
64
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
+ ]
65
71
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} " )
You can’t perform that action at this time.
0 commit comments