code: def mock_data(request):
      """Parametrized fixture to test both `USE_FILES` and `USE_CODE` policy."""
      return functools.partial(
          tfds.testing.mock_data, policy=request.param
      )
  
imports: import functools
  import pickle
  import numpy
  import pytest
  import tensorflow
  import tensorflow_datasets
label_desc: Use image node_id 1 for calling a global function with example usage: mock_data(request) and returns: functools