diff --git a/mediapipe/tasks/ios/test/text/language_detector/MPPLanguageDetectorTests.mm b/mediapipe/tasks/ios/test/text/language_detector/MPPLanguageDetectorTests.mm index f1186e68d0..ad0b0eb92a 100644 --- a/mediapipe/tasks/ios/test/text/language_detector/MPPLanguageDetectorTests.mm +++ b/mediapipe/tasks/ios/test/text/language_detector/MPPLanguageDetectorTests.mm @@ -113,6 +113,10 @@ - (void)testClassifyWithL2CModelSucceeds { NSArray *expectedRuLanguagePredictions = @[ [[MPPLanguagePrediction alloc] initWithLanguageCode:@"ru" probability:0.9933616f] ]; + [self assertResultsOfDetectLanguageOfText:ruText + usingLanguageDetector:languageDetector + approximatelyEqualsExpectedLanguagePredictions:expectedRuLanguagePredictions]; + NSString *zhText = @"分久必合合久必分"; NSArray *expectedZhLanguagePredictions = @[ [[MPPLanguagePrediction alloc] initWithLanguageCode:@"zh" probability:0.505424f],