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))
