def test_sort_with_key(simple_int_csv): processor = CSVProcessor(simple_int_csv, types=(int, int, int)) processor.sort(key=1) assert processor.csv == [[4, 5, 6], [0, 5, 7], [1, 8, 3]] assert processor.header == ['col1', 'col2', 'col3']
def test_sort_with_key_multi(multiple_types_csv): processor = CSVProcessor(multiple_types_csv, types=[int, str, float]) processor.sort(key=1) assert processor.csv == [[2, "'str1'", 7.9], [1, "'str2'", 6.2], [0, "'str3'", 13.1]] assert processor.header == ['col1', 'col2', 'col3']
def test_sort_without_key_multiple_types(multiple_types_csv): processor = CSVProcessor(multiple_types_csv, types=(int, str, float)) processor.sort() assert processor.csv == [[0, "'str3'", 13.1], [1, "'str2'", 6.2], [2, "'str1'", 7.9]] assert processor.header == ['col1', 'col2', 'col3']
def test_sort_without_key(simple_int_csv): processor = CSVProcessor(simple_int_csv, types=[int, int, int]) processor.sort() assert processor.csv == [[0, 5, 7], [1, 8, 3], [4, 5, 6]] assert processor.header == ['col1', 'col2', 'col3']