class_docstr: Tests for `_tf_style_*` decorators.
  
  The class decorators, `tf_style_encoding_stage` and
  `tf_style_adaptive_encoding_stage` are tested by the example implementations
  of `EncodingStageInterface` and `AdaptiveEncodingStageInterface`, in
  `test_utils.py`.
class_docstr_tok: ["Tests","for","`","_tf_style_","*","`","decorators",".","The","class","decorators",",","`","tf_style_encoding_stage","`","and","`","tf_style_adaptive_encoding_stage","`","are","tested","by","the","example","implementations","of","`","EncodingStageInterface","`","and","`","AdaptiveEncodingStageInterface","`",",","in","`","test_utils.py","`","."]
code: def test_tf_style_adaptive_encoding_stage_raises(self, bad_stage):
      with self.assertRaises(TypeError):
          encoding_stage.tf_style_adaptive_encoding_stage(bad_stage)
  
imports: from __future__ import absolute_import
  from __future__ import division
  from __future__ import print_function
  from absl.testing import parameterized
  import mock
  import numpy
  import tensorflow
  from tensorflow_model_optimization.python.core.internal.tensor_encoding.core import encoding_stage
  from tensorflow_model_optimization.python.core.internal.tensor_encoding.testing import test_utils
label_desc: Use image node_id 10 for calling the TFStyleEncodeDecodeTest obj's underlying member method code with example usage: obj.test_tf_style_adaptive_encoding_stage_raises(bad_stage) without return types