code: def test_tir_schedule_creation():
      # Tests:
      # - Schedule.__init__ for PrimFunc and IRModule
      # - Schedule.mod
      # - Schedule.state
      sch_1 = tir.Schedule(matmul, debug_mask="all")
      sch_2 = tir.Schedule(IRModule({"main": matmul}), debug_mask="all")
      assert sch_1.mod["main"].same_as(sch_2.mod["main"])
      assert sch_1.state.mod["main"].same_as(sch_2.state.mod["main"])
  
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 6 for calling a global function with example usage: test_tir_schedule_creation() without return types