def test_fmap(): t = Tree(1, [ Tree(2, [ Tree(3) , Tree(4) ]) , Tree(5, [ Tree(6, [ Tree(7) ]) ]) ]) expected = Tree(2, [ Tree(4, [ Tree(6) , Tree(8) ]) , Tree(10, [ Tree(12, [ Tree(14) ]) ]) ]) assert_equals(expected, t.fmap(lambda x: x*2))
def test_fmap_no_children(): t = Tree(1) assert_equals(Tree(2), t.fmap(lambda x: x*2))