code: def testBrownianMotion(self):
      dataset = data.brownian_motion_missing_middle_observations()
      self.assertEqual((30,), dataset["observed_locs"].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 5 for calling the DataTest obj's underlying member method code with example usage: obj.testBrownianMotion() without return types