code: def _test_checkpoint(self, scheduler_states, lr_scheduler):
      for epoch, epoch_scheduler_state in scheduler_states.items():
          filename = self.checkpoint_filename.format(epoch=epoch)
          lr_scheduler.load_state(filename)
          saved_scheduler_state = torch_to_numpy(
              lr_scheduler.scheduler.state_dict()
          )
  
          self.assertEqual(epoch_scheduler_state, saved_scheduler_state)
  
file_path: poutyne/tests/framework/callbacks/test_lr_scheduler_checkpoint.py
imports: import os
  import unittest
  from tempfile import TemporaryDirectory
  from unittest import TestCase
  import torch
  import torch.nn
  from poutyne import ExponentialLR, LRSchedulerCheckpoint, Model, ReduceLROnPlateau, torch_to_numpy
  from tests.framework.tools import some_data_generator
label_desc: Use image node_id 9 for calling the OptimizerCheckpointTest obj's underlying member method code with example usage: obj._test_checkpoint(scheduler_states, lr_scheduler) without return types