code: def test_use_transformed_axes(
      self,
      schedule,
      expected_loop_order,
      transform_A,
      transform_B,
      after_transform,
  ):
      s = schedule["schedule"]
      A, B = schedule["tensors"]
  
      func = tvm.lower(s, [A, B])["main"]
      self.compare_tir_loop_order(func.body, 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 8 for calling the TestTransformedSchedules obj's underlying member method code with example usage: obj.test_use_transformed_axes(schedule, expected_loop_order, transform_A, transform_B, after_transform) without return types