def test_get_node_labels_by_gene_id(self):
con = mock_DatabaseConnector()
con.insert_many(
"genes",
[
{"head": 1, "tail": 2},
{"head": 1, "tail": 3},
{"head": 3, "tail": 2},
{"head": 1, "tail": 4},
{"head": 4, "tail": 3},
{"head": 5, "tail": 2}
]
)
gene_repo = GeneRepository(con)
edge = gene_repo.get_node_labels_by_gene_id(0)
self.assertTupleEqual(
edge,
(1, 2),
"GeneRepository didn't find the appropriate endpoints for gene id."
)
edge = gene_repo.get_node_labels_by_gene_id(4)
self.assertTupleEqual(
edge,
(4, 3),
"GeneRepository didn't find the appropriate endpoints for gene id."
)
