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