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 41 matching lines...) Expand 10 before | Expand all | Expand 10 after Loading... |
52 counter += 1 | 52 counter += 1 |
53 self.assertEqual(counter, len(self.__set)) | 53 self.assertEqual(counter, len(self.__set)) |
54 | 54 |
55 # Now test removal. | 55 # Now test removal. |
56 for i in range(RANGE): | 56 for i in range(RANGE): |
57 dummy_var = create_dummy_var(i) | 57 dummy_var = create_dummy_var(i) |
58 self.__set.remove(dummy_var) | 58 self.__set.remove(dummy_var) |
59 self.assertEqual(len(self.__set), RANGE-i-1) | 59 self.assertEqual(len(self.__set), RANGE-i-1) |
60 self.assertFalse(dummy_var in self.__set) | 60 self.assertFalse(dummy_var in self.__set) |
61 | 61 |
| 62 # Test reset(), for a range of ranges. |
| 63 for subrange in range(RANGE): |
| 64 for i in range(subrange): |
| 65 self.__set.add(create_dummy_var(i)) |
| 66 self.assertEqual(len(self.__set), subrange) |
| 67 self.__set.reset() |
| 68 self.assertEqual(len(self.__set), 0) |
| 69 |
| 70 |
62 def main(): | 71 def main(): |
63 loader = unittest.TestLoader() | 72 loader = unittest.TestLoader() |
64 suite = loader.loadTestsFromTestCase(OrderedSetTest) | 73 suite = loader.loadTestsFromTestCase(OrderedSetTest) |
65 unittest.TextTestRunner(verbosity=2).run(suite) | 74 unittest.TextTestRunner(verbosity=2).run(suite) |
66 | 75 |
67 if __name__ == "__main__": | 76 if __name__ == "__main__": |
68 main() | 77 main() |
69 | 78 |
OLD | NEW |