Skip to content
Prev Previous commit
Next Next commit
Unnecessary generator - rewrite as a set comprehension
  • Loading branch information
hugovk committed Mar 18, 2018
commit db0dfa07e34ed80bfe0ce389da946755ada13c5d
2 changes: 1 addition & 1 deletion git/test/test_fun.py
Original file line number Diff line number Diff line change
Expand Up @@ -211,7 +211,7 @@ def assert_entries(entries, num_entries, has_conflict=False):
def _assert_tree_entries(self, entries, num_trees):
for entry in entries:
assert len(entry) == num_trees
paths = set(e[2] for e in entry if e)
paths = {e[2] for e in entry if e}

# only one path per set of entries
assert len(paths) == 1
Expand Down
4 changes: 2 additions & 2 deletions git/test/test_index.py
Original file line number Diff line number Diff line change
Expand Up @@ -147,8 +147,8 @@ def _cmp_tree_index(self, tree, index):
blist.append(blob)
# END for each blob in tree
if len(blist) != len(index.entries):
iset = set(k[0] for k in index.entries.keys())
bset = set(b.path for b in blist)
iset = {k[0] for k in index.entries.keys()}
bset = {b.path for b in blist}
raise AssertionError("CMP Failed: Missing entries in index: %s, missing in tree: %s" %
(bset - iset, iset - bset))
# END assertion message
Expand Down
2 changes: 1 addition & 1 deletion git/test/test_tree.py
Original file line number Diff line number Diff line change
Expand Up @@ -88,7 +88,7 @@ def test_traverse(self):

# trees and blobs
assert len(set(trees) | set(root.trees)) == len(trees)
assert len(set(b for b in root if isinstance(b, Blob)) | set(root.blobs)) == len(root.blobs)
assert len({b for b in root if isinstance(b, Blob)} | set(root.blobs)) == len(root.blobs)
subitem = trees[0][0]
assert "/" in subitem.path
assert subitem.name == osp.basename(subitem.path)
Expand Down