update test method

pytorch · Jun 18, 2020 · a55646b · a55646b
1 parent 7e1d516
commit a55646b
Showing 1 changed file with 4 additions and 2 deletions.
diff --git a/test/test_models.py b/test/test_models.py
@@ -3,8 +3,10 @@
 import torch
 from torchaudio.models import Wav2Letter, _MelResNet
 
+from . import common_utils
 
-class TestWav2Letter(unittest.TestCase):
+
+class TestWav2Letter(common_utils.TorchaudioTestCase):
 
     def test_waveform(self):
         batch_size = 2
@@ -33,7 +35,7 @@ def test_mfcc(self):
         assert out.size() == (batch_size, num_classes, 2)
 
 
-class TestMelResNet(unittest.TestCase):
+class TestMelResNet(common_utils.TorchaudioTestCase):
 
     def test_waveform(self):