class_docstr: Class to test Jax integration with classical Jacobian.
class_docstr_tok: ["Class","to","test","Jax","integration","with","classical","Jacobian","."]
docstr: Test ``classical_jacobian`` with ``argnum=Sequence[int]`` and JAX.
docstr_tok: ["Test","``","classical_jacobian","``","with","``","argnum=Sequence","[","int","]","``","and","JAX","."]
code: def test_jax_with_sequence_argnum(
      self, circuit, args, expected_jac, argnum, diff_method, interface
  ):
      r"""Test ``classical_jacobian`` with ``argnum=Sequence[int]`` and JAX."""
      import jax.numpy as jnp
  
      args = tuple((jnp.array(arg) for arg in args))
      dev = qml.device("default.qubit", wires=2)
      qnode = qml.QNode(
          circuit, dev, interface=interface, diff_method=diff_method
      )
      jac = classical_jacobian(qnode, argnum=argnum)(*args)
  
      if len(argnum) > 1:
          expected_jac = expected_jac[-1]
      print(argnum)
      print(jac, expected_jac)
      assert len(jac) == len(expected_jac)
      for _jac, _expected_jac in zip(jac, expected_jac):
          assert np.allclose(_jac, _expected_jac)
  
imports: import pytest
  import numpy
  import pennylane
  from pennylane import numpy
  from pennylane.gradients.classical_jacobian import classical_jacobian
label_desc: Use image node_id 4 for calling the TestJax obj's underlying member method code with example usage: obj.test_jax_with_sequence_argnum(circuit, args, expected_jac, argnum, diff_method, interface) without return types