Changes in / [4368223:e86676b]

File:

• tests/python/fft.py (modified) (1 diff)

tests/python/fft.py

@@ -83 +83 @@
       for channel in range(1, channels):
         self.assertEqual(0., cvec_read_norm(fftgrain, channel, index))
-        self.assertEqual(0., cvec_read_phas(fftgrain, channel, index))
-
     # check norm and phase == 0 in first first and last bin of first channel
-    # check unwrap2pi(phas) ~= pi everywhere but in first and last bin
+    self.assertCloseEnough((buf_size-1)*some_constant, cvec_read_norm(fftgrain, 0, 0))
     self.assertCloseEnough(0., cvec_read_phas(fftgrain, 0, 0))
-    for index in range(1,buf_size/2):
+    self.assertCloseEnough(0., cvec_read_norm(fftgrain, 0, buf_size/2+1))
+    self.assertCloseEnough(0., cvec_read_phas(fftgrain, 0, buf_size/2+1))
+    # check unwrap2pi(phas) ~= pi everywhere but in first bin
+    for index in range(1,buf_size/2+1):
        self.assertCloseEnough(math.pi, aubio_unwrap2pi(cvec_read_phas(fftgrain, 0, index)))
-    self.assertCloseEnough(0., cvec_read_phas(fftgrain, 0, buf_size/2))
-    self.assertCloseEnough(0., cvec_read_phas(fftgrain, 0, buf_size/2+1))
-
-    self.assertCloseEnough((buf_size-1)*some_constant, cvec_read_norm(fftgrain, 0, 0))
-    for index in range(1,buf_size/2+1):
-       self.assertCloseEnough(some_constant, abs(cvec_read_norm(fftgrain, 0, index)))
-    self.assertCloseEnough(0., cvec_read_norm(fftgrain, 0, buf_size/2+1))
-
+       self.assertCloseEnough(some_constant, cvec_read_norm(fftgrain, 0, index))
     del fftgrain
     del input