code: def dry_static_energy(heights, temperature):
      r
      return (mpconsts.g * heights + mpconsts.Cp_d * temperature).to('kJ/kg')
docs: r"""Calculate the dry static energy of parcels.
  
      This function will calculate the dry static energy following the first two terms of
      equation 3.72 in [Hobbs2006]_.
  
      Notes
      -----
      .. math::\text{dry static energy} = c_{pd} * T + gz
  
      * :math:`T` is temperature
      * :math:`z` is height
  
      Parameters
      ----------
      heights : array-like
          Atmospheric height
      temperature : array-like
          Atmospheric temperature
  
      Returns
      -------
      `pint.Quantity`
          The dry static energy