Type Traversable.traverse() better, start types of submodule

2 files changed, 14 insertions, 1 deletions

diff --git a/test/test_commit.py b/test/test_commit.py
index 2fe80530..34b91ac7 100644
--- a/test/test_commit.py
+++ b/test/test_commit.py
@@ -179,6 +179,19 @@ class TestCommit(TestCommitSerialization):
 
         # at some point, both iterations should stop
         self.assertEqual(list(bfirst)[-1], first)
+
+        stoptraverse = self.rorepo.commit("254d04aa3180eb8b8daf7b7ff25f010cd69b4e7d").traverse(ignore_self=0,
+                                                                                               as_edge=True)
+        stoptraverse_list = list(stoptraverse)
+        for itemtup in stoptraverse_list:
+            self.assertIsInstance(itemtup, (tuple)) and self.assertEqual(len(itemtup), 2)  # as_edge=True -> tuple
+            src, item = itemtup
+            self.assertIsInstance(item, Commit)
+            if src:
+                self.assertIsInstance(src, Commit)
+            else:
+                self.assertIsNone(src)   # ignore_self=0 -> first is (None, Commit)
+
         stoptraverse = self.rorepo.commit("254d04aa3180eb8b8daf7b7ff25f010cd69b4e7d").traverse(as_edge=True)
         self.assertEqual(len(next(stoptraverse)), 2)
 
diff --git a/test/test_tree.py b/test/test_tree.py
index 49b34c5e..0607d8e3 100644
--- a/test/test_tree.py
+++ b/test/test_tree.py
@@ -8,7 +8,7 @@ from io import BytesIO
 import sys
 from unittest import skipIf
 
-from git import (
+from git.objects import (
     Tree,
     Blob
 )