From 998259682f2cba30a0cc4a606f486585a80adf5f Mon Sep 17 00:00:00 2001
From: moto <855818+mthrok@users.noreply.github.com>
Date: Fri, 13 Nov 2020 17:10:38 +0000
Subject: [PATCH] Add test for #1021

---
 .../sox_effect/dataset_test.py                | 21 +++++++++++++++++++
 1 file changed, 21 insertions(+)

diff --git a/test/torchaudio_unittest/sox_effect/dataset_test.py b/test/torchaudio_unittest/sox_effect/dataset_test.py
index 647e4d2a81e..050b0638ef7 100644
--- a/test/torchaudio_unittest/sox_effect/dataset_test.py
+++ b/test/torchaudio_unittest/sox_effect/dataset_test.py
@@ -2,6 +2,7 @@
 import platform
 from unittest import skipIf
 from typing import List, Tuple
+from concurrent.futures import ProcessPoolExecutor
 
 import numpy as np
 import torch
@@ -122,3 +123,23 @@ def test_apply_effects_tensor(self):
         )
         for batch in loader:
             assert batch.shape == (32, 2, 2 * sample_rate)
+
+
+@skipIfNoExtension
+class TestProcessPoolExecutor(TempDirMixin, PytorchTestCase):
+    @staticmethod
+    def _speed():
+        sample_rate = 16000
+        wav = get_whitenoise(n_channels=1, sample_rate=sample_rate, duration=1, dtype='float')
+        effects = [
+            ['speed', '1.03756523535464655'],
+            ['rate', f'{sample_rate}'],
+        ]
+        return torchaudio.sox_effects.apply_effects_tensor(wav, sample_rate, effects)[0]
+
+    def test_executor(self):
+        """Test that apply_effects_tensor with speed + rate does not crush
+
+        https://github.com/pytorch/audio/issues/1021
+        """
+        ProcessPoolExecutor(1).submit(TestProcessPoolExecutor._speed).result()