def test_get_policy_returns_cached(self): ps = PolicyService(self.saved_model) first_policy = ps.GetPolicy(self.page, MockGRPCServicerContext()) second_policy = ps.GetPolicy(self.page, MockGRPCServicerContext())
def test_get_policy(self): ps = PolicyService(self.saved_model) policy = ps.GetPolicy(self.page, MockGRPCServicerContext()) assert policy assert isinstance(policy, policy_service_pb2.Policy)