Reverted removal of a line in MPPLanguageDetectorTests.mm

google-ai-edge · Dec 22, 2023 · ba8316a · ba8316a
1 parent a7f3321
commit ba8316a
Showing 1 changed file with 4 additions and 0 deletions.
diff --git 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<MPPLanguagePrediction *> *expectedRuLanguagePredictions =
       @[ [[MPPLanguagePrediction alloc] initWithLanguageCode:@"ru" probability:0.9933616f] ];
 
+  [self assertResultsOfDetectLanguageOfText:ruText
+                                usingLanguageDetector:languageDetector
+       approximatelyEqualsExpectedLanguagePredictions:expectedRuLanguagePredictions];
+
   NSString *zhText = @"分久必合合久必分";
   NSArray<MPPLanguagePrediction *> *expectedZhLanguagePredictions = @[
     [[MPPLanguagePrediction alloc] initWithLanguageCode:@"zh" probability:0.505424f],