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"])