From 3528e494a24379873d1cc35022a3d88b5d771654 Mon Sep 17 00:00:00 2001
From: Mehdi Amini <aminim@google.com>
Date: Fri, 8 May 2020 19:32:09 -0700
Subject: [PATCH] Add missing ASSERT_EQ on status after API call in
 c_api_unified_experimental_test

PiperOrigin-RevId: 310671562
Change-Id: Id0b07d6889340d631f6144f8fa6dd3f3309b3776
---
 tensorflow/c/eager/c_api_unified_experimental_test.cc | 1 +
 1 file changed, 1 insertion(+)

diff --git a/tensorflow/c/eager/c_api_unified_experimental_test.cc b/tensorflow/c/eager/c_api_unified_experimental_test.cc
index 170b82333d8..8c9aa97ea8f 100644
--- a/tensorflow/c/eager/c_api_unified_experimental_test.cc
+++ b/tensorflow/c/eager/c_api_unified_experimental_test.cc
@@ -131,6 +131,7 @@ TEST(UnifedCAPI, TestBasicGraph) {
   string fn_name = "double";
   TF_AbstractFunction* func = TF_ExecutionContextToFunction(
       graph_ctx, fn_name.c_str(), 1, placeholder_t, 1, output_t, status.get());
+  ASSERT_EQ(TF_OK, TF_GetCode(status.get())) << TF_Message(status.get());
   TF_DeleteAbstractTensor(placeholder_t);
   TF_DeleteAbstractTensor(output_t);