def test_map_pairs():
src = np.asarray([[0, 1], [1, 1], [2, 1], [3, 1], [4, 1]], dtype=np.int)
data = np.arange(len(src), dtype=int)
pairs = np.asarray([[1, 1], [3, 1]], dtype=int)
out = map_pairs_to_values((src, data), pairs)
assert np.all(out == [1, 3])
def test_map_pairs_with_sorter_keyword():
src = np.asarray([[0, 1], [1, 1], [3, 1], [2, 1], [4, 1]], dtype=np.int)
data = np.array([0, 1, 3, 2, 4], dtype=int)
pairs = np.asarray([[1, 1], [3, 1]], dtype=int)
out = map_pairs_to_values((src, data), pairs, sorter=np.argsort(src[:, 0]))
assert np.all(out == [1, 3])
def test_map_pairs_all_missing():
src = np.asarray([[0, 1], [1, 1], [2, 1], [3, 1], [4, 1]], dtype=int)
data = np.arange(len(src), dtype=int)
pairs = np.asarray([[5, 1], [1, 42]], dtype=int)
out = map_pairs_to_values((src, data), pairs)
assert np.all(out == [-1, -1])
def test_map_pairs_with_out_keyword():
src = np.asarray([[0, 1], [1, 1], [2, 1], [3, 1], [4, 1]], dtype=np.int)
data = np.arange(len(src), dtype=int)
pairs = np.asarray([[1, 1], [3, 1]], dtype=int)
out = np.empty(len(pairs), dtype=np.int)
rtn = map_pairs_to_values((src, data), pairs, out=out)
assert rtn is out
assert np.all(out == [1, 3])
def test_map_pairs_big_data():
n_src_pairs = 2**20
src = np.random.randint(n_src_pairs * 2, size=n_src_pairs * 2,
dtype=int).reshape((-1, 2))
src = src[np.lexsort((src[:, 1], src[:, 0]))]
data = np.arange(n_src_pairs, dtype=int)
ids = np.random.randint(n_src_pairs, size=n_src_pairs // 10)
pairs = src[ids]
out = map_pairs_to_values((src, data), pairs)
assert_array_equal(out, ids)