Method setup

asv_bench/benchmarks/rolling.py:307–318

(self, method, window_kwargs)

    )

    def setup(self, method, window_kwargs):
        N = 1000
        window, kwargs = window_kwargs
        df = pd.DataFrame(
            {
                class="st">"A": [str(i) for i in range(N)] * 10,
                class="st">"B": list(range(N)) * 10,
            }
        )
        if isinstance(kwargs.get(class="st">"window", None), str):
            df.index = pd.date_range(start=class="st">"1900-01-01", freq=class="st">"1min", periods=N * 10)
        self.groupby_window = getattr(df.groupby(class="st">"A"), window)(**kwargs)

    def time_method(self, method, window_kwargs):
        getattr(self.groupby_window, method)()