Fix rst doctests in units package