code: def test_te_loop_order(self, schedule, expected_loop_order):
      s = schedule["schedule"]
      A, B = schedule["tensors"]
      iter_vars = schedule["iter_vars"]
  
      # No reduction axis, so all leaf_iter_vars are over the data
      # array, and should have the new iteration variables.
      extents = [
          int(iter_var.dom.extent) for iter_var in s[B].leaf_iter_vars
      ]
      np.testing.assert_array_equal(extents, expected_loop_order)
  
      # layout_transform should return the new iteration variables.
      extents = [int(iter_var.dom.extent) for iter_var in iter_vars]
      np.testing.assert_array_equal(extents, expected_loop_order)
  
imports: import functools
  import sys
  import pytest
  import numpy
  import tvm
  import tvm.testing
  from tvm import te
  from tvm.tir.stmt_functor import post_order_visit
  from tvm.driver.build_module import schedule_to_module
label_desc: Use image node_id 7 for calling the TestTransformedSchedules obj's underlying member method code with example usage: obj.test_te_loop_order(schedule, expected_loop_order) without return types