-
Notifications
You must be signed in to change notification settings - Fork 0
/
test_split_after.py
55 lines (48 loc) · 1.89 KB
/
test_split_after.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
from unittest import TestCase
from split_after import split_after
class SplitAfterTests(TestCase):
def test_start_with_sep(self):
actual = list(split_after('xooxoo', lambda c: c == 'x'))
expected = [['x'], ['o', 'o', 'x'], ['o', 'o']]
self.assertEqual(actual, expected)
def test_end_with_sep(self):
actual = list(split_after('ooxoox', lambda c: c == 'x'))
expected = [['o', 'o', 'x'], ['o', 'o', 'x']]
self.assertEqual(actual, expected)
def test_no_sep(self):
actual = list(split_after('ooo', lambda c: c == 'x'))
expected = [['o', 'o', 'o']]
self.assertEqual(actual, expected)
def test_max_split(self):
for args, expected in [
(
('a,b,c,d', lambda c: c == ',', -1),
([['a', ','], ['b', ','], ['c', ','], ['d']])
),
(
('a,b,c,d', lambda c: c == ',', 0),
([['a', ','], ['b', ','], ['c', ','], ['d']])
),
(
('a,b,c,d', lambda c: c == ',', 1),
([['a', ',', 'b', ',', 'c', ',', 'd']])
),
(
('a,b,c,d', lambda c: c == ',', 2),
([['a', ','], ['b', ','], ['c', ',', 'd']])
),
(
('a,b,c,d', lambda c: c == ',', 10),
([['a', ','], ['b', ','], ['c', ','], ['d']])
),
(
('a,b,c,d', lambda c: c == '@', 2),
([['a', ',', 'b', ',', 'c', ',', 'd']])
),
(
('a,b,c,d', lambda c: c != ',', 2),
([['a'], [',', 'b'], [',', 'c', ',', 'd']])
)
]:
actual = list(split_after(*args))
self.assertEqual(actual, expected)