def METHOD_NAME(setup_cluster):
    import dask
    def inc(x):
        return x + 1
    a = dask.delayed(inc)(1)
    task_mars_persist = dask.delayed(inc)(a.persist(scheduler=mars_scheduler))
    task_dask_persist = dask.delayed(inc)(a.persist())
    assert task_dask_persist.compute() == task_mars_persist.compute(
        scheduler=mars_scheduler
    )