1 files changed, 6 insertions, 10 deletions
diff --git a/tests/test-revlog-ancestry.py b/tests/test-revlog-ancestry.py
index 0834646..e5c2a42 100644
--- a/tests/test-revlog-ancestry.py
+++ b/tests/test-revlog-ancestry.py
@@ -47,31 +47,27 @@ if __name__ == '__main__':
 
     # Ancestors
     print 'Ancestors of 5'
-    for r in repo.changelog.ancestors([5]):
+    for r in repo.changelog.ancestors(5):
         print r,
 
     print '\nAncestors of 6 and 5'
-    for r in repo.changelog.ancestors([6, 5]):
+    for r in repo.changelog.ancestors(6, 5):
         print r,
 
     print '\nAncestors of 5 and 4'
-    for r in repo.changelog.ancestors([5, 4]):
-        print r,
-
-    print '\nAncestors of 7, stop at 6'
-    for r in repo.changelog.ancestors([7], 6):
+    for r in repo.changelog.ancestors(5, 4):
         print r,
 
     # Descendants
     print '\n\nDescendants of 5'
-    for r in repo.changelog.descendants([5]):
+    for r in repo.changelog.descendants(5):
         print r,
 
     print '\nDescendants of 5 and 3'
-    for r in repo.changelog.descendants([5, 3]):
+    for r in repo.changelog.descendants(5, 3):
         print r,
 
     print '\nDescendants of 5 and 4'
-    for r in repo.changelog.descendants([5, 4]):
+    for r in repo.changelog.descendants(5, 4):
         print r,