def test_count_pathologies(self):
"""Test counting pathologies in the graph."""
graph = BELGraph()
a, b, c, d = protein(n(), n()), protein(n(), n()), pathology(n(), n()), pathology(n(), n())
graph.add_association(a, c, n(), n())
graph.add_association(a, d, n(), n())
graph.add_association(b, d, n(), n())
pathology_counter = count_pathologies(graph)
self.assertIn(c, pathology_counter)
self.assertIn(d, pathology_counter)
self.assertEqual(1, pathology_counter[c])
self.assertEqual(2, pathology_counter[d])
top_pathology_counter = get_top_pathologies(graph, count=1)
self.assertEqual(1, len(top_pathology_counter))
node, count = top_pathology_counter[0]
self.assertEqual(d, node)
self.assertEqual(2, count)
def test_count_pathologies(self):
"""Test counting pathologies in the graph."""
graph = BELGraph()
a, b = (protein(namespace='HGNC', name=n()) for _ in range(2))
c, d = (pathology(namespace='DOID', name=n()) for _ in range(2))
graph.add_association(a, c, citation=n(), evidence=n())
graph.add_association(a, d, citation=n(), evidence=n())
graph.add_association(b, d, citation=n(), evidence=n())
pathology_counter = count_pathologies(graph)
self.assertIn(c, pathology_counter)
self.assertIn(d, pathology_counter)
self.assertEqual(1, pathology_counter[c])
self.assertEqual(2, pathology_counter[d])
top_pathology_counter = get_top_pathologies(graph, n=1)
self.assertEqual(1, len(top_pathology_counter))
node, count = top_pathology_counter[0]
self.assertEqual(d, node)
self.assertEqual(2, count)