# Copyright © 2020 Arm Ltd and Contributors. All rights reserved. | |
# SPDX-License-Identifier: MIT | |
import os | |
import numpy as np | |
from context import audio_capture | |
def test_audio_file(test_data_folder): | |
audio_file = os.path.join(test_data_folder, "myVoiceIsMyPassportVerifyMe04.wav") | |
capture = audio_capture.AudioCapture(audio_capture.ModelParams("")) | |
buffer = capture.from_audio_file(audio_file) | |
audio_data = next(buffer) | |
assert audio_data.shape == (47712,) | |
assert audio_data.dtype == np.float32 |