| # 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 |