diff options
Diffstat (limited to 'networkx/algorithms/centrality/tests/test_current_flow_betweenness_centrality.py')
-rw-r--r-- | networkx/algorithms/centrality/tests/test_current_flow_betweenness_centrality.py | 27 |
1 files changed, 13 insertions, 14 deletions
diff --git a/networkx/algorithms/centrality/tests/test_current_flow_betweenness_centrality.py b/networkx/algorithms/centrality/tests/test_current_flow_betweenness_centrality.py index 69b06cb9..eea05f43 100644 --- a/networkx/algorithms/centrality/tests/test_current_flow_betweenness_centrality.py +++ b/networkx/algorithms/centrality/tests/test_current_flow_betweenness_centrality.py @@ -1,7 +1,6 @@ import pytest import networkx as nx -from networkx.testing import almost_equal from networkx import edge_current_flow_betweenness_centrality as edge_current_flow from networkx import approximate_current_flow_betweenness_centrality as approximate_cfbc @@ -17,19 +16,19 @@ class TestFlowBetweennessCentrality: b = nx.current_flow_betweenness_centrality(G, normalized=True) b_answer = {0: 0.25, 1: 0.25, 2: 0.25, 3: 0.25} for n in sorted(G): - assert almost_equal(b[n], b_answer[n]) + assert b[n] == pytest.approx(b_answer[n], abs=1e-7) G.add_edge(0, 1, weight=0.5, other=0.3) b = nx.current_flow_betweenness_centrality(G, normalized=True, weight=None) for n in sorted(G): - assert almost_equal(b[n], b_answer[n]) + assert b[n] == pytest.approx(b_answer[n], abs=1e-7) wb_answer = {0: 0.2222222, 1: 0.2222222, 2: 0.30555555, 3: 0.30555555} b = nx.current_flow_betweenness_centrality(G, normalized=True, weight="weight") for n in sorted(G): - assert almost_equal(b[n], wb_answer[n]) + assert b[n] == pytest.approx(wb_answer[n], abs=1e-7) wb_answer = {0: 0.2051282, 1: 0.2051282, 2: 0.33974358, 3: 0.33974358} b = nx.current_flow_betweenness_centrality(G, normalized=True, weight="other") for n in sorted(G): - assert almost_equal(b[n], wb_answer[n]) + assert b[n] == pytest.approx(wb_answer[n], abs=1e-7) def test_K4(self): """Betweenness centrality: K4""" @@ -40,7 +39,7 @@ class TestFlowBetweennessCentrality: ) b_answer = {0: 0.75, 1: 0.75, 2: 0.75, 3: 0.75} for n in sorted(G): - assert almost_equal(b[n], b_answer[n]) + assert b[n] == pytest.approx(b_answer[n], abs=1e-7) def test_P4_normalized(self): """Betweenness centrality: P4 normalized""" @@ -48,7 +47,7 @@ class TestFlowBetweennessCentrality: b = nx.current_flow_betweenness_centrality(G, normalized=True) b_answer = {0: 0, 1: 2.0 / 3, 2: 2.0 / 3, 3: 0} for n in sorted(G): - assert almost_equal(b[n], b_answer[n]) + assert b[n] == pytest.approx(b_answer[n], abs=1e-7) def test_P4(self): """Betweenness centrality: P4""" @@ -56,7 +55,7 @@ class TestFlowBetweennessCentrality: b = nx.current_flow_betweenness_centrality(G, normalized=False) b_answer = {0: 0, 1: 2, 2: 2, 3: 0} for n in sorted(G): - assert almost_equal(b[n], b_answer[n]) + assert b[n] == pytest.approx(b_answer[n], abs=1e-7) def test_star(self): """Betweenness centrality: star""" @@ -65,7 +64,7 @@ class TestFlowBetweennessCentrality: b = nx.current_flow_betweenness_centrality(G, normalized=True) b_answer = {"a": 1.0, "b": 0.0, "c": 0.0, "d": 0.0} for n in sorted(G): - assert almost_equal(b[n], b_answer[n]) + assert b[n] == pytest.approx(b_answer[n], abs=1e-7) def test_solvers2(self): """Betweenness centrality: alternate solvers""" @@ -76,7 +75,7 @@ class TestFlowBetweennessCentrality: ) b_answer = {0: 0.75, 1: 0.75, 2: 0.75, 3: 0.75} for n in sorted(G): - assert almost_equal(b[n], b_answer[n]) + assert b[n] == pytest.approx(b_answer[n], abs=1e-7) class TestApproximateFlowBetweennessCentrality: @@ -149,7 +148,7 @@ class TestEdgeFlowBetweennessCentrality: b_answer = dict.fromkeys(G.edges(), 0.25) for (s, t), v1 in b_answer.items(): v2 = b.get((s, t), b.get((t, s))) - assert almost_equal(v1, v2) + assert v1 == pytest.approx(v2, abs=1e-7) def test_K4_normalized(self): """Edge flow betweenness centrality: K4""" @@ -158,7 +157,7 @@ class TestEdgeFlowBetweennessCentrality: b_answer = dict.fromkeys(G.edges(), 0.75) for (s, t), v1 in b_answer.items(): v2 = b.get((s, t), b.get((t, s))) - assert almost_equal(v1, v2) + assert v1 == pytest.approx(v2, abs=1e-7) def test_C4(self): """Edge flow betweenness centrality: C4""" @@ -167,7 +166,7 @@ class TestEdgeFlowBetweennessCentrality: b_answer = {(0, 1): 1.25, (0, 3): 1.25, (1, 2): 1.25, (2, 3): 1.25} for (s, t), v1 in b_answer.items(): v2 = b.get((s, t), b.get((t, s))) - assert almost_equal(v1, v2) + assert v1 == pytest.approx(v2, abs=1e-7) def test_P4(self): """Edge betweenness centrality: P4""" @@ -176,4 +175,4 @@ class TestEdgeFlowBetweennessCentrality: b_answer = {(0, 1): 1.5, (1, 2): 2.0, (2, 3): 1.5} for (s, t), v1 in b_answer.items(): v2 = b.get((s, t), b.get((t, s))) - assert almost_equal(v1, v2) + assert v1 == pytest.approx(v2, abs=1e-7) |