code: def test_neuron_selection_layer():
      neuron_selection_array = tf.constant([[0, 3], [1, 1]])
      inputs = klayers.Input(shape=(6,))
      outputs = ilayers.NeuronSelection()(
          [inputs, neuron_selection_array]
      )
      model = kmodels.Model(
          inputs=inputs, outputs=outputs, name="TestNeuronSelection"
      )
  
      x = np.array([[1, 2, 3, 6, 5, 4], [7, 42, 9, 10, 11, 12]])
      out = model.predict(x)
      assert np.all(out == np.array([6, 42]))
  
file_path: innvestigate/tests/backend/test_neuron_selection.py
imports: from __future__ import annotations
  import numpy
  import pytest
  import tensorflow
  import tensorflow.keras.layers
  import tensorflow.keras.models
  import innvestigate.layers
label_desc: Use image node_id 3 for calling a global function with example usage: test_neuron_selection_layer() without return types