fix tests

iterative · Nov 2, 2020 · 0aeba39 · 0aeba39
1 parent ff73586
commit 0aeba39
Showing 1 changed file with 4 additions and 4 deletions.
diff --git a/tests/func/test_stage_resolver.py b/tests/func/test_stage_resolver.py
@@ -260,17 +260,17 @@ def test_foreach_loop_dict(tmp_dir, dvc):
     d = {
         "stages": {
             "build": {
-                "foreach": iterable,
-                "in": {"cmd": "python script.py ${item}"},
+                "foreach": iterable["models"],
+                "in": {"cmd": "python script.py ${item.thresh}"},
             }
         }
     }
 
     resolver = DataResolver(dvc, PathInfo(str(tmp_dir)), d)
     assert resolver.resolve() == {
         "stages": {
-            f"build-{key}": {"cmd": f"python script.py {item}"}
-            for key, item in iterable.items()
+            f"build-{key}": {"cmd": f"python script.py {item['thresh']}"}
+            for key, item in iterable["models"].items()
         }
     }