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)