def test_model_init(self): model_name = 'modelname' def initial_func(a=1): return a model = Model(model_name,initial_func) def policy_func(a=1): return a policy = Policy('test', function=policy_func, unknown='a') model.model_init(policy) self.assertEqual(policy, model.policy) self.assertEqual(model.function, policy_func) with self.assertRaises(AttributeError): model.unknown
def test_model_init(self): model_name = 'modelname' def initial_func(a=1): return a model = Model(model_name,initial_func) def policy_func(a=1): return a policy = Policy('test', function=policy_func, unknown='a') model.model_init(policy) self.assertEqual(policy, model.policy) self.assertEqual(model.function, policy_func) with self.assertRaises(AttributeError): model.unknown