def test_get_columns(simple_int_csv):
processor = CSVProcessor(simple_int_csv, types=[int, int, int])
assert processor.get_columns((0, 1)) == [[1, 8], [4, 5], [0, 5]]
def test_get_columns(simple_int_csv):
processor = CSVProcessor(simple_int_csv, types=(int, int, int))
assert processor.get_columns((1, 2)) == [[8, 3], [5, 6], [5, 7]]
def test_get_columns_multi(multiple_types_csv):
processor = CSVProcessor(multiple_types_csv, types=[int, str, float])
assert processor.get_columns((0, 1)) == [[0, "'str3'"], [1, "'str2'"],
[2, "'str1'"]]
def test_get_columns_multiple_types(multiple_types_csv):
processor = CSVProcessor(multiple_types_csv, types=(int, str, float))
assert processor.get_columns((1, 2)) == [["'str3'", 13.1], ["'str2'", 6.2], ["'str1'", 7.9]]