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