class_docstr: Tests for purity transform
docstr: Test purity for a QNode with jax-jit interface.
docstr_tok: ["Test","purity","for","a","QNode","with","jax-jit","interface","."]
code: def test_IsingXX_qnode_purity_jax_jit(
      self, device, param, wires, is_partial, interface
  ):
      """Test purity for a QNode with jax-jit interface."""
  
      import jax
      import jax.numpy as jnp
  
      jax.config.update("jax_enable_x64", True)
  
      dev = qml.device(device, wires=2)
  
      @qml.qnode(dev, interface=interface)
      def circuit_state(x):
          qml.IsingXX(x, wires=[0, 1])
          return qml.state()
  
      purity = jax.jit(qml.qinfo.purity(circuit_state, wires=wires))(
          jnp.array(param)
      )
      expected_purity = (
          expected_purity_ising_xx(param) if is_partial else 1
      )
      assert qml.math.allclose(purity, expected_purity)
  
imports: import pytest
  import pennylane
  from pennylane import numpy
label_desc: Use image node_id 10 for calling the TestPurity obj's underlying member method code with example usage: obj.test_IsingXX_qnode_purity_jax_jit(device, param, wires, is_partial, interface) and returns: qml