code: def test_output_handler_output_transform():
      wrapper = OutputHandler("tag", output_transform=lambda x: x)
      logger = NeptuneLogger(
          project="tests/dry-run",
          mode="debug",
      )
      mock_engine = MagicMock()
      mock_engine.state = State()
      mock_engine.state.output = 12345
      mock_engine.state.iteration = 123
  
      wrapper(mock_engine, logger, Events.ITERATION_STARTED)
      assert_logger_called_once_with(logger, "tag/output", 12345)
      logger.stop()
  
      wrapper = OutputHandler(
          "another_tag", output_transform=lambda x: {"loss": x}
      )
      logger = NeptuneLogger(
          project="tests/dry-run",
          mode="debug",
      )
  
      wrapper(mock_engine, logger, Events.ITERATION_STARTED)
      assert_logger_called_once_with(logger, "another_tag/loss", 12345)
      logger.stop()
  
imports: import math
  import warnings
  from unittest.mock import MagicMock
  import pytest
  import torch
  from ignite.contrib.handlers.neptune_logger import global_step_from_engine, GradsScalarHandler, NeptuneLogger, NeptuneSaver, OptimizerParamsHandler, OutputHandler, WeightsScalarHandler
  from ignite.engine import Engine, Events, State
label_desc: Use image node_id 5 for calling a global function with example usage: test_output_handler_output_transform() without return types