class_docstr: Test the state_vector() method of QutritBasisState operation.
class_docstr_tok: ["Test","the","state_vector","(",")","method","of","QutritBasisState","operation","."]
docstr: Tests that given an array of some type, the resulting state_vector is also that type.
docstr_tok: ["Tests","that","given","an","array","of","some","type",",","the","resulting","state_vector","is","also","that","type","."]
code: def test_QutritBasisState_state_vector_preserves_parameter_type(
      self, interface
  ):
      """Tests that given an array of some type, the resulting state_vector is also that type."""
      basis_op = qml.QutritBasisState(
          qml.math.array([0, 1], like=interface), wires=[1, 2]
      )
      assert (
          qml.math.get_interface(basis_op.state_vector()) == interface
      )
      assert (
          qml.math.get_interface(
              basis_op.state_vector(wire_order=[0, 1, 2])
          )
          == interface
      )
  
file_path: pennylane/tests/ops/qutrit/test_qutrit_state_prep.py
imports: import pytest
  import pennylane
  from pennylane import numpy
  from pennylane.wires import WireError
label_desc: Use image node_id 3 for calling the TestStateVector obj's underlying member method code with example usage: obj.test_QutritBasisState_state_vector_preserves_parameter_type(interface) without return types