with self.assertRaisesRegex(exception, message):
      array_ops.one_hot(
          ragged_indices,
          depth,