from qutip import *

from qutip.cyQ.spmatfuncs import cy_expect

N = 50

op = rand_herm(N)
ket = rand_ket(N)
rho = ket2dm(ket)

ket_data = ket.full()
op_data = op.data

sop = spre(op)
sop_data = sop.data

rho_vec = operator_to_vector(rho)
rho_vec_data = rho_vec.full()

expect(op, ket)

expect(op, rho)

cy_expect(op_data.data, op_data.indices, op_data.indptr, ket_data, 1)

expect_psi(op_data, ket_data, 1)

expect_rho_vec(sop_data, rho_vec_data, 1)

%%timeit

expect(op, ket)

%%timeit

cy_expect(op_data.data, op_data.indices, op_data.indptr, ket_data, 1)

%%timeit

expect_psi(op_data, ket_data, 1)

%%timeit

expect(op, rho)

%%timeit

expect_rho_vec(sop_data, rho_vec_data)

from qutip.ipynbtools import version_table

version_table()