code: def test_add_mean_shift_to_first_layer(load_mnist_model_tf):
      model = TensorFlowModel(
          model=load_mnist_model_tf, channel_first=False
      )
      shift = 0.2
      X = np.random.random((32, 28, 28, 1))
      X_shift = X + shift
      new_model = model.add_mean_shift_to_first_layer(
          shift, X[:1].shape
      )
      a1 = model.model(X)
      a2 = new_model(X_shift)
      assert np.all(np.isclose(a1, a2, atol=1e-04))
  
file_path: Quantus/tests/helpers/test_tf_model.py
imports: from functools import reduce
  from operator import and_
  from typing import Union
  import tensorflow
  import numpy
  import pytest
  from pytest_lazyfixture import lazy_fixture
  from scipy.special import softmax
  from quantus.helpers.model.tf_model import TensorFlowModel
label_desc: Use image node_id 6 for calling a global function with example usage: test_add_mean_shift_to_first_layer(load_mnist_model_tf) without return types