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
103
104
105
106
107
108
109
|
#!/usr/bin/python
# Copyright 2014 Google Inc.
#
# Use of this source code is governed by a BSD-style license that can be
# found in the LICENSE file.
"""
Test OrderedSet.
"""
import sys
import test_variables
import unittest
sys.path.append(test_variables.GYP_GEN_DIR)
from vars_dict_lib import OrderedSet
def create_dummy_var(i):
return 'dummy_var' + str(i)
class OrderedSetTest(unittest.TestCase):
def setUp(self):
self.__set = OrderedSet()
def test_methods(self):
"""Test methods on OrderedSet.
"""
RANGE = 10
for i in range(RANGE):
dummy_var = create_dummy_var(i)
# Add to the list. This should succeed.
self.__set.add(dummy_var)
self.assertEqual(len(self.__set), i+1)
self.assertTrue(dummy_var in self.__set)
self.assertEqual(self.__set[i], dummy_var)
# Now attempt to add it again. This should fail.
self.__set.add(dummy_var)
self.assertEqual(len(self.__set), i+1)
self.assertEqual(self.__set[i], dummy_var)
# Test iterator.
counter = 0
for set_member in self.__set:
self.assertEqual(create_dummy_var(counter), set_member)
counter += 1
self.assertEqual(counter, len(self.__set))
# Now test removal.
for i in range(RANGE):
dummy_var = create_dummy_var(i)
self.__set.remove(dummy_var)
self.assertEqual(len(self.__set), RANGE-i-1)
self.assertFalse(dummy_var in self.__set)
# Test reset(), for a range of ranges.
for subrange in range(RANGE):
for i in range(subrange):
self.__set.add(create_dummy_var(i))
self.assertEqual(len(self.__set), subrange)
self.__set.reset()
self.assertEqual(len(self.__set), 0)
def test_set(self):
"""Test OrderedSet.set().
"""
# Create a set with dummy values.
my_set = OrderedSet()
RANGE = 10
for i in range(RANGE):
my_set.add(create_dummy_var(i))
my_len = len(my_set)
self.assertEqual(my_len, RANGE)
# Copy it to another set.
other_set = OrderedSet()
self.assertEqual(len(other_set), 0)
other_set.set(my_set)
# Both sets should contain the same values, in the same order.
iterator = iter(my_set)
for item in other_set:
self.assertTrue(item == iterator.next())
with self.assertRaises(StopIteration):
iterator.next()
self.assertEqual(my_len, len(other_set))
# But the sets are different. Changing one will not affect the other.
self.assertFalse(other_set is my_set)
other_var = 'something_else'
other_set.add(other_var)
self.assertEqual(my_len + 1, len(other_set))
self.assertEqual(my_len, len(my_set))
self.assertNotIn(other_var, my_set)
def main():
loader = unittest.TestLoader()
suite = loader.loadTestsFromTestCase(OrderedSetTest)
unittest.TextTestRunner(verbosity=2).run(suite)
if __name__ == "__main__":
main()
|
