Как добавить подсветку синтаксиса Python в файл уценки в vim?

Question

Для Vim, как мне добавить подсветку синтаксиса Python к файлам уценки?

контекст

Я создаю документ уценки, чтобы показать, как использовать проект. Довольно простая вещь

if __name__ == "__main__":
    result_queue = find_data_corruption()

    corruption_data = []
    while not result_queue.finished():
        if result_queue.has_results():
            corruption_data.append(result_queue.get())

Так что я сижу здесь, пишу пример кода в этом файле уценки, и я действительно хочу подсветку синтаксиса, потому что я не ненавижу себя.