mirror of
https://github.com/Textualize/textual.git
synced 2025-10-17 02:38:12 +03:00
17 lines
605 B
Python
17 lines
605 B
Python
import pytest
|
|
from typing import cast
|
|
from textual.widgets import Tree
|
|
from textual.widgets._tree import NodeID
|
|
|
|
|
|
def test_get_tree_node_by_id() -> None:
|
|
"""It should be possible to get a TreeNode by its ID."""
|
|
tree = Tree[None]("Anakin")
|
|
child = tree.root.add("Leia")
|
|
grandchild = child.add("Ben")
|
|
assert tree.get_node_by_id(tree.root.id).id == tree.root.id
|
|
assert tree.get_node_by_id(child.id).id == child.id
|
|
assert tree.get_node_by_id(grandchild.id).id == grandchild.id
|
|
with pytest.raises(Tree.UnknownID):
|
|
tree.get_node_by_id(cast(NodeID, grandchild.id + 1000))
|