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