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
|
import pickle
import unittest
from Orange.misc.collections import frozendict, natural_sorted, DictMissingConst
class TestFrozenDict(unittest.TestCase):
def test_removed_methods(self):
d = frozendict({"a": 12})
self.assertRaises(AttributeError, d.clear)
self.assertRaises(AttributeError, d.pop, "a")
self.assertRaises(AttributeError, d.popitem)
self.assertRaises(AttributeError, d.update, {"b": 5})
self.assertRaises(AttributeError, d.setdefault, "b", 5)
with self.assertRaises(AttributeError):
del d["a"]
with self.assertRaises(AttributeError):
d["a"] = 13
self.assertEqual(d, {"a": 12})
def test_functions_as_dict(self):
d = frozendict({"a": 12, "b": 13})
self.assertEqual(len(d), 2)
self.assertEqual(d["a"], 12)
self.assertEqual(d.get("a"), 12)
self.assertEqual(d.get("c", 14), 14)
self.assertEqual(set(d), {"a", "b"})
self.assertEqual(set(d.keys()), {"a", "b"})
self.assertEqual(set(d.values()), {12, 13})
self.assertEqual(set(d.items()), {("a", 12), ("b", 13)})
class TestUtils(unittest.TestCase):
def test_natural_sorted(self):
data = [
"something1",
"something20",
"something2",
"something12"
]
res = [
"something1",
"something2",
"something12",
"something20"
]
self.assertListEqual(res, natural_sorted(data))
def test_natural_sorted_text(self):
data = ["b", "aa", "c", "dd"]
res = ["aa", "b", "c", "dd"]
self.assertListEqual(res, natural_sorted(data))
def test_natural_sorted_numbers_str(self):
data = ["1", "20", "2", "12"]
res = ["1", "2", "12", "20"]
self.assertListEqual(res, natural_sorted(data))
def test_natural_sorted_numbers(self):
data = [1, 20, 2, 12]
res = [1, 2, 12, 20]
self.assertListEqual(res, natural_sorted(data))
class TestDictMissingConst(unittest.TestCase):
def test_dict_missing(self):
d = DictMissingConst("<->", {1: 1, 2: 2})
self.assertEqual(d[1], 1)
self.assertEqual(d[-1], "<->")
# d[-1] must not grow the dict
self.assertEqual(len(d), 2)
self.assertEqual(d, DictMissingConst("<->", {1: 1, 2: 2}))
self.assertNotEqual(
DictMissingConst("A", {1: 1}), DictMissingConst("B", {1: 1}),
)
dc = pickle.loads(pickle.dumps(d))
self.assertEqual(d, dc)
self.assertEqual(dict(d), dict(dc))
self.assertEqual(d.missing, dc.missing)
self.assertEqual(d[object()], dc[object()])
if __name__ == "__main__":
unittest.main()
|
