diff options
Diffstat (limited to 'gnuradio-runtime/python/gnuradio/gr_unittest.py')
| -rw-r--r-- | gnuradio-runtime/python/gnuradio/gr_unittest.py | 19 |
1 files changed, 19 insertions, 0 deletions
diff --git a/gnuradio-runtime/python/gnuradio/gr_unittest.py b/gnuradio-runtime/python/gnuradio/gr_unittest.py index 6177f0f32f..ebd47019a5 100644 --- a/gnuradio-runtime/python/gnuradio/gr_unittest.py +++ b/gnuradio-runtime/python/gnuradio/gr_unittest.py @@ -110,6 +110,25 @@ class TestCase(unittest.TestCase): self.assertComplexAlmostEqual2(x, y, abs_eps, rel_eps, msg) for (x, y) in zip(a, b) ]) + + + def assertSequenceEqualGR(self, data_in, data_out): + """ + Note this function exists because of this bug: https://bugs.python.org/issue19217 + Calling self.assertEqual(seqA, seqB) can hang if seqA and seqB are not equal. + """ + if len(data_in) != len(data_out): + print('Lengths do not match: {:d} -- {:d}'.format(len(data_in), len(data_out))) + self.assertTrue(len(data_in) == len(data_out)) + total_miscompares = 0 + for idx, item in enumerate(zip(data_in, data_out)): + if item[0] != item[1]: + total_miscompares += 1 + print('Miscompare at: {:d} ({:d} -- {:d})'.format(idx, item[0], item[1])) + if total_miscompares > 0: + print('Total miscompares: {:d}'.format(total_miscompares)) + self.assertTrue(total_miscompares == 0) + def waitFor( self, |