Skip to content

Commit 22642f1

Browse files
committed
fix: refactor test_server_terminate to improve readability of patch context
tests/unit/test_lit_server.py:320:10: SyntaxError: Cannot use parentheses within a `with` statement on Python 3.8 (syntax was added in Python 3.9)
1 parent 260acd8 commit 22642f1

1 file changed

Lines changed: 3 additions & 5 deletions

File tree

tests/unit/test_lit_server.py

Lines changed: 3 additions & 5 deletions
Original file line numberDiff line numberDiff line change
@@ -317,11 +317,9 @@ def test_server_terminate():
317317
server.verify_worker_status = MagicMock()
318318
server._transport = MagicMock()
319319

320-
with (
321-
patch("litserve.server.LitServer._init_manager", return_value=MagicMock()) as mock_init_manager,
322-
patch("litserve.server.LitServer._start_server", side_effect=Exception("mocked error")) as mock_start,
323-
patch("litserve.server.LitServer.launch_inference_worker", return_value=([MagicMock()])) as mock_launch,
324-
):
320+
with patch("litserve.server.LitServer._init_manager", return_value=MagicMock()) as mock_init_manager, \
321+
patch("litserve.server.LitServer._start_server", side_effect=Exception("mocked error")) as mock_start, \
322+
patch("litserve.server.LitServer.launch_inference_worker", return_value=([MagicMock()])) as mock_launch:
325323
with pytest.raises(Exception, match="mocked error"):
326324
server.run(port=8001)
327325

0 commit comments

Comments
 (0)