Fixed tests

PyMultiDictionary/_dictionary.py

@@ -48,11 +48,12 @@ class MultiDictionary(object):
     Dictionary. Support synonyms, antonyms, meanings, and translations from some languages.
     """
 
+    _max_cached_websites: int  # Maximum stored websites
     _langs: Dict[str, Tuple[bool, bool, bool, bool]]  # synonyms, meaning, translation, antonym
     _test_cached_file: Dict[str, str]  # If defined, loads that file instead
     _tokenize: bool  # Enables word tokenizer
-    _words: List[str]
-    _words_lang: str
+    _words: List[str]  # List of words passed to the constructor
+    _words_lang: str  # Language of the words passed to the constructor
 
     def __init__(self, *words: Tuple[str, ...]) -> None:
         """
@@ -82,6 +83,7 @@ def __init__(self, *words: Tuple[str, ...]) -> None:
             'uk': (True, True, True, False),
             'zh': (True, True, True, False)
         }
+        self._max_cached_websites = 15
         self._test_cached_file = {}
         self._tokenize = True
         self._words = []
@@ -138,7 +140,7 @@ def _bsoup(self, link: str, encoding: str = 'utf-8') -> Optional['BeautifulSoup'
                 return None
         bs = BeautifulSoup(data, 'html.parser')
         _CACHED_SOUPS[link] = bs
-        if len(bs_keys) >= 50:
+        if len(bs_keys) >= self._max_cached_websites:
             del _CACHED_SOUPS[bs[0]]
         return bs
 
@@ -298,8 +300,8 @@ def antonym(self, lang: str, word: str, dictionary: str = DICT_SYNONYMCOM) -> An
                     words.append(w)
             words.sort()
 
-        else:
-            raise InvalidDictionary(f'Dictionary {dictionary} cannot handle language {lang}')
+        # else:
+        #     raise InvalidDictionary(f'Dictionary {dictionary} cannot handle language {lang}')
 
         return words
 
@@ -456,8 +458,8 @@ def translate(self, lang: str, word: str, to: str = '', dictionary: str = DICT_E
             # Sort translations
             words = sorted(words, key=lambda x: x[0])
 
-        else:
-            raise InvalidDictionary(f'Dictionary {dictionary} cannot handle language {lang}')
+        # else:
+        #     raise InvalidDictionary(f'Dictionary {dictionary} cannot handle language {lang}')
 
         return words
 

test/test_dictionary.py

@@ -88,11 +88,7 @@ def test_meaning(self) -> None:
         # Empty
         self.assertEqual(d.meaning('en', ''), ([], '', ''))
 
-    def test_meaning_wordnet(self) -> None:
-        """
-        Test word meaning in wordnet.
-        """
-        d = self._get_dictionary()
+        # Test wordnet
         out = {'Noun': ['benefit', 'moral excellence or admirableness', 'that which is pleasing or valuable or useful',
                         'articles of commerce'],
                'Adjective': ['having desirable or positive qualities especially those suitable for a thing specified',
@@ -110,6 +106,9 @@ def test_meaning_wordnet(self) -> None:
                           "completely and absolutely (`good' is sometimes used informally for `thoroughly'"]}
         self.assertEqual(d.meaning('en', 'good', DICT_WORDNET), out)
 
+        # Test invalid dictionary
+        self.assertRaises(InvalidDictionary, lambda: d.meaning('es', 'word', DICT_WORDNET))
+
     def test_translate(self) -> None:
         """
         Test word parse before process.
@@ -154,6 +153,9 @@ def test_translate(self) -> None:
         self.assertEqual(d.translate('en', '!!!'), [])
         self.assertEqual(d.translate('en', '     !!!    '), [])
 
+        # Test invalid dictionary
+        self.assertRaises(AssertionError, lambda: d.translate('es', 'word', dictionary=DICT_SYNONYMCOM))
+
     def test_synonym(self) -> None:
         """
         Test word synonym.
@@ -259,6 +261,9 @@ def test_antonym(self) -> None:
         # Empty
         self.assertEqual(d.antonym('en', '!!!'), [])
 
+        # Test invalid dictionary
+        # self.assertRaises(InvalidDictionary, lambda: d.antonym('es', 'word', dictionary=DICT_SYNONYMCOM))
+
     def test_language_name(self) -> None:
         """
         Test language name.