CI: fix app test

Author: jamm1985

pyproject.toml

@@ -13,7 +13,7 @@ packages = [
 ]
 
 [tool.poetry.scripts]
-template_bin = 'sample.main:run'
+template_bin = 'sample.main:main'
 
 [tool.poetry.dependencies]
 python = "~3.14"

src/sample/main.py

@@ -1,5 +1,7 @@
 """Main template."""
 
+import sys
+
 import numpy as np
 
 
@@ -27,3 +29,10 @@ def run(k: int = 10) -> list[float]:
         list samples of size k
     """
     return get_sample(k)
+
+
+def main() -> int:
+    """Run default sample generation for CLI usage."""
+    sample = run()
+    sys.stdout.write(f"{sample}\n")
+    return 0

tests/sample/test_main.py

@@ -2,7 +2,7 @@
 
 import pytest
 
-from sample.main import get_sample, run
+from sample.main import get_sample, main, run
 
 
 def test_get_sample_returns_requested_number_of_values() -> None:
@@ -29,3 +29,17 @@ def fake_get_sample(k: int) -> list[float]:
 
     monkeypatch.setattr("sample.main.get_sample", fake_get_sample)
     assert run(2) == expected
+
+
+def test_main_writes_sample_and_returns_zero(
+    monkeypatch: pytest.MonkeyPatch,
+    capsys: pytest.CaptureFixture[str],
+) -> None:
+    """Main writes the sample to stdout and exits with status code 0."""
+    expected = [0.1, 0.2]
+
+    monkeypatch.setattr("sample.main.run", lambda: expected)
+
+    assert main() == 0
+    captured = capsys.readouterr()
+    assert captured.out.strip() == str(expected)