Make test scripts runnable without being modules (#188)

Author: jamarshon

build_tools/travis/test_script.sh

@@ -18,7 +18,7 @@ run_tests() {
   for FILE in $TEST_FILES; do
     # run each file on a separate process. if one fails, just keep going and
     # return the final exit status.
-    python -m unittest -v $FILE
+    python -m pytest -v $FILE
     STATUS=$?
     EXIT_STATUS="$(($EXIT_STATUS+STATUS))"
   done

requirements.txt

@@ -12,3 +12,6 @@ flake8
 # Used for comparison of outputs in tests
 librosa>=0.4.3
 scipy
+
+# Unit tests with pytest
+pytest

test/__init__.py

@@ -1,13 +1,13 @@
 import unittest
-import test.common_utils
+import common_utils
 import torch
 import torchaudio
 import math
 import os
 
 
 class Test_LoadSave(unittest.TestCase):
-    test_dirpath, test_dir = test.common_utils.create_temp_assets_dir()
+    test_dirpath, test_dir = common_utils.create_temp_assets_dir()
     test_filepath = os.path.join(test_dirpath, "assets",
                                  "steam-train-whistle-daniel_simon.mp3")
 

test/test.py

@@ -1,5 +1,5 @@
 import unittest
-import test.common_utils
+import common_utils
 import torch
 import torch.nn as nn
 from torch.utils.data import Dataset, DataLoader
@@ -10,7 +10,7 @@
 
 class TORCHAUDIODS(Dataset):
 
-    test_dirpath, test_dir = test.common_utils.create_temp_assets_dir()
+    test_dirpath, test_dir = common_utils.create_temp_assets_dir()
 
     def __init__(self):
         self.asset_dirpath = os.path.join(self.test_dirpath, "assets")

test/test_dataloader.py

@@ -2,13 +2,13 @@
 import torch
 import torchaudio.kaldi_io as kio
 import unittest
-import test.common_utils
+import common_utils
 
 
 class KaldiIOTest(unittest.TestCase):
     data1 = [[1, 2, 3], [11, 12, 13], [21, 22, 23]]
     data2 = [[31, 32, 33], [41, 42, 43], [51, 52, 53]]
-    test_dirpath, test_dir = test.common_utils.create_temp_assets_dir()
+    test_dirpath, test_dir = common_utils.create_temp_assets_dir()
 
     def _test_helper(self, file_name, expected_data, fn, expected_dtype):
         """ Takes a file_name to the input data and a function fn to extract the

test/test_kaldi_io.py

@@ -1,5 +1,5 @@
 import unittest
-import test.common_utils
+import common_utils
 import torch
 import torchaudio
 from torchaudio.legacy import save, load
@@ -8,7 +8,7 @@
 
 
 class Test_LoadSave(unittest.TestCase):
-    test_dirpath, test_dir = test.common_utils.create_temp_assets_dir()
+    test_dirpath, test_dir = common_utils.create_temp_assets_dir()
     test_filepath = os.path.join(test_dirpath, "assets",
                                  "steam-train-whistle-daniel_simon.mp3")
 

test/test_legacy.py

@@ -1,13 +1,13 @@
 import unittest
-import test.common_utils
+import common_utils
 import torch
 import torchaudio
 import math
 import os
 
 
 class Test_SoxEffectsChain(unittest.TestCase):
-    test_dirpath, test_dir = test.common_utils.create_temp_assets_dir()
+    test_dirpath, test_dir = common_utils.create_temp_assets_dir()
     test_filepath = os.path.join(test_dirpath, "assets",
                                  "steam-train-whistle-daniel_simon.mp3")
 

test/test_sox_effects.py

@@ -7,7 +7,7 @@
 from torchaudio.common_utils import IMPORT_LIBROSA, IMPORT_SCIPY
 import torchaudio.transforms as transforms
 import unittest
-import test.common_utils
+import common_utils
 
 if IMPORT_LIBROSA:
     import librosa
@@ -26,9 +26,9 @@ class Tester(unittest.TestCase):
     sig.unsqueeze_(1)  # (64000, 1)
     sig = (sig * volume * 2**31).long()
     # file for stereo stft test
-    test_dirpath, test_dir = test.common_utils.create_temp_assets_dir()
-    test_filepath = os.path.join(test_dirpath, "assets",
-                                 "steam-train-whistle-daniel_simon.mp3")
+    test_dirpath, test_dir = common_utils.create_temp_assets_dir()
+    test_filepath = os.path.join(test_dirpath, 'assets',
+                                 'steam-train-whistle-daniel_simon.mp3')
 
     def test_scale(self):