def test_random_geometric_pos_inf_norm(self):
pos = [[0.1, 0.1], [0.2, 0.2], [0.3, 0.3]]
graph = retworkx.random_geometric_graph(3,
0.11,
pos=pos,
p=float("inf"))
self.assertEqual(set(graph.edge_list()), {(0, 1), (1, 2)})
def test_random_geometric_pos_num_nodes_incomp(self):
with self.assertRaises(ValueError):
retworkx.random_geometric_graph(3, 0.15, pos=[[0.5, 0.5]])
def test_random_geometric_num_nodes_invalid(self):
with self.assertRaises(ValueError):
retworkx.random_geometric_graph(0, 1.0)
def test_random_geometric_pos(self):
pos = [[0.1, 0.1], [0.2, 0.2], [0.3, 0.3]]
graph = retworkx.random_geometric_graph(3, 0.15, pos=pos)
self.assertEqual(set(graph.edge_list()), {(0, 1), (1, 2)})
for i in range(3):
self.assertEqual(graph[i]["pos"], pos[i])
def test_random_geometric_dim(self):
graph = retworkx.random_geometric_graph(10, 0.5, dim=3)
self.assertEqual(len(graph[0]["pos"]), 3)
def test_random_geometric_same_seed(self):
# with other arguments equal, same seed results in same graph
graph_s1 = retworkx.random_geometric_graph(20, 0.5, seed=10)
graph_s2 = retworkx.random_geometric_graph(20, 0.5, seed=10)
self.assertEqual(graph_s1.edge_list(), graph_s2.edge_list())
def test_random_geometric_complete(self):
r = 1.42 # > sqrt(2)
graph = retworkx.random_geometric_graph(10, r)
self.assertEqual(len(graph), 10)
self.assertEqual(len(graph.edges()), 45)
def test_random_geometric_empty(self):
graph = retworkx.random_geometric_graph(20, 0)
self.assertEqual(len(graph), 20)
self.assertEqual(len(graph.edges()), 0)
