def test_prompt_already_with_space_extra_spaces(prompt): original = '<span class="gp">{} </span> foo'.format(prompt) assert style_space_after_prompt(original) == original
def test_multiple_spaces_after_prompt(prompt): original = '<span class="gp">{}</span> foo'.format(prompt) expected = '<span class="gp">{} </span> foo'.format(prompt) assert style_space_after_prompt(original) == expected