code: def _get_mpi_info(self):
          
  
          rank = self.comm.Get_rank()
          size = self.comm.Get_size()
          return rank, size
docs: get basic MPI info
  
          Returns
          -------
          comm : Intracomm
              Returns MPI communication group
  
          rank : integer
              Returns the rank of this process
  
          size : integer
              Returns total number of processes