code: def test_wrong_output_shape(self):
      f = jnp.vectorize(jnp.dot, signature="(n,m),(m,k)->(n)")
      with self.assertRaisesRegex(
          ValueError, r"output shape \(2, 2\) does not match"
      ):
          f(jnp.zeros((2, 2)), jnp.zeros((2, 2)))
  
imports: from functools import partial
  from absl.testing import absltest
  import numpy
  import jax
  from jax import numpy
  from jax._src import test_util
  from jax import config
label_desc: Use image node_id 15 for calling the VectorizeTest obj's underlying member method code with example usage: obj.test_wrong_output_shape() without return types