def test_register_args_not_implemented(self):
     registry = Registry()
     with self.assertRaises(NotImplementedError):
         registry.register_agent(ArgsNotImplemented)
 def test_register_invalid_class(self):
     registry = Registry()
     with self.assertRaises(AssertionError):
         registry.register_agent(NotSubclass)