code: def test_buffer_flatten_uses_axis_separators():
      """Flattening to N-d physical buffers uses the axis separators"""
      buf = tvm.tir.decl_buffer([4, 16, 32], axis_separators=[2])
      flat = buf.get_flattened_buffer()
      tvm.ir.assert_structural_equal(flat.axis_separators, [1])
      tvm.ir.assert_structural_equal(flat.shape, [4 * 16, 32])
  
file_path: tvm/tests/python/tir-base/test_tir_buffer.py
imports: import pytest
  import tvm
  import tvm.testing
  from tvm import te
  from tvm.tir import Buffer
  import numpy
label_desc: Use image node_id 12 for calling a global function with example usage: test_buffer_flatten_uses_axis_separators() without return types