def test_digitize_bins_decreasing():
x = np.sin(np.linspace(0, 10 * np.pi, 1000))
bins = np.linspace(1, -1, 10)
indices_np = np.digitize(x, bins)
indices_mp = mp.digitize(x, bins)
assert indices_np.dtype == indices_mp.dtype
assert indices_np.ndim == indices_mp.ndim
assert indices_np.shape == indices_mp.shape
assert np.array_equal(indices_np, indices_mp)
def test_digitize_bins_decreasing_right(right):
x = np.array([1, 1, 2, 2, 3, 4])
bins = np.linspace(4, 1, 4)
indices_np = np.digitize(x, bins, right)
indices_mp = mp.digitize(x, bins, right)
assert indices_np.dtype == indices_mp.dtype
assert indices_np.ndim == indices_mp.ndim
assert indices_np.shape == indices_mp.shape
assert np.array_equal(indices_np, indices_mp)
def test_digitize_x_list():
x = [1, 1, 2, 2, 3, 4]
bins = np.linspace(1, 4, 4)
indices_np = np.digitize(x, bins)
indices_mp = mp.digitize(x, bins)
assert indices_np.dtype == indices_mp.dtype
assert indices_np.ndim == indices_mp.ndim
assert indices_np.shape == indices_mp.shape
assert np.array_equal(indices_np, indices_mp)