def test_entities_tostring(analysis):
assert analysis.entities_tostring([
Entity(name="General Motors",
entity_type="ORGANIZATION",
metadata={
"mid": "/m/035nm",
"wikipedia_url":
"http://en.wikipedia.org/wiki/General_Motors"
},
salience=0.33838183,
mentions=["General Motors"]),
Entity(name="jobs",
entity_type="OTHER",
metadata={"wikipedia_url": None},
salience=0.31634554,
mentions=["jobs"])
]) == ('[{name: "General Motors",'
' entity_type: "ORGANIZATION",'
' wikipedia_url: "http://en.wikipedia.org/wiki/General_Motors",'
' metadata: {"mid": "/m/035nm"},'
' salience: 0.33838183,'
' mentions: ["General Motors"]}, '
'{name: "jobs",'
' entity_type: "OTHER",'
' wikipedia_url: None,'
' metadata: {},'
' salience: 0.31634554,'
' mentions: ["jobs"]}]')
assert analysis.entities_tostring([]) == "[]"
