def METHOD_NAME(self, num_wires, wire_order, one_position):
    """Tests that QutritBasisState state_vector returns kets as expected."""
    basis_op = qml.QutritBasisState([0, 1], wires=[1, 2])
    ket = basis_op.state_vector(wire_order=wire_order)
    assert qml.math.shape(ket) == (3,) * num_wires
    assert ket[one_position] == 1
    ket[one_position] = 0  # everything else should be zero, as we assert below
    assert np.allclose(np.zeros((3,) * num_wires), ket)