Function _make_tree

tests/knowledge/test_tree.py:26–81

Three-level tree: root ├── a │ ├── a1 │ └── a2 └── b

()

def _make_tree() -> _SampleTree:
    """Three-level tree:

    root
    ├── a
    │   ├── a1
    │   └── a2
    └── b
    """
    a1_id, a2_id = uuid4(), uuid4()
    a_id = uuid4()
    b_id = uuid4()
    root_id = uuid4()
    a1 = TreeNode(
        node_id=a1_id,
        parent_id=a_id,
        position=0,
        title="A1",
        summary="leaf a1",
    )
    a2 = TreeNode(
        node_id=a2_id,
        parent_id=a_id,
        position=1,
        title="A2",
        summary="leaf a2",
    )
    a = TreeNode(
        node_id=a_id,
        parent_id=root_id,
        position=0,
        title="A",
        summary="branch a",
        children_ids=[a1_id, a2_id],
    )
    b = TreeNode(
        node_id=b_id,
        parent_id=root_id,
        position=1,
        title="B",
        summary="leaf b",
    )
    root = TreeNode(
        node_id=root_id,
        parent_id=None,
        position=0,
        title="Root",
        summary="root summary",
        children_ids=[a_id, b_id],
    )
    return _SampleTree(
        as_of=_now(),
        source=_src(),
        root_node_id=root_id,
        nodes={n.node_id: n for n in [root, a, b, a1, a2]},
    )