def test_clear_nocallback(self): ld = LabelledDict(label='all_objects', key_attr='prop1') obj1 = MyTestClass('a', 'b') obj2 = MyTestClass('d', 'b') ld.add(obj1) ld.add(obj2) ld.clear() self.assertEqual(ld, dict())
def test_clear_callback(self): self.signal = set() def func(v): self.signal.add(v) ld = LabelledDict(label='all_objects', key_attr='prop1', remove_callable=func) obj1 = MyTestClass('a', 'b') obj2 = MyTestClass('d', 'b') ld.add(obj1) ld.add(obj2) ld.clear() self.assertSetEqual(self.signal, {obj2, obj1}) self.assertEqual(ld, dict())