perf(strings): optimize anagram signature using frequency counts

strings/anagrams.py

@@ -6,27 +6,32 @@
 
 
 def signature(word: str) -> str:
-    """Return a word sorted
+    """
+    Return a word's frequency-based signature.
+
     >>> signature("test")
-    'estt'
+    'e1s1t2'
     >>> signature("this is a test")
-    '   aehiisssttt'
+    ' 3a1e1h1i2s3t3'
     >>> signature("finaltest")
-    'aefilnstt'
+    'a1e1f1i1l1n1s1t2'
     """
-    return "".join(sorted(word))
+    freq = collections.Counter(word)
+    return "".join(f"{ch}{freq[ch]}" for ch in sorted(freq))
 
 
 def anagram(my_word: str) -> list[str]:
-    """Return every anagram of the given word
+    """
+    Return every anagram of the given word from the dictionary.
+
     >>> anagram('test')
     ['sett', 'stet', 'test']
     >>> anagram('this is a test')
     []
     >>> anagram('final')
     ['final']
     """
-    return word_by_signature[signature(my_word)]
+    return word_by_signature.get(signature(my_word), [])
 
 
 data: str = Path(__file__).parent.joinpath("words.txt").read_text(encoding="utf-8")
@@ -39,6 +44,6 @@ def anagram(my_word: str) -> list[str]:
 if __name__ == "__main__":
     all_anagrams = {word: anagram(word) for word in word_list if len(anagram(word)) > 1}
 
-    with open("anagrams.txt", "w") as file:
-        file.write("all_anagrams = \n ")
+    with open("anagrams.txt", "w", encoding="utf-8") as file:
+        file.write("all_anagrams = \n")
         file.write(pprint.pformat(all_anagrams))