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
|
# Copyright 2009-2013 Jaap Karssenberg <jaap.karssenberg@gmail.com>
import tests
from copy import copy
from zim.base import LastDefinedOrderedDict, MovingWindowIter
from zim.base.naturalsort import natural_sort, natural_sorted
class TestNaturalSorting(tests.TestCase):
def runTest(self):
input = [
'a', 'Aa', 'AA', # (these last 2 should be swapped due to case)
'1.1 foo', '10.1.1 bar', '2.1 dus', '1.01 foo',
'foo2bar', 'foo10bar', 'foo01bar',
]
wanted = [
'1.01 foo', '1.1 foo', '2.1 dus', '10.1.1 bar',
'a', 'AA', 'Aa',
'foo01bar', 'foo2bar', 'foo10bar',
]
# TODO: add utf-8 test data and set matching locale
result = natural_sorted(input)
self.assertEqual(result, wanted)
self.assertTrue(id(result) != id(input))
result = copy(input)
natural_sort(result)
self.assertEqual(result, wanted)
input = [(1, 'b'), (2, 'a')]
wanted = [(2, 'a'), (1, 'b')]
result = natural_sorted(input, key=lambda t: t[1])
self.assertEqual(result, wanted)
self.assertTrue(id(result) != id(input))
class TestLastDefinedOrderedDict(tests.TestCase):
def runTest(self):
items = [('foo', 1), ('bar', 2), ('baz', 3)]
mydict = LastDefinedOrderedDict(items)
self.assertIsInstance(repr(mydict), str)
self.assertEqual(list(mydict.items()), items)
self.assertEqual(list(mydict), [i[0] for i in items])
self.assertEqual(list(mydict.keys()), [i[0] for i in items])
mydict['bar'] = 'X'
mydict.setdefault('foo', 'dus')
items = [('foo', 1), ('bar', 'X'), ('baz', 3)]
self.assertEqual(list(mydict.items()), items)
self.assertEqual(list(mydict), [i[0] for i in items])
self.assertEqual(list(mydict.keys()), [i[0] for i in items])
del mydict['bar']
mydict['bar'] = 'Y'
items = [('foo', 1), ('baz', 3), ('bar', 'Y')]
self.assertEqual(list(mydict.items()), items)
self.assertEqual(list(mydict), [i[0] for i in items])
self.assertEqual(list(mydict.keys()), [i[0] for i in items])
mydict.pop('foo')
mydict.setdefault('foo', 'dus')
items = [('baz', 3), ('bar', 'Y'), ('foo', 'dus')]
self.assertEqual(list(mydict.items()), items)
self.assertEqual(list(mydict), [i[0] for i in items])
self.assertEqual(list(mydict.keys()), [i[0] for i in items])
class TestMovingWindowIterBuffer(tests.TestCase):
def runTest(self):
mylist = ['a', 'b', 'c', 'd']
myiter = MovingWindowIter(mylist)
self.assertEqual(iter(myiter), myiter, 'MovingWindowIter should be an iter, not an iterable')
seen = []
n = len(mylist)
for i, t in enumerate(myiter):
seen.append(t[1])
if i == 0:
self.assertEqual(t, (None, mylist[0], mylist[1]))
self.assertFalse(myiter.last)
elif i == n - 1:
self.assertEqual(t, (mylist[-2], mylist[-1], None))
self.assertTrue(myiter.last)
else:
self.assertEqual(t, (mylist[i - 1], mylist[i], mylist[i + 1]))
self.assertFalse(myiter.last)
self.assertEqual(seen, mylist)
|
