class_docstr: Test ``qml.qjit`` with Catalyst
class_docstr_tok: ["Test","``","qml.qjit","``","with","Catalyst"]
docstr: Test pytree return values.
docstr_tok: ["Test","pytree","return","values","."]
code: def test_return_value_dict(self):
      """Test pytree return values."""
      dev = qml.device("lightning.qubit", wires=2)
  
      @qml.qnode(dev)
      def circuit1(params):
          qml.RX(params[0], wires=0)
          qml.RX(params[1], wires=1)
          return {
              "w0": qml.expval(qml.PauliZ(0)),
              "w1": qml.expval(qml.PauliZ(1)),
          }
  
      jitted_fn = qml.qjit(circuit1)
  
      params = [0.2, 0.6]
      expected = {"w0": 0.98006658, "w1": 0.82533561}
      result = jitted_fn(params)
      assert isinstance(result, dict)
      assert jnp.allclose(result["w0"], expected["w0"])
      assert jnp.allclose(result["w1"], expected["w1"])
  
imports: from unittest.mock import patch
  import pytest
  import pennylane
  from pennylane.compiler.compiler import CompileError
  from pennylane import numpy
  from jax import numpy
  from jax.core import ShapedArray
label_desc: Use image node_id 8 for calling the TestCatalyst obj's underlying member method code with example usage: obj.test_return_value_dict() and returns: dict