class_docstr: Test that the right method between _apply_channel and _apply_channel_tensordot
  is chosen.
docstr: Test with a JAX array as device state.
code: def test_with_jax_state(self, mocker, op, exp_method, dev_wires):
      """Test with a JAX array as device state."""
  
      methods = ["_apply_channel", "_apply_channel_tensordot"]
      del methods[methods.index(exp_method)]
      unexp_method = methods[0]
      spy_exp = mocker.spy(DefaultMixed, exp_method)
      spy_unexp = mocker.spy(DefaultMixed, unexp_method)
      dev = qml.device("default.mixed", wires=dev_wires)
      state = np.zeros((2**dev_wires, 2**dev_wires))
      state[0, 0] = 1.0
      dev._state = jnp.array(state).reshape([2] * (2 * dev_wires))
      dev._apply_operation(op)
  
      spy_unexp.assert_not_called()
      spy_exp.assert_called_once()
  
imports: from functools import partial
  import pytest
  import numpy
  import pennylane
  from pennylane import numpy
  from pennylane.devices.default_mixed import DefaultMixed
label_desc: Use image node_id 2 for calling the TestApplyChannelMethodChoice obj's underlying member method code with example usage: obj.test_with_jax_state(mocker, op, exp_method, dev_wires) without return types