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))