code: def test_starts_at_correct_position(self):
      policy_iter = iter(self.policy)
      # The first three steps are ignored
      # assert next(policy_iter) == {"lr": 0}
      # assert next(policy_iter) == {"lr": 0}
      # assert next(policy_iter) == {"lr": 1}
      assert next(policy_iter) == {"lr": 1}
      assert next(policy_iter) == {"lr": 2}
      assert next(policy_iter) == {"lr": 2}
  
      with self.assertRaises(StopIteration):
          next(policy_iter)
  
imports: import unittest
  from poutyne import OptimizerPolicy, Phase, cosinespace, linspace, one_cycle_phases, sgdr_phases
label_desc: Use image node_id 2 for calling the TestOptimizerPolicyRestart obj's underlying member method code with example usage: obj.test_starts_at_correct_position() without return types