diff options
| author | mormj <mormjb@gmail.com> | 2020-10-30 10:59:50 -0400 |
|---|---|---|
| committer | Marcus Müller <marcus@hostalia.de> | 2020-10-30 17:52:53 +0100 |
| commit | 7a0948ba85758fba1cc3858ef99bfa600dcc7416 (patch) | |
| tree | 610d7f9d773a193562def6df2d4b50f1bb3b3f86 /gr-fft/python/fft | |
| parent | 12192ee7d58de95ddca35a3e93bfc172bdb5c820 (diff) | |
qa: run autopep8 formatting on qa python files
find ./ -iname qa*.py | xargs autopep8 --in-place -a -a
mostly formats whitespace and gets rid of trailing semicolons
Diffstat (limited to 'gr-fft/python/fft')
| -rw-r--r-- | gr-fft/python/fft/qa_fft.py | 81 | ||||
| -rw-r--r-- | gr-fft/python/fft/qa_goertzel.py | 8 | ||||
| -rw-r--r-- | gr-fft/python/fft/qa_window.py | 9 |
3 files changed, 85 insertions, 13 deletions
diff --git a/gr-fft/python/fft/qa_fft.py b/gr-fft/python/fft/qa_fft.py index 36af441fd2..c1b8bbdf40 100644 --- a/gr-fft/python/fft/qa_fft.py +++ b/gr-fft/python/fft/qa_fft.py @@ -20,10 +20,71 @@ from gnuradio import gr, gr_unittest, fft, blocks # windowed = fft(src_data(:).*hamming(32)) # reverse_window_shift = ifft(fftshift(forward.*hamming(32))) -primes = (2, 3, 5, 7, 11, 13, 17, 19, 23, 29, 31, 37, 41, 43, 47, 53, - 59, 61, 67, 71, 73, 79, 83, 89, 97, 101, 103, 107, 109, 113, 127, 131, - 137, 139, 149, 151, 157, 163, 167, 173, 179, 181, 191, 193, 197, 199, 211, 223, - 227, 229, 233, 239, 241, 251, 257, 263, 269, 271, 277, 281, 283, 293, 307, 311) +primes = ( + 2, + 3, + 5, + 7, + 11, + 13, + 17, + 19, + 23, + 29, + 31, + 37, + 41, + 43, + 47, + 53, + 59, + 61, + 67, + 71, + 73, + 79, + 83, + 89, + 97, + 101, + 103, + 107, + 109, + 113, + 127, + 131, + 137, + 139, + 149, + 151, + 157, + 163, + 167, + 173, + 179, + 181, + 191, + 193, + 197, + 199, + 211, + 223, + 227, + 229, + 233, + 239, + 241, + 251, + 257, + 263, + 269, + 271, + 277, + 281, + 283, + 293, + 307, + 311) primes_transformed = ((4377 + 4516j), (-1706.1268310546875 + 1638.4256591796875j), @@ -73,7 +134,8 @@ class test_fft(gr_unittest.TestCase): abs_eps=1e-9, rel_eps=4e-4) def test_forward(self): - src_data = tuple([complex(primes[2 * i], primes[2 * i + 1]) for i in range(self.fft_size)]) + src_data = tuple([complex(primes[2 * i], primes[2 * i + 1]) + for i in range(self.fft_size)]) expected_result = primes_transformed src = blocks.vector_source_c(src_data) @@ -88,7 +150,8 @@ class test_fft(gr_unittest.TestCase): def test_reverse(self): src_data = tuple([x / self.fft_size for x in primes_transformed]) - expected_result = tuple([complex(primes[2 * i], primes[2 * i + 1]) for i in range(self.fft_size)]) + expected_result = tuple( + [complex(primes[2 * i], primes[2 * i + 1]) for i in range(self.fft_size)]) src = blocks.vector_source_c(src_data) s2v = blocks.stream_to_vector(gr.sizeof_gr_complex, self.fft_size) @@ -103,7 +166,8 @@ class test_fft(gr_unittest.TestCase): def test_multithreaded(self): # Same test as above, only use 2 threads src_data = tuple([x / self.fft_size for x in primes_transformed]) - expected_result = tuple([complex(primes[2 * i], primes[2 * i + 1]) for i in range(self.fft_size)]) + expected_result = tuple( + [complex(primes[2 * i], primes[2 * i + 1]) for i in range(self.fft_size)]) nthreads = 2 src = blocks.vector_source_c(src_data) @@ -117,7 +181,8 @@ class test_fft(gr_unittest.TestCase): self.assert_fft_ok2(expected_result, result_data) def test_window(self): - src_data = tuple([complex(primes[2 * i], primes[2 * i + 1]) for i in range(self.fft_size)]) + src_data = tuple([complex(primes[2 * i], primes[2 * i + 1]) + for i in range(self.fft_size)]) expected_result = ((2238.9174 + 2310.4750j), (-1603.7416 - 466.7420j), (116.7449 - 70.8553j), diff --git a/gr-fft/python/fft/qa_goertzel.py b/gr-fft/python/fft/qa_goertzel.py index ee4af1bb40..b7eb1f168a 100644 --- a/gr-fft/python/fft/qa_goertzel.py +++ b/gr-fft/python/fft/qa_goertzel.py @@ -13,6 +13,7 @@ from math import pi, cos from gnuradio import gr, gr_unittest, fft, blocks + class test_goertzel(gr_unittest.TestCase): def setUp(self): @@ -22,7 +23,7 @@ class test_goertzel(gr_unittest.TestCase): self.tb = None def make_tone_data(self, rate, freq): - return [cos(2*pi*x*freq/rate) for x in range(rate)] + return [cos(2 * pi * x * freq / rate) for x in range(rate)] def transform(self, src_data, rate, freq): src = blocks.vector_source_f(src_data, False) @@ -32,7 +33,7 @@ class test_goertzel(gr_unittest.TestCase): self.tb.run() return dst.data() - def test_001(self): # Measure single tone magnitude + def test_001(self): # Measure single tone magnitude rate = 8000 freq = 100 bin = freq @@ -41,7 +42,7 @@ class test_goertzel(gr_unittest.TestCase): actual_result = abs(self.transform(src_data, rate, bin)[0]) self.assertAlmostEqual(expected_result, actual_result, places=4) - def test_002(self): # Measure off frequency magnitude + def test_002(self): # Measure off frequency magnitude rate = 8000 freq = 100 bin = freq / 2 @@ -50,5 +51,6 @@ class test_goertzel(gr_unittest.TestCase): actual_result = abs(self.transform(src_data, rate, bin)[0]) self.assertAlmostEqual(expected_result, actual_result, places=4) + if __name__ == '__main__': gr_unittest.run(test_goertzel) diff --git a/gr-fft/python/fft/qa_window.py b/gr-fft/python/fft/qa_window.py index f5afd2640d..ba133f6cb4 100644 --- a/gr-fft/python/fft/qa_window.py +++ b/gr-fft/python/fft/qa_window.py @@ -14,6 +14,7 @@ import numpy from gnuradio import gr_unittest from gnuradio import fft + class test_window(gr_unittest.TestCase): """ Unit tests for fft.window @@ -29,9 +30,13 @@ class test_window(gr_unittest.TestCase): """ Verify window normalization """ - win = fft.window.build(fft.win_type.WIN_BLACKMAN_hARRIS, 21, normalize=True) - power = numpy.sum([x*x for x in win])/len(win) + win = fft.window.build( + fft.win_type.WIN_BLACKMAN_hARRIS, + 21, + normalize=True) + power = numpy.sum([x * x for x in win]) / len(win) self.assertAlmostEqual(power, 1.0) + if __name__ == '__main__': gr_unittest.run(test_window) |