import pytest from cli import cli def test_cli_main(capsys): output = "Hello\n" cli.main() captured = capsys.readouterr() assert captured.out == output