mirror of
https://github.com/simonw/files-to-prompt.git
synced 2025-10-23 00:02:47 +03:00
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:
@@ -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
|
||||
|
||||
Reference in New Issue
Block a user