fixup! Add test for pytorch#1021

mthrok · Nov 13, 2020 · 840f0ce · 840f0ce
1 parent 6808cfa
commit 840f0ce
Showing 1 changed file with 3 additions and 3 deletions.
diff --git a/test/torchaudio_unittest/sox_effect/dataset_test.py b/test/torchaudio_unittest/sox_effect/dataset_test.py
@@ -126,9 +126,7 @@ def test_apply_effects_tensor(self):
 
 
 def speed(path):
-    sample_rate = 8000
-    wav = get_whitenoise(n_channels=1, sample_rate=sample_rate, duration=1, dtype='float')
-    # wav, sample_rate = torchaudio.load(path)
+    wav, sample_rate = torchaudio.load(path)
     effects = [
         ['speed', '1.03756523535464655'],
         ['rate', f'{sample_rate}'],
@@ -138,6 +136,8 @@ def speed(path):
 
 @skipIfNoExtension
 class TestProcessPoolExecutor(TempDirMixin, PytorchTestCase):
+    backend = "sox_io"
+
     def setUp(self):
         sample_rate = 16000
         self.flist = []