docstr: Checks that creating a dataset works.
code: def testCreateDataset(self):
      """Checks that creating a dataset works."""
      # Technically this is private functionality, but we don't have it tested
      # elsewhere.
      model = plasma_spectroscopy.PlasmaSpectroscopy(**_test_dataset())
      model2 = plasma_spectroscopy.PlasmaSpectroscopy(
          **model._sample_dataset(tfp_test_util.test_seed())[1]
      )
      self.validate_log_prob_and_transforms(
          model2,
          sample_transformation_shapes=dict(
              identity=model.dtype._replace(
                  amplitude=[16],
                  temperature=[16],
                  velocity=[16],
                  shift=[],
              )
          ),
      )
  
file_path: probability/spinoffs/inference_gym/inference_gym/targets/plasma_spectroscopy_test.py
imports: from absl.testing import parameterized
  import numpy
  from tensorflow_probability.python.internal import test_util
  from inference_gym.internal import test_util
  from inference_gym.targets import plasma_spectroscopy
label_desc: Use image node_id 2 for calling the PlasmaSpectroscopyTest obj's underlying member method code with example usage: obj.testCreateDataset() without return types