瀏覽代碼

Extract get_deps test helper

Signed-off-by: Aanand Prasad <[email protected]>
Aanand Prasad 9 年之前
父節點
當前提交
3720b50c3b
共有 1 個文件被更改,包括 6 次插入2 次删除
  1. 6 2
      tests/unit/parallel_test.py

+ 6 - 2
tests/unit/parallel_test.py

@@ -22,6 +22,10 @@ deps = {
 }
 
 
+def get_deps(obj):
+    return deps[obj]
+
+
 def test_parallel_execute():
     results = parallel_execute(
         objects=[1, 2, 3, 4, 5],
@@ -44,7 +48,7 @@ def test_parallel_execute_with_deps():
         func=process,
         get_name=lambda obj: obj,
         msg="Processing",
-        get_deps=lambda obj: deps[obj],
+        get_deps=get_deps,
     )
 
     assert sorted(log) == sorted(objects)
@@ -67,7 +71,7 @@ def test_parallel_execute_with_upstream_errors():
         func=process,
         get_name=lambda obj: obj,
         msg="Processing",
-        get_deps=lambda obj: deps[obj],
+        get_deps=get_deps,
     )
 
     assert log == [cache]