def test_drop_column(simple_int_csv):
    processor = CSVProcessor(simple_int_csv, types=[int, int, int])
    processor.drop_column(2)

    assert processor.csv == [[1, 8], [4, 5], [0, 5]]
    assert processor.header == ['col1', 'col2']
def test_drop_column_multiple_types(multiple_types_csv):
    processor = CSVProcessor(multiple_types_csv, types=(int, str, float))

    processor.drop_column(2)
    assert processor.csv == [[0, "'str3'"], [1, "'str2'"], [2, "'str1'"]]
def test_drop_column_multi(multiple_types_csv):
    processor = CSVProcessor(multiple_types_csv, types=[int, str, float])
    processor.drop_column(2)

    assert processor.csv == [[0, "'str3'"], [1, "'str2'"], [2, "'str1'"]]
    assert processor.header == ['col1', 'col2']
def test_drop_column(simple_int_csv):
    processor = CSVProcessor(simple_int_csv, types=(int, int, int))

    processor.drop_column(2)
    assert processor.csv == [[1, 8], [4, 5], [0, 5]]