Update tests

Signed-off-by: Ahdra Merali <[email protected]>
kedro-org · Feb 1, 2024 · 4581373 · 4581373
1 parent 1bbd161
commit 4581373
Showing 1 changed file with 2 additions and 4 deletions.
diff --git a/tests/ipython/test_ipython.py b/tests/ipython/test_ipython.py
@@ -500,10 +500,8 @@ def test_get_lambda_function_body(self, lambda_node):
 
     def test_get_nested_function_body(self):
         func_strings = """def nested_function(input):
-    # return not input
-    print(not input)
-# return nested_function(dummy_input)
-print(nested_function(dummy_input))"""
+    return not input
+return nested_function(dummy_input)"""
 
         result = _prepare_function_body(dummy_nested_function)
         assert result == func_strings