OLD | NEW |
(Empty) | |
| 1 #!/usr/bin/python |
| 2 |
| 3 # Copyright 2014 Google Inc. |
| 4 # |
| 5 # Use of this source code is governed by a BSD-style license that can be |
| 6 # found in the LICENSE file. |
| 7 |
| 8 """ |
| 9 Test OrderedSet. |
| 10 """ |
| 11 |
| 12 import helpers |
| 13 import sys |
| 14 import unittest |
| 15 |
| 16 sys.path.append(helpers.GYP_GEN_DIR) |
| 17 |
| 18 from vars_dict_lib import OrderedSet |
| 19 |
| 20 |
| 21 def create_dummy_var(i): |
| 22 return 'dummy_var' + str(i) |
| 23 |
| 24 |
| 25 class OrderedSetTest(unittest.TestCase): |
| 26 |
| 27 def setUp(self): |
| 28 self.__set = OrderedSet() |
| 29 |
| 30 def test_methods(self): |
| 31 """ |
| 32 Test methods on OrderedSet. |
| 33 """ |
| 34 RANGE = 10 |
| 35 for i in range(RANGE): |
| 36 dummy_var = create_dummy_var(i) |
| 37 # Add to the list. This should succeed. |
| 38 self.__set.add(dummy_var) |
| 39 self.assertEqual(len(self.__set), i+1) |
| 40 self.assertTrue(dummy_var in self.__set) |
| 41 self.assertEqual(self.__set[i], dummy_var) |
| 42 |
| 43 # Now attempt to add it again. This should fail. |
| 44 self.__set.add(dummy_var) |
| 45 self.assertEqual(len(self.__set), i+1) |
| 46 self.assertEqual(self.__set[i], dummy_var) |
| 47 |
| 48 # Test iterator. |
| 49 counter = 0 |
| 50 for set_member in self.__set: |
| 51 self.assertEqual(create_dummy_var(counter), set_member) |
| 52 counter += 1 |
| 53 self.assertEqual(counter, len(self.__set)) |
| 54 |
| 55 # Now test removal. |
| 56 for i in range(RANGE): |
| 57 dummy_var = create_dummy_var(i) |
| 58 self.__set.remove(dummy_var) |
| 59 self.assertEqual(len(self.__set), RANGE-i-1) |
| 60 self.assertFalse(dummy_var in self.__set) |
| 61 |
| 62 def main(): |
| 63 loader = unittest.TestLoader() |
| 64 suite = loader.loadTestsFromTestCase(OrderedSetTest) |
| 65 unittest.TextTestRunner(verbosity=2).run(suite) |
| 66 |
| 67 if __name__ == "__main__": |
| 68 main() |
| 69 |
OLD | NEW |