summaryrefslogtreecommitdiff
path: root/gr-blocks/python/blocks/qa_add_mult_v.py
diff options
context:
space:
mode:
authormormj <mormjb@gmail.com>2020-10-30 10:59:50 -0400
committerMarcus Müller <marcus@hostalia.de>2020-10-30 17:52:53 +0100
commit7a0948ba85758fba1cc3858ef99bfa600dcc7416 (patch)
tree610d7f9d773a193562def6df2d4b50f1bb3b3f86 /gr-blocks/python/blocks/qa_add_mult_v.py
parent12192ee7d58de95ddca35a3e93bfc172bdb5c820 (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-blocks/python/blocks/qa_add_mult_v.py')
-rw-r--r--gr-blocks/python/blocks/qa_add_mult_v.py195
1 files changed, 127 insertions, 68 deletions
diff --git a/gr-blocks/python/blocks/qa_add_mult_v.py b/gr-blocks/python/blocks/qa_add_mult_v.py
index dd872ee8ce..39f11bc4be 100644
--- a/gr-blocks/python/blocks/qa_add_mult_v.py
+++ b/gr-blocks/python/blocks/qa_add_mult_v.py
@@ -11,6 +11,7 @@
from gnuradio import gr, gr_unittest, blocks
+
class test_add_mult_v(gr_unittest.TestCase):
def setUp(self):
@@ -20,7 +21,7 @@ class test_add_mult_v(gr_unittest.TestCase):
self.tb = None
def help_ss(self, size, src_data, exp_data, op):
- for s in zip(list(range(len (src_data))), src_data):
+ for s in zip(list(range(len(src_data))), src_data):
src = blocks.vector_source_s(s[1])
srcv = blocks.stream_to_vector(gr.sizeof_short, size)
self.tb.connect(src, srcv)
@@ -34,7 +35,7 @@ class test_add_mult_v(gr_unittest.TestCase):
self.assertEqual(exp_data, result_data)
def help_ii(self, size, src_data, exp_data, op):
- for s in zip(list(range(len (src_data))), src_data):
+ for s in zip(list(range(len(src_data))), src_data):
src = blocks.vector_source_i(s[1])
srcv = blocks.stream_to_vector(gr.sizeof_int, size)
self.tb.connect(src, srcv)
@@ -48,7 +49,7 @@ class test_add_mult_v(gr_unittest.TestCase):
self.assertEqual(exp_data, result_data)
def help_ff(self, size, src_data, exp_data, op):
- for s in zip(list(range(len (src_data))), src_data):
+ for s in zip(list(range(len(src_data))), src_data):
src = blocks.vector_source_f(s[1])
srcv = blocks.stream_to_vector(gr.sizeof_float, size)
self.tb.connect(src, srcv)
@@ -62,7 +63,7 @@ class test_add_mult_v(gr_unittest.TestCase):
self.assertEqual(exp_data, result_data)
def help_cc(self, size, src_data, exp_data, op):
- for s in zip(list(range(len (src_data))), src_data):
+ for s in zip(list(range(len(src_data))), src_data):
src = blocks.vector_source_c(s[1])
srcv = blocks.stream_to_vector(gr.sizeof_gr_complex, size)
self.tb.connect(src, srcv)
@@ -118,10 +119,10 @@ class test_add_mult_v(gr_unittest.TestCase):
# add_vXX
def test_add_vss_one(self):
- src1_data = [1,]
- src2_data = [2,]
- src3_data = [3,]
- expected_result = [6,]
+ src1_data = [1, ]
+ src2_data = [2, ]
+ src3_data = [3, ]
+ expected_result = [6, ]
op = blocks.add_ss(1)
self.help_ss(1, (src1_data, src2_data, src3_data), expected_result, op)
@@ -134,10 +135,10 @@ class test_add_mult_v(gr_unittest.TestCase):
self.help_ss(5, (src1_data, src2_data, src3_data), expected_result, op)
def test_add_vii_one(self):
- src1_data = [1,]
- src2_data = [2,]
- src3_data = [3,]
- expected_result = [6,]
+ src1_data = [1, ]
+ src2_data = [2, ]
+ src3_data = [3, ]
+ expected_result = [6, ]
op = blocks.add_ii(1)
self.help_ii(1, (src1_data, src2_data, src3_data), expected_result, op)
@@ -150,10 +151,10 @@ class test_add_mult_v(gr_unittest.TestCase):
self.help_ii(5, (src1_data, src2_data, src3_data), expected_result, op)
def test_add_vff_one(self):
- src1_data = [1.0,]
- src2_data = [2.0,]
- src3_data = [3.0,]
- expected_result = [6.0,]
+ src1_data = [1.0, ]
+ src2_data = [2.0, ]
+ src3_data = [3.0, ]
+ expected_result = [6.0, ]
op = blocks.add_ff(1)
self.help_ff(1, (src1_data, src2_data, src3_data), expected_result, op)
@@ -166,27 +167,47 @@ class test_add_mult_v(gr_unittest.TestCase):
self.help_ff(5, (src1_data, src2_data, src3_data), expected_result, op)
def test_add_vcc_one(self):
- src1_data = [1.0+2.0j,]
- src2_data = [3.0+4.0j,]
- src3_data = [5.0+6.0j,]
- expected_result = [9.0+12j,]
+ src1_data = [1.0 + 2.0j, ]
+ src2_data = [3.0 + 4.0j, ]
+ src3_data = [5.0 + 6.0j, ]
+ expected_result = [9.0 + 12j, ]
op = blocks.add_cc(1)
self.help_cc(1, (src1_data, src2_data, src3_data), expected_result, op)
def test_add_vcc_five(self):
- src1_data = [1.0+2.0j, 3.0+4.0j, 5.0+6.0j, 7.0+8.0j, 9.0+10.0j]
- src2_data = [11.0+12.0j, 13.0+14.0j, 15.0+16.0j, 17.0+18.0j, 19.0+20.0j]
- src3_data = [21.0+22.0j, 23.0+24.0j, 25.0+26.0j, 27.0+28.0j, 29.0+30.0j]
- expected_result = [33.0+36.0j, 39.0+42.0j, 45.0+48.0j, 51.0+54.0j, 57.0+60.0j]
+ src1_data = [
+ 1.0 + 2.0j,
+ 3.0 + 4.0j,
+ 5.0 + 6.0j,
+ 7.0 + 8.0j,
+ 9.0 + 10.0j]
+ src2_data = [
+ 11.0 + 12.0j,
+ 13.0 + 14.0j,
+ 15.0 + 16.0j,
+ 17.0 + 18.0j,
+ 19.0 + 20.0j]
+ src3_data = [
+ 21.0 + 22.0j,
+ 23.0 + 24.0j,
+ 25.0 + 26.0j,
+ 27.0 + 28.0j,
+ 29.0 + 30.0j]
+ expected_result = [
+ 33.0 + 36.0j,
+ 39.0 + 42.0j,
+ 45.0 + 48.0j,
+ 51.0 + 54.0j,
+ 57.0 + 60.0j]
op = blocks.add_cc(5)
self.help_cc(5, (src1_data, src2_data, src3_data), expected_result, op)
# add_const_vXX
def test_add_const_vss_one(self):
- src_data = [1,]
+ src_data = [1, ]
op = blocks.add_const_vss((2,))
- exp_data = [3,]
+ exp_data = [3, ]
self.help_const_ss(src_data, exp_data, op)
def test_add_const_vss_five(self):
@@ -196,9 +217,9 @@ class test_add_mult_v(gr_unittest.TestCase):
self.help_const_ss(src_data, exp_data, op)
def test_add_const_vii_one(self):
- src_data = [1,]
+ src_data = [1, ]
op = blocks.add_const_vii((2,))
- exp_data = [3,]
+ exp_data = [3, ]
self.help_const_ii(src_data, exp_data, op)
def test_add_const_vii_five(self):
@@ -208,9 +229,9 @@ class test_add_mult_v(gr_unittest.TestCase):
self.help_const_ii(src_data, exp_data, op)
def test_add_const_vff_one(self):
- src_data = [1.0,]
+ src_data = [1.0, ]
op = blocks.add_const_vff((2.0,))
- exp_data = [3.0,]
+ exp_data = [3.0, ]
self.help_const_ff(src_data, exp_data, op)
def test_add_const_vff_five(self):
@@ -220,24 +241,39 @@ class test_add_mult_v(gr_unittest.TestCase):
self.help_const_ff(src_data, exp_data, op)
def test_add_const_vcc_one(self):
- src_data = [1.0+2.0j,]
- op = blocks.add_const_vcc((2.0+3.0j,))
- exp_data = [3.0+5.0j,]
+ src_data = [1.0 + 2.0j, ]
+ op = blocks.add_const_vcc((2.0 + 3.0j,))
+ exp_data = [3.0 + 5.0j, ]
self.help_const_cc(src_data, exp_data, op)
def test_add_const_vcc_five(self):
- src_data = [1.0+2.0j, 3.0+4.0j, 5.0+6.0j, 7.0+8.0j, 9.0+10.0j]
- op = blocks.add_const_vcc((11.0+12.0j, 13.0+14.0j, 15.0+16.0j, 17.0+18.0j, 19.0+20.0j))
- exp_data = [12.0+14.0j, 16.0+18.0j, 20.0+22.0j, 24.0+26.0j, 28.0+30.0j]
+ src_data = [
+ 1.0 + 2.0j,
+ 3.0 + 4.0j,
+ 5.0 + 6.0j,
+ 7.0 + 8.0j,
+ 9.0 + 10.0j]
+ op = blocks.add_const_vcc(
+ (11.0 + 12.0j,
+ 13.0 + 14.0j,
+ 15.0 + 16.0j,
+ 17.0 + 18.0j,
+ 19.0 + 20.0j))
+ exp_data = [
+ 12.0 + 14.0j,
+ 16.0 + 18.0j,
+ 20.0 + 22.0j,
+ 24.0 + 26.0j,
+ 28.0 + 30.0j]
self.help_const_cc(src_data, exp_data, op)
# multiply_vXX
def test_multiply_vss_one(self):
- src1_data = [1,]
- src2_data = [2,]
- src3_data = [3,]
- expected_result = [6,]
+ src1_data = [1, ]
+ src2_data = [2, ]
+ src3_data = [3, ]
+ expected_result = [6, ]
op = blocks.multiply_ss(1)
self.help_ss(1, (src1_data, src2_data, src3_data), expected_result, op)
@@ -250,10 +286,10 @@ class test_add_mult_v(gr_unittest.TestCase):
self.help_ss(5, (src1_data, src2_data, src3_data), expected_result, op)
def test_multiply_vii_one(self):
- src1_data = [1,]
- src2_data = [2,]
- src3_data = [3,]
- expected_result = [6,]
+ src1_data = [1, ]
+ src2_data = [2, ]
+ src3_data = [3, ]
+ expected_result = [6, ]
op = blocks.multiply_ii(1)
self.help_ii(1, (src1_data, src2_data, src3_data), expected_result, op)
@@ -266,10 +302,10 @@ class test_add_mult_v(gr_unittest.TestCase):
self.help_ii(5, (src1_data, src2_data, src3_data), expected_result, op)
def test_multiply_vff_one(self):
- src1_data = [1.0,]
- src2_data = [2.0,]
- src3_data = [3.0,]
- expected_result = [6.0,]
+ src1_data = [1.0, ]
+ src2_data = [2.0, ]
+ src3_data = [3.0, ]
+ expected_result = [6.0, ]
op = blocks.multiply_ff(1)
self.help_ff(1, (src1_data, src2_data, src3_data), expected_result, op)
@@ -282,27 +318,43 @@ class test_add_mult_v(gr_unittest.TestCase):
self.help_ff(5, (src1_data, src2_data, src3_data), expected_result, op)
def test_multiply_vcc_one(self):
- src1_data = [1.0+2.0j,]
- src2_data = [3.0+4.0j,]
- src3_data = [5.0+6.0j,]
- expected_result = [-85+20j,]
+ src1_data = [1.0 + 2.0j, ]
+ src2_data = [3.0 + 4.0j, ]
+ src3_data = [5.0 + 6.0j, ]
+ expected_result = [-85 + 20j, ]
op = blocks.multiply_cc(1)
self.help_cc(1, (src1_data, src2_data, src3_data), expected_result, op)
def test_multiply_vcc_five(self):
- src1_data = [1.0+2.0j, 3.0+4.0j, 5.0+6.0j, 7.0+8.0j, 9.0+10.0j]
- src2_data = [11.0+12.0j, 13.0+14.0j, 15.0+16.0j, 17.0+18.0j, 19.0+20.0j]
- src3_data = [21.0+22.0j, 23.0+24.0j, 25.0+26.0j, 27.0+28.0j, 29.0+30.0j]
- expected_result = [-1021.0+428.0j, -2647.0+1754.0j, -4945.0+3704.0j, -8011.0+6374.0j, -11941.0+9860.0j]
+ src1_data = [
+ 1.0 + 2.0j,
+ 3.0 + 4.0j,
+ 5.0 + 6.0j,
+ 7.0 + 8.0j,
+ 9.0 + 10.0j]
+ src2_data = [
+ 11.0 + 12.0j,
+ 13.0 + 14.0j,
+ 15.0 + 16.0j,
+ 17.0 + 18.0j,
+ 19.0 + 20.0j]
+ src3_data = [
+ 21.0 + 22.0j,
+ 23.0 + 24.0j,
+ 25.0 + 26.0j,
+ 27.0 + 28.0j,
+ 29.0 + 30.0j]
+ expected_result = [-1021.0 + 428.0j, -2647.0 + 1754.0j, - \
+ 4945.0 + 3704.0j, -8011.0 + 6374.0j, -11941.0 + 9860.0j]
op = blocks.multiply_cc(5)
self.help_cc(5, (src1_data, src2_data, src3_data), expected_result, op)
# multiply_const_vXX
def test_multiply_const_vss_one(self):
- src_data = [2,]
+ src_data = [2, ]
op = blocks.multiply_const_vss((3,))
- exp_data = [6,]
+ exp_data = [6, ]
self.help_const_ss(src_data, exp_data, op)
def test_multiply_const_vss_five(self):
@@ -312,9 +364,9 @@ class test_add_mult_v(gr_unittest.TestCase):
self.help_const_ss(src_data, exp_data, op)
def test_multiply_const_vii_one(self):
- src_data = [2,]
+ src_data = [2, ]
op = blocks.multiply_const_vii((3,))
- exp_data = [6,]
+ exp_data = [6, ]
self.help_const_ii(src_data, exp_data, op)
def test_multiply_const_vii_five(self):
@@ -324,9 +376,9 @@ class test_add_mult_v(gr_unittest.TestCase):
self.help_const_ii(src_data, exp_data, op)
def test_multiply_const_vff_one(self):
- src_data = [2.0,]
+ src_data = [2.0, ]
op = blocks.multiply_const_vff((3.0,))
- exp_data = [6.0,]
+ exp_data = [6.0, ]
self.help_const_ff(src_data, exp_data, op)
def test_multiply_const_vff_five(self):
@@ -336,15 +388,22 @@ class test_add_mult_v(gr_unittest.TestCase):
self.help_const_ff(src_data, exp_data, op)
def test_multiply_const_vcc_one(self):
- src_data = [1.0+2.0j,]
- op = blocks.multiply_const_vcc((2.0+3.0j,))
- exp_data = [-4.0+7.0j,]
+ src_data = [1.0 + 2.0j, ]
+ op = blocks.multiply_const_vcc((2.0 + 3.0j,))
+ exp_data = [-4.0 + 7.0j, ]
self.help_const_cc(src_data, exp_data, op)
def test_multiply_const_vcc_five(self):
- src_data = [1.0+2.0j, 3.0+4.0j, 5.0+6.0j, 7.0+8.0j, 9.0+10.0j]
- op = blocks.multiply_const_vcc((11.0+12.0j, 13.0+14.0j, 15.0+16.0j, 17.0+18.0j, 19.0+20.0j))
- exp_data = [-13.0+34.0j, -17.0+94.0j, -21.0+170.0j, -25.0+262.0j, -29.0+370.0j]
+ src_data = [
+ 1.0 + 2.0j,
+ 3.0 + 4.0j,
+ 5.0 + 6.0j,
+ 7.0 + 8.0j,
+ 9.0 + 10.0j]
+ op = blocks.multiply_const_vcc(
+ (11.0 + 12.0j, 13.0 + 14.0j, 15.0 + 16.0j, 17.0 + 18.0j, 19.0 + 20.0j))
+ exp_data = [-13.0 + 34.0j, -17.0 + 94.0j, -
+ 21.0 + 170.0j, -25.0 + 262.0j, -29.0 + 370.0j]
self.help_const_cc(src_data, exp_data, op)