def test_remap_node_ids_unweighted_graph_raises_warning(self):
with warnings.catch_warnings(record=True) as warnings_context_manager:
graph = nx.florentine_families_graph()
gus.remap_node_ids(graph)
self.assertEqual(len(warnings_context_manager), 1)
self.assertTrue(
issubclass(warnings_context_manager[0].category, UserWarning))
self.assertTrue("Graph has at least one unweighted edge" in str(
warnings_context_manager[0].message))
def test_remap_node_ids_graph_has_same_edges_but_remapped(self):
graph = nx.Graph()
graph.add_edge(0, 1, weight=10)
graph.add_edge(1, "someid", weight=100)
new_graph, new_node_ids = gus.remap_node_ids(graph)
self._assert_graphs_are_equivalent(graph, new_graph, new_node_ids)
def test_remap_node_ids_invalid_typ_raises_typeerror(self):
invalid_types = [str, int, list]
for type in invalid_types:
with pytest.raises(TypeError):
gus.remap_node_ids(graph=type())