def test_comet(self, n=100, k=10):
graph = graphs.Comet(n, k)
self.assertEqual(graph.n_vertices, n)
self.assertEqual(graph.n_edges, n - 1)
self.assertEqual(graph.dw[0], k)
graph = graphs.Comet(7, 4)
adjacency = [
[0, 1, 1, 1, 1, 0, 0], # center
[1, 0, 0, 0, 0, 0, 0], # branch
[1, 0, 0, 0, 0, 0, 0], # branch
[1, 0, 0, 0, 0, 0, 0], # branch
[1, 0, 0, 0, 0, 1, 0], # tail
[0, 0, 0, 0, 1, 0, 1], # tail
[0, 0, 0, 0, 0, 1, 0], # tail
]
np.testing.assert_array_equal(graph.W.toarray(), adjacency)
coords = [[0, 0], [0, 1], [-1, 0], [0, -1], [1, 0], [2, 0], [3, 0]]
np.testing.assert_allclose(graph.coords, coords, atol=1e-10)
# Comet generalizes Path.
g1 = graphs.Comet(n, 0)
g2 = graphs.Path(n)
np.testing.assert_array_equal(g1.W.toarray(), g2.W.toarray())
# Comet generalizes Star.
g1 = graphs.Comet(n, n - 1)
g2 = graphs.Star(n)
np.testing.assert_array_equal(g1.W.toarray(), g2.W.toarray())
def test_path(self, n=5):
graph = graphs.Path(n, directed=False)
adjacency = [
[0, 1, 0, 0, 0],
[1, 0, 1, 0, 0],
[0, 1, 0, 1, 0],
[0, 0, 1, 0, 1],
[0, 0, 0, 1, 0],
]
coords = [[0, 0], [1, 0], [2, 0], [3, 0], [4, 0]]
np.testing.assert_array_equal(graph.W.toarray(), adjacency)
np.testing.assert_array_equal(graph.coords, coords)
graph = graphs.Path(n, directed=True)
adjacency = [
[0, 1, 0, 0, 0],
[0, 0, 1, 0, 0],
[0, 0, 0, 1, 0],
[0, 0, 0, 0, 1],
[0, 0, 0, 0, 0],
]
np.testing.assert_array_equal(graph.W.toarray(), adjacency)
np.testing.assert_array_equal(graph.coords, coords)
def test_networkx_signal_export(self):
graph = graphs.BarabasiAlbert(N=100, seed=42)
rs = np.random.RandomState(42)
signal1 = rs.normal(size=graph.N)
signal2 = rs.normal(size=graph.N)
graph.set_signal(signal1, "signal1")
graph.set_signal(signal2, "signal2")
graph_nx = graph.to_networkx()
for i in range(graph.N):
self.assertEqual(graph_nx.node[i]["signal1"], signal1[i])
self.assertEqual(graph_nx.node[i]["signal2"], signal2[i])
# invalid signal type
graph = graphs.Path(3)
graph.set_signal(np.array(['a', 'b', 'c']), 'sig')
self.assertRaises(ValueError, graph.to_networkx)
def test_graphtool_signal_export(self):
g = graphs.Logo()
rs = np.random.RandomState(42)
s = rs.normal(size=g.N)
s2 = rs.normal(size=g.N)
g.set_signal(s, "signal1")
g.set_signal(s2, "signal2")
g_gt = g.to_graphtool()
# Check the signals on all nodes
for i, v in enumerate(g_gt.vertices()):
self.assertEqual(g_gt.vertex_properties["signal1"][v], s[i])
self.assertEqual(g_gt.vertex_properties["signal2"][v], s2[i])
# invalid signal type
graph = graphs.Path(3)
graph.set_signal(np.array(['a', 'b', 'c']), 'sig')
self.assertRaises(TypeError, graph.to_graphtool)
def test_path(self):
graphs.Path()
def test_Path():
G = graphs.Path()
def test_Path():
G = graphs.Path()
needed_attributes_testing(G)
