def test_map_tree_idents():
"""Verify that certain things do and don't work"""
lst = [1, 2, 3]
assert map_tree(lambda x: x, lst) == lst
t = (1, 2, 3)
assert map_tree(lambda x: x, t) == t
def test_postorder_transformation():
def coerce_to_int(o):
return int(o)
inp = dict(a=2.3, b=4.4)
with pytest.raises(TypeError):
map_tree(coerce_to_int, inp)
assert dict(a=2, b=4) == map_tree(coerce_to_int, inp, postorder=True)
assert dict(p=7, g=[2, 3, 4]) == map_tree(coerce_to_int,
dict(p=7.4, g=[2.01, 3.01,
4.01]),
postorder=True)
def test_compose():
itx = compose(lambda s: s + "i", str, int)
tx = type_dispatched_transform({float: itx})
d = dict(p=[1.2, 3.4, 8.8])
assert map_tree(tx, d) == dict(p=["1i", "3i", "8i"])
def test_type_dispatched_path_transform():
pathed_int_tx = make_path_only_transform(("floats_to_int", ), int)
typed_tx = type_dispatched_transform({float: pathed_int_tx, int: str})
o = dict(floats_to_int=[1.2, 2.3, 3.4],
ints_to_str=[1, 2, 3],
other_floats=(1.2, 2.2))
assert map_tree(typed_tx, o) == dict(floats_to_int=[1, 2, 3],
ints_to_str=["1", "2", "3"],
other_floats=(1.2, 2.2))
def test_map_tree_type_dispatched_tx():
tx = type_dispatched_transform({Decimal: decimal_to_number, float: int})
d = dict(one=[
dict(two=tuple([dict(three=Decimal("3")), "blah"]), twotwo=2, ff=4.2)
],
oneone="string")
dr = map_tree(tx, d)
assert isinstance(dr["one"][0]["two"][0]["three"], int)
assert dr == dict(
one=[dict(two=tuple([dict(three=3), "blah"]), twotwo=2, ff=4)],
oneone="string")
def test_path_only_transform():
tp = make_path_only_transform(("a", "four"), str)
inp = dict(a=[dict(four=[1, 2, 3]), [8, 8, dict(four=(3, 3, 3))]],
b=[2, 4, 6],
c=dict(four=[2, 3, 4]))
out = map_tree(tp, inp)
assert out == dict(
a=[dict(four="[1, 2, 3]"), [8, 8, dict(four="(3, 3, 3)")]],
b=[2, 4, 6],
c=dict(four=[2, 3, 4]),
)