diff --git a/test/torchaudio_unittest/transforms/autograd_test_impl.py b/test/torchaudio_unittest/transforms/autograd_test_impl.py index 7c368445a8..b8fe6c1863 100644 --- a/test/torchaudio_unittest/transforms/autograd_test_impl.py +++ b/test/torchaudio_unittest/transforms/autograd_test_impl.py @@ -137,6 +137,16 @@ def test_amplitude_to_db(self): waveform = get_whitenoise(sample_rate=sample_rate, duration=0.05, n_channels=2) self.assert_grad(transform, [waveform]) + def test_melscale(self): + sample_rate = 8000 + n_fft = 400 + n_mels = n_fft // 2 + 1 + transform = T.MelScale(sample_rate=sample_rate, n_mels=n_mels) + spec = get_spectrogram( + get_whitenoise(sample_rate=sample_rate, duration=0.05, n_channels=2), + n_fft=n_fft, power=1) + self.assert_grad(transform, [spec]) + @parameterized.expand([(1.5, "amplitude"), (2, "power"), (10, "db")]) def test_vol(self, gain, gain_type): sample_rate = 8000