test_backend.py 1.38 KB
Newer Older
1
2
import torchaudio

moto's avatar
moto committed
3
from . import common_utils
4

moto's avatar
moto committed
5
6

class BackendSwitchMixin:
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
    """Test set/get_audio_backend works"""
    backend = None
    backend_module = None

    def test_switch(self):
        torchaudio.set_audio_backend(self.backend)
        if self.backend is None:
            assert torchaudio.get_audio_backend() is None
        else:
            assert torchaudio.get_audio_backend() == self.backend
        assert torchaudio.load == self.backend_module.load
        assert torchaudio.load_wav == self.backend_module.load_wav
        assert torchaudio.save == self.backend_module.save
        assert torchaudio.info == self.backend_module.info


moto's avatar
moto committed
23
class TestBackendSwitch_NoBackend(BackendSwitchMixin, common_utils.TorchaudioTestCase):
24
25
26
27
    backend = None
    backend_module = torchaudio.backend.no_backend


28
@common_utils.skipIfNoExtension
moto's avatar
moto committed
29
class TestBackendSwitch_SoX(BackendSwitchMixin, common_utils.TorchaudioTestCase):
30
31
32
33
    backend = 'sox'
    backend_module = torchaudio.backend.sox_backend


moto's avatar
moto committed
34
35
36
37
38
39
@common_utils.skipIfNoExtension
class TestBackendSwitch_SoXIO(BackendSwitchMixin, common_utils.TorchaudioTestCase):
    backend = 'sox_io'
    backend_module = torchaudio.backend.sox_io_backend


40
@common_utils.skipIfNoModule('soundfile')
moto's avatar
moto committed
41
class TestBackendSwitch_soundfile(BackendSwitchMixin, common_utils.TorchaudioTestCase):
42
43
    backend = 'soundfile'
    backend_module = torchaudio.backend.soundfile_backend