def test_unregister_removes_item_from_collection(self): key = "ABC" lc = Registry() TestClass = self.get_test_class(key) lc[key] = TestClass lc.unregister(TestClass) self.assertNotIn(key, lc)
def test_unregister_returns_none_when_class_isnt_already_registered(self): # as opposed to blowing up on a KeyError lc = Registry() TestClass = self.get_test_class("ABC") self.assertEqual(None, lc.unregister(TestClass))