def METHOD_NAME(
    self, atomic_db: AtomicDatabaseAPI
) -> None:
    class CustomException(Exception):
        pass
    atomic_db[b"key-1"] = b"origin"
    with pytest.raises(CustomException):
        with atomic_db.atomic_batch() as batch:
            batch.delete(b"key-1")
            batch.set(b"key-2", b"val-2")
            raise CustomException("pretend something went wrong")
    with atomic_db.atomic_batch() as batch:
        assert batch[b"key-1"] == b"origin"
        assert b"key-2" not in batch