def test_load_folder_with_non_audio_stereo(self, shared_datadir):
file_name = (shared_datadir / 'audio').resolve()
audio = AudioBuffer.load_folder(file_name, mono=False)
assert len(audio) == 2
assert audio[0].get_audio().shape == (44100,)
assert audio[0].get_sample_rate() == 44100
assert audio[0].file_name == 'test_sine.wav'
assert audio[1].get_audio().shape == (2,44100)
assert audio[1].get_sample_rate() == 44100
assert audio[1].file_name == 'test_sine_stereo.wav'
def test_load_folder_resample(self, shared_datadir):
file_name = (shared_datadir / 'audio/test_folder').resolve()
audio = AudioBuffer.load_folder(file_name, sample_rate=22050)
assert len(audio) == 3
assert audio[0].get_audio().shape == (22050,)
assert audio[0].get_sample_rate() == 22050
assert audio[0].file_name == 'test_1.wav'
assert audio[1].get_audio().shape == (22050,)
assert audio[1].get_sample_rate() == 22050
assert audio[1].file_name == 'test_2.wav'
assert audio[2].get_audio().shape == (22050,)
assert audio[2].get_sample_rate() == 22050
assert audio[2].file_name == 'test_3.wav'
def test_load_folder_exception(self, shared_datadir):
file_name = (shared_datadir / 'audio/non_existent').resolve()
with pytest.raises(Exception) as exc_info:
audio = AudioBuffer.load_folder(file_name)
assert exc_info.type is ValueError
assert exc_info.value.args[0] == '%s is not a directory' % file_name
