def METHOD_NAME(PointT):
    p = PointT(x=1, y=2, meta={"abc": True})
    leaves = jax.tree_util.tree_leaves(p)
    assert leaves == [1, 2]
    new_p = jax.tree_map(lambda x: x + x, p)
    assert new_p == PointT(x=2, y=4, meta={"abc": True})