diff --git a/strings/split.py b/strings/split.py index ed194ec69c2f..fe09a92063f8 100644 --- a/strings/split.py +++ b/strings/split.py @@ -17,18 +17,15 @@ def split(string: str, separator: str = " ") -> list: >>> split(";abbb;;c;", separator=';') ['', 'abbb', '', 'c', ''] - """ - - split_words = [] - last_index = 0 - for index, char in enumerate(string): - if char == separator: - split_words.append(string[last_index:index]) - last_index = index + 1 - if index + 1 == len(string): - split_words.append(string[last_index : index + 1]) - return split_words + >>> split("a--b--c", separator="--") + Traceback (most recent call last): + ... + ValueError: separator must be a single character + """ + if len(separator) != 1: + raise ValueError("separator must be a single character") + return string.split(separator) if __name__ == "__main__":