def METHOD_NAME(self, mocker):
    optimizer = mocker.stub()
    optimizer.param_groups = [{"lr": 1}]
    lr_scheduler = StageLRScheduler.from_config(optimizer, LR_SCHEDULER_PARAMS)
    lr_scheduler._init_lr = 2.5e-4
    lr_scheduler._num_epochs = 10
    assert lr_scheduler.current_epoch == -1
    assert lr_scheduler.current_step == -1
    lr_scheduler.epoch_step()
    lr_scheduler.step()
    assert lr_scheduler.current_epoch == 0
    assert lr_scheduler.current_step == 0
    lr_scheduler.epoch_step()
    lr_scheduler.step()
    assert lr_scheduler.current_epoch == 1
    assert lr_scheduler.current_step == 1
    lr_scheduler.reset(LR_SCHEDULER_PARAMS.base_lr, LR_SCHEDULER_PARAMS.num_epochs)
    assert lr_scheduler.current_epoch == 0
    assert lr_scheduler.current_step == 0