test fix

Textualize · Oct 1, 2024 · fd7b326 · fd7b326
1 parent 3936deb
commit fd7b326
Show file tree

Hide file tree

Showing 2 changed files with 4 additions and 2 deletions.
diff --git a/rich/console.py b/rich/console.py
@@ -1005,7 +1005,8 @@ def size(self) -> ConsoleDimensions:
  width: Optional[int] = None
  height: Optional[int] = None
 
- for file_descriptor in _STD_STREAMS_OUTPUT if WINDOWS else _STD_STREAMS:
+ streams = _STD_STREAMS_OUTPUT if WINDOWS else _STD_STREAMS
+ for file_descriptor in streams:
  try:
  width, height = os.get_terminal_size(file_descriptor)
  except (AttributeError, ValueError, OSError): # Probably not a terminal

diff --git a/tests/test_panel.py b/tests/test_panel.py
@@ -111,6 +111,7 @@ def test_title_text() -> None:
  console = Console(
  file=io.StringIO(),
  width=50,
+ height=20,
  legacy_windows=False,
  force_terminal=True,
  color_system="truecolor",
@@ -119,7 +120,7 @@ def test_title_text() -> None:
 
  result = console.file.getvalue()
  print(repr(result))
- expected = "╭───────────────────────────────────\x1b[31m title \x1b[0m────────────────────────────────────╮\n│ Hello, World  │\n╰──────────────────────────────────\x1b[1;35m subtitle \x1b[0m──────────────────────────────────╯\n"
+ expected = "╭────────────────────\x1b[31m title \x1b[0m─────────────────────╮\n│ Hello, World │\n╰───────────────────\x1b[1;35m subtitle \x1b[0m───────────────────╯\n"
  assert result == expected