def test_remove_nodelim_with_nodelim(): lines = [(1, 'nodelim'), (2, '1 2 3'), (3, 'and_no_delim')] result = list(wcut.suppress_no_delim_lines(lines, ' ')) expected = [(2, '1 2 3')] assert result == expected
def test_remove_preheader_and_no_delim(): lines = [(1, 'preheader'), (2, '1 2 3'), (3, 'nodelim')] lines = wcut.suppress_preheader_lines(lines, 2) lines = wcut.suppress_no_delim_lines(lines, ' ') result = list(lines) expected = [(2, '1 2 3')] assert result == expected
def test_remove_nodelim_when_all_have_delim(): lines = [(1, '1 2 3'), (2, '3 4 5')] result = list(wcut.suppress_no_delim_lines(lines, ' ')) expected = [(1, '1 2 3'), (2, '3 4 5')] assert result == expected