code: def test_conditional_mean_large_alpha_is_round(self):
      # We don't care what happens exactly near integer values:
      for offset in range(-5, 5):
          x = tf.linspace(offset + 0.001, offset + 0.999, 100)
          y = round_ops.soft_round_conditional_mean(x, alpha=5000.0)
          self.assertAllClose(tf.math.round(x), y, atol=0.001)
  
file_path: compression/tensorflow_compression/python/ops/round_ops_test.py
imports: from absl.testing import parameterized
  import tensorflow
  from tensorflow_compression.python.ops import round_ops
label_desc: Use image node_id 6 for calling the SoftRoundTest obj's underlying member method code with example usage: obj.test_conditional_mean_large_alpha_is_round() without return types