From d9d40163342c1721ecfdde79e9ce36168dfaa921 Mon Sep 17 00:00:00 2001
From: Prianka Liz Kariat <prianka.kariat@codeandtheory.com>
Date: Tue, 19 Sep 2023 20:06:25 +0530
Subject: [PATCH] Uncommented live stream test in iOS image segmenter tests

---
 .../image_segmenter/MPPImageSegmenterTests.mm | 61 +++++++++----------
 1 file changed, 29 insertions(+), 32 deletions(-)

diff --git a/mediapipe/tasks/ios/test/vision/image_segmenter/MPPImageSegmenterTests.mm b/mediapipe/tasks/ios/test/vision/image_segmenter/MPPImageSegmenterTests.mm
index 739871ff..e400d66a 100644
--- a/mediapipe/tasks/ios/test/vision/image_segmenter/MPPImageSegmenterTests.mm
+++ b/mediapipe/tasks/ios/test/vision/image_segmenter/MPPImageSegmenterTests.mm
@@ -368,47 +368,44 @@ double softIOU(const float *mask1, const float *mask2, size_t size) {
   }
 }
 
-// - (void)testSegmentWithOutOfOrderTimestampsAndLiveStreamModeFails {
-//  MPPImageSegmenterOptions *options =
-//       [self imageSegmenterOptionsWithModelFileInfo:kImageSegmenterModelFileInfo];
-//   options.runningMode = MPPRunningModeLiveStream;
-//   options.imageSegmenterLiveStreamDelegate = self;
+- (void)testSegmentWithOutOfOrderTimestampsAndLiveStreamModeFails {
+  MPPImageSegmenterOptions *options =
+      [self imageSegmenterOptionsWithModelFileInfo:kImageSegmenterModelFileInfo];
+  options.runningMode = MPPRunningModeLiveStream;
+  options.imageSegmenterLiveStreamDelegate = self;
 
-//   XCTestExpectation *expectation = [[XCTestExpectation alloc]
-//       initWithDescription:@"segmentWithOutOfOrderTimestampsAndLiveStream"];
+  XCTestExpectation *expectation = [[XCTestExpectation alloc]
+      initWithDescription:@"segmentWithOutOfOrderTimestampsAndLiveStream"];
 
-//   expectation.expectedFulfillmentCount = 1;
+  expectation.expectedFulfillmentCount = 1;
 
-//   MPPImageSegmenter *imageSegmenter = [self createImageSegmenterWithOptionsSucceeds:options];
+  MPPImageSegmenter *imageSegmenter = [self createImageSegmenterWithOptionsSucceeds:options];
 
-//   _outOfOrderTimestampTestDict = @{
-//     kLiveStreamTestsDictImageSegmenterKey : imageSegmenter,
-//     kLiveStreamTestsDictExpectationKey : expectation
-//   };
+  _outOfOrderTimestampTestDict = @{
+    kLiveStreamTestsDictImageSegmenterKey : imageSegmenter,
+    kLiveStreamTestsDictExpectationKey : expectation
+  };
 
-//   MPPImage *image = [MPPImage imageWithFileInfo:kCatImageFileInfo];
-//   XCTAssertNotNil(image);
+  MPPImage *image = [MPPImage imageWithFileInfo:kCatImageFileInfo];
+  XCTAssertNotNil(image);
 
-//   XCTAssertTrue([imageSegmenter segmentAsyncImage:image timestampInMilliseconds:1 error:nil]);
+  XCTAssertTrue([imageSegmenter segmentAsyncImage:image timestampInMilliseconds:1 error:nil]);
 
-//   NSError *error;
-//   XCTAssertFalse([imageSegmenter segmentAsyncImage:image
-//                                 timestampInMilliseconds:0
-//                                                   error:&error]);
+  NSError *error;
+  XCTAssertFalse([imageSegmenter segmentAsyncImage:image timestampInMilliseconds:0 error:&error]);
 
-//   NSError *expectedError =
-//       [NSError errorWithDomain:kExpectedErrorDomain
-//                           code:MPPTasksErrorCodeInvalidArgumentError
-//                       userInfo:@{
-//                         NSLocalizedDescriptionKey :
-//                             @"INVALID_ARGUMENT: Input timestamp must be monotonically
-//                             increasing."
-//                       }];
-//   AssertEqualErrors(error, expectedError);
+  NSError *expectedError =
+      [NSError errorWithDomain:kExpectedErrorDomain
+                          code:MPPTasksErrorCodeInvalidArgumentError
+                      userInfo:@{
+                        NSLocalizedDescriptionKey :
+                            @"INVALID_ARGUMENT: Input timestamp must be monotonically increasing."
+                      }];
+  AssertEqualErrors(error, expectedError);
 
-//   NSTimeInterval timeout = 0.5f;
-//   [self waitForExpectations:@[ expectation ] timeout:timeout];
-// }
+  NSTimeInterval timeout = 0.5f;
+  [self waitForExpectations:@[ expectation ] timeout:timeout];
+}
 
 - (void)testSegmentWithLiveStreamModeSucceeds {
   MPPImageSegmenterOptions *options =