Python CSVProcessor.drop_column Examples

Example #1

File: csv_processor_test.py Project: krzysiekjezierski/zaliczenie_wsb

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

Example #2

File: test_csv_processor.py Project: psobczak/csv_processor

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

Example #3

File: csv_processor_test.py Project: krzysiekjezierski/zaliczenie_wsb

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

Example #4

File: test_csv_processor.py Project: psobczak/csv_processor

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