diff --git a/generate_parameter_library_py/generate_parameter_library_py/generate_markdown.py b/generate_parameter_library_py/generate_parameter_library_py/generate_markdown.py index f39708d..4a266df 100644 --- a/generate_parameter_library_py/generate_parameter_library_py/generate_markdown.py +++ b/generate_parameter_library_py/generate_parameter_library_py/generate_markdown.py @@ -67,6 +67,14 @@ def __init__(self, validation: ValidationFunction): self.validation = validation + @staticmethod + def format_arguments(arguments): + if not arguments: + return '' + if len(arguments) == 1: + return str(arguments[0]) + return str(arguments) + def get_validation_type(self, function_base_name): if function_base_name in self.sentence_conventions: return self.sentence_conventions[function_base_name] @@ -77,9 +85,9 @@ def __str__(self): arguments = self.validation.arguments validation = self.get_validation_type(self.validation.function_base_name) if validation.__contains__('VALUES'): - validation = validation.replace('VALUES', str(arguments[0])) + validation = validation.replace('VALUES', self.format_arguments(arguments)) elif arguments: - validation += ': ' + str(arguments[0]) + validation += ': ' + self.format_arguments(arguments) return ' - ' + validation