parameterize power

pytorch · Nov 6, 2020 · a305fb8 · a305fb8
1 parent 627e8a5
commit a305fb8
Showing 1 changed file with 8 additions and 2 deletions.
diff --git a/test/torchaudio_unittest/functional/functional_impl.py b/test/torchaudio_unittest/functional/functional_impl.py
@@ -33,7 +33,13 @@ def test_clamp(self):
 
 
 class Spectrogram(common_utils.TestBaseMixin):
-    def test_grad_at_zero(self):
+    @parameterized.expand([
+        (0, ),
+        (1, ),
+        (2, ),
+        (3, ),
+    ])
+    def test_grad_at_zero(self, power):
         """The gradient of power spectrogram should not be non but zero near x=0
 
         https://github.com/pytorch/audio/issues/993
@@ -46,7 +52,7 @@ def test_grad_at_zero(self):
             n_fft=2048,
             hop_length=None,
             win_length=None,
-            power=1,
+            power=power,
             normalized=False,
         )
         spec.sum().backward()