v0.0-641-g4d325ff: Only disable formatting on leaf-partitions that don't fit.

Previously, space-preservation was too aggressive and affected larger regions
of code than was intended, resulting in indentations not being adjusted
correctly in several cases.

fixes #508

PiperOrigin-RevId: 334466721