code: def setUp(self):
      super(PixelCnnTest, self).setUp()
      self.batch_shape = (2,)
      self.image_shape = tf.TensorShape((8, 8, 3))
      self.high = 10
      self.low = 0
      self.h_shape = None
      self.num_logistic_mix = 1
  
imports: import numpy
  import tensorflow.compat.v2
  from tensorflow_probability.python.distributions import pixel_cnn
  from tensorflow_probability.python.internal import test_util
  from tensorflow_probability.python.internal import tf_keras
  from tensorflow_probability.python.math import gradient
label_desc: Use image node_id 1 for calling the PixelCnnTest obj's underlying member method code with example usage: obj.setUp() without return types