code: def test_dense_sample():
      # Expected values are obtained from the previous implementation
      run_dense_sample(
          iterations=100,
          expected=[[0.66723955, 1.6672393], [2.6672382, 3.6672382]],
          optimizer=AdaBelief(lr=1e-3, 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 3 for calling a global function with example usage: test_dense_sample() without return types