diff options
author | 2017-04-11 08:52:06 -0800 | |
---|---|---|
committer | 2017-04-11 10:05:47 -0700 | |
commit | 5c8acccfc9e90d694a8394f5522097bfe87379b2 (patch) | |
tree | 14a599c7ca3e73bcb09cd565a07da1aa87ef20d5 /tensorflow/compiler/xla/tests/map_test.cc | |
parent | 315443bbb01719a9e417826c4a077c47056bc071 (diff) |
Using GMock matchers in XLA tests.
Change: 152823724
Diffstat (limited to 'tensorflow/compiler/xla/tests/map_test.cc')
-rw-r--r-- | tensorflow/compiler/xla/tests/map_test.cc | 8 |
1 files changed, 4 insertions, 4 deletions
diff --git a/tensorflow/compiler/xla/tests/map_test.cc b/tensorflow/compiler/xla/tests/map_test.cc index 2433c5653a..3cfa89e2e7 100644 --- a/tensorflow/compiler/xla/tests/map_test.cc +++ b/tensorflow/compiler/xla/tests/map_test.cc @@ -25,6 +25,7 @@ limitations under the License. #include "tensorflow/compiler/xla/literal_util.h" #include "tensorflow/compiler/xla/shape_util.h" #include "tensorflow/compiler/xla/statusor.h" +#include "tensorflow/compiler/xla/test.h" #include "tensorflow/compiler/xla/test_helpers.h" #include "tensorflow/compiler/xla/tests/client_library_test_base.h" #include "tensorflow/compiler/xla/tests/literal_test_util.h" @@ -33,7 +34,6 @@ limitations under the License. #include "tensorflow/compiler/xla/xla.pb.h" #include "tensorflow/compiler/xla/xla_data.pb.h" #include "tensorflow/core/platform/stream_executor_no_cuda.h" -#include "tensorflow/core/platform/test.h" #include "tensorflow/core/platform/types.h" namespace xla { @@ -529,9 +529,9 @@ TEST_F(MapTest, MapOperantionWithBuildError) { StatusOr<Computation> computation_status = builder.Build(); ASSERT_TRUE(!computation_status.ok()); - EXPECT_MATCH(computation_status.status().ToString(), - testing::HasSubstr("error from: ErrorAdd: binary op with " - "different element types: f32[] and u16[]")); + EXPECT_THAT(computation_status.status().ToString(), + ::testing::HasSubstr("error from: ErrorAdd: binary op with " + "different element types: f32[] and u16[]")); } // MapTest disables inline and algsimp. MapTestWithFullOpt runs all |