Changeset e7556a1

Timestamp:

Sep 16, 2017, 11:46:30 PM (7 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

Children:

b99e2a5

Parents:

c96e6c0

Message:

python/tests/test_dct.py: add comments

File:

• python/tests/test_dct.py (modified) (3 diffs)

python/tests/test_dct.py

@@ -6 +6 @@
 import aubio
 
+precomputed_arange = [ 9.89949512, -6.44232273,  0., -0.67345482, 0.,
+        -0.20090288, 0., -0.05070186]
+
+precomputed_some_ones = [ 4.28539848,  0.2469689,  -0.14625292, -0.58121818,
+        -0.83483052, -0.75921834, -0.35168475,  0.24087936,
+        0.78539824, 1.06532764,  0.97632152,  0.57164496, 0.03688532,
+        -0.39446154, -0.54619485, -0.37771079]
+
 class aubio_dct(TestCase):
 
     def test_init(self):
-        """ check aubio.dct() can be created """
+        """ test that aubio.dct() is created with expected size """
         a_dct = aubio.dct()
         self.assertEqual(a_dct.size, 1024)
@@ -20 +28 @@
         >>> precomputed = dct(a_in, norm='ortho')
         """
-        precomputed = [ 9.89949512, -6.44232273,  0., -0.67345482, 0.,
-                -0.20090288, 0., -0.05070186]
+        N = len(precomputed_arange)
         a_dct = aubio.dct(8)
         a_in = np.arange(8).astype('float32')
-        a_expected = aubio.fvec(precomputed)
+        a_expected = aubio.fvec(precomputed_arange)
         assert_almost_equal(a_dct(a_in), a_expected, decimal=6)
 
     def test_some_ones(self):
         """ test that dct(somevector) is computed correctly """
-        precomputed = [ 4.28539848,  0.2469689,  -0.14625292, -0.58121818,
-                -0.83483052, -0.75921834, -0.35168475,  0.24087936,
-                0.78539824, 1.06532764,  0.97632152,  0.57164496, 0.03688532,
-                -0.39446154, -0.54619485, -0.37771079]
         a_dct = aubio.dct(16)
         a_in = np.ones(16).astype('float32')
         a_in[1] = 0
         a_in[3] = np.pi
-        a_expected = aubio.fvec(precomputed)
+        a_expected = aubio.fvec(precomputed_some_ones)
         assert_almost_equal(a_dct(a_in), a_expected, decimal=7)
 
     def test_reconstruction(self):
+        """ test that some_ones vector can be recontructed """
         a_dct = aubio.dct(16)
         a_in = np.ones(16).astype('float32')
@@ -50 +54 @@
 
     def test_negative_size(self):
+        """ test that creation fails with a negative size """
         with self.assertRaises(ValueError):
             aubio.dct(-1)
 
     def test_wrong_size(self):
+        """ test that creation fails with a non power-of-two size """
         # supports for non 2** fft sizes only when compiled with fftw3
         try: