code: def test_running_migrations_generates_expected_schema(
      tmp_path, expected_schema_file, db_url
  ):
      """Test that migrating an existing database generates the desired schema."""
      engine = sqlalchemy.create_engine(db_url)
      InitialBase.metadata.create_all(engine)
      invoke_cli_runner(mlflow.db.commands, ["upgrade", db_url])
      generated_schema_file = tmp_path.joinpath("generated-schema.sql")
      dump_db_schema(db_url, generated_schema_file)
      _assert_schema_files_equal(
          generated_schema_file, expected_schema_file
      )
  
imports: import os
  import sqlite3
  import pytest
  import sqlalchemy
  from alembic import command
  from alembic.autogenerate import compare_metadata
  from alembic.migration import MigrationContext
  from alembic.script import ScriptDirectory
  import mlflow.db
  from mlflow.exceptions import MlflowException
  from mlflow.store.db.base_sql_model import Base
  from mlflow.store.db.utils import _get_alembic_config, _verify_schema
  from mlflow.store.tracking.dbmodels.initial_models import Base
  from mlflow.store.tracking.sqlalchemy_store import SqlAlchemyStore
  from tests.integration.utils import invoke_cli_runner
  from tests.store.dump_schema import dump_db_schema
label_desc: Use image node_id 5 for calling a global function with example usage: test_running_migrations_generates_expected_schema(tmp_path, expected_schema_file, db_url) without return types