docstr: Checks that creating a dataset works.
docstr_tok: ["Checks","that","creating","a","dataset","works","."]
code: def testCreateDataset(self, num_test_points):
      """Checks that creating a dataset works."""
      # Technically this is private functionality, but we don't have it tested
      # elsewhere.
      if not tf.executing_eagerly():
          self.skipTest(
              "This is Eager only for now due to _sample_dataset being "
              "Eager-only."
          )
      model = item_response_theory.ItemResponseTheory(
          **_test_dataset(num_test_points)
      )
      model2 = item_response_theory.ItemResponseTheory(
          **model._sample_dataset(tfp_test_util.test_seed())
      )
      self.validate_log_prob_and_transforms(
          model2,
          sample_transformation_shapes=dict(
              identity={
                  "mean_student_ability": [],
                  "centered_student_ability": [20],
                  "question_difficulty": [10],
              },
              test_nll=[],
              per_example_test_nll=[num_test_points],
          ),
      )
  
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 3 for calling the ItemResponseTheoryTest obj's underlying member method code with example usage: obj.testCreateDataset(num_test_points) without return types