From 93bfee96c483058d5288533a5f55ef4947b9b09d Mon Sep 17 00:00:00 2001
From: Marcus Müller <mmueller@gnuradio.org>
Date: Fri, 16 Jul 2021 20:33:33 +0200
Subject: digital: at least superficially test chunk_to_symbols dimensions>1
MIME-Version: 1.0
Content-Type: text/plain; charset=UTF-8
Content-Transfer-Encoding: 8bit
Signed-off-by: Marcus Müller <mmueller@gnuradio.org>
---
gr-digital/python/digital/qa_chunks_to_symbols.py | 44 +++++++++++++++++++++++
1 file changed, 44 insertions(+)
diff --git a/gr-digital/python/digital/qa_chunks_to_symbols.py b/gr-digital/python/digital/qa_chunks_to_symbols.py
index 1616f29e08..5b0ae8e04e 100644
--- a/gr-digital/python/digital/qa_chunks_to_symbols.py
+++ b/gr-digital/python/digital/qa_chunks_to_symbols.py
@@ -56,6 +56,50 @@ class test_chunks_to_symbols(gr_unittest.TestCase):
actual_result = dst.data()
self.assertEqual(expected_result, actual_result)
+ def test_bf_2d(self):
+ maxval = 4
+ dimensions = 2
+ const = list(range(maxval * dimensions))
+ src_data = [v * 13 % maxval for v in range(maxval)]
+ expected_result = []
+ for data in src_data:
+ for i in range(dimensions):
+ expected_result += [const[data * dimensions + i]]
+
+ self.assertEqual(len(src_data) * dimensions, len(expected_result))
+ src = blocks.vector_source_b(src_data)
+ op = digital.chunks_to_symbols_bf(const, dimensions)
+
+ dst = blocks.vector_sink_f()
+ self.tb.connect(src, op)
+ self.tb.connect(op, dst)
+ self.tb.run()
+
+ actual_result = dst.data()
+ self.assertEqual(expected_result, actual_result)
+
+ def test_bf_3d(self):
+ maxval = 48
+ dimensions = 3
+ const = list(range(maxval * dimensions))
+ src_data = [v * 13 % maxval for v in range(maxval)]
+ expected_result = []
+ for data in src_data:
+ for i in range(dimensions):
+ expected_result += [const[data * dimensions + i]]
+
+ self.assertEqual(len(src_data) * dimensions, len(expected_result))
+ src = blocks.vector_source_b(src_data)
+ op = digital.chunks_to_symbols_bf(const, dimensions)
+
+ dst = blocks.vector_sink_f()
+ self.tb.connect(src, op)
+ self.tb.connect(op, dst)
+ self.tb.run()
+
+ actual_result = dst.data()
+ self.assertEqual(expected_result, actual_result)
+
def test_ic_003(self):
const = [1 + 0j, 0 + 1j,
-1 + 0j, 0 - 1j]
--
cgit v1.2.3