mypy

Author: willmcgugan

tests/devtools/test_devtools.py

@@ -6,6 +6,8 @@
 from rich.align import Align
 from rich.console import Console
 from rich.segment import Segment
+from rich.table import Table
+from rich.text import Text
 
 from textual_dev.renderables import DevConsoleLog, DevConsoleNotice
 
@@ -43,19 +45,24 @@ def test_log_message_render(console) -> None:
         severity=0,
     )
     table = next(iter(message.__rich_console__(console, console.options)))
+    assert isinstance(table, Table)
 
     assert len(table.rows) == 1
 
     columns = list(table.columns)
     left_cells = list(columns[0].cells)
     left = left_cells[0]
     right_cells = list(columns[1].cells)
+    assert isinstance(right_cells[0], Align)
     right: Align = right_cells[0]
 
     # Since we can't guarantee the timezone the tests will run in...
     local_time = datetime.fromtimestamp(TIMESTAMP)
     string_timestamp = local_time.time()
 
+    assert isinstance(left, Text)
+    assert isinstance(right.renderable, Text)
+
     assert left.plain == f"[{string_timestamp}] UNDEFINED"
     assert right.align == "right"
     assert "hello.py:123" in right.renderable
@@ -64,6 +71,7 @@ def test_log_message_render(console) -> None:
 def test_internal_message_render(console):
     message = DevConsoleNotice("hello")
     rule = next(iter(message.__rich_console__(console, console.options)))
+    assert isinstance(rule, Rule)
     assert rule.title == "hello"
     assert rule.characters == "─"