code: def testRadonIndiana(self):
      dataset = data.radon_indiana(train_fraction=0.75)
      for k, v in dataset.items():
          if k.startswith("train_"):
              self.assertEqual((1388,), tf.convert_to_tensor(v).shape)
          if k.startswith("test_"):
              self.assertEqual((463,), tf.convert_to_tensor(v).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 8 for calling the DataTest obj's underlying member method code with example usage: obj.testRadonIndiana() without return types