def test_add_edges(add_text, add_doc, add_citation):
"""
Text_Graph#add_edges() should register edges from the citation table.
"""
t1 = add_text()
t2 = add_text()
t3 = add_text()
t4 = add_text()
t5 = add_text()
t6 = add_text()
d1 = add_doc()
d2 = add_doc()
d3 = add_doc()
# Texts 1-4 in d1.
add_citation(document=d1, text=t1)
add_citation(document=d1, text=t2)
add_citation(document=d1, text=t3)
add_citation(document=d1, text=t4)
# Texts 2-5 in d2.
add_citation(document=d2, text=t2)
add_citation(document=d2, text=t3)
add_citation(document=d2, text=t4)
add_citation(document=d2, text=t5)
# Texts 3-6 in d3.
add_citation(document=d3, text=t3)
add_citation(document=d3, text=t4)
add_citation(document=d3, text=t5)
add_citation(document=d3, text=t6)
g = Text_Graph()
g.add_edges()
assert g.graph.edge[t1.id][t2.id]['weight'] == 1
assert g.graph.edge[t2.id][t3.id]['weight'] == 2
assert g.graph.edge[t3.id][t4.id]['weight'] == 3
assert g.graph.edge[t4.id][t5.id]['weight'] == 2
assert g.graph.edge[t5.id][t6.id]['weight'] == 1
def test_hydrate_nodes(add_text, add_doc, add_citation):
"""
Text_Graph#hydrate_nodes() should load titles and authors onto nodes.
"""
t1 = add_text(title='title1', authors=['author1', 'author2'])
t2 = add_text(title='title2', authors=['author3', 'author4'])
doc = add_doc()
add_citation(document=doc, text=t1)
add_citation(document=doc, text=t2)
g = Text_Graph()
g.add_edges()
g.hydrate_nodes()
assert g.graph.node[t1.id]['authors'] == t1.pretty('authors')
assert g.graph.node[t2.id]['authors'] == t2.pretty('authors')
assert g.graph.node[t1.id]['title'] == t1.pretty('title')
assert g.graph.node[t2.id]['title'] == t2.pretty('title')