def test_parse_returns(style):
it = split_section(doc[style])
next(it)
next(it)
next(it)
section, body, style = next(it)
markdown, type = parse_returns(body, style)
assert markdown == "e\n\nf"
assert type == "int"
def test_parse_parameter(style):
it = split_section(doc[style])
next(it)
section, body, style = next(it)
it = split_parameter(body)
lines = next(it)
assert len(lines) == 4 if style == "goole" else 5
name, markdown, type = parse_parameter(lines, style)
assert name == "x"
assert markdown == "The first\nparameter\n\nwith type."
assert type == "int"
lines = next(it)
assert len(lines) == 4 if style == "goole" else 5
name, markdown, type = parse_parameter(lines, style)
assert name == "y"
assert markdown == "The second\nparameter\n\nwithout type."
assert type == ""
def test_split_section(style):
it = split_section(doc[style])
section, body, style = next(it)
assert section == ""
assert body == "a\n\n b\n\nc"
section, body, style = next(it)
assert section == "Parameters"
if style == "google":
assert body.startswith("x (int)")
else:
assert body.startswith("x : int")
assert body.endswith("\n without type.")
section, body, style = next(it)
assert section == "Raises"
if style == "google":
assert body.startswith("ValueError: a")
else:
assert body.startswith("ValueError\n a")
assert body.endswith("\n d")