Add README.md and fix a bug

wkingfly · Aug 7, 2014 · a0ddaf7 · a0ddaf7
1 parent 212a2b3
commit a0ddaf7
Show file tree

Hide file tree

Showing 2 changed files with 6 additions and 8 deletions.
diff --git a/kmp/kmp.py b/kmp/kmp.py
@@ -1,5 +1,3 @@
-import pdb
-
 class strFind:
     def __init__(self, child, matrix):
         self.child = list(child)
@@ -42,8 +40,6 @@ def find(self):
             if i == self.child[j]:
                 equals += 1
                 j += 1
-                if equals == 6:
-                    pdb.set_trace()
                 if equals == len(self.child):
                     return True
             else:
@@ -52,5 +48,7 @@ def find(self):
                 else:
                     move_num = 1
                 self.matrix = self.matrix[move_num:len(self.matrix)]
-                self.find()
-        return False
+                if len(self.matrix) < len(self.child):
+                    return False     
+
+                return self.find()
diff --git a/test/test_kmp.py b/test/test_kmp.py
@@ -6,7 +6,7 @@
 class Test_strFind(unittest.TestCase):
 
     def setUp(self):
-        matrix = 'iacdabcabcdabcabcix' 
+        matrix = 'acdabcabcdabcabcx' 
         child = 'abcabcx'
         self.sF = kmp.strFind(child, matrix)
     def tearDown(self):
@@ -18,7 +18,7 @@ def test_part_match(self):
         self.assertEqual(self.sF.part_list, l)
 
     def test_find(self):
-        self.assertFalse(self.sF.find())
+        self.assertTrue(self.sF.find())
 
 
 if __name__ == '__main__':