| 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 |