def test_qsd_codegen_operator_basis():
a = Destroy(hs=hs1)
ad = a.dag()
s = LocalSigma(1, 0, hs=hs2)
sd = s.dag()
circuit = SLH(identity_matrix(0), [], a * ad + s + sd)
codegen = QSDCodeGen(circuit)
ob = codegen._operator_basis_lines(indent=0)
assert dedent(ob).strip() == dedent("""
IdentityOperator Id0(0);
IdentityOperator Id1(1);
AnnihilationOperator A0(0);
FieldTransitionOperator S1_0_1(0,1,1);
FieldTransitionOperator S1_1_0(1,0,1);
Operator Id = Id0*Id1;
Operator Ad0 = A0.hc();
""").strip()
circuit = SLH(identity_matrix(0), [], ad)
codegen = QSDCodeGen(circuit)
ob = codegen._operator_basis_lines(indent=0)
assert dedent(ob).strip() == dedent("""
IdentityOperator Id0(0);
AnnihilationOperator A0(0);
Operator Id = Id0;
Operator Ad0 = A0.hc();
""").strip()
def test_qsd_codegen_hamiltonian():
k = symbols(r'\kappa', positive=True)
x = symbols(r'\chi', real=True)
c = symbols("c", real=True)
a = Destroy(hs=hs1)
H = x * (a * a + a.dag() * a.dag()) + (c * a.dag() + c.conjugate() * a)
L = sqrt(k) * a
slh = SLH(identity_matrix(1), [L], H)
codegen = QSDCodeGen(circuit=slh, num_vals={x: 2., c: 1, k: 2})
codegen._operator_basis_lines(indent=0)
scode = codegen._hamiltonian_lines(indent=0)
assert scode.strip() == (r'Operator H = ((c) * (Ad0) + (c) * (A0) + (chi) '
r'* (((Ad0 * Ad0) + (A0 * A0))));')
def test_qsd_codegen_observables(caplog, slh_Sec6, slh_Sec6_vals):
A2 = Destroy(hs=hs1)
Sp = LocalSigma(1, 0, hs=hs2)
Sm = Sp.dag()
codegen = QSDCodeGen(circuit=slh_Sec6, num_vals=slh_Sec6_vals)
with pytest.raises(QSDCodeGenError) as excinfo:
scode = codegen._observables_lines(indent=0)
assert "Must register at least one observable" in str(excinfo.value)
name = 'a_1 sigma_10^[2]'
codegen.add_observable(Sp * A2 * Sm * Sp, name=name)
filename = codegen._observables[name][1]
assert filename == 'a_1_sigma_10_2.out'
codegen.add_observable(Sp * A2 * Sm * Sp, name=name)
assert 'Overwriting existing operator' in caplog.text
with pytest.raises(ValueError) as exc_info:
codegen.add_observable(Sp * A2 * A2 * Sm * Sp, name="xxxx" * 20)
assert "longer than limit" in str(exc_info.value)
name = 'A2^2'
codegen.add_observable(Sp * A2 * A2 * Sm * Sp, name=name)
assert name in codegen._observables
filename = codegen._observables[name][1]
assert filename == 'A2_2.out'
with pytest.raises(ValueError) as exc_info:
codegen.add_observable(A2, name='A2_2')
assert "Cannot generate unique filename" in str(exc_info.value)
with pytest.raises(ValueError) as exc_info:
codegen.add_observable(A2, name="A2\t2")
assert "invalid characters" in str(exc_info.value)
with pytest.raises(ValueError) as exc_info:
codegen.add_observable(A2, name="A" * 100)
assert "longer than limit" in str(exc_info.value)
with pytest.raises(ValueError) as exc_info:
codegen.add_observable(A2, name="()")
assert "Cannot generate filename" in str(exc_info.value)
codegen = QSDCodeGen(circuit=slh_Sec6, num_vals=slh_Sec6_vals)
codegen.add_observable(Sp * A2 * Sm * Sp, name="X1")
codegen.add_observable(Sm * Sp * A2 * Sm, name="X2")
assert codegen._observables["X2"] == (Sm * Sp * A2 * Sm, 'X2.out')
codegen.add_observable(A2, name="A2")
assert codegen._observables["A2"] == (A2, 'A2.out')
scode = codegen._observables_lines(indent=0)
assert dedent(scode).strip() == dedent(r'''
const int nOfOut = 3;
Operator outlist[nOfOut] = {
(A1 * S2_1_0),
(A1 * S2_0_1),
A1
};
char *flist[nOfOut] = {"X1.out", "X2.out", "A2.out"};
int pipe[4] = {1,2,3,4};
''').strip()
# Note how the observables have been simplified
assert Sp * A2 * Sm * Sp == Sp * A2
assert codegen._operator_str(Sp * A2) == '(A1 * S2_1_0)'
assert Sm * Sp * A2 * Sm == Sm * A2
assert codegen._operator_str(Sm * A2) == '(A1 * S2_0_1)'
# If the oberservables introduce new operators or symbols, these should
# extend the existing ones
P1 = LocalSigma(1, 1, hs=hs2)
zeta = symbols("zeta", real=True)
codegen.add_observable(zeta * P1, name="P1")
assert P1 in codegen._local_ops
assert str(codegen._qsd_ops[P1]) == 'S2_1_1'
assert zeta in codegen.syms
codegen.num_vals.update({zeta: 1.0})
assert 'zeta' in codegen._parameters_lines(indent=0)
assert str(codegen._qsd_ops[P1]) in codegen._operator_basis_lines(indent=0)
assert Sp * A2 in set(codegen.observables)
assert Sm * A2 in set(codegen.observables)
assert zeta * P1 in set(codegen.observables)
assert list(codegen.observable_names) == ['X1', 'X2', 'A2', 'P1']
assert codegen.get_observable('X1') == Sp * A2 * Sm * Sp