code: def test_uniform_is_special_case(self):
      # With the scale parameter going to zero, the density should approach a
      # unit-width uniform distribution.
      df = round_adapters.NoisySoftRoundedDeepFactorized(
          init_scale=1e-3
      )
      x = tf.linspace(-1.0, 1.0, 10)
      self.assertAllClose(df.prob(x), [0, 0, 0, 1, 1, 1, 1, 0, 0, 0])
  
imports: from absl.testing import parameterized
  import numpy
  import tensorflow
  import tensorflow_probability
  from tensorflow_compression.python.distributions import deep_factorized
  from tensorflow_compression.python.distributions import round_adapters
label_desc: Use image node_id 1 for calling the NoisySoftRoundedDeepFactorizedTest obj's underlying member method code with example usage: obj.test_uniform_is_special_case() without return types