def test_qnode_intergration():
    'Test a simple use of qnode with a JAX interface and non-JAX device'
    dev = qml.device('default.mixed', wires=2)

    @qml.qnode(dev, interface='jax')
    def circuit(weights):
        qml.RX(weights[0], wires=0)
        qml.RZ(weights[1], wires=1)
        return qml.expval((qml.PauliZ(0) @ qml.PauliZ(1)))
    weights = jnp.array([0.1, 0.2])
    val = circuit(weights)
    assert ('DeviceArray' in val.__repr__())
Test a simple use of qnode with a JAX interface and non-JAX device
def test_qnode_intergration():
    
    dev = qml.device('default.mixed', wires=2)

    @qml.qnode(dev, interface='jax')
    def circuit(weights):
        qml.RX(weights[0], wires=0)
        qml.RZ(weights[1], wires=1)
        return qml.expval((qml.PauliZ(0) @ qml.PauliZ(1)))
    weights = jnp.array([0.1, 0.2])
    val = circuit(weights)
    assert ('DeviceArray' in val.__repr__())