def test_remove(self): sc = ISubscriptionContainer(self.root) sub1 = AccountSubscription('user1', 'instant', 'en') sub2 = AccountSubscription('user2', 'instant', 'en') sc.add(sub1) sc.add(sub2) self.assertEqual(len(list(sc)), 2) enum = dict(sc.list_with_keys()) self.assertTrue(enum[1] is sub1) sc.remove(1) self.assertEqual(len(list(sc)), 1) self.assertTrue(list(sc)[0] is sub2)