10. Popular Words / Вирішуємо задачі на Python CheckIO Українською

# ----------------------        tasks/popular_words.py ---------------

from collections import defaultdict, Counter


def popular_words(text: str, words: list) -> dict:

    res = {}

    text = text.lower().split()

    for word in words:
        res[word] = text.count(word)

    return res


def popular_words_list(text: str, words: list) -> dict:

    text = text.lower().split()
    return dict([word, text.count(word)] for word in words)


def popular_words_comp(text: str, words: list) -> dict:

    text = text.lower().split()
    return {word: text.count(word) for word in words}


def popular_words_perf(text: str, words: list) -> dict:

    words = set(words)

    res = dict.fromkeys(words, 0)
    for word in text.lower().split():
        if word in words:
            res[word] = res[word] + 1

    return res


def popular_words_all(text: str, words: list) -> dict:

    res = {}
    for word in text.lower().split():
        res[word] = res.get(word, 0) + 1

    return {word: res.get(word, 0) for word in words}


def popular_words_defaultdict(text: str, words: list) -> dict:

    res = defaultdict(int)
    for word in text.lower().split():
        res[word] += 1

    return {word: res[word] for word in words}


def popular_words_counter(text: str, words: list) -> dict:
    res = Counter(text.lower().split())
    return {word: res[word] for word in words}



# --------------  tests/test_popular_words.py ------------------

import pytest

from tasks.popular_words import (popular_words, popular_words_list,
                                 popular_words_comp, popular_words_perf,
                                 popular_words_all, popular_words_defaultdict,
                                 popular_words_counter)


@pytest.mark.parametrize("callable", [
    popular_words,
    popular_words_list,
    popular_words_comp,
    popular_words_perf,
    popular_words_all,
    popular_words_defaultdict,
    popular_words_counter])
def test_popular_words(callable):
    assert callable('''
    When I was One
    I had just begun
    When I was Two
    I was nearly new
    ''', ['i', 'was', 'three', 'near']) == {
            'i': 4,
            'was': 3,
            'three': 0,
            'near': 0
        }