From 346659f47f6533e318d276cb2425b00130e4f8f7 Mon Sep 17 00:00:00 2001 From: Dave Pearson Date: Mon, 9 Jan 2023 13:28:25 +0000 Subject: [PATCH] Move typing of the tree into its own class --- tests/tree/test_tree_messages.py | 10 +++++++--- 1 file changed, 7 insertions(+), 3 deletions(-) diff --git a/tests/tree/test_tree_messages.py b/tests/tree/test_tree_messages.py index 1537c8c00..67620d70e 100644 --- a/tests/tree/test_tree_messages.py +++ b/tests/tree/test_tree_messages.py @@ -6,6 +6,10 @@ from textual.widgets import Tree from textual.message import Message +class MyTree(Tree[None]): + pass + + class TreeApp(App[None]): """Test tree app.""" @@ -15,11 +19,11 @@ class TreeApp(App[None]): def compose(self) -> ComposeResult: """Compose the child widgets.""" - yield Tree[None]("Root") + yield MyTree("Root") def on_mount(self) -> None: - self.query_one(Tree[None]).root.add("Child") - self.query_one(Tree[None]).focus() + self.query_one(MyTree).root.add("Child") + self.query_one(MyTree).focus() def record(self, event: Message) -> None: self.messages.append(event.__class__.__name__)