pyo3/examples/word-count/word_count/__init__.py

from .word_count import search, search_sequential, search_sequential_allow_threads

__all__ = [
    "search_py",
    "search",
    "search_sequential",
    "search_sequential_allow_threads",
]


def search_py(contents, needle):
    total = 0
    for line in contents.split():
        words = line.split(" ")
        for word in words:
            if word == needle:
                total += 1
    return total