code: def fix_get_cifar10_data():
      """
      Get the first 10 samples of the cifar10 test set
  
      :return: First 10 sample/label pairs of the cifar10 test dataset.
      """
      nb_test = 10
  
      (_, _), (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]
      return np.moveaxis(x_test, [3], [1]).astype(np.float32), y_test
  
file_path: adversarial-robustness-toolbox/tests/defences/trainer/test_certified_adversarial_trainer.py
imports: import pytest
  import numpy
  import random
  import torch
  from art.utils import load_dataset
  from art.estimators.certification.deep_z import PytorchDeepZ
  from art.attacks.evasion.projected_gradient_descent.projected_gradient_descent import ProjectedGradientDescent
  from art.defences.trainer import AdversarialTrainerCertifiedPytorch
  from tests.utils import ARTTestException
  from tests.utils import get_image_classifier_pt, get_cifar10_image_classifier_pt
label_desc: Use image node_id 2 for calling a global function with example usage: fix_get_cifar10_data() and returns: np,  y_test