code: def test_step_schedule(backend_default):
      """
      Test the StepSchedule class
      """
      step_config = [1, 4, 5]
      change = [0.1, 0.3, 0.4]
      sch = StepSchedule(step_config=step_config, change=change)
  
      target_lr = [1.0, 0.1, 0.1, 0.1, 0.3, 0.4, 0.4, 0.4, 0.4]
  
      for e, lr in enumerate(target_lr):
          assert allclose_with_out(
              lr, sch.get_learning_rate(learning_rate=1.0, epoch=e)
          )
  
file_path: neon/tests/test_schedule.py
imports: import numpy
  from neon.optimizers import Schedule, ExpSchedule, PowerSchedule, StepSchedule, ShiftSchedule
  from utils import allclose_with_out
label_desc: Use image node_id 2 for calling a global function with example usage: test_step_schedule(backend_default) without return types