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)