Updating media rate hertz

aws-amplify · Dec 23, 2024 · d2ff9ca · d2ff9ca
1 parent 13b1ca2
commit d2ff9ca
Showing 1 changed file with 18 additions and 5 deletions.
diff --git a/AWSTranscribeStreamingTests/AWSTranscribeStreamingSwiftTests.swift b/AWSTranscribeStreamingTests/AWSTranscribeStreamingSwiftTests.swift
@@ -44,18 +44,31 @@ class AWSTranscribeStreamingSwiftTests: XCTestCase {
     }
 
     func testStreamingExampleEnglish() throws {
-        try transcribeAudio(languageCode: .enUS, fileName: "hello_world")
+        try transcribeAudio(
+            languageCode: .enUS,
+            fileName: "hello_world",
+            mediaSampleRateHertz: 8000
+        )
     }
 
     func testStreamingExampleDeutsche() throws {
-        try transcribeAudio(languageCode: .deDE, fileName: "guten_tag")
+        try transcribeAudio(
+            languageCode: .deDE,
+            fileName: "guten_tag"
+        )
     }
 
     func testStreamingExampleEspañol() throws {
-        try transcribeAudio(languageCode: .esES, fileName: "hola_mundo")
+        try transcribeAudio(
+            languageCode: .esES,
+            fileName: "hola_mundo"
+        )
     }
 
-    private func transcribeAudio(languageCode: AWSTranscribeStreamingLanguageCode, fileName: String) throws {
+    private func transcribeAudio(
+        languageCode: AWSTranscribeStreamingLanguageCode,
+        fileName: String,
+        mediaSampleRateHertz: NSNumber = 2400) throws {
         let bundle = Bundle(for: AWSTranscribeStreamingSwiftTests.self)
         guard let audioPath = bundle.path(forResource: fileName, ofType: "wav") else {
             XCTFail("Can't find audio path: \(fileName).wav")
@@ -72,7 +85,7 @@ class AWSTranscribeStreamingSwiftTests: XCTestCase {
 
         request.languageCode = languageCode
         request.mediaEncoding = .pcm
-        request.mediaSampleRateHertz = 24000
+        request.mediaSampleRateHertz = mediaSampleRateHertz
 
         // Set up delegate and its expectations
         let delegate = MockTranscribeStreamingClientDelegate()