def test_drop_columns_multi(multiple_types_csv):
processor = CSVProcessor(multiple_types_csv, types=[int, str, float])
processor.drop_columns((0, 2))
assert processor.csv == [["'str3'"], ["'str2'"], ["'str1'"]]
assert processor.header == ['col2']
def test_drop_columns(simple_int_csv):
processor = CSVProcessor(simple_int_csv, types=[int, int, int])
processor.drop_columns((0, 2))
assert processor.csv == [[8], [5], [5]]
assert processor.header == ['col2']
def test_drop_columns_multiple_types(multiple_types_csv):
processor = CSVProcessor(multiple_types_csv, types=(int, str, float))
processor.drop_columns((0, 1))
assert processor.csv == [[13.1], [6.2], [7.9]]
assert processor.header == ['col3']