aaaand the alloc_to becomes a dict comprehension

hjwp · hjwp · commit 9257c6f3a754 · 2019-02-05T10:15:06.000Z
diff --git a/domain_model.py b/domain_model.py
@@ -9,11 +9,11 @@ def skus(d):
 
 
 def allocate_to(order, source):
-    allocations = {}
-    for sku, quantity in order.items():
-        if source.get(sku, 0) > quantity:
-            allocations[sku] = source
-    return allocations
+    return {
+        sku: source
+        for sku, quantity in order.items()
+        if source.get(sku, 0) > quantity
+    }
 
 
 def allocate(order, stock, shipments):
