def test_iter(self):
objects = OrderedSet()
for i in xrange(10):
objects.add(i)
it = iter(objects)
for i in xrange(10):
self.assertEqual(i, it.next())
self.assertRaises(StopIteration, it.next)
def test_remove(self):
objects = OrderedSet([1, 2, 3])
self.assertEqual(3, len(objects))
objects.remove(2)
self.assertEqual(2, len(objects))
self.assertEqual(1, objects[0])
self.assertEqual(3, objects[1])
objects.remove(3)
self.assertEqual(1, len(objects))
self.assertEqual(1, objects[0])
objects.add(1)
self.assertEqual(1, len(objects))
self.assertEqual(1, objects[0])
objects.remove(1)
self.assertEqual(0, len(objects))
def test_some_elements(self):
objects = OrderedSet()
self.assertEqual(0, len(objects))
objects.add(1)
objects.add("Hello")
objects.add((1, 2, 3))
self.assertEqual(3, len(objects))
self.assert_(1 in objects)
self.assert_("Hello" in objects)
self.assert_((1, 2, 3) in objects)
self.assert_(2 not in objects)
self.assert_("HELLO" not in objects)
self.assertEqual(1, objects[0])
self.assertEqual("Hello", objects[1])
self.assertEqual((1, 2, 3), objects[2])
