summaryrefslogtreecommitdiff
path: root/networkx/algorithms/centrality/tests/test_current_flow_betweenness_centrality.py
diff options
context:
space:
mode:
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.py27
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)