tests.py

test_cases = [
    {
        "data": ["hello sun", "hello moon", "hello sun"],
        "word1": "hello",
        "word2": "sun",
        "result": {"count": 3, "co_occurrence_count": 2}
    },
]