code: def test_skipgrams():
      # test with no window size and binary labels
      couples, labels = sequence.skipgrams(
          np.arange(3), vocabulary_size=3
      )
      for couple in couples:
          assert couple[0] in [0, 1, 2] and couple[1] in [0, 1, 2]
  
      # test window size and categorical labels
      couples, labels = sequence.skipgrams(
          np.arange(5),
          vocabulary_size=5,
          window_size=1,
          categorical=True,
      )
      for couple in couples:
          assert couple[0] - couple[1] <= 3
      for label in labels:
          assert len(label) == 2
  
file_path: keras-preprocessing/tests/sequence_test.py
imports: from math import ceil
  import numpy
  import pytest
  from numpy.testing import assert_allclose, assert_equal, assert_raises
  from keras_preprocessing import sequence
label_desc: Use image node_id 5 for calling a global function with example usage: test_skipgrams() without return types