def test_parse_docs_has_yaml():
def f():
"""
Hello
name: kk
email: [email protected]
"""
assert parse_docs(f.__doc__, marks=[]) == {
"$desc": "Hello\n\nname: kk\nemail: [email protected]"
}
assert parse_docs(f.__doc__, marks=["unknown"]) == {
"$desc": "Hello\n\nname: kk\nemail: [email protected]"
}
assert parse_docs(f.__doc__, marks=["email"]) == {
"$desc": "Hello\n\nname: kk",
"email": "*****@*****.**"
}
assert parse_docs(f.__doc__, marks=["name"]) == {
"$desc": "Hello",
"name": "kk",
"email": "*****@*****.**"
}
assert parse_docs(f.__doc__, marks=["name", "email"]) == {
"$desc": "Hello",
"name": "kk",
"email": "*****@*****.**"
}
def test_parse_docs():
def f_no_docs():
pass
assert parse_docs(f_no_docs.__doc__, ["$input"]) == {}
def f_no_content():
""""""
assert parse_docs(f_no_content.__doc__, ["$input"]) == {"$desc": ""}
def f_empty_content():
""" """
assert parse_docs(f_empty_content.__doc__, ["$input"]) == {"$desc": ""}
def f_no_yaml():
"""Hello World"""
assert parse_docs(f_no_yaml.__doc__, ["$input"]) == {
"$desc": "Hello World"
}
def f_no_marks():
"""
Hello World
No marks
"""
assert parse_docs(f_no_marks.__doc__, ["$input"]) == {
"$desc": "Hello World\n\nNo marks"
}
def test_parse_docs_invalid_yaml():
def f_invalid_syntax():
"""
$input:
name: kk
email: [email protected]
"""
with pytest.raises(yaml.YAMLError):
parse_docs(f_invalid_syntax.__doc__, ["$input"])
