Skip to content

Commit db0dfa0

Browse files
committedMar 18, 2018
Unnecessary generator - rewrite as a set comprehension
1 parent 87a6441 commit db0dfa0

File tree

3 files changed

+4
-4
lines changed

3 files changed

+4
-4
lines changed
 

‎git/test/test_fun.py

+1-1
Original file line numberDiff line numberDiff line change
@@ -211,7 +211,7 @@ def assert_entries(entries, num_entries, has_conflict=False):
211211
def _assert_tree_entries(self, entries, num_trees):
212212
for entry in entries:
213213
assert len(entry) == num_trees
214-
paths = set(e[2] for e in entry if e)
214+
paths = {e[2] for e in entry if e}
215215

216216
# only one path per set of entries
217217
assert len(paths) == 1

‎git/test/test_index.py

+2-2
Original file line numberDiff line numberDiff line change
@@ -147,8 +147,8 @@ def _cmp_tree_index(self, tree, index):
147147
blist.append(blob)
148148
# END for each blob in tree
149149
if len(blist) != len(index.entries):
150-
iset = set(k[0] for k in index.entries.keys())
151-
bset = set(b.path for b in blist)
150+
iset = {k[0] for k in index.entries.keys()}
151+
bset = {b.path for b in blist}
152152
raise AssertionError("CMP Failed: Missing entries in index: %s, missing in tree: %s" %
153153
(bset - iset, iset - bset))
154154
# END assertion message

‎git/test/test_tree.py

+1-1
Original file line numberDiff line numberDiff line change
@@ -88,7 +88,7 @@ def test_traverse(self):
8888

8989
# trees and blobs
9090
assert len(set(trees) | set(root.trees)) == len(trees)
91-
assert len(set(b for b in root if isinstance(b, Blob)) | set(root.blobs)) == len(root.blobs)
91+
assert len({b for b in root if isinstance(b, Blob)} | set(root.blobs)) == len(root.blobs)
9292
subitem = trees[0][0]
9393
assert "/" in subitem.path
9494
assert subitem.name == osp.basename(subitem.path)

0 commit comments

Comments
 (0)
Please sign in to comment.