Add test case for constant folding/model pruning on the input to a `tf.cond()`.

This adds coverage for the previously broken case where using a `tf.constant()`
as the predicate for a conditional would be optimized incorrectly.

PiperOrigin-RevId: 234664039