Merge pull request #5382 from priankakariatyml:ios-audio-data-test

PiperOrigin-RevId: 631205189
google-ai-edge · May 6, 2024 · fff374d · fff374d
2 parents fccb895 + 9db5fcf
commit fff374d
Show file tree

Hide file tree

Showing 3 changed files with 478 additions and 1 deletion.
diff --git a/mediapipe/tasks/ios/test/audio/core/BUILD b/mediapipe/tasks/ios/test/audio/core/BUILD
@@ -92,3 +92,34 @@ ios_unit_test(
         ":MPPAudioRecordObjcTestLibrary",
     ],
 )
+
+objc_library(
+    name = "MPPAudioDataObjcTestLibrary",
+    testonly = 1,
+    srcs = ["MPPAudioDataTests.mm"],
+    copts = [
+        "-ObjC++",
+        "-std=c++17",
+        "-x objective-c++",
+    ],
+    data = [
+        "//mediapipe/tasks/testdata/audio:test_audio_clips",
+    ],
+    tags = TFL_DEFAULT_TAGS,
+    deps = [
+        "//mediapipe/tasks/ios/audio/core:MPPAudioData",
+        "//mediapipe/tasks/ios/common:MPPCommon",
+        "//mediapipe/tasks/ios/test/audio/core/utils:AVAudioPCMBufferTestUtils",
+        "//mediapipe/tasks/ios/test/utils:MPPFileInfo",
+    ],
+)
+
+ios_unit_test(
+    name = "MPPAudioDataObjcTest",
+    minimum_os_version = MPP_TASK_MINIMUM_OS_VERSION,
+    runner = tflite_ios_lab_runner("IOS_LATEST"),
+    tags = TFL_DEFAULT_TAGS + TFL_DISABLED_SANITIZER_TAGS,
+    deps = [
+        ":MPPAudioDataObjcTestLibrary",
+    ],
+)