'Return TopicPartitions for which the broker is a leader.
Arguments:
broker_id (int): node id for a broker
Returns:
set: {TopicPartition, ...}'

def partitions_for_broker(self, broker_id):

    return self._broker_partitions.get(broker_id)
