code: def _assert_lowered_main(
      mod, main_func, call_type, print_script=False
  ):
      mod, config = _plan_devices(mod)
      mod = AOTLowerMain("test_mod", config, call_type)(mod)
      if print_script:
          print(mod["__tvm_main__"].script())
  
      assert_structural_equal(mod["__tvm_main__"], main_func)
  
imports: import numpy
  import pytest
  import tvm
  import tvm.testing
  from tvm.ir import assert_structural_equal
  from tvm.relay.backend.aot import AOTLowerMain, CallType
  from tvm.script import tir
label_desc: Use image node_id 4 for calling a global function with example usage: _assert_lowered_main(mod, main_func, call_type, print_script) without return types