code: def small_mock_mnist_data():
      return _mock_mnist_data(SMALL_MOCK_IMAGE_COUNT)
  
file_path: petastorm/examples/mnist/tests/conftest.py
imports: import numpy
  import pytest
label_desc: Use image node_id 2 for calling a global function with example usage: small_mock_mnist_data() and returns: _mock_mnist_data