code: def test_df_get_chunks(use_batch, size, n_chunks):
      table = pa.table({"x": list(range(size))})
      if use_batch:
          table = table.to_batches()[0]
      df = table.__dataframe__()
      chunks = list(df.get_chunks(n_chunks))
      assert len(chunks) == n_chunks
      assert sum(chunk.num_rows() for chunk in chunks) == size
  
file_path: arrow/python/pyarrow/tests/interchange/test_interchange_spec.py
imports: import ctypes
  import hypothesis
  import hypothesis.strategies
  import numpy
  import pyarrow
  import pyarrow.tests.strategies
  import pytest
label_desc: Use image node_id 7 for calling a global function with example usage: test_df_get_chunks(use_batch, size, n_chunks) without return types