def METHOD_NAME(self):
    reduce_lr = callbacks.ReduceLROnPlateau(
        patience=1,
        factor=0.1,
        monitor="val_loss",
        min_delta=100,
        cooldown=2,
    )
    self.model.fit(
        self.x_train,
        self.y_train,
        validation_data=(self.x_test, self.y_test),
        callbacks=[reduce_lr],
        epochs=4,
    )
    # With a cooldown of 2 epochs, we should only reduce the LR every other
    # epoch, so after 4 epochs we will have reduced 2 times.
    self.assertAllClose(self.model.optimizer.learning_rate.value, 0.001)