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)
