summaryrefslogtreecommitdiff
path: root/tests/test_row_at.py
diff options
context:
space:
mode:
Diffstat (limited to 'tests/test_row_at.py')
-rw-r--r--tests/test_row_at.py11
1 files changed, 10 insertions, 1 deletions
diff --git a/tests/test_row_at.py b/tests/test_row_at.py
index 6b7c379..82e3270 100644
--- a/tests/test_row_at.py
+++ b/tests/test_row_at.py
@@ -1,6 +1,15 @@
+from src.program import Program
+
+
def test_row_at(capsys, row_at_program):
row_at_program.run()
assert capsys.readouterr()[0].startswith(
"('1919', 'Ten Days that Shook the World', 'John Reed')"
- ) \ No newline at end of file
+ )
+
+
+def test_out_of_bounds_row_at(capsys):
+ Program(args=['-f', './tests/test_table.csv', '-c', 'row-at', '5']).run()
+
+ assert capsys.readouterr()[0] == 'None\n' \ No newline at end of file