diff --git a/kitty_tests/multicell.py b/kitty_tests/multicell.py index f3bd77019..5824b0e4f 100644 --- a/kitty_tests/multicell.py +++ b/kitty_tests/multicell.py @@ -618,10 +618,13 @@ def as_lists(x): for i, (e, a) in enumerate(zip(as_lists(bytes(expected)), as_lists(actual))): self.ae(e, a, f'Row: {i}') - def ast(*expected, as_ansi=False, strip_trailing_whitespace=False): + def ast(*expected, strip_trailing_whitespace=False, as_ansi=False): actual = s.text_for_selection(as_ansi, strip_trailing_whitespace) self.ae(expected, actual) + def asa(*expected, strip_trailing_whitespace=False): + ast(*expected, as_ansi=True, strip_trailing_whitespace=strip_trailing_whitespace) + s.reset() s.draw('a'), multicell(s, 'b', width=2), s.draw('c') ss(p(), p(x=1, in_left_half_of_cell=False))