From 4861d33e9ab4468bb9f194b220175a6770c12023 Mon Sep 17 00:00:00 2001
From: Sebastian Wehling-Benatelli <sebastian.wehling@rub.de>
Date: Sun, 16 Apr 2023 09:58:51 +0200
Subject: [PATCH] bugfix: add tests to key_for_set_value

---
 pylot/core/util/utils.py | 10 ++++++++--
 1 file changed, 8 insertions(+), 2 deletions(-)

diff --git a/pylot/core/util/utils.py b/pylot/core/util/utils.py
index e04118e1..eaeaf2a0 100644
--- a/pylot/core/util/utils.py
+++ b/pylot/core/util/utils.py
@@ -634,13 +634,19 @@ def is_iterable(obj):
 def key_for_set_value(d):
     """
     takes a dictionary and returns the first key for which's value the
-    boolean is True
+    boolean representation is True
     :param d: dictionary containing values
     :type d: dict
     :return: key to the first non-False value found; None if no value's
     boolean equals True
-    :rtype:
+    :rtype: bool or NoneType
+
+    >>> key_for_set_value({'one': 0, 'two': 1})
+    'two'
+    >>> print(key_for_set_value({1: 0, 2: False}))
+    None
     """
+    assert type(d) is dict, "Function only defined for inputs of type 'dict'."
     r = None
     for k, v in d.items():
         if v: