code: def test_shapes(self, shape):
      samples = discrete_gaussian_utils.sample_discrete_gaussian(
          100, shape
      )
      samples = self.evaluate(samples)
      self.assertAllEqual(samples.shape, shape)
  
imports: import collections
  import fractions
  import math
  import random
  from absl.testing import parameterized
  import numpy
  import tensorflow
  from tensorflow_privacy.privacy.dp_query import discrete_gaussian_utils
label_desc: Use image node_id 4 for calling the DiscreteGaussianUtilsTest obj's underlying member method code with example usage: obj.test_shapes(shape) without return types