forked from wireservice/csvkit
/
test_csvsort.py
executable file
·105 lines (71 loc) · 3.23 KB
/
test_csvsort.py
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
101
102
103
104
#!/usr/bin/env python
# -*- coding: utf-8 -*-
import six
try:
import unittest2 as unittest
except ImportError:
import unittest
from csvkit import CSVKitReader
from csvkit.utilities.csvsort import CSVSort
from csvkit.exceptions import ColumnIdentifierError, RequiredHeaderError
class TestCSVSort(unittest.TestCase):
def test_sort_string_reverse(self):
args = ['-c', '1', '-r', 'examples/testxls_converted.csv']
output_file = six.StringIO()
utility = CSVSort(args, output_file)
utility.main()
input_file = six.StringIO(output_file.getvalue())
reader = CSVKitReader(input_file)
test_order = [u'text', u'Unicode! Σ', u'This row has blanks', u'Chicago Tribune', u'Chicago Sun-Times', u'Chicago Reader']
new_order = [six.text_type(r[0]) for r in reader]
self.assertEqual(test_order, new_order)
def test_sort_date(self):
args = ['-c', '2', 'examples/testxls_converted.csv']
output_file = six.StringIO()
utility = CSVSort(args, output_file)
utility.main()
input_file = six.StringIO(output_file.getvalue())
reader = CSVKitReader(input_file)
test_order = [u'text', u'This row has blanks', u'Unicode! Σ', u'Chicago Tribune', u'Chicago Sun-Times', u'Chicago Reader']
new_order = [six.text_type(r[0]) for r in reader]
self.assertEqual(test_order, new_order)
def test_invalid_column(self):
args = ['-c', '0', 'examples/dummy.csv']
output_file = six.StringIO()
utility = CSVSort(args, output_file)
self.assertRaises(ColumnIdentifierError, utility.main)
def test_invalid_options(self):
args = ['-n', '--no-header-row', 'examples/dummy.csv']
output_file = six.StringIO()
utility = CSVSort(args, output_file)
self.assertRaises(RequiredHeaderError, utility.main)
def test_no_header_row(self):
args = ['--no-header-row', '-c', '1', '-r', 'examples/no_header_row3.csv']
output_file = six.StringIO()
utility = CSVSort(args, output_file)
utility.main()
input_file = six.StringIO(output_file.getvalue())
reader = CSVKitReader(input_file)
test_order = ['column1', '4', '1']
new_order = [six.text_type(r[0]) for r in reader]
self.assertEqual(test_order, new_order)
def test_no_inference(self):
args = ['--no-inference', '-c', '1', 'examples/test_literal_order.csv']
output_file = six.StringIO()
utility = CSVSort(args, output_file)
utility.main()
input_file = six.StringIO(output_file.getvalue())
reader = CSVKitReader(input_file)
test_order = [u'a', u'192', u'27', u'3']
new_order = [six.text_type(r[0]) for r in reader]
self.assertEqual(test_order, new_order)
def test_sort_ints_and_nulls(self):
args = ['-c', '2', 'examples/sort_ints_nulls.csv']
output_file = six.StringIO()
utility = CSVSort(args, output_file)
utility.main()
input_file = six.StringIO(output_file.getvalue())
reader = CSVKitReader(input_file)
test_order = ['b', '', '1', '2']
new_order = [six.text_type(r[1]) for r in reader]
self.assertEqual(test_order, new_order)