slight tidy

iterative · Mar 7, 2024 · 47e8401 · 47e8401
1 parent c33102b
commit 47e8401
Show file tree

Hide file tree

Showing 2 changed files with 3 additions and 3 deletions.
diff --git a/shtab/__init__.py b/shtab/__init__.py
@@ -129,8 +129,8 @@ def complete2pattern(opt_complete, shell: str, choice_type2fn) -> str:
 
 
 def wordify(string: str) -> str:
-    r"""Replace non-word chars [\W] with underscores [_]"""
-    return re.sub(r"\W", "_", string)
+    """Replace non-word chars [\\W] with underscores [_]"""
+    return re.sub("\\W", "_", string)
 
 
 def get_public_subcommands(sub):

diff --git a/tests/test_shtab.py b/tests/test_shtab.py
@@ -256,8 +256,8 @@ def test_subparser_slashes(shell, caplog):
         shell.compgen('-W "${_shtab_test_pos_0_choices[*]}"', "s", "sub/cmd")
         shell.test('-z "${_shtab_test_COMPGEN-}"')
     elif shell == "zsh":
-        # make sure the slash was properly substituted to avoid syntax errors
         assert "_shtab_test_sub/cmd" not in completion
+        assert "_shtab_test_sub_cmd" in completion
 
 
 @fix_shell