Update fuzzyfinder function to accept integer inputs

Matheus Rosa · Matheus Rosa · commit 1a2d468f2e2a · 2015-10-06T23:06:43.000-03:00
diff --git a/fuzzyfinder/main.py b/fuzzyfinder/main.py
@@ -16,6 +16,7 @@ def fuzzyfinder(text, collection):
             input.
     """
     suggestions = []
+    text = str(text) if not isinstance(text, str) else text
     pat = '.*?'.join(map(re.escape, text))
     regex = re.compile(pat)
     for item in collection:
diff --git a/tests/test_fuzzyfinder.py b/tests/test_fuzzyfinder.py
@@ -19,6 +19,8 @@ def collection():
             'user_group.doc',
             'users.txt',
             'accounts.txt',
+            '123.py',
+            'test123test.py'
             ]
 
 def test_substring_match(collection):
@@ -50,3 +52,9 @@ def test_fuzzy_match_greedy(collection):
     results = fuzzyfinder(text, collection)
     expected = ['user_group.doc', 'users.txt', 'api_user.doc']
     assert list(results) == expected
+
+def test_fuzzy_integer_input(collection):
+    text = 123
+    results = fuzzyfinder(text, collection)
+    expected = ['123.py', 'test123test.py']
+    assert list(results) == expected
