code: def test_integration(self):
      train_gen = some_data_generator(20)
      valid_gen = some_data_generator(20)
      model_restore = BestModelRestore(monitor="val_loss", verbose=True)
      self.model.fit_generator(
          train_gen,
          valid_gen,
          epochs=10,
          steps_per_epoch=5,
          callbacks=[model_restore],
      )
  
file_path: poutyne/tests/framework/callbacks/test_best_model_restore.py
imports: import unittest
  from unittest import TestCase
  import torch
  import torch.nn
  from poutyne import BestModelRestore, Model, torch_to_numpy
  from tests.framework.tools import some_data_generator
label_desc: Use image node_id 2 for calling the BestModelRestoreTest obj's underlying member method code with example usage: obj.test_integration() without return types