code: def testGetInputWords(self, word_counts, expected_counts):
      max_token_length = 10
      reserved_tokens = ["<unk>", "<s>", "</s>"]
      new_counts = learner.get_input_words(
          word_counts, reserved_tokens, max_token_length
      )
  
      self.assertEqual(new_counts, expected_counts)
  
imports: import collections
  import logging
  from absl.testing import absltest
  from absl.testing import parameterized
  import numpy
  from tensorflow.python.data.ops import dataset_ops
  from tensorflow.python.ops.ragged import ragged_string_ops
  from tensorflow_text.tools.wordpiece_vocab import wordpiece_tokenizer_learner_lib
label_desc: Use image node_id 1 for calling the GetInputWordsTest obj's underlying member method code with example usage: obj.testGetInputWords(word_counts, expected_counts) without return types