def test_mutual_info_score_big_array(use_handle, input_range): a, b, _, _ = generate_random_labels(lambda rd: rd.randint(*input_range, int(10e4), dtype=np.int32)) score = score_mutual_info(a, b, use_handle) ref = sk_mutual_info_score(a, b) np.testing.assert_almost_equal(score, ref, decimal=4)
def test_mutual_info_score_many_blocks(use_handle, input_range, n_samples): a, b, _, _ = generate_random_labels(lambda rd: rd.randint(*input_range, n_samples, dtype=np.int32)) score = score_mutual_info(a, b, use_handle) ref = sk_mutual_info_score(a, b) np.testing.assert_almost_equal(score, ref, decimal=4)
def test_mutual_info_score(use_handle, input_labels): score = score_mutual_info(*input_labels, use_handle) ref = sk_mutual_info_score(*input_labels) np.testing.assert_almost_equal(score, ref, decimal=4)