def test_riaa(self): """ Test biquad riaa filter, compare to SoX implementation """ noise_filepath = common_utils.get_asset_path('whitenoise.wav') E = torchaudio.sox_effects.SoxEffectsChain() E.set_input_file(noise_filepath) E.append_effect_to_chain("riaa") sox_output_waveform, sr = E.sox_build_flow_effects() waveform, sample_rate = torchaudio.load(noise_filepath, normalization=True) output_waveform = F.riaa_biquad(waveform, sample_rate) self.assertEqual(output_waveform, sox_output_waveform, atol=1e-4, rtol=1e-5)
def test_riaa(self): """ Test biquad riaa filter, compare to SoX implementation """ noise_filepath = os.path.join(self.test_dirpath, "assets", "whitenoise.wav") E = torchaudio.sox_effects.SoxEffectsChain() E.set_input_file(noise_filepath) E.append_effect_to_chain("riaa") sox_output_waveform, sr = E.sox_build_flow_effects() waveform, sample_rate = torchaudio.load(noise_filepath, normalization=True) output_waveform = F.riaa_biquad(waveform, sample_rate) assert torch.allclose(sox_output_waveform, output_waveform, atol=1e-4) _test_torchscript_functional(F.riaa_biquad, waveform, sample_rate)
def test_riaa(self): """ Test biquad riaa filter, compare to SoX implementation """ E = torchaudio.sox_effects.SoxEffectsChain() E.set_input_file(self.noise_filepath) E.append_effect_to_chain("riaa") sox_output_waveform, sr = E.sox_build_flow_effects() output_waveform = F.riaa_biquad(self.noise_waveform, self.NOISE_SAMPLE_RATE) self.assertEqual(output_waveform, sox_output_waveform, atol=1e-4, rtol=1e-5)
def func(tensor): sample_rate = 44100 return F.riaa_biquad(tensor, sample_rate)
def test_riaa(self): sample_rate = 44100 data, path = self.get_whitenoise(sample_rate) result = F.riaa_biquad(data, sample_rate) self.assert_sox_effect(result, path, ['riaa'])