code: def test_assert_all_nan_input_inf(self):
      a = tf.convert_to_tensor(
          value=np.full((10, 10, 10), np.inf), dtype=self.dtype
      )
      with self.assertRaisesRegex(
          AssertionError, "Arrays are not equal"
      ):
          self.assertAllNan(a)
  
imports: import collections
  from absl import flags
  from absl.testing import flagsaver
  from absl.testing import parameterized
  import numpy
  import tensorflow.compat.v1
  import tensorflow.compat.v2
  from tensorflow_probability.python.internal import dtype_util
  from tensorflow_probability.python.internal import test_combinations
  from tensorflow_probability.python.internal import test_util
  from tensorflow_probability.python.util.seed_stream import SeedStream
label_desc: Use image node_id 9 for calling the _TestCaseTest obj's underlying member method code with example usage: obj.test_assert_all_nan_input_inf() without return types