code: def test_serialization():
      optimizer = AdaBelief(
          lr=1e-3,
          total_steps=10000,
          warmup_proportion=0.1,
          min_lr=1e-5,
          rectify=False,
      )
      config = tf.keras.optimizers.serialize(optimizer)
      new_optimizer = tf.keras.optimizers.deserialize(config)
      assert new_optimizer.get_config() == optimizer.get_config()
  
imports: import numpy
  import pytest
  from packaging.version import Version
  import tensorflow.compat.v2
  from tensorflow_addons.optimizers import AdaBelief, Lookahead
label_desc: Use image node_id 16 for calling a global function with example usage: test_serialization() without return types