code: def test_skip_gram_sample_non_string_input():
      """Tests skip-gram with non-string input."""
      input_tensor = tf.constant([1, 2, 3], dtype=tf.dtypes.int16)
      tokens, labels = text.skip_gram_sample(
          input_tensor, min_skips=1, max_skips=1
      )
      expected_tokens, expected_labels = _split_tokens_labels(
          [(1, 2), (2, 1), (2, 3), (3, 2)]
      )
      np.testing.assert_equal(
          np.asanyarray(expected_tokens), tokens.numpy()
      )
      np.testing.assert_equal(
          np.asanyarray(expected_labels), labels.numpy()
      )
  
imports: import csv
  import os
  import tempfile
  import numpy
  import pytest
  import tensorflow
  from tensorflow_addons import text
  from tensorflow_addons.text import skip_gram_ops
label_desc: Use image node_id 10 for calling a global function with example usage: test_skip_gram_sample_non_string_input() without return types