Add --line-numbers flag (#38)

* add --line-numbers flag
* -n/--line-numbers in README, refs #38

Co-authored-by: Simon Willison <swillison@gmail.com>
This commit is contained in:
Dan Clayton
2025-02-14 03:10:13 +00:00
committed by GitHub
parent 8c89fe24a8
commit bc05005e3b
3 changed files with 76 additions and 7 deletions

View File

@@ -279,3 +279,31 @@ Contents of file2.txt
---
"""
assert expected.strip() == actual.strip()
def test_line_numbers(tmpdir):
runner = CliRunner()
with tmpdir.as_cwd():
os.makedirs("test_dir")
test_content = "First line\nSecond line\nThird line\nFourth line\n"
with open("test_dir/multiline.txt", "w") as f:
f.write(test_content)
result = runner.invoke(cli, ["test_dir"])
assert result.exit_code == 0
assert "1 First line" not in result.output
assert test_content in result.output
result = runner.invoke(cli, ["test_dir", "-n"])
assert result.exit_code == 0
assert "1 First line" in result.output
assert "2 Second line" in result.output
assert "3 Third line" in result.output
assert "4 Fourth line" in result.output
result = runner.invoke(cli, ["test_dir", "--line-numbers"])
assert result.exit_code == 0
assert "1 First line" in result.output
assert "2 Second line" in result.output
assert "3 Third line" in result.output
assert "4 Fourth line" in result.output