Use dict comprehensions where obviously useful

fabric · Oct 18, 2017 · 82dcc9e · 82dcc9e
1 parent 1c937e3
commit 82dcc9e
Showing 1 changed file with 3 additions and 9 deletions.
diff --git a/tests/group.py b/tests/group.py
@@ -96,9 +96,7 @@ def returns_results_mapping(self):
             g = SerialGroup.from_connections(cxns)
             result = g.run("whatever", hide=True)
             ok_(isinstance(result, GroupResult))
-            expected = {}
-            for cxn in cxns:
-                expected[cxn] = cxn.run.return_value
+            expected = {x: x.run.return_value for x in cxns}
             eq_(result, expected)
             eq_(result.succeeded, expected)
             eq_(result.failed, {})
@@ -163,9 +161,7 @@ def queue_used_to_return_results(self, Queue):
             # Execute & inspect results
             g = ThreadingGroup.from_connections(cxns)
             results = g.run(*self.args, **self.kwargs)
-            expected = {}
-            for cxn in cxns:
-                expected[cxn] = cxn.run.return_value
+            expected = {x: x.run.return_value for x in cxns}
             eq_(results, expected)
             # Make sure queue was used as expected within worker &
             # ThreadingGroup.run()
@@ -213,9 +209,7 @@ def returns_results_mapping(self):
             g = ThreadingGroup.from_connections(cxns)
             result = g.run("whatever", hide=True)
             ok_(isinstance(result, GroupResult))
-            expected = {}
-            for cxn in cxns:
-                expected[cxn] = cxn.run.return_value
+            expected = {x: x.run.return_value for x in cxns}
             eq_(result, expected)
             eq_(result.succeeded, expected)
             eq_(result.failed, {})