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

Fixed formatting of MPPPoseLandmarkerTests.mm

Browse Source
This commit is contained in:
Prianka Liz Kariat 2023-11-20 22:37:43 +05:30
parent 972e3d81c0
commit 447f9cc452
1 changed files with 11 additions and 10 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

21
mediapipe/tasks/ios/test/vision/pose_landmarker/MPPPoseLandmarkerTests.mm
View File

@ -75,7 +75,8 @@ static NSString *const kLiveStreamTestsDictExpectationKey = @"expectation";
} }
- (void)testDetectWithOptionsSucceeds { - (void)testDetectWithOptionsSucceeds {
MPPPoseLandmarkerOptions *options = [self poseLandmarkerOptionsWithModelFileInfo:kPoseLandmarkerBundleAssetFileInfo]; MPPPoseLandmarkerOptions *options =
[self poseLandmarkerOptionsWithModelFileInfo:kPoseLandmarkerBundleAssetFileInfo];
MPPPoseLandmarker *poseLandmarker = [self createPoseLandmarkerWithOptionsSucceeds:options]; MPPPoseLandmarker *poseLandmarker = [self createPoseLandmarkerWithOptionsSucceeds:options];
[self assertResultsOfDetectInImageWithFileInfo:kPoseImageFileInfo [self assertResultsOfDetectInImageWithFileInfo:kPoseImageFileInfo
@ -85,13 +86,14 @@ static NSString *const kLiveStreamTestsDictExpectationKey = @"expectation";
} }
- (void)testDetectWithEmptyResultsSucceeds { - (void)testDetectWithEmptyResultsSucceeds {
MPPPoseLandmarkerOptions *options = [self poseLandmarkerOptionsWithModelFileInfo:kPoseLandmarkerBundleAssetFileInfo]; MPPPoseLandmarkerOptions *options =
[self poseLandmarkerOptionsWithModelFileInfo:kPoseLandmarkerBundleAssetFileInfo];
MPPPoseLandmarker *poseLandmarker = [self createPoseLandmarkerWithOptionsSucceeds:options]; MPPPoseLandmarker *poseLandmarker = [self createPoseLandmarkerWithOptionsSucceeds:options];
[self assertResultsOfDetectInImageWithFileInfo:kNoPoseImageFileInfo [self
usingPoseLandmarker:poseLandmarker assertResultsOfDetectInImageWithFileInfo:kNoPoseImageFileInfo
approximatelyEqualsPoseLandmarkerResult:[MPPPoseLandmarkerTests usingPoseLandmarker:poseLandmarker
emptyPoseLandmarkerResult]]; approximatelyEqualsPoseLandmarkerResult:[MPPPoseLandmarkerTests emptyPoseLandmarkerResult]];
} }
- (void)testCreatePoseLandmarkerFailsWithDelegateInNonLiveStreamMode { - (void)testCreatePoseLandmarkerFailsWithDelegateInNonLiveStreamMode {
@ -432,7 +434,6 @@ static NSString *const kLiveStreamTestsDictExpectationKey = @"expectation";
// Expects to have the same number of poses detected. // Expects to have the same number of poses detected.
[self assertMultiPoseLandmarks:poseLandmarkerResult.landmarks [self assertMultiPoseLandmarks:poseLandmarkerResult.landmarks
areApproximatelyEqualToExpectedMultiPoseLandmarks:expectedPoseLandmarkerResult.landmarks]; areApproximatelyEqualToExpectedMultiPoseLandmarks:expectedPoseLandmarkerResult.landmarks];
[self assertLandmarksAreVisibleAndPresentInPoseLandmarkerResult:poseLandmarkerResult]; [self assertLandmarksAreVisibleAndPresentInPoseLandmarkerResult:poseLandmarkerResult];
} }
@ -440,8 +441,8 @@ static NSString *const kLiveStreamTestsDictExpectationKey = @"expectation";
- (void)assertMultiPoseLandmarks:(NSArray<NSArray<MPPNormalizedLandmark *> *> *)multiPoseLandmarks - (void)assertMultiPoseLandmarks:(NSArray<NSArray<MPPNormalizedLandmark *> *> *)multiPoseLandmarks
areApproximatelyEqualToExpectedMultiPoseLandmarks: areApproximatelyEqualToExpectedMultiPoseLandmarks:
(NSArray<NSArray<MPPNormalizedLandmark *> *> *)expectedMultiPoseLandmarks { (NSArray<NSArray<MPPNormalizedLandmark *> *> *)expectedMultiPoseLandmarks {
XCTAssertEqual(multiPoseLandmarks.count, expectedMultiPoseLandmarks.count); XCTAssertEqual(multiPoseLandmarks.count, expectedMultiPoseLandmarks.count);
if (multiPoseLandmarks.count == 0) { if (multiPoseLandmarks.count == 0) {
return; return;
} }
@ -475,4 +476,4 @@ static NSString *const kLiveStreamTestsDictExpectationKey = @"expectation";
} }
} }
@end @end