def METHOD_NAME(self, tol):
    """Test reduced_dm jitting for QNode."""
    import jax.numpy as jnp
    from jax import jit
    angle = jnp.array(0.1)
    dev = qml.device("default.qubit", wires=2)
    @qml.qnode(dev, interface="jax-jit")
    def circuit(x):
        qml.IsingXX(x, wires=[0, 1])
        return qml.state()
    density_matrix = jit(qml.qinfo.reduced_dm(circuit, wires=[0]))(angle)
    expected_density_matrix = [[np.cos(angle / 2) ** 2, 0], [0, np.sin(angle / 2) ** 2]]
    assert np.allclose(density_matrix, expected_density_matrix, atol=tol, rtol=0)