Skip to content

Commit f0e89ff

Browse files
committed
changed strstr check to HasSubstr
1 parent fa4a879 commit f0e89ff

File tree

1 file changed

+8
-8
lines changed

1 file changed

+8
-8
lines changed

tensorflow_lite_support/c/test/task/vision/image_classifier_test.cc

Lines changed: 8 additions & 8 deletions
Original file line numberDiff line numberDiff line change
@@ -33,6 +33,7 @@ namespace {
3333

3434
using ::tflite::support::StatusOr;
3535
using ::tflite::task::JoinPath;
36+
using ::testing::HasSubstr;
3637

3738
constexpr 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

Comments
 (0)