summaryrefslogtreecommitdiff
path: root/tests/test_delete_at.py
diff options
context:
space:
mode:
Diffstat (limited to 'tests/test_delete_at.py')
-rw-r--r--tests/test_delete_at.py11
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