class_docstr: Tests for methods in piecewise module.
class_docstr_tok: ["Tests","for","methods","in","piecewise","module","."]
docstr: Tests that `jump_locations` and `values` should have the same batch.
docstr_tok: ["Tests","that","`","jump_locations","`","and","`","values","`","should","have","the","same","batch","."]
code: def test_invalid_jump_batch_shape(self, dtype):
      """Tests that `jump_locations` and `values` should have the same batch."""
      jump_locations = np.array([[0.1, 10], [2.0, 10]])
      values = tf.constant([[[3, 4, 5], [3, 4, 5]]], dtype=dtype)
      with self.assertRaises(ValueError):
          piecewise.PiecewiseConstantFunc(
              jump_locations, values, dtype=dtype
          )
  
imports: from absl.testing import parameterized
  import numpy
  import tensorflow.compat.v2
  import tf_quant_finance
  from tensorflow.python.framework import test_util
label_desc: Use image node_id 10 for calling the Piecewise obj's underlying member method code with example usage: obj.test_invalid_jump_batch_shape(dtype) without return types