diff options
Diffstat (limited to 'tensorflow/core/framework/tensor_test.cc')
-rw-r--r-- | tensorflow/core/framework/tensor_test.cc | 16 |
1 files changed, 10 insertions, 6 deletions
diff --git a/tensorflow/core/framework/tensor_test.cc b/tensorflow/core/framework/tensor_test.cc index 4d8d378b84..3c3b457e74 100644 --- a/tensorflow/core/framework/tensor_test.cc +++ b/tensorflow/core/framework/tensor_test.cc @@ -834,20 +834,24 @@ TEST(SummarizeValue, INT32) { Tensor x = MkTensor<int>(DT_INT32, TensorShape({5}), {1, 2, 3, 4, 0}); EXPECT_EQ("1 2 3 4 0", x.SummarizeValue(16)); x = MkTensor<int>(DT_INT32, TensorShape({2, 2}), {1, 2, 3, 4, 0}); - EXPECT_EQ("1 2 3 4", x.SummarizeValue(16)); + EXPECT_EQ("[1 2][3 4]", x.SummarizeValue(16)); x = MkTensor<int>(DT_INT32, TensorShape({2, 2, 1, 1}), {1, 2, 3, 4, 0}); - EXPECT_EQ("1 2 3 4", x.SummarizeValue(16)); - EXPECT_EQ("1 2 3...", x.SummarizeValue(3)); + EXPECT_EQ("[[[1]][[2]]][[[3]][[4]]]", x.SummarizeValue(16)); + EXPECT_EQ("[[[1]][[2]]][[[3]]]...", x.SummarizeValue(3)); + x = MkTensor<int>(DT_INT32, TensorShape({0}), {}); + EXPECT_EQ("", x.SummarizeValue(16)); } TEST(SummarizeValue, FLOAT) { Tensor x = MkTensor<float>(DT_FLOAT, TensorShape({5}), {1, 2, 3, 4, 0}); EXPECT_EQ("1 2 3 4 0", x.SummarizeValue(16)); x = MkTensor<float>(DT_FLOAT, TensorShape({2, 2}), {1, 2, 3, 4, 0}); - EXPECT_EQ("1 2 3 4", x.SummarizeValue(16)); + EXPECT_EQ("[1 2][3 4]", x.SummarizeValue(16)); x = MkTensor<float>(DT_FLOAT, TensorShape({2, 2, 1, 1}), {1, 2, 3, 4, 0}); - EXPECT_EQ("1 2 3 4", x.SummarizeValue(16)); - EXPECT_EQ("1 2 3...", x.SummarizeValue(3)); + EXPECT_EQ("[[[1]][[2]]][[[3]][[4]]]", x.SummarizeValue(16)); + EXPECT_EQ("[[[1]][[2]]][[[3]]]...", x.SummarizeValue(3)); + x = MkTensor<float>(DT_FLOAT, TensorShape({0}), {}); + EXPECT_EQ("", x.SummarizeValue(16)); } TEST(SummarizeValue, BOOL) { |