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

Added FreeMemory test for GestureRecognizerResult

Browse Source
This commit is contained in:
Kinar 2023-11-29 03:17:35 -08:00
parent 6ed5e3d0df
commit d19d5a50be
1 changed files with 34 additions and 8 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

42
mediapipe/tasks/c/components/containers/gesture_recognizer_result_converter_test.cc
View File

@ -21,10 +21,9 @@ limitations under the License.
namespace mediapipe::tasks::c::components::containers { namespace mediapipe::tasks::c::components::containers {
TEST(GestureRecognizerResultConverterTest, ConvertsCustomResult) { void InitGestureRecognizerResult(
::mediapipe::tasks::vision::gesture_recognizer::GestureRecognizerResult ::mediapipe::tasks::vision::gesture_recognizer::GestureRecognizerResult*
cpp_result; cpp_result) {
// Initialize gestures // Initialize gestures
Classification classification_for_gestures; Classification classification_for_gestures;
classification_for_gestures.set_index(0); classification_for_gestures.set_index(0);
@ -33,7 +32,7 @@ TEST(GestureRecognizerResultConverterTest, ConvertsCustomResult) {
classification_for_gestures.set_display_name("gesture_display_name_1"); classification_for_gestures.set_display_name("gesture_display_name_1");
ClassificationList gestures_list; ClassificationList gestures_list;
*gestures_list.add_classification() = classification_for_gestures; *gestures_list.add_classification() = classification_for_gestures;
cpp_result.gestures.push_back(gestures_list); cpp_result->gestures.push_back(gestures_list);
// Initialize handedness // Initialize handedness
Classification classification_for_handedness; Classification classification_for_handedness;
@ -43,7 +42,7 @@ TEST(GestureRecognizerResultConverterTest, ConvertsCustomResult) {
classification_for_handedness.set_display_name("handeness_display_name_1"); classification_for_handedness.set_display_name("handeness_display_name_1");
ClassificationList handedness_list; ClassificationList handedness_list;
*handedness_list.add_classification() = classification_for_handedness; *handedness_list.add_classification() = classification_for_handedness;
cpp_result.handedness.push_back(handedness_list); cpp_result->handedness.push_back(handedness_list);
// Initialize hand_landmarks // Initialize hand_landmarks
NormalizedLandmark normalized_landmark; NormalizedLandmark normalized_landmark;
@ -52,7 +51,7 @@ TEST(GestureRecognizerResultConverterTest, ConvertsCustomResult) {
normalized_landmark.set_z(0.3f); normalized_landmark.set_z(0.3f);
NormalizedLandmarkList normalized_landmark_list; NormalizedLandmarkList normalized_landmark_list;
*normalized_landmark_list.add_landmark() = normalized_landmark; *normalized_landmark_list.add_landmark() = normalized_landmark;
cpp_result.hand_landmarks.push_back(normalized_landmark_list); cpp_result->hand_landmarks.push_back(normalized_landmark_list);
// Initialize hand_world_landmarks // Initialize hand_world_landmarks
Landmark landmark; Landmark landmark;
@ -62,7 +61,13 @@ TEST(GestureRecognizerResultConverterTest, ConvertsCustomResult) {
LandmarkList landmark_list; LandmarkList landmark_list;
*landmark_list.add_landmark() = landmark; *landmark_list.add_landmark() = landmark;
cpp_result.hand_world_landmarks.push_back(landmark_list); cpp_result->hand_world_landmarks.push_back(landmark_list);
}
TEST(GestureRecognizerResultConverterTest, ConvertsCustomResult) {
::mediapipe::tasks::vision::gesture_recognizer::GestureRecognizerResult
cpp_result;
InitGestureRecognizerResult(&cpp_result);
GestureRecognizerResult c_result; GestureRecognizerResult c_result;
CppConvertToGestureRecognizerResult(cpp_result, &c_result); CppConvertToGestureRecognizerResult(cpp_result, &c_result);
@ -119,4 +124,25 @@ TEST(GestureRecognizerResultConverterTest, ConvertsCustomResult) {
CppCloseGestureRecognizerResult(&c_result); CppCloseGestureRecognizerResult(&c_result);
} }
TEST(GestureRecognizerResultConverterTest, FreesMemory) {
::mediapipe::tasks::vision::gesture_recognizer::GestureRecognizerResult
cpp_result;
InitGestureRecognizerResult(&cpp_result);
GestureRecognizerResult c_result;
CppConvertToGestureRecognizerResult(cpp_result, &c_result);
EXPECT_NE(c_result.gestures, nullptr);
EXPECT_NE(c_result.handedness, nullptr);
EXPECT_NE(c_result.hand_landmarks, nullptr);
EXPECT_NE(c_result.hand_world_landmarks, nullptr);
CppCloseGestureRecognizerResult(&c_result);
EXPECT_EQ(c_result.gestures, nullptr);
EXPECT_EQ(c_result.handedness, nullptr);
EXPECT_EQ(c_result.hand_landmarks, nullptr);
EXPECT_EQ(c_result.hand_world_landmarks, nullptr);
}
} // namespace mediapipe::tasks::c::components::containers } // namespace mediapipe::tasks::c::components::containers