From 6219973839570ac92b74cf5e379182c6cf65abae Mon Sep 17 00:00:00 2001 From: Will McGugan Date: Sat, 27 Feb 2021 19:57:28 +0000 Subject: [PATCH] test fixes --- rich/padding.py | 18 +++++++++++++----- 1 file changed, 13 insertions(+), 5 deletions(-) diff --git a/rich/padding.py b/rich/padding.py index 45fd7b4e..ec6556fc 100644 --- a/rich/padding.py +++ b/rich/padding.py @@ -99,16 +99,24 @@ class Padding(JupyterMixin): _Segment = Segment left = _Segment(" " * self.left, style) if self.left else None - right = _Segment(f'{" " * self.right}\n', style) + right = ( + [_Segment(f'{" " * self.right}', style), _Segment.line()] + if self.right + else [_Segment.line()] + ) blank_line: Optional[List[Segment]] = None if self.top: blank_line = [_Segment(f'{" " * width}\n', style)] yield from blank_line * self.top - for line in lines: - if left is not None: + if left: + for line in lines: yield left - yield from line - yield right + yield from line + yield from right + else: + for line in lines: + yield from line + yield from right if self.bottom: blank_line = blank_line or [_Segment(f'{" " * width}\n', style)] yield from blank_line * self.bottom