update test

Author: Ji Chen

test/test_models.py

@@ -1,5 +1,3 @@
-import unittest
-
 import torch
 from torchaudio.models import Wav2Letter, _MelResNet, _UpsampleNetwork, _WaveRNN
 
@@ -107,6 +105,3 @@ def test_waveform(self):
         out = model(x, mels)
 
         assert out.size() == (batch_size, hop_length * (n_time - kernel_size + 1), 2 ** n_bits)
-
-if __name__ == '__main__':
-    unittest.main()