def test_bin_1D():
# set up simple data
x = np.linspace(0, 1, 100)
y = np.arange(100)
nx = 10
# make call
edges, val, count = core.bin_1D(x, y, nx)
# check that values are as expected
assert_array_almost_equal(edges,
np.linspace(0, 1, nx + 1, endpoint=True))
assert_array_almost_equal(val,
np.sum(y.reshape(nx, -1), axis=1))
assert_array_equal(count,
np.ones(nx) * 10)
def test_bin_1D_2():
"""
Test for appropriate default value handling
"""
# set up simple data
x = np.linspace(0, 1, 100)
y = np.arange(100)
nx = core._defaults["bins"]
min_x = None
max_x = None
# make call
edges, val, count = core.bin_1D(x=x, y=y, nx=nx, min_x=min_x, max_x=max_x)
# check that values are as expected
assert_array_almost_equal(edges,
np.linspace(0, 1, nx + 1, endpoint=True))
assert_array_almost_equal(val,
np.sum(y.reshape(nx, -1), axis=1))
assert_array_equal(count,
np.ones(nx))