correct expected repr of dummy_func_args

WaylonWalker · Oct 7, 2020 · 58636d9 · 58636d9
1 parent eb98122
commit 58636d9
Showing 1 changed file with 6 additions and 4 deletions.
diff --git a/tests/pipeline/test_node.py b/tests/pipeline/test_node.py
@@ -359,15 +359,17 @@ def partial_inconsistent_input_size():
         (
             inconsistent_input_args,
             (
-                r"Inputs of '<function dummy_func_args at .*>' "
-                r"expected \[\'args\'\], but got {\'a\': \'A\'}"
+                r"Inputs of '"
+                r"<function inconsistent_input_args.<locals>.dummy_func_args"
+                r" at .*>' expected \[\'kwargs\'\], but got A"
             ),
         ),
         (
             inconsistent_input_kwargs,
             (
-                r"Inputs of '<function dummy_func_args at .*>' "
-                r"expected \[\'kwargs\'\], but got A"
+                r"Inputs of '"
+                r"<function inconsistent_input_kwargs.<locals>.dummy_func_args"
+                r" at .*>' expected \[\'kwargs\'\], but got A"
             ),
         ),
         (