tests/test_row_at.py


1
2
3
4
5
6
7
8
9
10
11
12
13
14
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')"
        )


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'