Fix CustomRegisterTest.

[riscv-tests.git] / debug / gdbserver.py

diff --git a/debug/gdbserver.py b/debug/gdbserver.py
index c362e1dc03dea441af582c093cbf490eba36101e..1d58f769c2dbe0718cd691e210d6ca0537611aec 100755 (executable)
--- a/debug/gdbserver.py
+++ b/debug/gdbserver.py
@@ -142,7 +142,8 @@ class CustomRegisterTest(SimpleRegisterTest):
         return self.target.implements_custom_test
 
     def check_custom(self, magic):
-        regs = self.gdb.info_registers("custom")
+        regs = {k: v for k, v in self.gdb.info_registers("all").iteritems()
+                if k.startswith("custom")}
         assertEqual(set(regs.keys()),
                 set(("custom1",
                     "custom12345",