Fix surround substring to surround all substrings (#212)

* surround_substring surround **all** substrings

* added unit tests

* ruff lint

---------

Co-authored-by: Jimmy Jensen <jsj@kapacity.dk>

quinn/keyword_finder.py

@@ -111,10 +111,7 @@ def surround_substring(input: str, substring: str, surround_start: str, surround
     :rtype: str
 
     """
-    index = input.find(substring)
-    res = ""
-    if index == -1:
-        res = input
-    else:
-        res = input[:index] + surround_start + substring + surround_end + input[(index + len(substring)) :]
-    return res
+    return input.replace(
+        substring,
+        surround_start + substring + surround_end,
+    )

tests/test_keyword_finder.py

@@ -17,4 +17,9 @@ def test_keyword_format():
 
 
 def test_surround_substring():
-    print(surround_substring("spark rdd stuff", "rdd", "**", "||"))
+
+    assert "spark **rdd|| stuff" == surround_substring("spark rdd stuff", "rdd", "**", "||")
+    assert "spark **rdd|| stuff with **rdd||" == surround_substring("spark rdd stuff with rdd", "rdd", "**", "||")
+    assert "spark **rdd||dd stuff" == surround_substring("spark rdddd stuff", "rdd", "**", "||")
+
+