From 52a9d49acef2ec124d52e7e223dc16e767af92c7 Mon Sep 17 00:00:00 2001
From: Marcus Müller <mmueller@gnuradio.org>
Date: Fri, 16 Jul 2021 20:52:00 +0200
Subject: digital: more details in constellation test
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_constellation.py | 15 ++++++++++++++-
 1 file changed, 14 insertions(+), 1 deletion(-)

(limited to 'gr-digital/python/digital')

diff --git a/gr-digital/python/digital/qa_constellation.py b/gr-digital/python/digital/qa_constellation.py
index 0e3460a142..d13dae376e 100644
--- a/gr-digital/python/digital/qa_constellation.py
+++ b/gr-digital/python/digital/qa_constellation.py
@@ -12,6 +12,7 @@
 import random
 import math
 from cmath import exp, pi, log, sqrt
+import numpy
 
 from gnuradio import gr, gr_unittest, digital, blocks
 from gnuradio.digital.utils import mod_codes
@@ -195,7 +196,19 @@ class test_constellation(gr_unittest.TestCase):
                 data = dst.data()
                 # Don't worry about cut off data for now.
                 first = constellation.bits_per_symbol()
-                self.assertEqual(self.src_data[first:len(data)], data[first:])
+                equality = all(numpy.equal(self.src_data[first:len(data)],
+                                           data[first:]))
+                if not equality:
+                    msg = "Constellations mismatched. " + \
+                        f"{type(constellation)}; " + \
+                        f"Differential? {differential}; " + \
+                        f"{len(constellation.points())} " +\
+                        "Constellation points: " + \
+                        f"{constellation.points()};"
+                    self.assertEqual(self.src_data[first:len(data)],
+                                     data[first:],
+                                     msg=msg)
+
 
     def test_soft_qpsk_gen(self):
         prec = 8
-- 
cgit v1.2.3