code: def testWhitespaceTokenization(self, input_text, expected_output):
      # TODO(b/194508683) Delete the check when TF1 is deprecated.
      if tf.__version__ < "2":
          return
  
      actual = text.whitespace_tokenization(input_text).to_tensor()
      expected = tf.constant(expected_output, dtype=tf.string)
      self.assertAllEqual(actual, expected)
  
imports: from absl.testing import parameterized
  import tensorflow
  from tensorflow_model_analysis.experimental.preprocessing_functions import text
label_desc: Use image node_id 1 for calling the TextTest obj's underlying member method code with example usage: obj.testWhitespaceTokenization(input_text, expected_output) without return types