TheAlgorithms-Python
52 строки · 1.6 Кб
1def snake_to_camel_case(input_str: str, use_pascal: bool = False) -> str:2"""3Transforms a snake_case given string to camelCase (or PascalCase if indicated)
4(defaults to not use Pascal)
5
6>>> snake_to_camel_case("some_random_string")
7'someRandomString'
8
9>>> snake_to_camel_case("some_random_string", use_pascal=True)
10'SomeRandomString'
11
12>>> snake_to_camel_case("some_random_string_with_numbers_123")
13'someRandomStringWithNumbers123'
14
15>>> snake_to_camel_case("some_random_string_with_numbers_123", use_pascal=True)
16'SomeRandomStringWithNumbers123'
17
18>>> snake_to_camel_case(123)
19Traceback (most recent call last):
20...
21ValueError: Expected string as input, found <class 'int'>
22
23>>> snake_to_camel_case("some_string", use_pascal="True")
24Traceback (most recent call last):
25...
26ValueError: Expected boolean as use_pascal parameter, found <class 'str'>
27"""
28
29if not isinstance(input_str, str):30msg = f"Expected string as input, found {type(input_str)}"31raise ValueError(msg)32if not isinstance(use_pascal, bool):33msg = f"Expected boolean as use_pascal parameter, found {type(use_pascal)}"34raise ValueError(msg)35
36words = input_str.split("_")37
38start_index = 0 if use_pascal else 139
40words_to_capitalize = words[start_index:]41
42capitalized_words = [word[0].upper() + word[1:] for word in words_to_capitalize]43
44initial_word = "" if use_pascal else words[0]45
46return "".join([initial_word, *capitalized_words])47
48
49if __name__ == "__main__":50from doctest import testmod51
52testmod()53