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
|
import sys
from unittest.mock import patch
from csvkit.utilities.csvcut import CSVCut, launch_new_instance
from tests.utils import ColumnsTests, CSVKitTestCase, EmptyFileTests, NamesTests
class TestCSVCut(CSVKitTestCase, ColumnsTests, EmptyFileTests, NamesTests):
Utility = CSVCut
def test_launch_new_instance(self):
with patch.object(sys, 'argv', [self.Utility.__name__.lower(), 'examples/dummy.csv']):
launch_new_instance()
def test_skip_lines(self):
self.assertRows(['--skip-lines', '3', '-c', '1,3', 'examples/test_skip_lines.csv'], [
['a', 'c'],
['1', '3'],
])
def test_simple(self):
self.assertRows(['-c', '1,3', 'examples/dummy.csv'], [
['a', 'c'],
['1', '3'],
])
def test_linenumbers(self):
self.assertRows(['-c', '1,3', '-l', 'examples/dummy.csv'], [
['line_number', 'a', 'c'],
['1', '1', '3'],
])
def test_unicode(self):
self.assertRows(['-c', '1,3', 'examples/test_utf8.csv'], [
['foo', 'baz'],
['1', '3'],
['4', 'ʤ'],
])
def test_with_gzip(self):
self.assertRows(['-c', '1,3', 'examples/dummy.csv.gz'], [
['a', 'c'],
['1', '3'],
])
def test_with_bzip2(self):
self.assertRows(['-c', '1,3', 'examples/dummy.csv.bz2'], [
['a', 'c'],
['1', '3'],
])
def test_exclude(self):
self.assertRows(['-C', '1,3', 'examples/dummy.csv'], [
['b'],
['2'],
])
def test_exclude_unknown_columns(self):
self.assertRows(['-C', '1,foo,42', 'examples/dummy.csv'], [
['b', 'c'],
['2', '3'],
])
def test_include_and_exclude(self):
self.assertRows(['-c', '1,3', '-C', '3', 'examples/dummy.csv'], [
['a'],
['1'],
])
def test_delete_empty(self):
self.assertRows(['-c', 'column_c', '--delete-empty-rows', 'examples/bad.csv'], [
['column_c'],
['17'],
])
def test_no_header_row(self):
self.assertRows(['-c', '2', '--no-header-row', 'examples/no_header_row.csv'], [
['b'],
['2'],
])
def test_ragged(self):
# Test that csvcut doesn't error when a row is short.
self.get_output(['-c', 'column_c', 'examples/bad.csv'])
def test_truncate(self):
# Test that csvcut truncates long rows.
self.assertRows(['-C', 'column_a,column_b', '--delete-empty-rows', 'examples/bad.csv'], [
['column_c'],
['17'],
])
def test_names_with_skip_lines(self):
self.assertLines(['--names', '--skip-lines', '3', 'examples/test_skip_lines.csv'], [
' 1: a',
' 2: b',
' 3: c',
])
def test_null_byte(self):
# Test that csvcut doesn't error on a null byte.
self.get_output(['-C', '', 'examples/null_byte.csv'])
|
