Exemplos de CSVProcessor.drop_column em Python

Exemplo n.º 1

Arquivo: csv_processor_test.py Projeto: 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']

Exemplo n.º 2

Arquivo: test_csv_processor.py Projeto: 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'"]]

Exemplo n.º 3

Arquivo: csv_processor_test.py Projeto: 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']

Exemplo n.º 4

Arquivo: test_csv_processor.py Projeto: 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]]