From: Paul Brossier <piem@piem.org>
Date: Sat, 30 Apr 2016 04:22:58 +0000 (+0200)
Subject: python/tests/test_fft.py: clean up, add tests for f.rdo input size
X-Git-Tag: 0.4.4~300^2~168
X-Git-Url: https://git.aubio.org/?a=commitdiff_plain;h=437ef0789c8c0c8ad782874424433c7d5fdbeef9;p=aubio.git

python/tests/test_fft.py: clean up, add tests for f.rdo input size
---

diff --git a/python/tests/test_fft.py b/python/tests/test_fft.py
index 257a571a..dec20081 100755
--- a/python/tests/test_fft.py
+++ b/python/tests/test_fft.py
@@ -140,18 +140,30 @@ class aubio_fft_test_case(TestCase):
         win_s = 1024
         f = fft(win_s)
         t = fvec(win_s + 1)
-        print f(t)
         with self.assertRaises(ValueError):
-            print f(t)
+            f(t)
 
     def test_small_input_timegrain(self):
         win_s = 1024
         f = fft(win_s)
         t = fvec(1)
-        print f(t).phas
         with self.assertRaises(ValueError):
             f(t)
 
+    def test_large_input_fftgrain(self):
+        win_s = 1024
+        f = fft(win_s)
+        s = cvec(win_s + 1)
+        with self.assertRaises(ValueError):
+            f.rdo(s)
+
+    def test_small_input_timegrain(self):
+        win_s = 1024
+        f = fft(win_s)
+        s = cvec(16)
+        with self.assertRaises(ValueError):
+            f.rdo(s)
+
 if __name__ == '__main__':
     from unittest import main
     main()