test: update test cases

tests/lambdas/test_lambdaSpeechToScore.py

@@ -1,5 +1,6 @@
 import json
 import os
+from pathlib import Path
 import platform
 import unittest
 
@@ -95,6 +96,10 @@ def check_output(self, output, expected_output, check_audio_files=False):
         output["real_transcripts_ipa"] = expected_output["real_transcripts_ipa"]
         if check_audio_files:
             audio_files = output["audio_files"]
+            audio_durations = output["audio_durations"]
+            for audio_duration in audio_durations:
+                assert isinstance(audio_duration, float)
+                assert audio_duration > 0
             for audio_file in audio_files:
                 path_audio_file = Path(audio_file)
                 app_logger.info(f"path_audio_file:{path_audio_file}.")
@@ -103,6 +108,10 @@ def check_output(self, output, expected_output, check_audio_files=False):
         
             output["audio_files"] = [*audio_files]
             expected_output["audio_files"] = [*audio_files]
+            n_durations = len(output["audio_durations"])
+            audio_durations = [x/4 + 0.5 for x in range(n_durations)]
+            output["audio_durations"] = [*audio_durations]
+            expected_output["audio_durations"] = [*audio_durations]
             app_logger.info(f"output audio_files:{output['audio_files']}.")
             app_logger.info(f"expected_output audio_files:{expected_output['audio_files']}.")
         self.assertDictEqual(output, expected_output)
@@ -210,6 +219,7 @@ class TestGetAccuracyFromRecordedAudio(unittest.TestCase):
             ipa_transcript,
             real_transcripts_ipa,
             num_words,
+            first_audio_file,
             dumped,
         ) = lambdaSpeechToScore.get_speech_to_score_tuple(
             real_text=text_dict[language],
@@ -224,7 +234,10 @@ class TestGetAccuracyFromRecordedAudio(unittest.TestCase):
         assert len(ipa_transcript.strip()) > 0
         assert len(real_transcripts_ipa.strip()) > 0
         assert num_words == 6
-        check_output(self, json.loads(dumped), expected_output[language], check_audio_files=True)
+        first_audio_file_path = Path(first_audio_file)
+        assert first_audio_file_path.exists() and first_audio_file_path.is_file()
+        json_loaded = json.loads(dumped)
+        check_output(self, json_loaded, expected_output[language], check_audio_files=True)
 
     def test_get_speech_to_score_tuple_en_ok(self):
         from aip_trainer.lambdas import lambdaSpeechToScore
@@ -238,6 +251,7 @@ class TestGetAccuracyFromRecordedAudio(unittest.TestCase):
             ipa_transcript,
             real_transcripts_ipa,
             num_words,
+            first_audio_file,
             dumped,
         ) = lambdaSpeechToScore.get_speech_to_score_tuple(
             real_text=text_dict[language],
@@ -252,6 +266,8 @@ class TestGetAccuracyFromRecordedAudio(unittest.TestCase):
         assert len(ipa_transcript.strip()) > 0
         assert len(real_transcripts_ipa.strip()) > 0
         assert num_words == 5
+        first_audio_file_path = Path(first_audio_file)
+        assert first_audio_file_path.exists() and first_audio_file_path.is_file()
         check_output(self, json.loads(dumped), expected_output[language], check_audio_files=True)
 
     def test_get_speech_to_score_dict__de_empty_input_text(self):