diff --git a/networkx/drawing/tests/test_layout.py b/networkx/drawing/tests/test_layout.py index a858570a2c04..6bba0defdfbc 100644 --- a/networkx/drawing/tests/test_layout.py +++ b/networkx/drawing/tests/test_layout.py @@ -307,12 +307,10 @@ def test_arc_layout(self): sizes = (0, 5, 7, 2, 8) G = nx.complete_multipartite_graph(*sizes) - vpos = nx.arc_layout(G) + vpos = nx.arc_layout(G, subset_key="subset") assert len(vpos) == len(G) - pytest.raises( - ValueError, nx.multipartite_layout, G, radius="non-sensible input" - ) + pytest.raises(TypeError, nx.multipartite_layout, G, radius="non-sensible input") def test_kamada_kawai_costfn_1d(self): costfn = nx.drawing.layout._kamada_kawai_costfn @@ -461,7 +459,7 @@ def test_multipartite_layout_nonnumeric_partition_labels(): def test_arc_layout_nonnumeric_paratition_labels(): """see gh-5123.""" - g = nx.graph() + g = nx.Graph() g.add_node(0, subset="s0") g.add_node(1, subset="s0") g.add_node(2, subset="s1")