def test_mm_pairwise():
for case in iter_testcases('pairwise'):
n_items = case["n_items"]
data = case["data"]
for params in (None, np.exp(RND.randn(n_items))):
assert np.allclose(case["ml_est"],
mm_pairwise(n_items, data, initial_params=params))
def test_opt_top1_json():
"""JSON test cases for top1-data ML estimator."""
for case in iter_testcases('top1'):
n_items = case["n_items"]
data = case["data"]
for method in ("BFGS", "Newton-CG"):
est = opt_top1(n_items, data, alpha=0.0, method=method)
assert np.allclose(case["ml_est"], est, atol=ATOL, rtol=RTOL)
def test_mm_top1():
"""JSON test cases for top-1 data."""
for case in iter_testcases('top1'):
n_items = case["n_items"]
data = case["data"]
for params in (None, RND.randn(n_items)):
est = mm_top1(n_items, data, initial_params=params)
assert np.allclose(case["ml_est"], est, atol=ATOL, rtol=RTOL)
def test_rank_centrality():
"""JSON test cases for Rank Centrality."""
for case in iter_testcases('pairwise'):
n_items = case["n_items"]
data = case["data"]
assert np.allclose(case["rc_est"],
rank_centrality(n_items, data),
atol=ATOL,
rtol=RTOL)
def test_ilsr_pairwise():
"""JSON test cases for I-LSR pairwise."""
for case in iter_testcases('pairwise'):
n_items = case["n_items"]
data = case["data"]
assert np.allclose(case["ml_est"],
ilsr_pairwise(n_items, data),
atol=ATOL,
rtol=RTOL)
def test_lsr_top1():
"""JSON test cases for LSR top1."""
for case in iter_testcases('top1'):
n_items = case["n_items"]
data = case["data"]
assert np.allclose(case["lsr_est"],
lsr_top1(n_items, data),
atol=ATOL,
rtol=RTOL)
def test_ilsr_rankings():
"""JSON test cases for I-LSR rankings."""
for case in iter_testcases('rankings'):
n_items = case["n_items"]
data = case["data"]
assert np.allclose(case["ml_est"],
ilsr_rankings(n_items, data),
atol=ATOL,
rtol=RTOL)
def test_ilsr_pairwise_dense():
"""JSON test cases for LSR pairwise."""
for case in iter_testcases('pairwise'):
n_items = case["n_items"]
mat = np.zeros((n_items, n_items), dtype=float)
for a, b in case["data"]:
mat[a, b] += 1
assert np.allclose(case["ml_est"],
ilsr_pairwise_dense(mat),
atol=ATOL,
rtol=RTOL)
def test_ilsr_pairwise():
for case in iter_testcases('pairwise'):
n_items = case["n_items"]
data = case["data"]
assert np.allclose(case["ml_est"], ilsr_pairwise(n_items, data))
