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):
RANGE = 10
for i in range(RANGE):
dummy_var = create_dummy_var(i)
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)
self.__set.add(dummy_var)
self.assertEqual(len(self.__set), i+1)
self.assertEqual(self.__set[i], dummy_var)
counter = 0
for set_member in self.__set:
self.assertEqual(create_dummy_var(counter), set_member)
counter += 1
self.assertEqual(counter, len(self.__set))
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)
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):
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)
other_set = OrderedSet()
self.assertEqual(len(other_set), 0)
other_set.set(my_set)
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))
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()