tests/test_delete_at.py


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


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"