Update counts

cooccurrence_count.py

@@ -36,7 +36,8 @@ Args:
     `word1`: The first word.
     `word2`: The second word.
 Returns:
-    count: The co-occurrence count of word1 and word2 in data.
+    count: The count of total sentences.
+    co_occurrence_count: The co-occurrence count of word1 and word2 in data.
 Examples:
     >>> data = ["hello sun","hello moon", "hello sun"]
     >>> c_count = evaluate.load("prb977/cooccurrence_count")

tests.py

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