Tidy up the disabled snapshot test

This commit is contained in:
Dave Pearson
2023-02-14 16:20:52 +00:00
parent b4f8a6b778
commit 26df6aeb00

View File

@@ -1,13 +1,26 @@
from textual.app import App, ComposeResult
from textual.containers import Vertical
from textual.containers import Vertical, Horizontal
from textual.widgets import (
Header, Footer, Button, DataTable, Input, ListView, ListItem, Label, Tree, TextLog
Header,
Footer,
Button,
DataTable,
Input,
ListView,
ListItem,
Label,
Tree,
TextLog,
)
class WidgetDisableTestApp( App[ None ] ):
class WidgetDisableTestApp(App[None]):
CSS = """
DataTable, ListView, Tree {
Horizontal {
height: auto;
}
DataTable, ListView, Tree, TextLog {
height: 2;
}
"""
@@ -15,30 +28,34 @@ class WidgetDisableTestApp( App[ None ] ):
@property
def data_table(self) -> DataTable:
data_table = DataTable()
data_table.add_columns( "Column 1", "Column 2", "Column 3", "Column 4" )
data_table.add_columns("Column 1", "Column 2", "Column 3", "Column 4")
data_table.add_rows(
[ ( str( n ), str( n * 10 ), str( n * 100 ), str( n * 1000 ) ) for n in range( 100 ) ]
[(str(n), str(n * 10), str(n * 100), str(n * 1000)) for n in range(100)]
)
return data_table
@property
def list_view( self ) -> ListView:
return ListView(
*[ ListItem( Label( f"This is list item {n}" ) ) for n in range( 20 ) ]
)
def list_view(self) -> ListView:
return ListView(*[ListItem(Label(f"This is list item {n}")) for n in range(20)])
@property
def test_tree( self ) -> Tree:
tree = Tree[None]( label="This is a test tree" )
for n in range( 10 ):
tree.root.add_leaf( f"Leaf {n}" )
def test_tree(self) -> Tree:
tree = Tree[None](label="This is a test tree")
for n in range(10):
tree.root.add_leaf(f"Leaf {n}")
tree.root.expand()
return tree
def compose( self ) -> ComposeResult:
def compose(self) -> ComposeResult:
yield Header()
yield Vertical(
Horizontal(
Button(),
Button(variant="primary"),
Button(variant="success"),
Button(variant="warning"),
Button(variant="error"),
),
self.data_table,
self.list_view,
self.test_tree,
@@ -54,5 +71,6 @@ class WidgetDisableTestApp( App[ None ] ):
self.query_one(TextLog).write("Hello, World!")
self.query_one("#test-container", Vertical).disabled = True
if __name__ == "__main__":
WidgetDisableTestApp().run()