rcovered_pk: fix c/h for partial in selftests

RfidResearchGroup · Dec 14, 2024 · 86de7f5 · 86de7f5
1 parent f26727e
commit 86de7f5
Showing 1 changed file with 2 additions and 1 deletion.
diff --git a/tools/recover_pk.py b/tools/recover_pk.py
@@ -229,7 +229,8 @@ def selftests():
         elif len(t['samples'])//2 == 1:
             recovereds = [(c, h) for c, h, pk in list(recovered)
                           if t['pk'].lower() == binascii.hexlify(pk).decode('utf8').lower()]
-            if len(recovereds) > 0:
+            if len(recovereds) == 1:
+                c, h = recovereds[0]
                 print("%15s/%-8s ( %s ) partial" % (c, h, color('ok', fg='green')))
             else:
                 succeeded = False