From 930489a82e56ddb1591dc4f91572258352053f05 Mon Sep 17 00:00:00 2001 From: Alpamys Date: Mon, 23 Mar 2026 21:39:04 +0500 Subject: [PATCH] Fix CI: strip ANSI codes in serve help assertions for Rich output Co-Authored-By: Claude Opus 4.6 (1M context) --- tests/test_vllm_serve.py | 16 +++++++++++++--- 1 file changed, 13 insertions(+), 3 deletions(-) diff --git a/tests/test_vllm_serve.py b/tests/test_vllm_serve.py index 0817928..2810da6 100644 --- a/tests/test_vllm_serve.py +++ b/tests/test_vllm_serve.py @@ -607,6 +607,13 @@ class TestTransformersBackendUnchanged: class TestServeCliRegistration: """Test serve command is properly registered.""" + @staticmethod + def _strip_ansi(text: str) -> str: + """Remove ANSI escape codes from text.""" + import re + + return re.sub(r"\x1b\[[0-9;]*m", "", text) + def test_serve_help_shows_backend(self): """soup serve --help should mention --backend flag.""" from typer.testing import CliRunner @@ -616,7 +623,8 @@ class TestServeCliRegistration: runner = CliRunner() result = runner.invoke(app, ["serve", "--help"]) assert result.exit_code == 0 - assert "--backend" in result.output + clean = self._strip_ansi(result.output) + assert "--backend" in clean def test_serve_help_shows_tensor_parallel(self): """soup serve --help should mention --tensor-parallel flag.""" @@ -627,7 +635,8 @@ class TestServeCliRegistration: runner = CliRunner() result = runner.invoke(app, ["serve", "--help"]) assert result.exit_code == 0 - assert "--tensor-parallel" in result.output + clean = self._strip_ansi(result.output) + assert "--tensor-parallel" in clean def test_serve_help_shows_gpu_memory(self): """soup serve --help should mention --gpu-memory flag.""" @@ -638,7 +647,8 @@ class TestServeCliRegistration: runner = CliRunner() result = runner.invoke(app, ["serve", "--help"]) assert result.exit_code == 0 - assert "--gpu-memory" in result.output + clean = self._strip_ansi(result.output) + assert "--gpu-memory" in clean # ============================================================