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)