def energy_prof(step):
diff, rad = diffs_prof(step)<EOL>adv, _ = advts_prof(step)<EOL>return (diff + np.append(adv, <NUM_LIT:0>)), rad<EOL>
Energy flux.

    This computation takes sphericity into account if necessary.

    Args:
        step (:class:`~stagpy.stagyydata._Step`): a step of a StagyyData
            instance.
    Returns:
        tuple of :class:`numpy.array`: the energy flux and the radial position
        at which it is evaluated.