def test_get_audio_paths(): dataset_path = './tests/data' dataset = Dataset(dataset_path) audio_paths = dataset.get_audio_paths() assert audio_paths[0] == './tests/data/audio' assert audio_paths[1] == ['./tests/data/audio/original'] audio_paths = dataset.get_audio_paths(22050) assert audio_paths[0] == './tests/data/audio22050' assert audio_paths[1] == ['./tests/data/audio22050/original']
def test_get_audio_paths(): dataset_path = "./tests/data" dataset = Dataset(dataset_path) audio_paths = dataset.get_audio_paths() assert audio_paths[0] == "./tests/data/audio" assert audio_paths[1] == ["./tests/data/audio/original"] audio_paths = dataset.get_audio_paths(22050) assert audio_paths[0] == "./tests/data/audio22050" assert audio_paths[1] == ["./tests/data/audio22050/original"]
def test_check_sampling_rate(): dataset_path = './tests/data' dataset = Dataset(dataset_path) sr = 22050 audio_path_sr = dataset.get_audio_paths(sr)[1][0] _clean(audio_path_sr) dataset.change_sampling_rate(sr) assert dataset.check_sampling_rate(sr)
def test_change_sampling_rate(sr): dataset_path = './tests/data' dataset = Dataset(dataset_path) audio_path = dataset.get_audio_paths()[0] audio_path_sr = dataset.get_audio_paths(sr)[1][0] _clean(audio_path_sr) dataset.change_sampling_rate(sr) for file_audio in audio_files: file_path = os.path.join(audio_path_sr, file_audio) file_data, file_sr = sf.read(file_path) length_seconds = len(file_data) / float(file_sr) file_path_original = os.path.join(audio_path, file_audio) file_data_original, file_sr_original = sf.read(file_path_original) length_seconds_original = len(file_data_original) / \ float(file_sr_original) assert np.allclose(length_seconds, length_seconds_original, rtol=0.0001, atol=0.0001) assert dataset.check_sampling_rate(sr)
def test_convert_to_wav(): dataset_path = './tests/data_aiff' dataset = Dataset(dataset_path) audio_path = dataset.get_audio_paths()[0] aiff_files = ['40722-8-0-7.aiff', '147764-4-7-0.aiff', '176787-5-0-0.aiff'] for wavfile in audio_files: wavpath = os.path.join(audio_path, wavfile) _clean(wavpath) dataset.convert_to_wav() for wavfile, aifffile in zip(audio_files, aiff_files): wavpath = os.path.join(audio_path, wavfile) data, sr = sf.read(wavpath) wavpath_orig = os.path.join('./tests/data/audio', wavfile) data_orig, sr_orig = sf.read(wavpath_orig) assert sr_orig == sr assert np.allclose(data_orig, data, rtol=0.0001, atol=0.0001) _clean(wavpath)