def test_add_nodes(add_text, add_citation):
"""
OSP_Graph#add_nodes() should register nodes for all texts.
"""
t1 = add_text(title='title1', surname='surname1')
t2 = add_text(title='title2', surname='surname2')
for i in range(3):
add_citation(text=t1)
for i in range(1):
add_citation(text=t2)
g = OSP_Graph()
g.add_nodes()
n1 = g.graph.node[t1.id]
n2 = g.graph.node[t2.id]
assert n1['label'] == t1.pretty('title')
assert n2['label'] == t2.pretty('title')
assert n1['author'] == t1.pretty('surname')
assert n2['author'] == t2.pretty('surname')
assert n1['count'] == 3
assert n2['count'] == 1
assert n1['score'] == 2/2
assert n2['score'] == 1/2
def test_trim_texts_by_count(add_text, add_doc, add_citation):
"""
OSP_Graph#trim_texts_by_count() should remove all texts with fewer than a
given number of total citations.
"""
docs = [
add_doc(),
add_doc(),
add_doc(),
add_doc(),
]
t1 = add_text()
t2 = add_text()
t3 = add_text()
t4 = add_text()
# 4 citations to t1.
for i in range(4):
add_citation(document=docs[i], text=t1)
# 3 citations to t2.
for i in range(3):
add_citation(document=docs[i], text=t2)
# 2 citations to t3.
for i in range(2):
add_citation(document=docs[i], text=t3)
# 1 citation to t4.
for i in range(1):
add_citation(document=docs[i], text=t4)
g = OSP_Graph()
g.add_edges()
g.add_nodes()
g.trim_texts_by_count(min_count=3)
assert g.graph.has_node(t1.id)
assert g.graph.has_node(t2.id)
assert not g.graph.has_node(t3.id)
assert not g.graph.has_node(t4.id)
