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']