code: def test_ema_with_model_fit(self):
      x_train = np.ones((1, 1)).astype("float32")
      y_train = np.zeros((1, 1)).astype("float32")
      optimizer = optimizers.SGD(
          learning_rate=0.1, use_ema=True, ema_momentum=0.9
      )
      model = models.Sequential(
          [layers.Dense(2, kernel_initializer="ones", use_bias=False)]
      )
      model.compile(loss="mse", optimizer=optimizer, run_eagerly=True)
      model.fit(x_train, y_train, batch_size=1, epochs=2)
      self.assertAllClose(
          optimizer._model_variables_moving_average[0].numpy(),
          [[0.891, 0.891]],
          atol=1e-5,
      )
      self.assertAllClose(
          model.trainable_variables[0].numpy(),
          [[0.891, 0.891]],
          atol=1e-5,
      )
  
file_path: keras/keras/optimizers/optimizer_test.py
imports: import numpy
  import pytest
  from keras import backend
  from keras import constraints
  from keras import layers
  from keras import models
  from keras import optimizers
  from keras import testing
label_desc: Use image node_id 3 for calling the OptimizerTest obj's underlying member method code with example usage: obj.test_ema_with_model_fit() without return types