code: def testSyntheticLogGaussianCoxModel(self):
      num_train_points = 100
  
      dataset = data.synthetic_log_gaussian_cox_process()
  
      self.assertEqual(
          (num_train_points, 2), dataset["train_locations"].shape
      )
      self.assertEqual(
          (num_train_points,), dataset["train_extents"].shape
      )
      self.assertEqual(
          (num_train_points,), dataset["train_counts"].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 10 for calling the DataTest obj's underlying member method code with example usage: obj.testSyntheticLogGaussianCoxModel() without return types