cassdlcm/mediapipe
1
0
Fork 0
Code Issues Pull Requests Packages Projects Releases Wiki Activity

Updated MPPImageClassifierTests.m to reflect new delegate name

Browse Source
This commit is contained in:
Prianka Liz Kariat 2023-05-04 16:17:37 +05:30
parent 244dcb727e
commit d6db8c7268
1 changed files with 5 additions and 5 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

10
mediapipe/tasks/ios/test/vision/image_classifier/MPPImageClassifierTests.m
View File

@ -56,7 +56,7 @@ static NSString *const kLiveStreamTestsDictExpectationKey = @"expectation";
XCTAssertEqual(imageClassifierResult.classificationResult.classifications.count, 1); \
XCTAssertEqual(imageClassifierResult.classificationResult.classifications[0].headIndex, 0);
@interface MPPImageClassifierTests : XCTestCase <MPPImageClassifierDelegate> {
@interface MPPImageClassifierTests : XCTestCase <MPPImageClassifierLiveStreamDelegate> {
NSDictionary *liveStreamSucceedsTestDict;
NSDictionary *outOfOrderTimestampTestDict;
}
@ -447,7 +447,7 @@ static NSString *const kLiveStreamTestsDictExpectationKey = @"expectation";
MPPImageClassifierOptions *options = [self imageClassifierOptionsWithModelName:kFloatModelName];
options.runningMode = runningModesToTest[i];
options.imageClassifierDelegate = self;
options.imageClassifierLiveStreamDelegate = self;
[self
assertCreateImageClassifierWithOptions:options
@ -558,7 +558,7 @@ static NSString *const kLiveStreamTestsDictExpectationKey = @"expectation";
MPPImageClassifierOptions *options = [self imageClassifierOptionsWithModelName:kFloatModelName];
options.runningMode = MPPRunningModeLiveStream;
options.imageClassifierDelegate = self;
options.imageClassifierLiveStreamDelegate = self;
MPPImageClassifier *imageClassifier = [self imageClassifierWithOptionsSucceeds:options];
@ -622,7 +622,7 @@ static NSString *const kLiveStreamTestsDictExpectationKey = @"expectation";
options.maxResults = 3;
options.runningMode = MPPRunningModeLiveStream;
options.imageClassifierDelegate = self;
options.imageClassifierLiveStreamDelegate = self;
XCTestExpectation *expectation = [[XCTestExpectation alloc]
initWithDescription:@"classifyWithOutOfOrderTimestampsAndLiveStream"];
@ -662,7 +662,7 @@ static NSString *const kLiveStreamTestsDictExpectationKey = @"expectation";
options.maxResults = maxResults;
options.runningMode = MPPRunningModeLiveStream;
options.imageClassifierDelegate = self;
options.imageClassifierLiveStreamDelegate = self;
NSInteger iterationCount = 100;