class_docstr: Integration tests for the vn_entropy measurement function.
class_docstr_tok: ["Integration","tests","for","the","vn_entropy","measurement","function","."]
docstr: Test entropy for a QNode gradient with Jax.
docstr_tok: ["Test","entropy","for","a","QNode","gradient","with","Jax","."]
code: def test_IsingXX_qnode_entropy_grad_jax(
      self, param, wires, base, diff_method, interface
  ):
      """Test entropy for a QNode gradient with Jax."""
      import jax
  
      dev = qml.device("default.qubit", wires=2)
  
      @qml.qnode(dev, interface=interface, diff_method=diff_method)
      def circuit_entropy(x):
          qml.IsingXX(x, wires=[0, 1])
          return qml.vn_entropy(wires=wires, log_base=base)
  
      grad_entropy = jax.grad(circuit_entropy)(jax.numpy.array(param))
      grad_expected_entropy = expected_entropy_grad_ising_xx(
          param
      ) / np.log(base)
  
      # higher tolerance for finite-diff method
      tol = 1e-8 if diff_method == "backprop" else 1e-5
  
      assert qml.math.allclose(
          grad_entropy, grad_expected_entropy, rtol=1e-04, atol=tol
      )
  
imports: import copy
  import numpy
  import pytest
  import pennylane
  from pennylane.measurements import VnEntropy, Shots
  from pennylane.measurements.vn_entropy import VnEntropyMP
  from pennylane.wires import Wires
label_desc: Use image node_id 9 for calling the TestIntegration obj's underlying member method code with example usage: obj.test_IsingXX_qnode_entropy_grad_jax(param, wires, base, diff_method, interface) and returns: qml