1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
|
import io
import sys
from unittest.mock import patch
from csvkit.utilities.csvsort import CSVSort, launch_new_instance
from tests.utils import ColumnsTests, CSVKitTestCase, EmptyFileTests, NamesTests, stdin_as_string
class TestCSVSort(CSVKitTestCase, ColumnsTests, EmptyFileTests, NamesTests):
Utility = CSVSort
def test_launch_new_instance(self):
with patch.object(sys, 'argv', [self.Utility.__name__.lower(), 'examples/dummy.csv']):
launch_new_instance()
def test_runs(self):
self.assertRows(['examples/test_utf8.csv'], [
['foo', 'bar', 'baz'],
['1', '2', '3'],
['4', '5', 'ʤ'],
])
def test_encoding(self):
self.assertRows(['-e', 'latin1', 'examples/test_latin1.csv'], [
['a', 'b', 'c'],
['1', '2', '3'],
['4', '5', '©'],
])
def test_sort_string_reverse(self):
reader = self.get_output_as_reader(['-c', '1', '-r', 'examples/testxls_converted.csv'])
test_order = ['text', 'Unicode! Σ', 'This row has blanks',
'Chicago Tribune', 'Chicago Sun-Times', 'Chicago Reader']
new_order = [str(r[0]) for r in reader]
self.assertEqual(test_order, new_order)
def test_sort_date(self):
reader = self.get_output_as_reader(['-c', '2', 'examples/testxls_converted.csv'])
test_order = ['text', 'Chicago Tribune', 'Chicago Sun-Times',
'Chicago Reader', 'This row has blanks', 'Unicode! Σ']
new_order = [str(r[0]) for r in reader]
self.assertEqual(test_order, new_order)
def test_ignore_case(self):
self.assertRows(['-i', 'examples/test_ignore_case.csv'], [
['a', 'b', 'c'],
['3', '2009-01-01', 'd'],
['20', '2001-01-01', 'c'],
['20', '2002-01-01', 'b'],
['100', '2003-01-01', 'a'],
['100', '2003-01-01', 'A'],
])
def test_no_blanks(self):
reader = self.get_output_as_reader(['examples/blanks.csv'])
test_order = [
['a', 'b', 'c', 'd', 'e', 'f'],
['', '', '', '', '', ''],
]
new_order = list(reader)
self.assertEqual(test_order, new_order)
def test_blanks(self):
reader = self.get_output_as_reader(['--blanks', 'examples/blanks.csv'])
test_order = [
['a', 'b', 'c', 'd', 'e', 'f'],
['', 'NA', 'N/A', 'NONE', 'NULL', '.'],
]
new_order = list(reader)
self.assertEqual(test_order, new_order)
def test_no_header_row(self):
self.assertRows(['--no-header-row', '--no-inference', 'examples/no_header_row.csv'], [
['a', 'b', 'c'],
['1', '2', '3'],
])
def test_no_inference(self):
reader = self.get_output_as_reader(['--no-inference', '-c', '1', 'examples/test_literal_order.csv'])
test_order = ['a', '192', '27', '3']
new_order = [str(r[0]) for r in reader]
self.assertEqual(test_order, new_order)
def test_sort_t_and_nulls(self):
reader = self.get_output_as_reader(['-c', '2', 'examples/sort_ints_nulls.csv'])
test_order = ['b', '1', '2', '']
new_order = [str(r[1]) for r in reader]
self.assertEqual(test_order, new_order)
def test_stdin(self):
input_file = io.BytesIO(b'a,b,c\n4,5,6\n1,2,3\n')
with stdin_as_string(input_file):
self.assertLines([], [
'a,b,c',
'1,2,3',
'4,5,6',
])
input_file.close()
|
