code: def _test_dataset(num_test_pairs=None):
      return dict(
          train_student_ids=np.arange(20),
          train_question_ids=(np.arange(20) % 10),
          train_correct=np.arange(20) % 2,
          test_student_ids=(
              np.arange(num_test_pairs) if num_test_pairs else None
          ),
          test_question_ids=(
              np.arange(num_test_pairs) % 10 if num_test_pairs else None
          ),
          test_correct=(
              np.arange(num_test_pairs) % 2 if num_test_pairs else None
          ),
      )
  
file_path: probability/spinoffs/inference_gym/inference_gym/targets/item_response_theory_test.py
imports: from absl.testing import parameterized
  import numpy
  import tensorflow.compat.v2
  from tensorflow_probability.python.internal import test_util
  from inference_gym.internal import test_util
  from inference_gym.targets import item_response_theory
label_desc: Use image node_id 1 for calling a global function with example usage: _test_dataset(num_test_pairs) and returns: dict