code: def testAverageWrongShape(self):
      with self.assertRaisesWithPredicateMatch(
          errors.InvalidArgumentError, r""
      ):
          np_math_ops.average(np.ones([2, 3]), weights=np.ones([2, 4]))
      with self.assertRaisesWithPredicateMatch(
          errors.InvalidArgumentError, r""
      ):
          np_math_ops.average(
              np.ones([2, 3]), axis=0, weights=np.ones([2, 4])
          )
      with self.assertRaisesWithPredicateMatch(
          errors.InvalidArgumentError, r""
      ):
          np_math_ops.average(
              np.ones([2, 3]), axis=0, weights=np.ones([])
          )
      with self.assertRaisesWithPredicateMatch(
          errors.InvalidArgumentError, r""
      ):
          np_math_ops.average(
              np.ones([2, 3]), axis=0, weights=np.ones([5])
          )
  
imports: import itertools
  from absl.testing import parameterized
  import numpy
  from tensorflow.python.framework import dtypes
  from tensorflow.python.framework import errors
  from tensorflow.python.framework import flexible_dtypes
  from tensorflow.python.framework import ops
  from tensorflow.python.framework import tensor
  from tensorflow.python.framework import test_util
  from tensorflow.python.framework import weak_tensor
  from tensorflow.python.ops import weak_tensor_ops
  from tensorflow.python.ops import weak_tensor_test_util
  from tensorflow.python.ops.numpy_ops import np_array_ops
  from tensorflow.python.ops.numpy_ops import np_arrays
  from tensorflow.python.ops.numpy_ops import np_math_ops
  from tensorflow.python.ops.numpy_ops import np_random
  from tensorflow.python.platform import googletest
label_desc: Use image node_id 10 for calling the MathTest obj's underlying member method code with example usage: obj.testAverageWrongShape() without return types