message: [TF2XLA] Fix the test docstring
diff: # See the License for the specific language governing permissions and
  # limitations under the License.
  # ==============================================================================
  -"""Tests for while loops in XLA."""
  +"""Tests for case statements in XLA."""
  from __future__ import absolute_import
  from __future__ import division
  