Split tests

test/test_srmr.py

@@ -8,22 +8,25 @@
 from scipy.io.matlab import loadmat
 import numpy as np
 
-def test_srmr():
-    fs = 16000
-    s = loadmat("test/test.mat")["s"][:,0]
+fs = 16000
+s = loadmat("test/test.mat")["s"][:,0]
+
+correct_ratios = loadmat("test/correct_ratios.mat")['correct_ratios'][0]
 
-    correct_ratios = loadmat("test/correct_ratios.mat")['correct_ratios'][0]
+
+def test_srmr():
     ratio, avg_energy = srmr(s, fs)
     assert np.allclose(ratio, correct_ratios[1], rtol=1e-6, atol=1e-12)
 
+def test_srmr_fast():
     ratio_norm_fast, avg_energy_norm_fast = srmr(s, fs, fast=True, norm=True, max_cf=30)
     assert np.allclose(ratio_norm_fast, correct_ratios[2], rtol=1e-6, atol=1e-12)
 
+def test_srmr_slow():
     ratio_slow, avg_energy_slow = srmr(s, fs, fast=False)
     assert np.allclose(ratio_slow, correct_ratios[0], rtol=1e-6, atol=1e-12)
 
+def test_srmr_slow_norm():
     ratio_norm, avg_energy_norm = srmr(s, fs, fast=False, norm=True, max_cf=30)
     assert np.allclose(ratio_norm, correct_ratios[3], rtol=1e-6, atol=1e-12)
 
-if __name__ == '__main__':
-    test_srmr()