commit bde18b71012189982d43ff91d69b4491480038fb
author Jean-Baptiste Lespiau <jblespiau@google.com> Fri Jun 03 11:15:50 2022 -0700
committer TensorFlower Gardener <gardener@tensorflow.org> Fri Jun 03 11:24:42 2022 -0700
tree 41321831dcef57e8ab980733f6dc9629d9b8d26a
parent c7d82d7a4de4b3989766f01142b20b3a61ab301b

Remove usages of errors::Code::NAME as it's going away.

PiperOrigin-RevId: 452804370

tensorflow/core/grappler/clusters/single_machine_test.cc

diff --git a/tensorflow/core/grappler/clusters/single_machine_test.cc b/tensorflow/core/grappler/clusters/single_machine_test.cc
index e914573..f3eab4b 100644
--- a/tensorflow/core/grappler/clusters/single_machine_test.cc
+++ b/tensorflow/core/grappler/clusters/single_machine_test.cc
@@ -629,7 +629,7 @@
   Status s = cluster.GetPeakMemoryUsage(&device_peak_memory);
   TF_CHECK_OK(cluster.Shutdown());
   ASSERT_FALSE(s.ok());
-  EXPECT_EQ(s.code(), errors::Code::INVALID_ARGUMENT);
+  EXPECT_TRUE(errors::IsInvalidArgument(s));
 }
 #endif
 

tensorflow/core/grappler/optimizers/meta_optimizer_test.cc

diff --git a/tensorflow/core/grappler/optimizers/meta_optimizer_test.cc b/tensorflow/core/grappler/optimizers/meta_optimizer_test.cc
index a63ba01..2d98bae 100644
--- a/tensorflow/core/grappler/optimizers/meta_optimizer_test.cc
+++ b/tensorflow/core/grappler/optimizers/meta_optimizer_test.cc
@@ -896,7 +896,7 @@
   MetaOptimizer optimizer_with_post_verifiers(nullptr, config_proto);
   Status status =
       optimizer_with_post_verifiers.Optimize(nullptr, item, &output);
-  EXPECT_EQ(status.code(), errors::Code::INVALID_ARGUMENT);
+  EXPECT_TRUE(errors::IsInvalidArgument(status));
   EXPECT_TRUE(absl::StrContains(
       status.error_message(),
       "NodeDef expected inputs 'float' do not match 3 inputs specified"));

tensorflow/core/grappler/verifiers/structure_verifier_test.cc

diff --git a/tensorflow/core/grappler/verifiers/structure_verifier_test.cc b/tensorflow/core/grappler/verifiers/structure_verifier_test.cc
index 763a088..667506b 100644
--- a/tensorflow/core/grappler/verifiers/structure_verifier_test.cc
+++ b/tensorflow/core/grappler/verifiers/structure_verifier_test.cc
@@ -24,6 +24,7 @@
 #include "tensorflow/core/framework/shape_inference.h"
 #include "tensorflow/core/lib/core/errors.h"
 #include "tensorflow/core/lib/core/status_test_util.h"
+#include "tensorflow/core/platform/errors.h"
 #include "tensorflow/core/platform/protobuf.h"
 #include "tensorflow/core/platform/test.h"
 
@@ -84,7 +85,7 @@
       "node { name: 't1' op: 'TestMul' input: [ 'input:0', 't2' ] }"
       "node { name: 't2' op: 'TestMul' input: [ 'input:1', 't1' ] }");
   Status status = verifier_->Verify(graph_);
-  EXPECT_EQ(status.code(), errors::Code::NOT_FOUND);
+  EXPECT_TRUE(errors::IsNotFound(status));
   EXPECT_TRUE(
       absl::StrContains(status.error_message(), "Op type not registered"));
 }
@@ -94,7 +95,7 @@
       "node { name: 'A' op: 'TestParams' }"
       "node { name: 'A' op: 'TestInput' }");
   Status status = verifier_->Verify(graph_);
-  EXPECT_EQ(status.code(), errors::Code::ALREADY_EXISTS);
+  EXPECT_TRUE(errors::IsAlreadyExists(status));
   EXPECT_TRUE(
       absl::StrContains(status.error_message(), "Node already exists:"));
 }
@@ -105,7 +106,7 @@
       "node { name: 't1' op: 'TestMul' input: [ 'input:0', 't2' ] }"
       "node { name: 't2' op: 'TestMul' input: [ 'input:1', 't1' ] }");
   Status status = verifier_->Verify(graph_);
-  EXPECT_EQ(status.code(), errors::Code::INVALID_ARGUMENT);
+  EXPECT_TRUE(errors::IsInvalidArgument(status));
   EXPECT_TRUE(
       absl::StrContains(status.error_message(),
                         "The graph couldn't be sorted in topological order"));