From 3cf97f151ddf7031cfaa84a4e48f2c3867e1829f Mon Sep 17 00:00:00 2001 From: Will McGugan Date: Sun, 4 Jul 2021 21:11:00 +0100 Subject: [PATCH] divide edge --- rich/segment.py | 5 +++-- tests/test_segment.py | 16 ++++++++++++++++ 2 files changed, 19 insertions(+), 2 deletions(-) diff --git a/rich/segment.py b/rich/segment.py index fee76130..2f334b14 100644 --- a/rich/segment.py +++ b/rich/segment.py @@ -528,8 +528,9 @@ class Segment(NamedTuple): yield split_segments[:] return - if split_segments: - yield split_segments[:] + yield [] + # if split_segments: + # yield split_segments[:] class Segments: diff --git a/tests/test_segment.py b/tests/test_segment.py index 63532266..8ec7b961 100644 --- a/tests/test_segment.py +++ b/tests/test_segment.py @@ -205,6 +205,22 @@ def test_divide_edge(): ] +def test_divide_edge_2(): + segments = [ + Segment("╭─"), + Segment( + "────── Placeholder ───────", + ), + Segment( + "─╮", + ), + ] + result = list(Segment.divide(segments, [30, 60])) + expected = [segments, []] + print(repr(result)) + assert result == expected + + @pytest.mark.parametrize( "text,split,result", [