Add missing tests

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