code: def fix_get_cifar10_data():
      """
      Get the first 128 samples of the cifar10 test set
  
      :return: First 128 sample/label pairs of the cifar10 test dataset.
      """
      nb_test = 128
  
      (_, _), (x_test, y_test), _, _ = load_dataset("cifar10")
      y_test = np.argmax(y_test, axis=1)
      x_test, y_test = x_test[:nb_test], y_test[:nb_test]
      x_test = np.transpose(
          x_test, (0, 3, 1, 2)
      )  # return in channels first format
      return x_test.astype(np.float32), y_test
  
file_path: adversarial-robustness-toolbox/tests/estimators/certification/test_derandomized_smoothing.py
imports: import os
  import pytest
  import numpy
  from art.utils import load_dataset
  from art.estimators.certification.derandomized_smoothing import PyTorchDeRandomizedSmoothing, TensorFlowV2DeRandomizedSmoothing
  from tests.utils import ARTTestException
label_desc: Use image node_id 2 for calling a global function with example usage: fix_get_cifar10_data() and returns: x_test,  y_test