From cc61b57dd1232b7bcf1f81eaa461a5388751cf53 Mon Sep 17 00:00:00 2001 From: Kelly Brazil Date: Fri, 26 Jan 2024 11:31:59 -0800 Subject: [PATCH] add blank line tests to slicer --- tests/test_jc_utils.py | 21 +++++++++++++++++++++ 1 file changed, 21 insertions(+) diff --git a/tests/test_jc_utils.py b/tests/test_jc_utils.py index 28f216dd..8f7df5ec 100644 --- a/tests/test_jc_utils.py +++ b/tests/test_jc_utils.py @@ -246,6 +246,27 @@ line4 expected = ['line2', 'line3'] self.assertEqual(list(jc.utils.line_slice(data, 1, -1)), expected) + def test_line_slice_string_blank_lines(self): + data = '''line1 +line2 + +line4 +line5 +''' + expected = 'line2\n\nline4' + self.assertEqual(jc.utils.line_slice(data, 1, 4), expected) + + def test_line_slice_iter_positive_blank_lines(self): + data = [ + 'line1', + 'line2', + '', + 'line4', + 'line5' + ] + expected = ['line2', '', 'line4'] + self.assertEqual(list(jc.utils.line_slice(data, 1, 4)), expected) + # need to mock shutil.get_terminal_size().columns or add a column parameter to test # def test_warning_message(self):