code: def test_get_output_blocks_single_output():
      sch = tir.Schedule(mod=matmul_relu, debug_mask="all")
      output_blocks = sch.get_output_blocks("root")
      assert (
          len(output_blocks) == 1
      ), "Unexpected number of blocks when 1 was expected"
      block = sch.get(output_blocks[0])
      assert block.name_hint == "relu"
      relu_block = sch.get_block("relu")
      assert sch.get(relu_block).same_as(block)
  
imports: import sys
  import pytest
  import tvm
  import tvm.testing
  from tvm import tir
  from tvm.ir import IRModule
  from tvm.script import tir
  from tvm.tir.schedule.testing import assert_structural_equal_ignore_global_symbol, verify_trace_roundtrip
label_desc: Use image node_id 20 for calling a global function with example usage: test_get_output_blocks_single_output() without return types