diff options
Diffstat (limited to 'gr-digital/python/digital/qa_glfsr_source.py')
| -rw-r--r-- | gr-digital/python/digital/qa_glfsr_source.py | 40 |
1 files changed, 25 insertions, 15 deletions
diff --git a/gr-digital/python/digital/qa_glfsr_source.py b/gr-digital/python/digital/qa_glfsr_source.py index 7505f5bdaa..dbe488399b 100644 --- a/gr-digital/python/digital/qa_glfsr_source.py +++ b/gr-digital/python/digital/qa_glfsr_source.py @@ -11,6 +11,7 @@ from gnuradio import gr, gr_unittest, digital, blocks + class test_glfsr_source(gr_unittest.TestCase): def setUp(self): @@ -22,7 +23,7 @@ class test_glfsr_source(gr_unittest.TestCase): def test_000_make_b(self): src = digital.glfsr_source_b(16) self.assertEqual(src.mask(), 0x8016) - self.assertEqual(src.period(), 2**16-1) + self.assertEqual(src.period(), 2**16 - 1) def test_001_degree_b(self): self.assertRaises(RuntimeError, @@ -31,53 +32,62 @@ class test_glfsr_source(gr_unittest.TestCase): lambda: digital.glfsr_source_b(33)) def test_002_correlation_b(self): - for degree in range(1,11): # Higher degrees take too long to correlate + for degree in range( + 1, 11): # Higher degrees take too long to correlate src = digital.glfsr_source_b(degree, False) - b2f = digital.chunks_to_symbols_bf((-1.0,1.0), 1) + b2f = digital.chunks_to_symbols_bf((-1.0, 1.0), 1) dst = blocks.vector_sink_f() - del self.tb # Discard existing top block + del self.tb # Discard existing top block self.tb = gr.top_block() self.tb.connect(src, b2f, dst) self.tb.run() self.tb.disconnect_all() actual_result = dst.data() R = auto_correlate(actual_result) - self.assertEqual(R[0], float(len(R))) # Auto-correlation peak at origin - for i in range(len(R)-1): - self.assertEqual(R[i+1], -1.0) # Auto-correlation minimum everywhere else + # Auto-correlation peak at origin + self.assertEqual(R[0], float(len(R))) + for i in range(len(R) - 1): + # Auto-correlation minimum everywhere else + self.assertEqual(R[i + 1], -1.0) def test_003_make_f(self): src = digital.glfsr_source_f(16) self.assertEqual(src.mask(), 0x8016) - self.assertEqual(src.period(), 2**16-1) + self.assertEqual(src.period(), 2**16 - 1) def test_004_degree_f(self): self.assertRaises(RuntimeError, lambda: digital.glfsr_source_f(0)) self.assertRaises(RuntimeError, lambda: digital.glfsr_source_f(33)) + def test_005_correlation_f(self): - for degree in range(1,11): # Higher degrees take too long to correlate + for degree in range( + 1, 11): # Higher degrees take too long to correlate src = digital.glfsr_source_f(degree, False) dst = blocks.vector_sink_f() - del self.tb # Discard existing top block + del self.tb # Discard existing top block self.tb = gr.top_block() self.tb.connect(src, dst) self.tb.run() actual_result = dst.data() R = auto_correlate(actual_result) - self.assertEqual(R[0], float(len(R))) # Auto-correlation peak at origin - for i in range(len(R)-1): - self.assertEqual(R[i+1], -1.0) # Auto-correlation minimum everywhere else + # Auto-correlation peak at origin + self.assertEqual(R[0], float(len(R))) + for i in range(len(R) - 1): + # Auto-correlation minimum everywhere else + self.assertEqual(R[i + 1], -1.0) + def auto_correlate(data): l = len(data) - R = [0,]*l + R = [0, ] * l for lag in range(l): for i in range(l): - R[lag] += data[i]*data[i-lag] + R[lag] += data[i] * data[i - lag] return R + if __name__ == '__main__': gr_unittest.run(test_glfsr_source) |