def test_tokenize_ansi(prompt_tokens, ansi_string_parts):
ansi_tokens = tokenize_ansi(prompt_tokens)
for token, text in zip(ansi_tokens, ansi_string_parts):
assert token[1] == text
def _invalidate():
new_prompt = "".join(self.tokens)
formatted_tokens = tokenize_ansi(
PygmentsTokens(partial_color_tokenize(new_prompt)))
setattr(self.session, self.name, formatted_tokens)
self.session.app.invalidate()