From d4d62c6efb5f22c4a935a0a5feb6ec13b7c82f2b Mon Sep 17 00:00:00 2001 From: HombreLaser Date: Wed, 13 Mar 2024 19:53:36 -0600 Subject: Add missing tests --- tests/test_row_at.py | 11 ++++++++++- 1 file changed, 10 insertions(+), 1 deletion(-) (limited to 'tests/test_row_at.py') 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 -- cgit v1.2.3