def test_compute_spectrogram(test_wav_file, output):
os.makedirs(os.path.dirname(output), exist_ok=True)
wav = tp.load_wav(test_wav_file)
ops.compute_spectrogram(wav.audio, output)
assert os.path.exists(output)
os.remove(output)
os.removedirs(os.path.dirname(output))
def test_should_skip_already_computed_spectrogram(test_wav_file, output):
os.makedirs(os.path.dirname(output), exist_ok=True)
wav = tp.load_wav(test_wav_file)
ops.compute_spectrogram(wav.audio, output)
created_time = os.path.getmtime(output)
ops.compute_spectrogram(wav.audio, output)
assert os.path.getmtime(output) == created_time
os.remove(output)
os.removedirs(os.path.dirname(output))
def test_should_create_output_parent_dir(test_wav_file, output):
wav = tp.load_wav(test_wav_file)
ops.compute_spectrogram(wav.audio, output)
assert os.path.exists(os.path.dirname(output))
os.remove(output)
os.removedirs(os.path.dirname(output))
def test_load_non_existing_wav(non_existing_test_wav_file):
with pytest.raises(FileNotFoundError):
tp.load_wav(non_existing_test_wav_file)
def test_load_wav(test_wav_file):
wav = tp.load_wav(test_wav_file)
assert wav.sample_rate.numpy() == 48000
assert len(wav.audio) >= 48000
def waveform(self):
if not os.path.exists(self.wav):
ops.extract_audio(self.raw, self.wav)
return tp.load_wav(self.wav)
def test_spectrogram(test_wav_file):
wav = tp.load_wav(test_wav_file)
spc = tp.spectrogram(wav.audio, 48000, 0.01, 0.5)
assert spc.shape == (257, 199)