def integrate(data, nint):
    """
    A simple integration for testing and benchmarking.

    Integrates along the time axis.

    Parameters
    ----------
    data : ndarray
        The data to integrate. Dimensions (time, baseline, channel,
        polarization).
    nint : int
         The number of consecutive time samples to combine.

    Returns
    -------
    ndarray
        The integrated data. Dimensions (time, baseline, channel,
        polarization).
    """
    (_nt, nbls, nchan, npol) = data.shape
    data = data.reshape(-1, nint, nbls, nchan, npol).mean(1)
    return data