code: def get_cluster_assignment(self):
          
          plan = self.get_cluster_plan()
          assignment = {}
          for elem in plan['partitions']:
              assignment[
                  (elem['topic'], elem['partition'])
              ] = elem['replicas']
  
          return assignment
docs: Fetch the cluster layout in form of assignment from zookeeper