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

    Get the first 10 samples of the cifar10 test set

    :return: First 10 sample/label pairs of the cifar10 test dataset.
    
tests/defences/trainer/test_certified_adversarial_trainer.py