code: def test_simple_script(input_code, output_code):
      input_ast = pasta.parse(input_code)
      output_ast = ASTTransformer().visit(input_ast)
      assert pasta.dump(output_ast) == output_code
  
imports: from __future__ import absolute_import
  import textwrap
  import pasta
  import pytest
  from sagemaker.cli.compatibility.v2.ast_transformer import ASTTransformer
label_desc: Use image node_id 3 for calling a global function with example usage: test_simple_script(input_code, output_code) without return types