diff options
author | 2016-12-22 15:38:30 -0800 | |
---|---|---|
committer | 2016-12-22 15:48:41 -0800 | |
commit | bed8383c27a0a7225e6fc7ff59a2cd6388fb4d09 (patch) | |
tree | b70cfc88f95f318195f8610ffb960e98604348d1 /tensorflow/c/c_api_test.cc | |
parent | 1e5bd8cdd62033d1f7ea928fcbec521bb48bb1f5 (diff) |
Merge changes from github.
Change: 142805270
Diffstat (limited to 'tensorflow/c/c_api_test.cc')
-rw-r--r-- | tensorflow/c/c_api_test.cc | 6 |
1 files changed, 6 insertions, 0 deletions
diff --git a/tensorflow/c/c_api_test.cc b/tensorflow/c/c_api_test.cc index 4b43b74ce6..4ea53ab230 100644 --- a/tensorflow/c/c_api_test.cc +++ b/tensorflow/c/c_api_test.cc @@ -211,6 +211,12 @@ TEST(CAPI, DataTypeEnum) { EXPECT_EQ(TF_UINT16, static_cast<TF_DataType>(tensorflow::DT_UINT16)); EXPECT_EQ(TF_COMPLEX128, static_cast<TF_DataType>(tensorflow::DT_COMPLEX128)); EXPECT_EQ(TF_HALF, static_cast<TF_DataType>(tensorflow::DT_HALF)); + EXPECT_EQ(TF_DataTypeSize(TF_DOUBLE), + tensorflow::DataTypeSize(tensorflow::DT_DOUBLE)); + EXPECT_EQ(TF_DataTypeSize(TF_STRING), + tensorflow::DataTypeSize(tensorflow::DT_STRING)); + // Test with invalid type; should always return 0 as documented + EXPECT_EQ(TF_DataTypeSize(static_cast<TF_DataType>(0)), 0); } TEST(CAPI, StatusEnum) { |