code: def test_all_zero_input_works(self):
      # Tests that encoding does not blow up with all-zero input. With
      # min_max=None, the derived min and max are identical, thus potential for
      # division by zero.
      stage = stages_impl.UniformQuantizationEncodingStage(
          bits=8, min_max=None
      )
      test_data = self.run_one_to_many_encode_decode(
          stage, lambda: tf.zeros([50])
      )
      self.assertAllEqual(
          np.zeros((50)).astype(np.float32), test_data.decoded_x
      )
  
imports: import itertools
  from absl.testing import parameterized
  import numpy
  import tensorflow
  from tensorflow_model_optimization.python.core.internal.tensor_encoding.stages import stages_impl
  from tensorflow_model_optimization.python.core.internal.tensor_encoding.testing import test_utils
label_desc: Use image node_id 10 for calling the UniformQuantizationEncodingStageStageTest obj's underlying member method code with example usage: obj.test_all_zero_input_works() without return types