OLD | NEW |
1 #!/usr/bin/python | 1 #!/usr/bin/python |
2 | 2 |
3 # Copyright 2014 Google Inc. | 3 # Copyright 2014 Google Inc. |
4 # | 4 # |
5 # Use of this source code is governed by a BSD-style license that can be | 5 # Use of this source code is governed by a BSD-style license that can be |
6 # found in the LICENSE file. | 6 # found in the LICENSE file. |
7 | 7 |
8 """ | 8 """ |
9 Test OrderedSet. | 9 Test OrderedSet. |
10 """ | 10 """ |
(...skipping 10 matching lines...) Expand all Loading... |
21 def create_dummy_var(i): | 21 def create_dummy_var(i): |
22 return 'dummy_var' + str(i) | 22 return 'dummy_var' + str(i) |
23 | 23 |
24 | 24 |
25 class OrderedSetTest(unittest.TestCase): | 25 class OrderedSetTest(unittest.TestCase): |
26 | 26 |
27 def setUp(self): | 27 def setUp(self): |
28 self.__set = OrderedSet() | 28 self.__set = OrderedSet() |
29 | 29 |
30 def test_methods(self): | 30 def test_methods(self): |
31 """ | 31 """Test methods on OrderedSet. |
32 Test methods on OrderedSet. | |
33 """ | 32 """ |
34 RANGE = 10 | 33 RANGE = 10 |
35 for i in range(RANGE): | 34 for i in range(RANGE): |
36 dummy_var = create_dummy_var(i) | 35 dummy_var = create_dummy_var(i) |
37 # Add to the list. This should succeed. | 36 # Add to the list. This should succeed. |
38 self.__set.add(dummy_var) | 37 self.__set.add(dummy_var) |
39 self.assertEqual(len(self.__set), i+1) | 38 self.assertEqual(len(self.__set), i+1) |
40 self.assertTrue(dummy_var in self.__set) | 39 self.assertTrue(dummy_var in self.__set) |
41 self.assertEqual(self.__set[i], dummy_var) | 40 self.assertEqual(self.__set[i], dummy_var) |
42 | 41 |
(...skipping 17 matching lines...) Expand all Loading... |
60 self.assertFalse(dummy_var in self.__set) | 59 self.assertFalse(dummy_var in self.__set) |
61 | 60 |
62 # Test reset(), for a range of ranges. | 61 # Test reset(), for a range of ranges. |
63 for subrange in range(RANGE): | 62 for subrange in range(RANGE): |
64 for i in range(subrange): | 63 for i in range(subrange): |
65 self.__set.add(create_dummy_var(i)) | 64 self.__set.add(create_dummy_var(i)) |
66 self.assertEqual(len(self.__set), subrange) | 65 self.assertEqual(len(self.__set), subrange) |
67 self.__set.reset() | 66 self.__set.reset() |
68 self.assertEqual(len(self.__set), 0) | 67 self.assertEqual(len(self.__set), 0) |
69 | 68 |
| 69 def test_set(self): |
| 70 """Test OrderedSet.set(). |
| 71 """ |
| 72 # Create a set with dummy values. |
| 73 my_set = OrderedSet() |
| 74 RANGE = 10 |
| 75 for i in range(RANGE): |
| 76 my_set.add(create_dummy_var(i)) |
| 77 my_len = len(my_set) |
| 78 self.assertEqual(my_len, RANGE) |
| 79 |
| 80 # Copy it to another set. |
| 81 other_set = OrderedSet() |
| 82 self.assertEqual(len(other_set), 0) |
| 83 other_set.set(my_set) |
| 84 |
| 85 # Both sets should contain the same values, in the same order. |
| 86 iterator = iter(my_set) |
| 87 for item in other_set: |
| 88 self.assertTrue(item == iterator.next()) |
| 89 with self.assertRaises(StopIteration): |
| 90 iterator.next() |
| 91 self.assertEqual(my_len, len(other_set)) |
| 92 |
| 93 # But the sets are different. Changing one will not affect the other. |
| 94 self.assertFalse(other_set is my_set) |
| 95 other_var = 'something_else' |
| 96 other_set.add(other_var) |
| 97 self.assertEqual(my_len + 1, len(other_set)) |
| 98 self.assertEqual(my_len, len(my_set)) |
| 99 self.assertNotIn(other_var, my_set) |
| 100 |
70 | 101 |
71 def main(): | 102 def main(): |
72 loader = unittest.TestLoader() | 103 loader = unittest.TestLoader() |
73 suite = loader.loadTestsFromTestCase(OrderedSetTest) | 104 suite = loader.loadTestsFromTestCase(OrderedSetTest) |
74 unittest.TextTestRunner(verbosity=2).run(suite) | 105 unittest.TextTestRunner(verbosity=2).run(suite) |
75 | 106 |
76 if __name__ == "__main__": | 107 if __name__ == "__main__": |
77 main() | 108 main() |
78 | 109 |
OLD | NEW |