From 698af805143475d2fb340e97c106ae9b2112afbe Mon Sep 17 00:00:00 2001 From: Ian Cordasco Date: Sun, 27 Mar 2016 14:32:21 -0500 Subject: [PATCH] Add test for FileProcessor#split_line --- tests/unit/test_file_processor.py | 18 ++++++++++++++++++ 1 file changed, 18 insertions(+) diff --git a/tests/unit/test_file_processor.py b/tests/unit/test_file_processor.py index 32da2cc..bfa3e55 100644 --- a/tests/unit/test_file_processor.py +++ b/tests/unit/test_file_processor.py @@ -138,3 +138,21 @@ def test_keyword_arguments_for_does_not_handle_attribute_errors(): with pytest.raises(AttributeError): file_processor.keyword_arguments_for(['fake']) + + +@pytest.mark.parametrize('unsplit_line, expected_lines', [ + ('line', []), + ('line 1\n', ['line 1']), + ('line 1\nline 2\n', ['line 1', 'line 2']), + ('line 1\n\nline 2\n', ['line 1', '', 'line 2']), +]) +def test_split_line(unsplit_line, expected_lines): + """Verify the token line spliting.""" + file_processor = processor.FileProcessor('-', options_from(), lines=[ + 'Line 1', + ]) + + actual_lines = list(file_processor.split_line((1, unsplit_line))) + assert expected_lines == actual_lines + + assert len(actual_lines) == file_processor.line_number