Make test output slightly more readable if slot missing (home-assista…

…nt#152)
cvladan · Dec 27, 2022 · e1b1ce8 · e1b1ce8
1 parent 8d859ca
commit e1b1ce8
Showing 1 changed file with 4 additions and 1 deletion.
diff --git a/tests/test_language_sentences.py b/tests/test_language_sentences.py
@@ -57,7 +57,10 @@ def do_test_language_sentences_file(
                 if not isinstance(slot_dict, dict):
                     slot_dict = {"value": slot_dict}
                 assert (
-                    slot_name in result.entities
+                    slot_name
+                    in
+                    # wrap it in a list to get more readable pytest assertion
+                    list(result.entities)
                 ), f"For '{sentence}' did not receive slot '{slot_name}'"
                 assert result.entities[slot_name].value == slot_dict["value"]