Unverified Commit 1ebfb3de authored by Caroline Chen's avatar Caroline Chen Committed by GitHub
Browse files

Fix functional_cpu_test test name generation (#1393)

parent 32cd700a
...@@ -13,7 +13,6 @@ from torchaudio_unittest.common_utils import ( ...@@ -13,7 +13,6 @@ from torchaudio_unittest.common_utils import (
TorchaudioTestCase, TorchaudioTestCase,
skipIfNoSox, skipIfNoSox,
) )
from torchaudio_unittest.backend.sox_io.common import name_func
from .functional_impl import Lfilter, Spectrogram from .functional_impl import Lfilter, Spectrogram
...@@ -254,17 +253,14 @@ class TestApplyCodec(TorchaudioTestCase): ...@@ -254,17 +253,14 @@ class TestApplyCodec(TorchaudioTestCase):
def test_wave(self): def test_wave(self):
self._smoke_test("wav", compression=None, check_num_frames=True) self._smoke_test("wav", compression=None, check_num_frames=True)
@parameterized.expand([(96,), (128,), (160,), (192,), (224,), (256,), (320,)], @parameterized.expand([(96,), (128,), (160,), (192,), (224,), (256,), (320,)])
name_func=name_func)
def test_mp3(self, compression): def test_mp3(self, compression):
self._smoke_test("mp3", compression, check_num_frames=False) self._smoke_test("mp3", compression, check_num_frames=False)
@parameterized.expand([(0,), (1,), (2,), (3,), (4,), (5,), (6,), (7,), (8,)], @parameterized.expand([(0,), (1,), (2,), (3,), (4,), (5,), (6,), (7,), (8,)])
name_func=name_func)
def test_flac(self, compression): def test_flac(self, compression):
self._smoke_test("flac", compression, check_num_frames=False) self._smoke_test("flac", compression, check_num_frames=False)
@parameterized.expand([(-1,), (0,), (1,), (2,), (3,), (3.6,), (5,), (10,)], @parameterized.expand([(-1,), (0,), (1,), (2,), (3,), (3.6,), (5,), (10,)])
name_func=name_func)
def test_vorbis(self, compression): def test_vorbis(self, compression):
self._smoke_test("vorbis", compression, check_num_frames=False) self._smoke_test("vorbis", compression, check_num_frames=False)
Markdown is supported
0% or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment