code: def test_main_logger_step_names():
      step_names = "iteration"
      logger = MainLogger(step_names=step_names)
      assert logger.step_names["Accuracy"] == "iteration"
      step_names = {"Accuracy": "evaluation", "Loss": "batch"}
      logger = MainLogger(step_names=step_names)
      assert logger.step_names["Accuracy"] == "evaluation"
      assert logger.step_names["Loss"] == "batch"
      assert logger.step_names["Epoch Time"] == "epoch"
  
file_path: livelossplot/tests/test_main_logger.py
imports: import re
  from livelossplot import MainLogger
label_desc: Use image node_id 6 for calling a global function with example usage: test_main_logger_step_names() without return types