def clean_function_name(name: str, snakeify: bool = True) -> str: """ Cleans a function name. Args: name: The function name to clean. snakeify: Whether to format the name to snake-case. Returns: name: The clean name. """ for i in ILLEGAL_CODE_CHARS: name = name.replace(i, '') if snakeify: name = camel_to_snake(name) name = name.replace('-', '_').replace('__', '_').strip('_') return name
def test_camel_to_snake(self): snake = tools.camel_to_snake('CamelCase') assert snake == 'camel_case'