code: def test_serialization():
      stoch_depth = StochasticDepth(survival_probability=0.5)
      serialized_stoch_depth = tf.keras.layers.serialize(stoch_depth)
      new_layer = tf.keras.layers.deserialize(serialized_stoch_depth)
      assert stoch_depth.get_config() == new_layer.get_config()
  
file_path: addons/tensorflow_addons/layers/tests/stochastic_depth_test.py
imports: import pytest
  import numpy
  import tensorflow
  from tensorflow_addons.layers.stochastic_depth import StochasticDepth
  from tensorflow_addons.utils import test_utils
label_desc: Use image node_id 3 for calling a global function with example usage: test_serialization() without return types