class_docstr: Test that the XYX decomposition is correct.
class_docstr_tok: ["Test","that","the","XYX","decomposition","is","correct","."]
docstr: Test that a one-qubit operation in JAX is correctly decomposed.
docstr_tok: ["Test","that","a","one-qubit","operation","in","JAX","is","correctly","decomposed","."]
code: def test_xyx_decomposition_jax(self, U, expected_params):
      """Test that a one-qubit operation in JAX is correctly decomposed."""
  
      import jax
  
      U = jax.numpy.array(U, dtype=jax.numpy.complex128)
      _test_decomposition(U, "XYX", typeof_gates_xyx, expected_params)
  
imports: from functools import reduce
  import pytest
  from gate_data import I, Z, S, T, H, X, Y, CNOT, SWAP
  import pennylane
  from pennylane import numpy
  from pennylane.wires import Wires
  from pennylane.ops.op_math.decompositions import one_qubit_decomposition
  from pennylane.ops.op_math.decompositions import two_qubit_decomposition
  from pennylane.ops.op_math.decompositions.two_qubit_unitary import _convert_to_su4, _su2su2_to_tensor_products, _compute_num_cnots
label_desc: Use image node_id 4 for calling the TestQubitUnitaryXYXDecomposition obj's underlying member method code with example usage: obj.test_xyx_decomposition_jax(U, expected_params) without return types