Fixed in operator in automata.py

aosingh · Jan 25, 2021 · 9c02bf6 · 9c02bf6
1 parent 8535dcb
commit 9c02bf6
Show file tree

Hide file tree

Showing 3 changed files with 15 additions and 2 deletions.
diff --git a/lexpy/_base/automata.py b/lexpy/_base/automata.py
@@ -40,6 +40,8 @@ def __contains__(self, word):
                 node = node[letter]
                 if node.eow and i == len(word) - 1:
                     return True
+            else:
+                return False
         return False
 
     def __contains_prefix(self, prefix):

diff --git a/lexpy/tests/test_dawg.py b/lexpy/tests/test_dawg.py
@@ -35,11 +35,17 @@ def test_word_in_dawg(self):
         self.dawg.reduce()
         self.assertTrue('ash' in self.dawg, "Word should be in dawg")
 
-    def test_word_not_int_dawg(self):
+    def test_word_not_int_dawg1(self):
         self.dawg = DAWG()
         self.dawg.add_all(['ash', 'ashley'])
         self.dawg.reduce()
         self.assertFalse('salary' in self.dawg, "Word should not be in dawg")
+
+    def test_word_not_int_dawg2(self):
+        self.dawg = DAWG()
+        self.dawg.add_all(['ash', 'ashley'])
+        self.dawg.reduce()
+        self.assertFalse('mash lolley' in self.dawg, "Word should not be in dawg")
 
 class TesDAWGWordInsert(unittest.TestCase):
 

diff --git a/lexpy/tests/test_trie.py b/lexpy/tests/test_trie.py
@@ -33,11 +33,16 @@ def test_word_in_trie(self):
         self.trie.add_all(['ash', 'ashley'])
         self.assertTrue('ash' in self.trie, "Word should be in trie")
 
-    def test_word_not_int_trie(self):
+    def test_word_not_int_trie1(self):
         self.trie = Trie()
         self.trie.add_all(['ash', 'ashley'])
         self.assertFalse('salary' in self.trie, "Word should not be in trie")
 
+    def test_word_not_int_trie2(self):
+        self.trie = Trie()
+        self.trie.add_all(['ash', 'ashley'])
+        self.assertFalse('mash lolley' in self.trie, "Word should not be in trie")
+
 
 class TesTrieWordInsert(unittest.TestCase):