Fix error message testing

jetavator · Mar 12, 2021 · 5591d69 · 5591d69
1 parent d1c477c
commit 5591d69
Show file tree

Hide file tree

Showing 2 changed files with 7 additions and 9 deletions.
diff --git a/features/steps/steps.py b/features/steps/steps.py
@@ -45,17 +45,17 @@ def step_impl(context, exception_type):
 
     def remove_whitespace(text):
         return " ".join(
-            line.strip()
-            for line in context.text.splitlines()
+            line.strip('"')
+            for line in text.splitlines()
         ).strip()
 
-    if remove_whitespace(context.text) != remove_whitespace(context.exception):
+    if remove_whitespace(context.text) != remove_whitespace(str(context.exception)):
         raise Exception(
             f"""
             Expected error message:
-            {context.text}
+            {remove_whitespace(context.text)}
             Got:
-            {context.exception}
+            {remove_whitespace(str(context.exception))}
             """
         )
 

diff --git a/wysdom/mixins/RegistersSubclasses.py b/wysdom/mixins/RegistersSubclasses.py
@@ -99,10 +99,8 @@ def registered_subclass(
                     ):
                         return matched_subclass
             raise KeyError(
-                f"""
-                The key '{name}' is ambiguous as it matches multiple proper subclasses of {cls}:
-                {matched_subclasses}
-                """)
+                f"The key '{name}' is ambiguous as it matches multiple proper subclasses of {cls}: "
+                f"{matched_subclasses}")
         return matched_subclasses[0]
 
     @classmethod