def __buildNeo4JMockDriver(self, N, M): driver = Driver() driver.getNodes = Mock(return_value = self.__generateNodes(N)) driver.getEdges = Mock(return_value = self.__generateEdges(M)) return driver