[sms] Rename sms.py to pysms.py.

PEP8 cleanup of test_sms.py.

1 files changed, 5 insertions, 13 deletions

diff --git a/tests/test_base.py b/tests/test_base.py
index ce4297e..1fe5881 100644
--- a/tests/test_base.py
+++ b/tests/test_base.py
@@ -1,7 +1,7 @@
 import os
 import numpy as np
-import scipy.io.wavfile as wavfile
 from nose.tools import assert_almost_equals
+import simpl
 import simpl.base as base
 
 float_precision = 5
@@ -74,9 +74,7 @@ class TestFrame(object):
 class TestPeakDetection(object):
     @classmethod
     def setup_class(cls):
-        cls.audio = wavfile.read(audio_path)[1]
-        cls.audio = np.asarray(cls.audio, dtype=np.double)
-        cls.audio /= np.max(cls.audio)
+        cls.audio = simpl.read_wav(audio_path)[0]
 
     def test_peak_detection(self):
         pd = base.PeakDetection()
@@ -89,9 +87,7 @@ class TestPeakDetection(object):
 class TestPartialTracking(object):
     @classmethod
     def setup_class(cls):
-        cls.audio = wavfile.read(audio_path)[1]
-        cls.audio = np.asarray(cls.audio, dtype=np.double)
-        cls.audio /= np.max(cls.audio)
+        cls.audio = simpl.read_wav(audio_path)[0]
 
     def test_partial_tracking(self):
         pd = base.PeakDetection()
@@ -107,9 +103,7 @@ class TestPartialTracking(object):
 class TestSynthesis(object):
     @classmethod
     def setup_class(cls):
-        cls.audio = wavfile.read(audio_path)[1]
-        cls.audio = np.asarray(cls.audio, dtype=np.double)
-        cls.audio /= np.max(cls.audio)
+        cls.audio = simpl.read_wav(audio_path)[0]
 
     def test_synthesis(self):
         pd = base.PeakDetection()
@@ -127,9 +121,7 @@ class TestSynthesis(object):
 class TestResidual(object):
     @classmethod
     def setup_class(cls):
-        cls.audio = wavfile.read(audio_path)[1]
-        cls.audio = np.asarray(cls.audio, dtype=np.double)
-        cls.audio /= np.max(cls.audio)
+        cls.audio = simpl.read_wav(audio_path)[0]
 
     def test_synthesis(self):
         pd = base.PeakDetection()