def test_get_column_multi(multiple_types_csv): processor = CSVProcessor(multiple_types_csv, types=[int, str, float]) assert processor.get_column(2) == [13.1, 6.2, 7.9]
def test_get_column(simple_int_csv): processor = CSVProcessor(simple_int_csv, types=[int, int, int]) assert processor.get_column(2) == [3, 6, 7]
def test_should_throw_when_column_out_of_bound(simple_int_csv): processor = CSVProcessor(simple_int_csv, types=(int, int, int)) with pytest.raises(CSVColumnException): processor.get_column(6) processor.get_column(-1)