summaryrefslogtreecommitdiff
path: root/gr-fec/python
diff options
context:
space:
mode:
authorJohannes Demel <ufcsy@student.kit.edu>2015-07-17 14:59:06 +0200
committerJohannes Demel <ufcsy@student.kit.edu>2015-07-17 14:59:06 +0200
commitb781fe746fb258af6ebc49499a76369665272f9d (patch)
tree57b2d2f8b50a4efa1d2749f3d1340716c19fe391 /gr-fec/python
parentad6d9ca46e5d6a24ba9625ab93e0c48b78f57e9e (diff)
fec: fixed QA for ber_bf.
Diffstat (limited to 'gr-fec/python')
-rw-r--r--gr-fec/python/fec/qa_ber_bf.py57
1 files changed, 29 insertions, 28 deletions
diff --git a/gr-fec/python/fec/qa_ber_bf.py b/gr-fec/python/fec/qa_ber_bf.py
index 7cd2871eff..3e7919e054 100644
--- a/gr-fec/python/fec/qa_ber_bf.py
+++ b/gr-fec/python/fec/qa_ber_bf.py
@@ -20,13 +20,13 @@
# Boston, MA 02110-1301, USA.
#
-from gnuradio import gr, gr_unittest
+from gnuradio import gr, gr_unittest, blocks
import fec_swig as fec
-import blocks_swig as blocks
-import numpy, copy
+import numpy
+import copy
-class test_ber_bf(gr_unittest.TestCase):
+class test_ber_bf(gr_unittest.TestCase):
def setUp(self):
self.tb = gr.top_block()
@@ -45,16 +45,16 @@ class test_ber_bf(gr_unittest.TestCase):
src0 = blocks.vector_source_b(data0)
src1 = blocks.vector_source_b(data1)
- op = fec.ber_bf(mode)
- dst = blocks.vector_sink_f()
+ op = fec.ber_bf(mode)
+ dst = blocks.vector_sink_f()
- self.tb.connect(src0, (op,0))
- self.tb.connect(src1, (op,1))
+ self.tb.connect(src0, (op, 0))
+ self.tb.connect(src1, (op, 1))
self.tb.connect(op, dst)
- self.tb.run()
+ self.tb.run()
data = dst.data()
- expected_result = [numpy.log10(1.0/(8.0*N)),]
+ expected_result = [numpy.log10(1.0 / (8.0 * N)), ]
self.assertFloatTuplesAlmostEqual(expected_result, data, 5)
@@ -70,16 +70,16 @@ class test_ber_bf(gr_unittest.TestCase):
src0 = blocks.vector_source_b(data0)
src1 = blocks.vector_source_b(data1)
- op = fec.ber_bf(mode, 1)
- dst = blocks.vector_sink_f()
+ op = fec.ber_bf(mode, 1)
+ dst = blocks.vector_sink_f()
- self.tb.connect(src0, (op,0))
- self.tb.connect(src1, (op,1))
+ self.tb.connect(src0, (op, 0))
+ self.tb.connect(src1, (op, 1))
self.tb.connect(op, dst)
- self.tb.run()
+ self.tb.run()
data = dst.data()
- expected_result = [numpy.log10(1.0/(8.0*N)),]
+ expected_result = [numpy.log10(1.0 / (8.0 * N)), ]
self.assertFloatTuplesAlmostEqual(expected_result, data, 5)
@@ -95,16 +95,16 @@ class test_ber_bf(gr_unittest.TestCase):
src0 = blocks.vector_source_b(data0)
src1 = blocks.vector_source_b(data1)
- op = fec.ber_bf(mode, 1, -2.0)
- dst = blocks.vector_sink_f()
+ op = fec.ber_bf(mode, 1, -2.0)
+ dst = blocks.vector_sink_f()
- self.tb.connect(src0, (op,0))
- self.tb.connect(src1, (op,1))
+ self.tb.connect(src0, (op, 0))
+ self.tb.connect(src1, (op, 1))
self.tb.connect(op, dst)
- self.tb.run()
+ self.tb.run()
data = dst.data()
- expected_result = [numpy.log10(8.0/(8.0*N)),]
+ expected_result = [numpy.log10(8.0 / (8.0 * N)), ]
self.assertFloatTuplesAlmostEqual(expected_result, data, 5)
@@ -121,21 +121,22 @@ class test_ber_bf(gr_unittest.TestCase):
src0 = blocks.vector_source_b(data0)
src1 = blocks.vector_source_b(data1)
- op = fec.ber_bf(mode, 10, -2.0)
- dst = blocks.vector_sink_f()
+ op = fec.ber_bf(mode, 10, -2.0)
+ dst = blocks.vector_sink_f()
- self.tb.connect(src0, (op,0))
- self.tb.connect(src1, (op,1))
+ self.tb.connect(src0, (op, 0))
+ self.tb.connect(src1, (op, 1))
self.tb.connect(op, dst)
- self.tb.run()
+ self.tb.run()
data = dst.data()
- expected_result = [-2.0,]
+ expected_result = [-2.0, ]
print data
print expected_result
self.assertFloatTuplesAlmostEqual(expected_result, data, 5)
+
if __name__ == '__main__':
gr_unittest.run(test_ber_bf, "test_ber_bf.xml")