Remove global audio backend from test (#3691)

mthrok · web-flow · commit a59e9ab848bf · 2023-11-09T17:42:59.000-05:00
global audio backend is removed thus this is no-op.
diff --git a/test/torchaudio_unittest/common_utils/__init__.py b/test/torchaudio_unittest/common_utils/__init__.py
@@ -1,5 +1,4 @@
 from .autograd_utils import use_deterministic_algorithms
-from .backend_utils import set_audio_backend
 from .case_utils import (
     disabledInCI,
     HttpServerMixin,
@@ -37,7 +36,6 @@
     "get_whitenoise",
     "get_sinusoid",
     "get_spectrogram",
-    "set_audio_backend",
     "TempDirMixin",
     "HttpServerMixin",
     "TestBaseMixin",
diff --git a/test/torchaudio_unittest/common_utils/backend_utils.py b/test/torchaudio_unittest/common_utils/backend_utils.py
diff --git a/test/torchaudio_unittest/common_utils/case_utils.py b/test/torchaudio_unittest/common_utils/case_utils.py
@@ -15,8 +15,6 @@
 from torchaudio._internal.module_utils import eval_env, is_module_available
 from torchaudio.utils.ffmpeg_utils import get_video_decoders, get_video_encoders
 
-from .backend_utils import set_audio_backend
-
 
 class TempDirMixin:
     """Mixin to provide easy access to temp dir"""
@@ -88,15 +86,13 @@ def get_url(self, *route):
 
 
 class TestBaseMixin:
-    """Mixin to provide consistent way to define device/dtype/backend aware TestCase"""
+    """Mixin to provide consistent way to define device/dtype aware TestCase"""
 
     dtype = None
     device = None
-    backend = None
 
     def setUp(self):
         super().setUp()
-        set_audio_backend(self.backend)
         torch.random.manual_seed(2434)
 
     @property
diff --git a/test/torchaudio_unittest/datasets/cmuarctic_test.py b/test/torchaudio_unittest/datasets/cmuarctic_test.py
@@ -47,7 +47,6 @@ def get_mock_dataset(root_dir):
 
 
 class TestCMUARCTIC(TempDirMixin, TorchaudioTestCase):
-    backend = "default"
 
     root_dir = None
     samples = []
diff --git a/test/torchaudio_unittest/datasets/commonvoice_test.py b/test/torchaudio_unittest/datasets/commonvoice_test.py
@@ -142,7 +142,6 @@ def _test_commonvoice(self, dataset):
 
 
 class TestCommonVoiceEN(BaseTestCommonVoice, TorchaudioTestCase):
-    backend = "default"
     root_dir = None
 
     @classmethod
@@ -160,7 +159,6 @@ def test_commonvoice_path(self):
 
 
 class TestCommonVoiceFR(BaseTestCommonVoice, TorchaudioTestCase):
-    backend = "default"
     root_dir = None
 
     @classmethod
diff --git a/test/torchaudio_unittest/datasets/dr_vctk_test.py b/test/torchaudio_unittest/datasets/dr_vctk_test.py
@@ -74,7 +74,6 @@ def get_mock_dataset(root_dir):
 
 
 class TestDRVCTK(TempDirMixin, TorchaudioTestCase):
-    backend = "default"
 
     root_dir = None
     samples = {}
diff --git a/test/torchaudio_unittest/datasets/fluentcommands_test.py b/test/torchaudio_unittest/datasets/fluentcommands_test.py
@@ -104,7 +104,6 @@ def get_mock_dataset(dataset_dir: str):
 
 class TestFluentSpeechCommands(TempDirMixin, TorchaudioTestCase):
     root_dir = None
-    backend = "default"
 
     mocked_train_samples = []
     mocked_valid_samples = []
diff --git a/test/torchaudio_unittest/datasets/gtzan_test.py b/test/torchaudio_unittest/datasets/gtzan_test.py
@@ -37,7 +37,6 @@ def get_mock_dataset(root_dir):
 
 
 class TestGTZAN(TempDirMixin, TorchaudioTestCase):
-    backend = "default"
 
     root_dir = None
     samples = []
diff --git a/test/torchaudio_unittest/datasets/iemocap_test.py b/test/torchaudio_unittest/datasets/iemocap_test.py
@@ -91,7 +91,6 @@ def get_mock_dataset(dataset_dir: str):
 
 class TestIemocap(TempDirMixin, TorchaudioTestCase):
     root_dir = None
-    backend = "default"
 
     all_samples = []
     script_samples = []
diff --git a/test/torchaudio_unittest/datasets/librilightlimited_test.py b/test/torchaudio_unittest/datasets/librilightlimited_test.py
@@ -69,7 +69,6 @@ def get_mock_datasets(root_dir):
 
 
 class TestLibriLightLimited(TempDirMixin, TorchaudioTestCase):
-    backend = "default"
 
     root_dir = None
     samples_10min = []
diff --git a/test/torchaudio_unittest/datasets/librimix_test.py b/test/torchaudio_unittest/datasets/librimix_test.py
@@ -65,7 +65,6 @@ def get_mock_dataset(root_dir: str, num_speaker: int):
 
 
 class TestLibriMix(TempDirMixin, TorchaudioTestCase):
-    backend = "default"
 
     root_dir = None
     samples_2spk = []
diff --git a/test/torchaudio_unittest/datasets/librispeech_test_impl.py b/test/torchaudio_unittest/datasets/librispeech_test_impl.py
@@ -49,7 +49,6 @@ def get_mock_dataset(root_dir):
 
 
 class LibriSpeechTestMixin(TempDirMixin):
-    backend = "default"
 
     root_dir = None
     samples = []
diff --git a/test/torchaudio_unittest/datasets/libritts_test.py b/test/torchaudio_unittest/datasets/libritts_test.py
@@ -41,7 +41,6 @@ def get_mock_dataset(root_dir):
 
 
 class TestLibriTTS(TempDirMixin, TorchaudioTestCase):
-    backend = "default"
 
     root_dir = None
     data = []
diff --git a/test/torchaudio_unittest/datasets/ljspeech_test.py b/test/torchaudio_unittest/datasets/ljspeech_test.py
@@ -45,7 +45,6 @@ def get_mock_dataset(root_dir):
 
 
 class TestLJSpeech(TempDirMixin, TorchaudioTestCase):
-    backend = "default"
 
     root_dir = None
     data, _transcripts, _normalized_transcript = [], [], []
diff --git a/test/torchaudio_unittest/datasets/musdb_hq_test.py b/test/torchaudio_unittest/datasets/musdb_hq_test.py
@@ -86,7 +86,6 @@ def get_mock_dataset(dataset_dir):
 
 class TestMusDB_HQ(TempDirMixin, TorchaudioTestCase):
     root_dir = None
-    backend = "default"
 
     train_all_samples = {}
     train_only_samples = {}
diff --git a/test/torchaudio_unittest/datasets/quesst14_test.py b/test/torchaudio_unittest/datasets/quesst14_test.py
@@ -104,7 +104,6 @@ def get_mock_dataset(dataset_dir):
 
 class TestQuesst14(TempDirMixin, TorchaudioTestCase):
     root_dir = None
-    backend = "default"
 
     utterances = {}
     dev_samples = {}
diff --git a/test/torchaudio_unittest/datasets/snips_test.py b/test/torchaudio_unittest/datasets/snips_test.py
@@ -82,7 +82,6 @@ def get_mock_datasets(dataset_dir):
 
 class TestSnips(TempDirMixin, TorchaudioTestCase):
     root_dir = None
-    backend = "default"
 
     train_samples = {}
     valid_samples = {}
diff --git a/test/torchaudio_unittest/datasets/speechcommands_test.py b/test/torchaudio_unittest/datasets/speechcommands_test.py
@@ -95,7 +95,6 @@ def get_mock_dataset(dataset_dir):
 
 
 class TestSpeechCommands(TempDirMixin, TorchaudioTestCase):
-    backend = "default"
 
     root_dir = None
     samples = []
diff --git a/test/torchaudio_unittest/datasets/tedlium_test.py b/test/torchaudio_unittest/datasets/tedlium_test.py
@@ -1,9 +1,8 @@
 import os
-import platform
 from pathlib import Path
 
 from torchaudio.datasets import tedlium
-from torchaudio_unittest.common_utils import get_whitenoise, save_wav, skipIfNoSox, TempDirMixin, TorchaudioTestCase
+from torchaudio_unittest.common_utils import get_whitenoise, save_wav, TempDirMixin, TorchaudioTestCase
 
 # Used to generate a unique utterance for each dummy audio file
 _UTTERANCES = [
@@ -84,7 +83,7 @@ def get_mock_dataset(dataset_dir):
     return mocked_samples
 
 
-class Tedlium(TempDirMixin):
+class Tedlium(TempDirMixin, TorchaudioTestCase):
     root_dir = None
     samples = {}
 
@@ -131,14 +130,3 @@ def test_tedlium_release3(self):
         release = "release3"
         dataset = tedlium.TEDLIUM(self.root_dir, release=release)
         self._test_tedlium(dataset, release)
-
-
-class TestTedliumSoundfile(Tedlium, TorchaudioTestCase):
-    backend = "soundfile"
-
-
-if platform.system() != "Windows":
-
-    @skipIfNoSox
-    class TestTedliumSoxIO(Tedlium, TorchaudioTestCase):
-        backend = "sox_io"
diff --git a/test/torchaudio_unittest/datasets/vctk_test.py b/test/torchaudio_unittest/datasets/vctk_test.py
@@ -57,7 +57,6 @@ def get_mock_dataset(root_dir):
 
 
 class TestVCTK(TempDirMixin, TorchaudioTestCase):
-    backend = "default"
 
     root_dir = None
     samples = []
diff --git a/test/torchaudio_unittest/datasets/voxceleb1_test.py b/test/torchaudio_unittest/datasets/voxceleb1_test.py
@@ -128,7 +128,6 @@ def get_mock_veri_dataset(root_dir: str, meta_file: str):
 
 class TestVoxCeleb1Identification(TempDirMixin, TorchaudioTestCase):
     root_dir = None
-    backend = "default"
 
     meta_file = "iden_list.txt"
     train_samples = {}
@@ -166,7 +165,6 @@ def testVoxCeleb1SubsetTest(self):
 
 class TestVoxCeleb1Verification(TempDirMixin, TorchaudioTestCase):
     root_dir = None
-    backend = "default"
 
     meta_file = "veri_test.txt"
     train_samples = {}
diff --git a/test/torchaudio_unittest/datasets/yesno_test.py b/test/torchaudio_unittest/datasets/yesno_test.py
@@ -23,7 +23,6 @@ def get_mock_data(root_dir, labels):
 
 
 class TestYesNo(TempDirMixin, TorchaudioTestCase):
-    backend = "default"
 
     root_dir = None
     data = []
diff --git a/test/torchaudio_unittest/example/souce_sepration/wsj0mix_test.py b/test/torchaudio_unittest/example/souce_sepration/wsj0mix_test.py
@@ -54,7 +54,6 @@ def _mock_dataset(root_dir, num_speaker):
 
 
 class TestWSJ0Mix2(TempDirMixin, TorchaudioTestCase):
-    backend = "default"
     root_dir = None
     expected = None
 
@@ -78,7 +77,6 @@ def test_wsj0mix(self):
 
 
 class TestWSJ0Mix3(TempDirMixin, TorchaudioTestCase):
-    backend = "default"
     root_dir = None
     expected = None
 
diff --git a/test/torchaudio_unittest/functional/batch_consistency_test.py b/test/torchaudio_unittest/functional/batch_consistency_test.py
@@ -25,8 +25,6 @@ def _name_from_args(func, _, params):
 class TestFunctional(common_utils.TorchaudioTestCase):
     """Test functions defined in `functional` module"""
 
-    backend = "default"
-
     def assert_batch_consistency(self, functional, inputs, atol=1e-6, rtol=1e-5, seed=42):
         n = inputs[0].size(0)
         for i in range(1, len(inputs)):
diff --git a/test/torchaudio_unittest/functional/functional_cpu_test.py b/test/torchaudio_unittest/functional/functional_cpu_test.py
@@ -24,8 +24,6 @@ class TestFunctionalFloat64(Functional, PytorchTestCase):
 
 @skipIfNoSox
 class TestApplyCodec(TorchaudioTestCase):
-    backend = "sox_io"
-
     def _smoke_test(self, format, compression, check_num_frames):
         """
         The purpose of this test suite is to verify that apply_codec functionalities do not exhibit
diff --git a/test/torchaudio_unittest/sox_effect/dataset_test.py b/test/torchaudio_unittest/sox_effect/dataset_test.py
@@ -135,8 +135,6 @@ def speed(path):
 
 @skipIfNoSox
 class TestProcessPoolExecutor(TempDirMixin, PytorchTestCase):
-    backend = "sox_io"
-
     def setUp(self):
         sample_rate = 16000
         self.flist = []
diff --git a/test/torchaudio_unittest/transforms/batch_consistency_test.py b/test/torchaudio_unittest/transforms/batch_consistency_test.py
@@ -10,8 +10,6 @@
 class TestTransforms(common_utils.TorchaudioTestCase):
     """Test suite for classes defined in `transforms` module"""
 
-    backend = "default"
-
     def assert_batch_consistency(self, transform, batch, *args, atol=1e-8, rtol=1e-5, seed=42, **kwargs):
         n = batch.size(0)
 
diff --git a/test/torchaudio_unittest/transforms/transforms_test.py b/test/torchaudio_unittest/transforms/transforms_test.py
@@ -8,7 +8,6 @@
 
 
 class Tester(common_utils.TorchaudioTestCase):
-    backend = "default"
 
     # create a sinewave signal for testing
     sample_rate = 16000
