codeflash/tests/test_help_banner.py
claude[bot] 8e2bab2e42 fix: add missing return type annotations to test functions
Co-authored-by: Aseem Saxena <aseembits93@users.noreply.github.com>
2026-04-06 23:12:44 +00:00

18 lines
574 B
Python

import subprocess
import sys
def test_help_displays_logo() -> None:
result = subprocess.run(
[sys.executable, "-c", "from codeflash.main import main; main()", "--help"], capture_output=True, text=True
)
assert result.returncode == 0
assert "codeflash.ai" in result.stdout
def test_help_short_flag_displays_logo() -> None:
result = subprocess.run(
[sys.executable, "-c", "from codeflash.main import main; main()", "-h"], capture_output=True, text=True
)
assert result.returncode == 0
assert "codeflash.ai" in result.stdout