def test_lobster(self):
import networkx as nx
import matplotlib.pyplot as plt
#g = nx.random_lobster(15, 0.8, 0.1)
g = nx.barbell_graph(7, 5)
#g = nx.erdos_renyi_graph(15, 0.2)
nx.draw_graphviz(g)
plt.savefig("/tmp/lobster.png")
print distancematrix.matrix_calls(g.edges(), 7)
def test_edges(self):
return
edges = []
with open('/tmp/edges') as f:
for line in f:
row = line.split()
edges.append((int(row[0]), int(row[1])))
print distancematrix.matrix_calls(edges, 10)
def test_simple(self):
pairs = [(1, 2),
(2, 3),
(7, 8),
(9, 10)]
self.assertEqual(distancematrix.matrix_calls(pairs),
([[1, 2, 3], [8, 7], [9, 10]], []))