diff options
Diffstat (limited to 'tests/test_delete_at.py')
-rw-r--r-- | tests/test_delete_at.py | 11 |
1 files changed, 10 insertions, 1 deletions
diff --git a/tests/test_delete_at.py b/tests/test_delete_at.py index 02471d1..662afd9 100644 --- a/tests/test_delete_at.py +++ b/tests/test_delete_at.py @@ -1,4 +1,13 @@ +from src.program import Program + + def test_delete_at(capsys, delete_at_program): delete_at_program.run() - assert capsys.readouterr()[0] == 'Row at 1 deleted\n'
\ No newline at end of file + assert capsys.readouterr()[0] == 'Row at 1 deleted\n' + + +def test_delete_at_out_of_bounds(capsys): + Program(args=['-f', './tests/test_table.csv', '-c', 'delete-at', '5']).run() + + assert capsys.readouterr()[0] == "Couldn't find row at 5\n"
\ No newline at end of file |