Merge pull request #33 from hitblast/optimize-utils

🔨 [tuning] Optimize utils using built-in func and list comprehension

avro/utils/count.py

@@ -1,8 +1,9 @@
+from avro import config
+
 # SPDX-License-Identifier: MIT
 
 
 # Import local modules.
-from avro import config
 
 
 # Functions.
@@ -11,24 +12,12 @@ def count_vowels(text: str) -> int:
     Count number of occurrences of vowels in a given string.
     """
 
-    count = 0
-
-    for i in text:
-        if i.lower() in config.AVRO_VOWELS:
-            count += 1
-
-    return count
+    return sum(1 for i in text if i.lower() in config.AVRO_VOWELS)
 
 
 def count_consonants(text: str) -> int:
     """
     Count number of occurrences of consonants in a given string.
     """
 
-    count = 0
-
-    for i in text:
-        if i.lower() in config.AVRO_CONSONANTS:
-            count += 1
-
-    return count
+    return sum(1 for i in text if i.lower() in config.AVRO_CONSONANTS)