code: def test_with_only_on_epoch_end_arg(self):
      on_epoch_end = Mock()
      lambda_callback = LambdaCallback(on_epoch_end=on_epoch_end)
  
      train_generator = some_data_tensor_generator(
          LambdaTest.batch_size
      )
      valid_generator = some_data_tensor_generator(
          LambdaTest.batch_size
      )
      test_generator = some_data_tensor_generator(LambdaTest.batch_size)
      logs = self.model.fit_generator(
          train_generator,
          valid_generator,
          epochs=LambdaTest.epochs,
          steps_per_epoch=LambdaTest.steps_per_epoch,
          validation_steps=LambdaTest.steps_per_epoch,
          callbacks=[lambda_callback],
      )
  
      num_steps = 10
      self.model.evaluate_generator(
          test_generator, steps=num_steps, callbacks=[lambda_callback]
      )
  
      expected_calls = [
          call(epoch_number, log)
          for epoch_number, log in enumerate(logs, 1)
      ]
      actual_calls = on_epoch_end.mock_calls
      self.assertEqual(len(expected_calls), len(actual_calls))
      self.assertEqual(expected_calls, actual_calls)
  
file_path: poutyne/tests/framework/callbacks/test_lambda.py
imports: from unittest.mock import Mock, call
  import torch
  import torch.nn
  from poutyne import Callback, LambdaCallback, Model
  from tests.framework.model.base import ModelFittingTestCase
  from tests.framework.tools import some_data_tensor_generator
label_desc: Use image node_id 3 for calling the LambdaTest obj's underlying member method code with example usage: obj.test_with_only_on_epoch_end_arg() without return types