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

Updated iOS hand landmarker tests

Browse Source
This commit is contained in:
Prianka Liz Kariat 2023-06-14 15:51:41 +05:30
parent dffca9e3b5
commit 0ae27fad37
1 changed files with 8 additions and 5 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

13
mediapipe/tasks/ios/test/vision/hand_landmarker/MPPHandLandmarkerTests.m
View File

@ -40,6 +40,9 @@ static ResourceFileInfo *const kExpectedPointingUpRotatedLandmarksFile =
static NSString *const kExpectedErrorDomain = @"com.google.mediapipe.tasks"; static NSString *const kExpectedErrorDomain = @"com.google.mediapipe.tasks";
static const float kLandmarksErrorTolerance = 0.03f; static const float kLandmarksErrorTolerance = 0.03f;
static NSString *const kLiveStreamTestsDictHandLandmarkerKey = @"gesture_recognizer";
static NSString *const kLiveStreamTestsDictExpectationKey = @"expectation";
#define AssertEqualErrors(error, expectedError) \ #define AssertEqualErrors(error, expectedError) \
XCTAssertNotNil(error); \ XCTAssertNotNil(error); \
XCTAssertEqualObjects(error.domain, expectedError.domain); \ XCTAssertEqualObjects(error.domain, expectedError.domain); \
@ -57,7 +60,7 @@ static const float kLandmarksErrorTolerance = 0.03f;
XCTAssertTrue(handLandmarkerResult.landmarks.count == 0); \ XCTAssertTrue(handLandmarkerResult.landmarks.count == 0); \
XCTAssertTrue(handLandmarkerResult.worldLandmarks.count == 0); XCTAssertTrue(handLandmarkerResult.worldLandmarks.count == 0);
@interface MPPHandLandmarkerTests : XCTestCase { @interface MPPHandLandmarkerTests : XCTestCase <MPPHandLandmarkerLiveStreamDelegate> {
NSDictionary<NSString *, id> *_liveStreamSucceedsTestDict; NSDictionary<NSString *, id> *_liveStreamSucceedsTestDict;
NSDictionary<NSString *, id> *_outOfOrderTimestampTestDict; NSDictionary<NSString *, id> *_outOfOrderTimestampTestDict;
} }
@ -335,7 +338,7 @@ static const float kLandmarksErrorTolerance = 0.03f;
MPPHandLandmarker *handLandmarker = MPPHandLandmarker *handLandmarker =
[self createHandLandmarkerWithOptionsSucceeds:options]; [self createHandLandmarkerWithOptionsSucceeds:options];
MPPImage *image = [self imageWithFileInfo:kFistImage]; MPPImage *image = [self imageWithFileInfo:kThumbUpImage];
NSError *liveStreamApiCallError; NSError *liveStreamApiCallError;
XCTAssertFalse([handLandmarker detectAsyncInImage:image XCTAssertFalse([handLandmarker detectAsyncInImage:image
@ -375,7 +378,7 @@ static const float kLandmarksErrorTolerance = 0.03f;
MPPHandLandmarker *handLandmarker = MPPHandLandmarker *handLandmarker =
[self createHandLandmarkerWithOptionsSucceeds:options]; [self createHandLandmarkerWithOptionsSucceeds:options];
MPPImage *image = [self imageWithFileInfo:kFistImage]; MPPImage *image = [self imageWithFileInfo:kThumbUpImage];
NSError *liveStreamApiCallError; NSError *liveStreamApiCallError;
XCTAssertFalse([handLandmarker detectAsyncInImage:image XCTAssertFalse([handLandmarker detectAsyncInImage:image
@ -414,7 +417,7 @@ static const float kLandmarksErrorTolerance = 0.03f;
MPPHandLandmarker *handLandmarker = MPPHandLandmarker *handLandmarker =
[self createHandLandmarkerWithOptionsSucceeds:options]; [self createHandLandmarkerWithOptionsSucceeds:options];
MPPImage *image = [self imageWithFileInfo:kFistImage]; MPPImage *image = [self imageWithFileInfo:kThumbUpImage];
NSError *imageApiCallError; NSError *imageApiCallError;
XCTAssertFalse([handLandmarker detectInImage:image error:&imageApiCallError]); XCTAssertFalse([handLandmarker detectInImage:image error:&imageApiCallError]);
@ -549,7 +552,7 @@ static const float kLandmarksErrorTolerance = 0.03f;
} }
- (void)handLandmarker:(MPPHandLandmarker *)handLandmarker - (void)handLandmarker:(MPPHandLandmarker *)handLandmarker
didFinishRecognitionWithResult:(MPPHandLandmarkerResult *)handLandmarkerResult didFinishDetectionWithResult:(MPPHandLandmarkerResult *)handLandmarkerResult
timestampInMilliseconds:(NSInteger)timestampInMilliseconds timestampInMilliseconds:(NSInteger)timestampInMilliseconds
error:(NSError *)error { error:(NSError *)error {
[self assertHandLandmarkerResult:handLandmarkerResult [self assertHandLandmarkerResult:handLandmarkerResult