How to use the numbagg.move_nanmean.func function in numbagg

To help you get started, we’ve selected a few numbagg examples, based on popular ways it is used in public projects.

Secure your code as it's written. Use Snyk Code to scan source code in minutes - no build needed - and fix issues immediately.

github shoyer / numbagg / numbagg / tests.py View on Github external
numbagg.nansum: np.nansum,
    numbagg.nanmean: np.nanmean,
    numbagg.nanstd: np.nanstd,
    numbagg.nanvar: np.nanvar,
    numbagg.nanmax: np.nanmax,
    numbagg.nanmin: np.nanmin,
    numbagg.count: lambda x, **kwargs: np.sum(~np.isnan(x), **kwargs),
}

argmax_reference_funcs = {
    numbagg.nanargmax: np.nanargmax,
    numbagg.nanargmin: np.nanargmin,
}

moving_references_funcs = {
    numbagg.move_nanmean: wrap_pure_python(numbagg.move_nanmean.func),
}


def allclose(actual, desired, **kwargs):
    if getattr(actual, 'shape', ()) != getattr(desired, 'shape', ()):
        return False
    return np.all(np.isclose(actual, desired, equal_nan=True, **kwargs))


def check_func(numbagg_func, ref_func, x, axis):
    actual = numbagg_func(x, axis=axis)
    desired = ref_func(x, axis=axis)
    assert allclose(actual, desired), (numbagg_func, axis)


def test_funcs():