code: def test_time_rolling(before, after):
      window = before
      expected = dts.compute().rolling(window).count()
  
      # String timedelta
      result = dts.map_overlap(
          lambda x: x.rolling(window).count(), before, after
      )
      assert_eq(result, expected)
  
      # Timedelta
      before = pd.Timedelta(before)
      after = pd.Timedelta(after)
      result = dts.map_overlap(
          lambda x: x.rolling(window).count(), before, after
      )
      assert_eq(result, expected)
  
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 21 for calling a global function with example usage: test_time_rolling(before, after) without return types