def METHOD_NAME(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