diff options
Diffstat (limited to 'networkx/classes/tests')
| -rw-r--r-- | networkx/classes/tests/test_function.py | 10 | ||||
| -rw-r--r-- | networkx/classes/tests/test_graph.py | 11 |
2 files changed, 16 insertions, 5 deletions
diff --git a/networkx/classes/tests/test_function.py b/networkx/classes/tests/test_function.py index 5c3fbe95..0eeb6c20 100644 --- a/networkx/classes/tests/test_function.py +++ b/networkx/classes/tests/test_function.py @@ -107,16 +107,16 @@ class TestFunction: ) G = self.G.copy() - nlist = [None] + nlist = ["node"] nx.add_path(G, nlist) assert edges_equal(G.edges(nlist), []) - assert nodes_equal(G, list(self.G) + [None]) + assert nodes_equal(G, list(self.G) + ["node"]) G = self.G.copy() - nlist = iter([None]) + nlist = iter(["node"]) nx.add_path(G, nlist) - assert edges_equal(G.edges([None]), []) - assert nodes_equal(G, list(self.G) + [None]) + assert edges_equal(G.edges(["node"]), []) + assert nodes_equal(G, list(self.G) + ["node"]) G = self.G.copy() nlist = [12] diff --git a/networkx/classes/tests/test_graph.py b/networkx/classes/tests/test_graph.py index 86744403..288a7efb 100644 --- a/networkx/classes/tests/test_graph.py +++ b/networkx/classes/tests/test_graph.py @@ -29,6 +29,17 @@ class BaseGraphTester: assert sorted(G.nodes()) == self.k3nodes assert sorted(G.nodes(data=True)) == [(0, {}), (1, {}), (2, {})] + def test_none_node(self): + G = self.Graph() + with pytest.raises(ValueError): + G.add_node(None) + with pytest.raises(ValueError): + G.add_nodes_from([None]) + with pytest.raises(ValueError): + G.add_edge(0, None) + with pytest.raises(ValueError): + G.add_edges_from([(0, None)]) + def test_has_node(self): G = self.K3 assert G.has_node(1) |