Stone_Red/dvc
1
0
Fork 0
mirror of https://github.com/Stone-Red-Code/dvc.git synced 2026-11-02 07:18:25 +01:00
Code Issues Packages Projects Releases Wiki Activity

parsing: fix foreach resolution with int keys (#8904)

Browse Source
This commit is contained in:
daniele
2023-03-20 18:40:30 +05:45
committed by GitHub
parent d1c5fcc1b8
commit 6c1b786564
2 changed files with 18 additions and 1 deletions
Download Patch File Download Diff File Expand all files Collapse all files
tests/func/parsing/test_foreach.py
+17
View File
@@ -44,6 +44,23 @@ def test_with_dict_data(tmp_dir, dvc):
assert not resolver.tracked_vars["build@model2"]
def test_with_dict_with_non_str_keys(tmp_dir, dvc):
resolver = DataResolver(dvc, tmp_dir.fs_path, {})
context = Context()
foreach_data = {2021: {"thresh": "foo"}, 2022: {"thresh": "bar"}}
data = {"foreach": foreach_data, "do": {"cmd": "echo ${key} ${item.thresh}"}}
definition = ForeachDefinition(resolver, context, "build", data)
assert definition.resolve_one("2021") == {"build@2021": {"cmd": "echo 2021 foo"}}
assert definition.resolve_one("2022") == {"build@2022": {"cmd": "echo 2022 bar"}}
# check that `foreach` item-key replacement didnot leave any leftovers.
assert not context
assert not resolver.tracked_vars["build@2021"]
assert not resolver.tracked_vars["build@2022"]
def test_with_composite_list(tmp_dir, dvc):
resolver = DataResolver(dvc, tmp_dir.fs_path, {})