def test_prepend(self): """Test the prepend method of OrderedSet.""" my_set = OrderedSet() my_set.prepend(1) my_set.prepend(2) my_set.prepend(3) x = 3 for element in my_set: assert_equal(element, x) x = x - 1 assert_equal(0, x, 'Expected set to have 3 items.') assert_raises(KeyError, my_set.prepend, 3)
def test_length(self): """Test the __length__ method of OrderedSet.""" my_set = OrderedSet() assert_equal(len(my_set), 0) my_set.append(1) assert_equal(len(my_set), 1) my_set.prepend(2) assert_equal(len(my_set), 2) my_set.insert_after(1, 3) assert_equal(len(my_set), 3) my_set.remove(2) assert_equal(len(my_set), 2)