diff options
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) { |