Changeset fb434c4 for python/tests

Timestamp:

Apr 30, 2016, 7:19:39 AM (9 years ago)

Author:

Paul Brossier <piem@piem.org>

Branches:

feature/autosink, feature/cnn, feature/cnn_org, feature/constantq, feature/crepe, feature/crepe_org, feature/pitchshift, feature/pydocstrings, feature/timestretch, fix/ffmpeg5, master, pitchshift, sampler, timestretch, yinfft+

Children:

43de50d

Parents:

7498e48

Message:

python/tests/test_phasevoc.py: cleanup, start using nose2

File:

• python/tests/test_phasevoc.py (modified) (4 diffs)

python/tests/test_phasevoc.py

@@ -5 +5 @@
 from numpy import array, shape
 from numpy.random import random
+from nose2.tools import params
 import numpy as np
 
-max_sq_error = 1.e-12
+if float_type == 'float32':
+    max_sq_error = 1.e-12
+else:
+    max_sq_error = 1.e-29
 
 def create_sine(hop_s, freq, samplerate):
     t = np.arange(hop_s).astype(float_type)
-    return np.sin( 2. * np.pi * t * freq / samplerate)
+    return np.sin( 2. * np.pi * freq * t / float(samplerate))
 
 def create_noise(hop_s):
@@ -42 +46 @@
             s = f(t)
             r = f.rdo(s)
-            assert_equal ( array(t), 0)
-            assert_equal ( s.norm, 0)
-            assert_equal ( s.phas, 0)
-            assert_equal ( r, 0)
+            assert_equal ( t, 0.)
+            assert_equal ( s.norm, 0.)
+            assert_equal ( s.phas, 0.)
+            assert_equal ( r, 0.)
 
     def test_resynth_8_steps_sine(self):
@@ -92 +96 @@
         self.reconstruction(sigin, hop_s, ratio)
 
+    @params(
+            ( 256, 8),
+            ( 256, 4),
+            ( 256, 2),
+            ( 512, 8),
+            ( 512, 4),
+            ( 512, 2),
+            (1024, 8),
+            (1024, 4),
+            (1024, 2),
+            (2048, 8),
+            (2048, 4),
+            (2048, 2),
+            (4096, 8),
+            (4096, 4),
+            (4096, 2),
+            (8192, 8),
+            (8192, 4),
+            (8192, 2),
+            )
+    def test_resynth_steps_noise(self, hop_s, ratio):
+        """ check the resynthesis of a random signal is correct """
+        sigin = create_noise(hop_s)
+        self.reconstruction(sigin, hop_s, ratio)
+
+    @params(
+            (44100,  256, 8,   441),
+            (44100,  256, 4,  1203),
+            (44100,  256, 2,  3045),
+            (44100,  512, 8,   445),
+            (44100,  512, 4,   445),
+            (44100,  512, 2,   445),
+            (44100, 1024, 8,   445),
+            (44100, 1024, 4,   445),
+            (44100, 1024, 2,   445),
+            ( 8000, 1024, 2,   445),
+            (22050, 1024, 2,   445),
+            (22050,  256, 8,   445),
+            (96000, 1024, 8, 47000),
+            (96000, 1024, 8,    20),
+            )
+    def test_resynth_steps_sine(self, samplerate, hop_s, ratio, freq):
+        """ check the resynthesis of a sine is correct """
+        sigin = create_sine(hop_s, freq, samplerate)
+        self.reconstruction(sigin, hop_s, ratio)
+
     def reconstruction(self, sigin, hop_s, ratio):
         buf_s = hop_s * ratio
@@ -105 +155 @@
 
 if __name__ == '__main__':
-  from unittest import main
-  main()
+    from nose2 import main
+    main()