def METHOD_NAME(self, atomic_db: AtomicDatabaseAPI) -> None:
    atomic_db.set(b"key-1", b"value-1")
    try:
        with atomic_db.atomic_batch() as batch:
            batch.set(b"key-1", b"new-value-1")
            batch.set(b"key-2", b"value-2")
            raise Exception
    except Exception:
        pass
    assert atomic_db.get(b"key-1") == b"value-1"
    with pytest.raises(KeyError):
        atomic_db[b"key-2"]