def test_histogram2d_x_list_y_list():
x = [1, 1, 2, 2, 3]
y = [1, 1, 2, 2, 3]
H_np, xedges_np, yedges_np = np.histogram2d(x, y)
H_mp, xedges_mp, yedges_mp = mp.histogram2d(x, y)
assert xedges_np.dtype == xedges_mp.dtype
assert xedges_np.ndim == xedges_mp.ndim
assert xedges_np.shape == xedges_mp.shape
assert np.allclose(xedges_np, xedges_mp)
assert yedges_np.dtype == yedges_mp.dtype
assert yedges_np.ndim == yedges_mp.ndim
assert yedges_np.shape == yedges_mp.shape
assert np.allclose(yedges_np, yedges_mp)
assert H_np.dtype == H_mp.dtype
assert H_np.ndim == H_mp.ndim
assert H_np.shape == H_mp.shape
assert H_np.sum() == H_mp.sum()
assert np.array_equal(H_np, H_mp)
def test_histogram2d_x_ndarray_y_ndarray():
x = np.sin(np.linspace(0, 10 * np.pi, 1000))
y = np.sin(np.linspace(0, 10 * np.pi, 1000))
H_np, xedges_np, yedges_np = np.histogram2d(x, y)
H_mp, xedges_mp, yedges_mp = mp.histogram2d(x, y)
assert xedges_np.dtype == xedges_mp.dtype
assert xedges_np.ndim == xedges_mp.ndim
assert xedges_np.shape == xedges_mp.shape
assert np.allclose(xedges_np, xedges_mp)
assert yedges_np.dtype == yedges_mp.dtype
assert yedges_np.ndim == yedges_mp.ndim
assert yedges_np.shape == yedges_mp.shape
assert np.allclose(yedges_np, yedges_mp)
assert H_np.dtype == H_mp.dtype
assert H_np.ndim == H_mp.ndim
assert H_np.shape == H_mp.shape
assert H_np.sum() == H_mp.sum()
assert np.array_equal(H_np, H_mp)
def test_histogram2d_bins_list_of_2_ints(bins):
rng = np.random.RandomState(42)
x = rng.normal(size=1000)
y = rng.normal(size=1000)
H_np, xedges_np, yedges_np = np.histogram2d(x, y, bins)
H_mp, xedges_mp, yedges_mp = mp.histogram2d(x, y, bins)
assert xedges_np.dtype == xedges_mp.dtype
assert xedges_np.ndim == xedges_mp.ndim
assert xedges_np.shape == xedges_mp.shape
assert np.allclose(xedges_np, xedges_mp)
assert yedges_np.dtype == yedges_mp.dtype
assert yedges_np.ndim == yedges_mp.ndim
assert yedges_np.shape == yedges_mp.shape
assert np.allclose(yedges_np, yedges_mp)
assert H_np.dtype == H_mp.dtype
assert H_np.ndim == H_mp.ndim
assert H_np.shape == H_mp.shape
assert H_np.sum() == H_mp.sum()
assert np.array_equal(H_np, H_mp)
def test_histogram2d_bins_list_mixed():
rng = np.random.RandomState(42)
x = rng.normal(size=1000)
y = rng.normal(size=1000)
bins = [np.linspace(-5, 5, 10), 100]
H_mp, xedges_mp, yedges_mp = mp.histogram2d(x, y, bins)