code: def test_buffer_flatten():
      """A buffer should flatten to a 1-d shape"""
      buf = tvm.tir.decl_buffer([16, 32])
      flat = buf.get_flattened_buffer()
      assert buf.data.same_as(flat.data)
      tvm.ir.assert_structural_equal(flat.shape, [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 10 for calling a global function with example usage: test_buffer_flatten() without return types