code: def test_map_overlap(npartitions, use_dask_input):
      ddf = df
      if use_dask_input:
          ddf = dd.from_pandas(df, npartitions)
  
      for before, after in [(0, 3), (3, 0), (3, 3), (0, 0)]:
          # DataFrame
          res = dd.map_overlap(
              shifted_sum, ddf, before, after, before, after, c=2
          )
          sol = shifted_sum(df, before, after, c=2)
          assert_eq(res, sol)
  
          # Series
          res = dd.map_overlap(
              shifted_sum, ddf.b, before, after, before, after, c=2
          )
          sol = shifted_sum(df.b, before, after, c=2)
          assert_eq(res, sol)
  
imports: from __future__ import annotations
  import contextlib
  import datetime
  import numpy
  import pandas
  import pytest
  import dask.dataframe
  from dask.dataframe._compat import PANDAS_GE_210
  from dask.dataframe.utils import assert_eq
label_desc: Use image node_id 2 for calling a global function with example usage: test_map_overlap(npartitions, use_dask_input) without return types