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

Paul Brossier committed 10 years ago

Commit 437ef071 parent 6014dc01 file · +15 −3Browse files

Showing the changes to python/tests (1 of 1 file). Show all

Unified · Split

@@ -140,18 +140,30 @@ class aubio_fft_test_case(TestCase): · 139 unmodified lines
140	140	win_s = 1024
141	141	f = fft(win_s)
142	142	t = fvec(win_s + 1)
143		− print f(t)
144	143	with self.assertRaises(ValueError):
145		− print f(t)
	144	+ f(t)
146	145
147	146	def test_small_input_timegrain(self):
148	147	win_s = 1024
149	148	f = fft(win_s)
150	149	t = fvec(1)
151		− print f(t).phas
152	150	with self.assertRaises(ValueError):
153	151	f(t)
154	152
	153	+ def test_large_input_fftgrain(self):
	154	+ win_s = 1024
	155	+ f = fft(win_s)
	156	+ s = cvec(win_s + 1)
	157	+ with self.assertRaises(ValueError):
	158	+ f.rdo(s)
	159	+
	160	+ def test_small_input_timegrain(self):
	161	+ win_s = 1024
	162	+ f = fft(win_s)
	163	+ s = cvec(16)
	164	+ with self.assertRaises(ValueError):
	165	+ f.rdo(s)
	166	+
155	167	if __name__ == '__main__':
156	168	from unittest import main
157	169	main()

Keyboard shortcuts