o پi@s2ddlmZmZddlmZmZGdddZdS)) BenchArgssend_one_prompt)is_in_ciwrite_github_step_summaryc@s&eZdZUeed<eed<ddZdS)SpecDecodingMixinbs_1_speed_thresaccept_length_threscCstt|jdddd}t|\}}td|dd|dtr4td|jd |dd |dd | ||j | ||j dS) N:i)portmax_new_tokensz acc_length=z.2fz speed=z### test_bs_1_speed (z ) acc_length=z speed=z token/s ) rintbase_urlsplitrprintrrmodel assertGreaterrr)selfargs acc_lengthspeedrV/home/ubuntu/.local/lib/python3.10/site-packages/sglang/test/kits/spec_decoding_kit.pytest_bs_1_speed s  z!SpecDecodingMixin.test_bs_1_speedN)__name__ __module__ __qualname__float__annotations__rrrrrrs  rN)sglang.test.send_onerrsglang.test.test_utilsrrrrrrrs