RomanNum3ral
/
soup
mirror of https://github.com/razor-ai/soup.git
1
0
Fork 0
Code Issues Packages Projects Releases Wiki Activity

Fix CI: strip ANSI codes in serve help assertions for Rich output 

Co-Authored-By: Claude Opus 4.6 (1M context) <noreply@anthropic.com>
This commit is contained in:
Alpamys 2026-03-23 21:39:04 +05:00
parent f1d039f223
commit 930489a82e
1 changed files with 13 additions and 3 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

16
tests/test_vllm_serve.py
View File

@ -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
# ============================================================