Nonstring marker handling (#36)

Co-authored-by: trag1c <[email protected]>

dahlia/utils.py

@@ -46,6 +46,9 @@ def _find_ansi_codes(string: str) -> set[str]:
 
 
 def _with_marker(marker: str) -> list[re.Pattern[str]]:
+    if not isinstance(marker, str):
+        msg = "The marker has to be a string"
+        raise TypeError(msg)
     if len(marker) != 1:
         msg = "The marker has to be a single character"
         raise ValueError(msg)

tests/test_api.py

@@ -59,6 +59,12 @@ def test_invalid_marker(marker: str) -> None:
         Dahlia(marker=marker)
 
 
+@pytest.mark.parametrize("marker", [123, [1], {"key": "value"}])
+def test_nonstring_marker(marker: str) -> None:
+    with pytest.raises(TypeError, match="The marker has to be a string"):
+        Dahlia(marker=marker)
+
+
 @pytest.mark.parametrize(("auto_reset", "expected"), [(True, "\x1b[0m"), (False, "")])
 def test_auto_reset(auto_reset: bool, expected: str) -> None:
     assert Dahlia(auto_reset=auto_reset).convert("") == expected