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.csvgrep import CSVGrep, launch_new_instance
from tests.utils import ColumnsTests, CSVKitTestCase, EmptyFileTests, NamesTests
class TestCSVGrep(CSVKitTestCase, ColumnsTests, EmptyFileTests, NamesTests):
Utility = CSVGrep
default_args = ['-c', '1', '-m', '1']
columns_args = ['-m', '1']
def test_launch_new_instance(self):
with patch.object(sys, 'argv', [self.Utility.__name__.lower()] + self.default_args + ['examples/dummy.csv']):
launch_new_instance()
def test_options(self):
for args, message in (
([], 'You must specify at least one column to search using the -c option.'),
(['-c', '1'], 'One of -r, -m or -f must be specified, unless using the -n option.'),
):
with self.subTest(args=args):
self.assertError(launch_new_instance, args, message)
def test_skip_lines(self):
self.assertRows(['--skip-lines', '3', '-c', '1', '-m', '1', 'examples/test_skip_lines.csv'], [
['a', 'b', 'c'],
['1', '2', '3'],
])
def test_match(self):
self.assertRows(['-c', '1', '-m', '1', 'examples/dummy.csv'], [
['a', 'b', 'c'],
['1', '2', '3'],
])
def test_any_match(self):
self.assertRows(['-c', '1,2,3', '-a', '-m', '1', 'examples/dummy.csv'], [
['a', 'b', 'c'],
['1', '2', '3'],
])
def test_match_utf8(self):
self.assertRows(['-c', '3', '-m', 'ʤ', 'examples/test_utf8.csv'], [
['foo', 'bar', 'baz'],
['4', '5', 'ʤ'],
])
def test_match_utf8_bom(self):
self.assertRows(['-c', '3', '-m', 'ʤ', 'examples/test_utf8_bom.csv'], [
['foo', 'bar', 'baz'],
['4', '5', 'ʤ'],
])
def test_no_match(self):
self.assertRows(['-c', '1', '-m', 'NO MATCH', 'examples/dummy.csv'], [
['a', 'b', 'c'],
])
def test_invert_match(self):
self.assertRows(['-c', '1', '-i', '-m', 'NO MATCH', 'examples/dummy.csv'], [
['a', 'b', 'c'],
['1', '2', '3'],
])
def test_re_match(self):
self.assertRows(['-c', '3', '-r', '^(3|9)$', 'examples/dummy.csv'], [
['a', 'b', 'c'],
['1', '2', '3'],
])
def test_re_match_utf8(self):
self.assertRows(['-c', '3', '-r', 'ʤ', 'examples/test_utf8.csv'], [
['foo', 'bar', 'baz'],
['4', '5', 'ʤ'],
])
def test_string_match(self):
self.assertRows(['-c', '1', '-m', 'ILLINOIS',
'examples/realdata/FY09_EDU_Recipients_by_State.csv'], [
['State Name', 'State Abbreviate', 'Code', 'Montgomery GI Bill-Active Duty',
'Montgomery GI Bill- Selective Reserve', 'Dependents\' Educational Assistance',
'Reserve Educational Assistance Program', 'Post-Vietnam Era Veteran\'s Educational Assistance Program',
'TOTAL', ''],
['ILLINOIS', 'IL', '17', '15,659', '2,491', '2,025', '1,770', '19', '21,964', ''],
])
def test_match_with_line_numbers(self):
self.assertRows(['-c', '1', '-m', 'ILLINOIS', '--linenumbers',
'examples/realdata/FY09_EDU_Recipients_by_State.csv'], [
['line_numbers', 'State Name', 'State Abbreviate', 'Code', 'Montgomery GI Bill-Active Duty',
'Montgomery GI Bill- Selective Reserve', 'Dependents\' Educational Assistance',
'Reserve Educational Assistance Program', 'Post-Vietnam Era Veteran\'s Educational Assistance Program',
'TOTAL', ''],
['14', 'ILLINOIS', 'IL', '17', '15,659', '2,491', '2,025', '1,770', '19', '21,964', ''],
])
def test_kwargs_with_line_numbers(self):
self.assertRows(['-t', '-c', '1', '-m', '1', '--linenumbers', 'examples/dummy.tsv'], [
['line_numbers', 'a', 'b', 'c'],
['1', '1', '2', '3'],
])
|
