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