code: def test_output_handler_output_transform():
      wrapper = OutputHandler("tag", output_transform=lambda x: x)
      mock_logger = MagicMock(spec=MLflowLogger)
      mock_logger.log_metrics = MagicMock()
  
      mock_engine = MagicMock()
      mock_engine.state = State()
      mock_engine.state.output = 12345
      mock_engine.state.iteration = 123
  
      wrapper(mock_engine, mock_logger, Events.ITERATION_STARTED)
  
      mock_logger.log_metrics.assert_called_once_with(
          {"tag output": 12345}, step=123
      )
  
      wrapper = OutputHandler(
          "another_tag", output_transform=lambda x: {"loss": x}
      )
      mock_logger = MagicMock(spec=MLflowLogger)
      mock_logger.log_metrics = MagicMock()
  
      wrapper(mock_engine, mock_logger, Events.ITERATION_STARTED)
      mock_logger.log_metrics.assert_called_once_with(
          {"another_tag loss": 12345}, step=123
      )
  
file_path: ignite/tests/ignite/contrib/handlers/test_mlflow_logger.py
imports: import sys
  from unittest.mock import call, MagicMock
  import pytest
  import torch
  from ignite.contrib.handlers.mlflow_logger import global_step_from_engine, MLflowLogger, OptimizerParamsHandler, OutputHandler
  from ignite.engine import Engine, Events, State
label_desc: Use image node_id 2 for calling a global function with example usage: test_output_handler_output_transform() without return types