code: def test_num_stepping_batches_with_max_steps(max_steps):
      """Test stepping batches with `max_steps`."""
      trainer = Trainer(max_steps=max_steps)
      model = BoringModel()
      trainer.fit(model)
      assert trainer.estimated_stepping_batches == max_steps
  
imports: import logging
  import os
  from unittest import mock
  import pytest
  import torch
  from lightning.pytorch import Trainer
  from lightning.pytorch.demos.boring_classes import BoringModel, RandomIterableDataset
  from lightning.pytorch.strategies import SingleDeviceXLAStrategy
  from torch.utils.data import DataLoader
  from tests_pytorch.conftest import mock_cuda_count
  from tests_pytorch.helpers.runif import RunIf
label_desc: Use image node_id 5 for calling a global function with example usage: test_num_stepping_batches_with_max_steps(max_steps) without return types