code: def test_fit_simple_linear_model():
      np.random.seed(0x2019)
      tf.random.set_seed(0x2019)
  
      x = np.random.standard_normal((10000, 3))
      w = np.random.standard_normal((3, 1))
      y = np.dot(x, w) + np.random.standard_normal((10000, 1)) * 1e-4
  
      model = tf.keras.models.Sequential()
      model.add(tf.keras.layers.Dense(input_shape=(3,), units=1))
      model.compile(Lookahead("sgd"), loss="mse")
  
      model.fit(x, y, epochs=3)
  
      x = np.random.standard_normal((100, 3))
      y = np.dot(x, w)
      predicted = model.predict(x)
  
      max_abs_diff = np.max(np.abs(predicted - y))
      assert max_abs_diff < 1e-3
  
file_path: addons/tensorflow_addons/optimizers/tests/lookahead_test.py
imports: import numpy
  import pytest
  import tensorflow
  from tensorflow_addons.optimizers import Lookahead
label_desc: Use image node_id 5 for calling a global function with example usage: test_fit_simple_linear_model() without return types