code: def testSampleWithBatchAtol(self):
      dist = det.Deterministic(0.0, atol=[0.1, 0.1], validate_args=True)
      for sample_shape in [(), (4,)]:
          sample = dist.sample(sample_shape, seed=test_util.test_seed())
          self.assertAllEqual(sample_shape + (2,), sample.shape)
          self.assertAllClose(
              np.zeros(sample_shape + (2,)).astype(np.float32),
              self.evaluate(sample),
          )
  
imports: from absl.testing import parameterized
  import numpy
  import tensorflow.compat.v1
  import tensorflow.compat.v2
  from tensorflow_probability.python.distributions import deterministic
  from tensorflow_probability.python.distributions import gamma
  from tensorflow_probability.python.distributions import kullback_leibler
  from tensorflow_probability.python.distributions import mvn_diag
  from tensorflow_probability.python.internal import tensorshape_util
  from tensorflow_probability.python.internal import test_util
label_desc: Use image node_id 17 for calling the DeterministicTest obj's underlying member method code with example usage: obj.testSampleWithBatchAtol() without return types