def test_select_unwrap():
context = Context({"dct": {"foo": "bar"}}, lst=[1, 2, 3], foo="foo")
assert context.select("dct.foo", unwrap=True) == "bar"
assert context.select("lst.0", unwrap=True) == 1
assert context.select("foo", unwrap=True) == "foo"
node = context.select("dct", unwrap=True)
assert isinstance(node, dict) and recurse_not_a_node(node)
assert node == {"foo": "bar"}
node = context.select("lst", unwrap=True)
assert isinstance(node, list) and recurse_not_a_node(node)
assert node == [1, 2, 3]
def test_node_value():
d = {"dct": {"foo": "bar"}, "lst": [1, 2, 3], "foo": "foo"}
context = Context(d)
assert isinstance(context, (Context, CtxDict))
assert isinstance(context["dct"], CtxDict)
assert isinstance(context["lst"], CtxList)
assert isinstance(context["foo"], Value)
assert isinstance(context["dct"]["foo"], Value)
assert isinstance(context["lst"][0], Value)
assert context.value == d
assert recurse_not_a_node(context.value)
assert isinstance(context.value["dct"], dict)
assert isinstance(context.value["lst"], list)
assert isinstance(context.value["foo"], str)
assert isinstance(context.value["dct"]["foo"], str)
assert isinstance(context.value["lst"][0], int)
assert isinstance(context["dct"].value, dict)
assert context["dct"]["foo"].value == "bar"
assert isinstance(context["lst"].value, list)
assert context["lst"][1].value == 2
assert context["foo"].value == "foo"
def test_resolve_collection():
from .test_stage_resolver import (
CONTEXT_DATA,
RESOLVED_DVC_YAML_DATA,
TEMPLATED_DVC_YAML_DATA,
)
context = Context(CONTEXT_DATA)
resolved = context.resolve(TEMPLATED_DVC_YAML_DATA)
assert resolved == RESOLVED_DVC_YAML_DATA
assert recurse_not_a_node(resolved)
