code: def test_stamp_api_creates_table(self):
      context = self.make_one(connection=self.connection)
      assert (
          "alembic_version"
          not in inspect(self.connection).get_table_names()
      )
  
      script = mock.Mock(
          _stamp_revs=lambda revision, heads: [
              _up(None, "a", True),
              _up(None, "b", True),
          ]
      )
  
      context.stamp(script, "b")
      eq_(context.get_current_heads(), ("a", "b"))
      assert (
          "alembic_version"
          in inspect(self.connection).get_table_names()
      )
  
imports: from sqlalchemy import Column
  from sqlalchemy import inspect
  from sqlalchemy import MetaData
  from sqlalchemy import String
  from sqlalchemy import Table
  from alembic import migration
  from alembic.testing import assert_raises
  from alembic.testing import assert_raises_message
  from alembic.testing import config
  from alembic.testing import eq_
  from alembic.testing import mock
  from alembic.testing.fixtures import TestBase
  from alembic.util import CommandError
label_desc: Use image node_id 18 for calling the TestMigrationContext obj's underlying member method code with example usage: obj.test_stamp_api_creates_table() without return types