Donate to e Foundation | Murena handsets with /e/OS | Own a part of Murena! Learn more

Commit a67e54d2 authored by Charlie Boutier's avatar Charlie Boutier Committed by Automerger Merge Worker
Browse files

Merge "Revert "Avatar: Add audio frequency count function to verify the... 

Merge "Revert "Avatar: Add audio frequency count function to verify the captured audio in test_music_audio_playback"" am: eec12315 am: 971b95a8 am: 48dcc3bb

Original change: https://android-review.googlesource.com/c/platform/packages/modules/Bluetooth/+/2634089



Change-Id: I982c4c5cbad0caf4abe9d029d51269b98b73dc8e
Signed-off-by: default avatarAutomerger Merge Worker <android-build-automerger-merge-worker@system.gserviceaccount.com>
parents 124ecd91 48dcc3bb

android/pandora/test/asha_test.py

+2 −18
Original line number Diff line number Diff line
@@ -46,7 +46,6 @@ HISYCNID: List[int] = [0x01, 0x02, 0x03, 0x04, 0x5, 0x6, 0x7, 0x8] 
COMPLETE_LOCAL_NAME: str = "Bumble"

AUDIO_SIGNAL_AMPLITUDE = 0.8

AUDIO_SIGNAL_SAMPLING_RATE = 44100

AUDIO_ASHA_SAMPLING_RATE = 16000

SINE_FREQUENCY = 440

SINE_DURATION = 0.1
@@ -237,26 +236,13 @@ class AshaTest(base_test.BaseTestClass): # type: ignore[misc] 
        s16le = (sine * 32767).astype('<i2')



        # Interleaved audio.

        stereo = np.zeros(s16le.size * 2, dtype=s16le.dtype)

        stereo = np.zeros(s16le.size * 2, dtype=sine.dtype)

        stereo[0::2] = s16le



        # Send 4 second of audio.

        for _ in range(0, int(4 / SINE_DURATION)):

            yield PlaybackAudioRequest(connection=connection, data=stereo.tobytes())



    def get_audio_frequency(self, audio_data: ByteString, start_time: int, end_time: int) -> float:

        data = np.frombuffer(bytes(audio_data), dtype=np.int16)

        start_point = int(AUDIO_ASHA_SAMPLING_RATE * start_time / 1000)

        end_point = int(AUDIO_ASHA_SAMPLING_RATE * end_time / 1000)

        length = (end_time - start_time) / 1000

        count = 0

        for i in range(start_point, end_point):

            # Count the cycles in the audio sine wave.

            if data[i] < 0 and data[i + 1] > 0:

                count += 1



        return count/length



    @avatar.parameterized(

        (RANDOM, Ear.LEFT),

        (RANDOM, Ear.RIGHT),
@@ -1176,9 +1162,7 @@ class AshaTest(base_test.BaseTestClass): # type: ignore[misc] 
        )



        assert_not_equal(len(audio_data), 0)

        # Take one second of audio after the first second.

        audio_frequency = self.get_audio_frequency(audio_data=audio_data, start_time=1000, end_time=2000)

        assert_equal(audio_frequency, SINE_FREQUENCY)

        # TODO(duoho): decode audio_data and verify the content





if __name__ == "__main__":