code: def testSyntheticItemResponseTheory(self):
      num_train_points = int(0.75 * 30012)
      num_test_points = 30012 - num_train_points
  
      dataset = data.synthetic_item_response_theory(train_fraction=0.75)
  
      self.assertEqual(
          (num_train_points,), dataset["train_student_ids"].shape
      )
      self.assertEqual(
          (num_train_points,), dataset["train_question_ids"].shape
      )
      self.assertEqual(
          (num_train_points,), dataset["train_correct"].shape
      )
      self.assertEqual(
          (num_test_points,), dataset["test_student_ids"].shape
      )
      self.assertEqual(
          (num_test_points,), dataset["test_question_ids"].shape
      )
      self.assertEqual(
          (num_test_points,), dataset["test_correct"].shape
      )
  
file_path: probability/spinoffs/inference_gym/inference_gym/internal/data_test.py
imports: import numpy
  import tensorflow.compat.v2
  from tensorflow_probability.python.internal import test_util
  from inference_gym.internal import data
  from inference_gym.internal import test_util
label_desc: Use image node_id 9 for calling the DataTest obj's underlying member method code with example usage: obj.testSyntheticItemResponseTheory() without return types