Add explicit put_{execution|context}_type method to the test_case_utils.
Add explicit put_{execution|context}_type method to the test_case_utils.

PiperOrigin-RevId: 489117158

tensorflow/tfx,tensorflow/tfx