def METHOD_NAME(self):
    # step should control how far apart consecutive windows are
    a = np.array([[0, 1, 2, 3, 4], [5, 6, 7, 8, 9]], dtype=np.int32)
    expected_result = np.array(
        [[[0, 1, 2], [2, 3, 4]], [[5, 6, 7], [7, 8, 9]]], dtype=np.int32
    )
    result = rolling_window(a, window=3, step=2, axis=1)
    self.assertArrayEqual(result, expected_result)