import tensorflow as tf

      qdist = distributions.QuantizedDistribution(
          base_dist_cls=distributions.Normal,
          lower_cutoff=1.0,
          upper_cutoff=10.0,
          mu=tf.zeros(batch_shape),
          sigma=tf.ones(batch_shape))

      self.assertEqual(batch_shape, qdist.get_batch_shape())
      self.assertAllEqual(batch_shape, qdist.batch_shape().eval())
      self.assertEqual((), qdist.get_event_shape())
