add some tests to appease coverage

pygraphviz · Aug 3, 2018 · 34c3cce · 34c3cce
1 parent b5df022
commit 34c3cce
Showing 1 changed file with 5 additions and 0 deletions.
diff --git a/pygraphviz/tests/test_graph.py b/pygraphviz/tests/test_graph.py
@@ -291,9 +291,14 @@ def test_edges(self):
                     [('1', '2'), ('2', '3')])
         assert_equal(sorted([tuple(sorted(e)) for e in A.in_edges_iter()]),
                     [('1', '2'), ('2', '3')])
+        assert_equal(sorted(A.edges(1)), [('1', '2')])
+        assert_equal(sorted(A.edges([1, 2])), [('1', '2'), ('2', '3')])
+        assert_equal(sorted(A.edges_iter(1)), [('1', '2')])
         assert_equal(sorted(A.out_edges(1)), [('1', '2')])
+        assert_equal(sorted(A.out_edges([1, 2])), [('1', '2'), ('2', '3')])
         assert_equal(sorted(A.out_edges_iter(1)), [('1', '2')])
         assert_equal(sorted(A.in_edges(2)), [('1', '2')])
+        assert_equal(sorted(A.in_edges([1, 2])), [('1', '2')])
         assert_equal(sorted(A.in_edges_iter(2)), [('1', '2')])
         assert_equal(A.predecessors(1), [])
         assert_equal(list(A.predecessors_iter(1)),[])