def METHOD_NAME(self, atomic_db: AtomicDatabaseAPI) -> None:
    with atomic_db.atomic_batch() as batch:
        batch.set(b"1", b"2")
        assert batch.get(b"1") == b"2"
    assert atomic_db.get(b"1") == b"2"