add edit_distance_of_list

Author: 刘欣

setup.py

@@ -90,7 +90,7 @@ def build_extensions(self):
 
 setup(
     name='pylcs',
-    version='0.0.5',
+    version='0.0.6',
     author='Meteorix',
     author_email='[email protected]',
     url='https://github.com/Meteorix/pylcs',

src/main.cpp

@@ -160,6 +160,17 @@ int levenshtein_distance(const string &str1, const string &str2) {
 }
 
 
+vector<int> levenshtein_distance_of_list(const string &str1, vector<string> &str_list){
+    int size = str_list.size();
+    vector<int> ls(size);
+    for (int i = 0; i < size; i++){
+        int l = levenshtein_distance(str1, str_list[i]);
+        ls[i] = l;
+    }
+    return ls;
+}
+
+
 namespace py = pybind11;
 
 
@@ -188,4 +199,11 @@ PYBIND11_MODULE(pylcs, m) {
         Same As levenshtein_distance(): Levenshtein Distance of Two Strings
     )pbdoc");
 
+    m.def("levenshtein_distance_of_list", &levenshtein_distance_of_list, R"pbdoc(
+        Levenshtein Distance of one string to a list of strings
+    )pbdoc");
+
+    m.def("edit_distance_of_list", &levenshtein_distance_of_list, R"pbdoc(
+        Levenshtein Distance of one string to a list of strings
+    )pbdoc");
 }

tests/test_pylcs.py

@@ -32,3 +32,8 @@ def test_edit_distance():
     assert pylcs.edit_distance("你好", "中国") == 2
     assert pylcs.edit_distance("aaa你好", "你好呀") == 4
 
+
+def test_edit_distance_of_list():
+    assert pylcs.edit_distance_of_list("aaa", ["bbb"] * 10) == [3] * 10
+    assert pylcs.edit_distance_of_list("aaa你好", ["你好呀"] * 10) == [4] * 10
+    assert pylcs.edit_distance_of_list("aaa你好", ["bbb", "你好呀"] * 10) == [5, 4] * 10