code: def test_compute_token_type_ids(self):
      separator = 101
      batch = torch.tensor(
          [
              [1, 2, 3, 4, 5, 6],
              [1, 2, 3, 101, 5, 6],
              [1, 101, 3, 4, 101, 6],
          ]
      )
      expected = torch.tensor(
          [[1, 1, 1, 1, 1, 1], [1, 1, 1, 0, 0, 0], [1, 0, 0, 0, 1, 1]]
      )
  
      result = compute_token_type_ids(batch, separator)
      np.testing.assert_array_equal(result, expected)
  
file_path: H2O/h2o_flexgen/benchmark/third_party/transformers/examples/research_projects/bertabs/test_utils_summarization.py
imports: import unittest
  import numpy
  import torch
  from .utils_summarization import build_mask, compute_token_type_ids, process_story, truncate_or_pad
label_desc: Use image node_id 11 for calling the SummarizationDataProcessingTest obj's underlying member method code with example usage: obj.test_compute_token_type_ids() without return types