def test_register_args_not_implemented(self):
registry = Registry()
with self.assertRaises(NotImplementedError):
registry.register_agent(ArgsNotImplemented)
def test_save_classes(self):
dummy_log_id_dir = "/tmp/adept_test/test_save_classes"
registry = Registry()
registry.register_env(DummyEnv)
registry.save_extern_classes(dummy_log_id_dir)
other = Registry()
other.load_extern_classes(dummy_log_id_dir)
env_cls = other.lookup_env("dummy")
env = env_cls()
env.reset()
def test_register_invalid_class(self):
registry = Registry()
with self.assertRaises(AssertionError):
registry.register_agent(NotSubclass)