def arg_p(self, arg): arg_line = arg["type"].split(", ") if len(arg_line) == 1: return "Required" else: p = arg_line[1] if p == "required": return "Required<br>(exactly 1)" elif p == "optional": if len(arg_line) == 3: return "Optional<br>" + capitalize(arg_line[2]) else: return "Optional" else: return p
def test_capitalize(): assert capitalize("word") == "Word" assert capitalize("x") == "X" assert capitalize("") == ""