code: def mock_udf_context(batch_length=10):
      from pyarrow._compute import _get_udf_context
  
      return _get_udf_context(pa.default_memory_pool(), batch_length)
  
imports: import pytest
  import numpy
  import pyarrow
  from pyarrow import compute
label_desc: Use image node_id 1 for calling a global function with example usage: mock_udf_context(batch_length) and returns: _get_udf_context