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("") == ""