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 os
  import pathlib
  import pytest
  import pyarrow
  import pyarrow.compute
  from pyarrow.lib import tobytes
  from pyarrow.lib import ArrowInvalid, ArrowNotImplementedError
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