From 82b131cf2afebbed3723df5b5dfd5cd820716f97 Mon Sep 17 00:00:00 2001 From: Yobmod Date: Wed, 30 Jun 2021 18:41:06 +0100 Subject: Type Traversable.traverse() better, start types of submodule --- test/test_commit.py | 13 +++++++++++++ test/test_tree.py | 2 +- 2 files changed, 14 insertions(+), 1 deletion(-) (limited to 'test') 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 ) -- cgit v1.2.1