From 162fe443d15f2e4b9fd781741ff7d8496843c149 Mon Sep 17 00:00:00 2001
From: Daniel Rojas <github@danielrojas.net>
Date: Fri, 5 Aug 2022 18:45:48 +0200
Subject: [PATCH] Add suggestions from #186

---
 src/wireviz/wv_utils.py | 7 ++++---
 1 file changed, 4 insertions(+), 3 deletions(-)

diff --git a/src/wireviz/wv_utils.py b/src/wireviz/wv_utils.py
index a2bff03f..58209e12 100644
--- a/src/wireviz/wv_utils.py
+++ b/src/wireviz/wv_utils.py
@@ -70,9 +70,10 @@ def expand(yaml_data):
 
 
 def get_single_key_and_value(d: dict):
-    k = list(d.keys())[0]
-    v = d[k]
-    return (k, v)
+    # used for defining a line in a harness' connection set
+    # E.g. for the YAML input `- X1: 1`
+    # this function returns a tuple in the form ("X1", "1")
+    return next(iter(d.items()))
 
 
 def int2tuple(inp):