Minor changes

word_forms/constants.py

@@ -1,6 +1,11 @@
-from collections import defaultdict
-from pathlib import Path
 import json
+from pathlib import Path
+from collections import defaultdict
+
+
+ABSOLUTE_PATH = Path(__file__).parent.absolute()
+ADJECTIVE_TO_ADVERB = json.loads((ABSOLUTE_PATH / "adj_to_adv.txt").read_text())
+
 
 class Verb(object):
     def __init__(self, verbs=None):
@@ -13,17 +18,11 @@ def update(self, verbs):
         self.verbs.update(verbs)
 
 
-verbs_fh = open(str(Path(__file__).parent.absolute() / "en-verbs.txt"))
-lines = verbs_fh.readlines()
-verbs_fh.close()
-
 CONJUGATED_VERB_DICT = defaultdict(Verb)
-for line in lines:
-    if line[0] != ";":
+with open(ABSOLUTE_PATH / "en-verbs.txt") as lines:
+    for line in lines:
+        if line.startswith(";"):
+            continue
         verbs = {string for string in line.strip().split(",") if string != ""}
         for verb in verbs:
             CONJUGATED_VERB_DICT[verb].update(verbs)
-
-adj_fh = open(str(Path(__file__).parent.absolute() / "adj_to_adv.txt"))
-ADJECTIVE_TO_ADVERB = json.load(adj_fh)
-adj_fh.close()

word_forms/lemmatizer.py

@@ -5,12 +5,7 @@ def lemmatize(word):
     """
     Out of all the related word forms of ``word``, return the smallest form that appears first in the dictionary
     """
-    forms = [word for pos_form in get_word_forms(word).values() for word in pos_form]
-    forms.sort()
-    forms.sort(key=len)
-    try:
-        return forms[0]
-    except IndexError:
+    forms = get_word_forms(word).values()
+    if not any(forms):
         raise ValueError("{} is not a real word".format(word))
-
-
+    return min([word for pos_form in forms for word in pos_form], key=len)