import pytest

pytest.importorskip("numpy")
pytest.importorskip("scipy")

import networkx as nx

communicability_fns = (nx.communicability, nx.communicability_exp)


@pytest.mark.parametrize("communicability_fn", communicability_fns)
def test_communicability(communicability_fn):
    expected = {
        0: {0: 1.5430806348152435, 1: 1.1752011936438012},
        1: {0: 1.1752011936438012, 1: 1.5430806348152435},
    }

    result = communicability_fn(nx.path_graph(2))

    for k1, val in result.items():
        for k2 in val:
            assert expected[k1][k2] == pytest.approx(result[k1][k2], abs=1e-7)


@pytest.mark.parametrize("communicability_fn", communicability_fns)
def test_communicability2(communicability_fn):
    answer = {
        ("1", "1"): 1.6445956054135658,
        ("1", "Albert"): 0.7430186221096251,
        ("1", "Aric"): 0.7430186221096251,
        ("1", "Dan"): 1.6208126320442937,
        ("1", "Franck"): 0.42639707170035257,
        ("Albert", "1"): 0.7430186221096251,
        ("Albert", "Albert"): 2.4368257358712189,
        ("Albert", "Aric"): 1.4368257358712191,
        ("Albert", "Dan"): 2.0472097037446453,
        ("Albert", "Franck"): 1.8340111678944691,
        ("Aric", "1"): 0.7430186221096251,
        ("Aric", "Albert"): 1.4368257358712191,
        ("Aric", "Aric"): 2.4368257358712193,
        ("Aric", "Dan"): 2.0472097037446457,
        ("Aric", "Franck"): 1.8340111678944691,
        ("Dan", "1"): 1.6208126320442937,
        ("Dan", "Albert"): 2.0472097037446453,
        ("Dan", "Aric"): 2.0472097037446457,
        ("Dan", "Dan"): 3.1306328496328168,
        ("Dan", "Franck"): 1.4860372442192515,
        ("Franck", "1"): 0.42639707170035257,
        ("Franck", "Albert"): 1.8340111678944691,
        ("Franck", "Aric"): 1.8340111678944691,
        ("Franck", "Dan"): 1.4860372442192515,
        ("Franck", "Franck"): 2.3876142275231915,
    }

    G1 = nx.Graph(
        [
            ("Franck", "Aric"),
            ("Aric", "Dan"),
            ("Dan", "Albert"),
            ("Albert", "Franck"),
            ("Dan", "1"),
            ("Franck", "Albert"),
        ]
    )

    result = communicability_fn(G1)
    for k1, val in result.items():
        for k2 in val:
            assert answer[(k1, k2)] == pytest.approx(result[k1][k2], abs=1e-7)