@@ -33,6 +33,7 @@ namespace {
3333
3434using ::tflite::support::StatusOr;
3535using ::tflite::task::JoinPath;
36+ using ::testing::HasSubstr;
3637
3738constexpr char kTestDataDirectory [] =
3839 " /tensorflow_lite_support/cc/test/testdata/task/"
@@ -59,7 +60,7 @@ TEST_F(ImageClassifierFromOptionsTest, FailsWithNullOptionsAndError) {
5960 ASSERT_NE (error, nullptr );
6061 EXPECT_EQ (error->code , kInvalidArgumentError );
6162 EXPECT_NE (error->message , nullptr );
62- EXPECT_NE ( strstr ( error->message , " Expected non null options" ), nullptr );
63+ EXPECT_THAT ( error->message , HasSubstr ( " Expected non null options" ));
6364
6465 TfLiteSupportErrorDelete (error);
6566}
@@ -85,7 +86,7 @@ TEST_F(ImageClassifierFromOptionsTest, FailsWithMissingModelPathAndError) {
8586 ASSERT_NE (error, nullptr );
8687 EXPECT_EQ (error->code , kInvalidArgumentError );
8788 EXPECT_NE (error->message , nullptr );
88- EXPECT_NE ( strstr ( error->message , " `base_options.model_file`" ), nullptr );
89+ EXPECT_THAT ( error->message , HasSubstr ( " `base_options.model_file`" ));
8990
9091 TfLiteSupportErrorDelete (error);
9192}
@@ -146,7 +147,7 @@ TEST_F(ImageClassifierFromOptionsTest,
146147 ASSERT_NE (error, nullptr );
147148 EXPECT_EQ (error->code , kInvalidArgumentError );
148149 EXPECT_NE (error->message , nullptr );
149- EXPECT_NE ( strstr ( error->message , " mutually exclusive options" ), nullptr );
150+ EXPECT_THAT ( error->message , HasSubstr ( " mutually exclusive options" ));
150151
151152 TfLiteSupportErrorDelete (error);
152153}
@@ -169,8 +170,7 @@ TEST(ImageClassifierNullClassifierClassifyTest,
169170 ASSERT_NE (error, nullptr );
170171 EXPECT_EQ (error->code , kInvalidArgumentError );
171172 EXPECT_NE (error->message , nullptr );
172- EXPECT_NE (strstr (error->message , " Expected non null image classifier" ),
173- nullptr );
173+ EXPECT_THAT (error->message , HasSubstr (" Expected non null image classifier" ));
174174
175175 TfLiteSupportErrorDelete (error);
176176}
@@ -237,7 +237,7 @@ TEST_F(ImageClassifierClassifyTest, FailsWithNullFrameBufferAndError) {
237237 ASSERT_NE (error, nullptr );
238238 EXPECT_EQ (error->code , kInvalidArgumentError );
239239 EXPECT_NE (error->message , nullptr );
240- EXPECT_NE ( strstr ( error->message , " Expected non null frame buffer" ), nullptr );
240+ EXPECT_THAT ( error->message , HasSubstr ( " Expected non null frame buffer" ));
241241
242242 TfLiteSupportErrorDelete (error);
243243}
@@ -261,7 +261,7 @@ TEST_F(ImageClassifierClassifyTest, FailsWithNullImageDataAndError) {
261261 ASSERT_NE (error, nullptr );
262262 EXPECT_EQ (error->code , kInvalidArgumentError );
263263 EXPECT_NE (error->message , nullptr );
264- EXPECT_NE ( strstr ( error->message , " Invalid stride information" ), nullptr );
264+ EXPECT_THAT ( error->message , HasSubstr ( " Invalid stride information" ));
265265
266266 TfLiteSupportErrorDelete (error);
267267}
@@ -324,7 +324,7 @@ TEST_F(ImageClassifierClassifyTest, FailsWithRoiOutsideImageBoundsAndError) {
324324 ASSERT_NE (error, nullptr );
325325 EXPECT_EQ (error->code , kInvalidArgumentError );
326326 EXPECT_NE (error->message , nullptr );
327- EXPECT_NE ( strstr ( error->message , " Invalid crop coordinates" ), nullptr );
327+ EXPECT_THAT ( error->message , HasSubstr ( " Invalid crop coordinates" ));
328328
329329 TfLiteSupportErrorDelete (error);
330330}
0 commit comments