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-analog/python/analog/qa_probe_avg_mag_sqrd.py | |
| 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-analog/python/analog/qa_probe_avg_mag_sqrd.py')
| -rw-r--r-- | gr-analog/python/analog/qa_probe_avg_mag_sqrd.py | 40 |
1 files changed, 31 insertions, 9 deletions
diff --git a/gr-analog/python/analog/qa_probe_avg_mag_sqrd.py b/gr-analog/python/analog/qa_probe_avg_mag_sqrd.py index cc2bc49099..721b9a6f6b 100644 --- a/gr-analog/python/analog/qa_probe_avg_mag_sqrd.py +++ b/gr-analog/python/analog/qa_probe_avg_mag_sqrd.py @@ -13,20 +13,24 @@ import math from gnuradio import gr, gr_unittest, analog, blocks + def avg_mag_sqrd_c(x, alpha): - y = [0,] + y = [0, ] for xi in x: - tmp = alpha*(xi.real*xi.real + xi.imag*xi.imag) + (1-alpha)*y[-1] + tmp = alpha * (xi.real * xi.real + xi.imag * + xi.imag) + (1 - alpha) * y[-1] y.append(tmp) return y + def avg_mag_sqrd_f(x, alpha): - y = [0,] + y = [0, ] for xi in x: - tmp = alpha*(xi*xi) + (1-alpha)*y[-1] + tmp = alpha * (xi * xi) + (1 - alpha) * y[-1] y.append(tmp) return y + class test_probe_avg_mag_sqrd(gr_unittest.TestCase): def setUp(self): @@ -37,8 +41,17 @@ class test_probe_avg_mag_sqrd(gr_unittest.TestCase): def test_c_001(self): alpha = 0.0001 - src_data = [1.0+1.0j, 2.0+2.0j, 3.0+3.0j, 4.0+4.0j, 5.0+5.0j, - 6.0+6.0j, 7.0+7.0j, 8.0+8.0j, 9.0+9.0j, 10.0+10.0j] + src_data = [ + 1.0 + 1.0j, + 2.0 + 2.0j, + 3.0 + 3.0j, + 4.0 + 4.0j, + 5.0 + 5.0j, + 6.0 + 6.0j, + 7.0 + 7.0j, + 8.0 + 8.0j, + 9.0 + 9.0j, + 10.0 + 10.0j] expected_result = avg_mag_sqrd_c(src_data, alpha)[-1] src = blocks.vector_source_c(src_data) @@ -52,8 +65,17 @@ class test_probe_avg_mag_sqrd(gr_unittest.TestCase): def test_cf_002(self): alpha = 0.0001 - src_data = [1.0+1.0j, 2.0+2.0j, 3.0+3.0j, 4.0+4.0j, 5.0+5.0j, - 6.0+6.0j, 7.0+7.0j, 8.0+8.0j, 9.0+9.0j, 10.0+10.0j] + src_data = [ + 1.0 + 1.0j, + 2.0 + 2.0j, + 3.0 + 3.0j, + 4.0 + 4.0j, + 5.0 + 5.0j, + 6.0 + 6.0j, + 7.0 + 7.0j, + 8.0 + 8.0j, + 9.0 + 9.0j, + 10.0 + 10.0j] expected_result = avg_mag_sqrd_c(src_data, alpha)[0:-1] src = blocks.vector_source_c(src_data) @@ -82,6 +104,6 @@ class test_probe_avg_mag_sqrd(gr_unittest.TestCase): result_data = op.level() self.assertAlmostEqual(expected_result, result_data, 5) + if __name__ == '__main__': gr_unittest.run(test_probe_avg_mag_sqrd) - |