code: def test_dense_sample_with_weight_decay():
      # Expected values are obtained from the previous implementation
      run_dense_sample(
          iterations=100,
          expected=[[0.66637343, 1.6653734], [2.6643748, 3.6633751]],
          optimizer=AdaBelief(
              lr=1e-3, weight_decay=0.01, rectify=False
          ),
      )
  
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 7 for calling a global function with example usage: test_dense_sample_with_weight_decay() without return types