Changeset 0b6d23d for python/tests/test_fft.py

Timestamp:

May 16, 2016, 5:08:18 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:

a6f9ebf

Parents:

58a5fb9

Message:

python/tests: fix most prospect warnings

File:

• python/tests/test_fft.py (modified) (6 diffs)

python/tests/test_fft.py

@@ -1 +1 @@
 #! /usr/bin/env python
 
+from unittest import main
 from numpy.testing import TestCase
 from numpy.testing import assert_equal, assert_almost_equal
+import numpy as np
 from aubio import fvec, fft, cvec
-from math import pi
+from math import pi, floor
+from random import random
 
 class aubio_fft_test_case(TestCase):
@@ -35 +38 @@
     def test_impulse(self):
         """ check the transform of one impulse at a random place """
-        from random import random
-        from math import floor
         win_s = 256
         i = int(floor(random()*win_s))
@@ -50 +51 @@
 
     def test_impulse_negative(self):
-        """ check the transform of one impulse at a random place """
-        from random import random
-        from math import floor
+        """ check the transform of a negative impulse at a random place """
         win_s = 256
-        i = 0
-        impulse = -10.
+        i = int(floor(random()*win_s))
+        impulse = -.1
         f = fft(win_s)
         timegrain = fvec(win_s)
+        timegrain[0] = 0
         timegrain[i] = impulse
         fftgrain = f ( timegrain )
         #self.plot_this ( fftgrain.phas )
-        assert_almost_equal ( fftgrain.norm, abs(impulse), decimal = 6 )
+        assert_almost_equal ( fftgrain.norm, abs(impulse), decimal = 5 )
         if impulse < 0:
             # phase can be pi or -pi, as it is not unwrapped
-            assert_almost_equal ( abs(fftgrain.phas[1:-1]) , pi, decimal = 6 )
+            #assert_almost_equal ( abs(fftgrain.phas[1:-1]) , pi, decimal = 6 )
             assert_almost_equal ( fftgrain.phas[0], pi, decimal = 6)
-            assert_almost_equal ( fftgrain.phas[-1], pi, decimal = 6)
+            assert_almost_equal ( np.fmod(fftgrain.phas[-1], pi), 0, decimal = 6)
         else:
-            assert_equal ( fftgrain.phas[1:-1] == 0, True)
-            assert_equal ( fftgrain.phas[0] == 0, True)
-            assert_equal ( fftgrain.phas[-1] == 0, True)
+            #assert_equal ( fftgrain.phas[1:-1] == 0, True)
+            assert_equal ( fftgrain.phas[0], 0)
+            assert_almost_equal ( np.fmod(fftgrain.phas[-1], pi), 0, decimal = 6)
         # now check the resynthesis
         synthgrain = f.rdo ( fftgrain )
@@ -96 +96 @@
         """ check running fft.rdo before fft.do works """
         win_s = 1024
-        impulse = pi
         f = fft(win_s)
         fftgrain = cvec(win_s)
@@ -178 +177 @@
             with self.assertRaises(RuntimeError):
                 fft(win_s)
-        except AssertionError as e:
+        except AssertionError:
             self.skipTest('creating aubio.fft with size %d did not fail' % win_s)
 
@@ -187 +186 @@
 
 if __name__ == '__main__':
-    from nose2 import main
     main()