class_docstr: Tests for discretization layers.
code: def setUp(self):
      tf.set_random_seed(1234)
      np.random.seed(123)
  
imports: from __future__ import absolute_import
  from __future__ import division
  from __future__ import print_function
  import numpy
  from tensor2tensor.layers import discretization
  from tensor2tensor.utils import test_utils
  import tensorflow.compat.v1
label_desc: Use image node_id 1 for calling the DiscretizationTest obj's underlying member method code with example usage: obj.setUp() without return types