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." )