def test_find_connecting_nodes(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) connecting_nodes = gene_repo.find_connecting_nodes(1, 2) self.assertListEqual( connecting_nodes, [3], "GeneRepository didn't find the right connecting nodes." )