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
