docstr: Tests batch diff works with axis argument use of exclusivity.
docstr_tok: ["Tests","batch","diff","works","with","axis","argument","use","of","exclusivity","."]
code: def test_batched_axis(self, exclusive, axis, dx_true):
      """Tests batch diff works with axis argument use of exclusivity."""
      x = tf.constant([[1, 2, 3, 4], [10, 20, 30, 40]])
      dx = self.evaluate(
          math.diff(x, order=1, exclusive=exclusive, axis=axis)
      )
      self.assertAllEqual(dx, dx_true)
  
file_path: tf-quant-finance/tf_quant_finance/math/diff_ops_test.py
imports: from absl.testing import parameterized
  import numpy
  import tensorflow.compat.v2
  from tf_quant_finance import math
  from tensorflow.python.framework import test_util
label_desc: Use image node_id 3 for calling the DiffOpsTest obj's underlying member method code with example usage: obj.test_batched_axis(exclusive, axis, dx_true) without return types