code: def partitions_for_topic(self, topic):
          
          if topic not in self._partitions:
              return None
          return set(self._partitions[topic].keys())
docs: Return set of all partitions for topic (whether available or not)
  
          Arguments:
              topic (str): topic to check for partitions
  
          Returns:
              set: {partition (int), ...}