import numpy as np
from slam.hamiltonian import ConversionGainSmush

U = ConversionGainSmush().construct_U(
    0, 0, np.pi / 2, 0, [3, 3, 3, 3, 3, 3], [0, 0, 0, 0, 0, 0]
)

from qutip import mesolve

from qutip import tensor, basis, qeye, sigmax, sigmaz

# initial state: start with one of the qubits in its excited state
psi0 = tensor(basis(N, 0), basis(2, 1), basis(2, 0))

from weylchamber import *

WeylChamber().plot()

from slam.utils.visualize import unitary_to_weyl

unitary_to_weyl(U)