File tree Expand file tree Collapse file tree 2 files changed +18
-0
lines changed Expand file tree Collapse file tree 2 files changed +18
-0
lines changed Original file line number Diff line number Diff line change
1
+ # SPDX-License-Identifier: Apache-2.0
2
+ from vllm .v1 .request import RequestStatus
3
+
4
+
5
+ def test_request_status_fmt_str ():
6
+ """Test that the string representation of RequestStatus is correct."""
7
+ assert f"{ RequestStatus .WAITING } " == "WAITING"
8
+ assert f"{ RequestStatus .WAITING_FOR_FSM } " == "WAITING_FOR_FSM"
9
+ assert f"{ RequestStatus .WAITING_FOR_REMOTE_KVS } " == "WAITING_FOR_REMOTE_KVS"
10
+ assert f"{ RequestStatus .RUNNING } " == "RUNNING"
11
+ assert f"{ RequestStatus .PREEMPTED } " == "PREEMPTED"
12
+ assert f"{ RequestStatus .FINISHED_STOPPED } " == "FINISHED_STOPPED"
13
+ assert f"{ RequestStatus .FINISHED_LENGTH_CAPPED } " == "FINISHED_LENGTH_CAPPED"
14
+ assert f"{ RequestStatus .FINISHED_ABORTED } " == "FINISHED_ABORTED"
15
+ assert f"{ RequestStatus .FINISHED_IGNORED } " == "FINISHED_IGNORED"
Original file line number Diff line number Diff line change @@ -171,6 +171,9 @@ class RequestStatus(enum.IntEnum):
171
171
FINISHED_ABORTED = enum .auto ()
172
172
FINISHED_IGNORED = enum .auto ()
173
173
174
+ def __str__ (self ):
175
+ return self .name
176
+
174
177
@staticmethod
175
178
def is_finished (status : "RequestStatus" ) -> bool :
176
179
return status > RequestStatus .PREEMPTED
You can’t perform that action at this time.
0 commit comments