def test_coupling_terms_handle_JW():
strength = 0.25
sites = []
L = 4
for i in range(L):
s = site.Site(spin_half.leg)
s.add_op("X_{i:d}".format(i=i), 2. * np.eye(2))
s.add_op("Y_{i:d}".format(i=i), 3. * np.eye(2), need_JW=True)
sites.append(s)
mc = MultiCouplingTerms(L)
# two-site terms
term = [("X_1", 1), ("X_0", 4)]
args = mc.coupling_term_handle_JW(strength, term, sites)
# args = i, j, op_i, op_j, op_str
assert args == (strength, 1, 4, "X_1", "X_0", "Id")
term = [("Y_1", 1), ("Y_0", 4)]
args = mc.coupling_term_handle_JW(strength, term, sites)
assert args == (strength, 1, 4, "Y_1 JW", "Y_0", "JW")
# switch order
term = [("Y_0", 4), ("Y_1", 1)]
term, sign = order_combine_term(term, sites)
assert term == [("Y_1", 1), ("Y_0", 4)]
args = mc.coupling_term_handle_JW(strength * sign, term, sites)
assert args == (-strength, 1, 4, "Y_1 JW", "Y_0", "JW")
# multi coupling
term = [("X_0", 0), ("X_1", 1), ("X_3", 3)]
args = mc.multi_coupling_term_handle_JW(strength, term, sites)
assert args == (strength, [0, 1, 3], ["X_0", "X_1", "X_3"], ["Id", "Id"])
term = [("X_0", 0), ("Y_1", 1), ("Y_3", 3)]
args = mc.multi_coupling_term_handle_JW(strength, term, sites)
assert args == (strength, [0, 1, 3], ["X_0", "Y_1 JW", "Y_3"], ["Id", "JW"])
term = [("Y_0", 0), ("X_1", 1), ("Y_3", 3), ("X_0", 4), ("Y_2", 6), ("Y_3", 7)]
args = mc.multi_coupling_term_handle_JW(strength, term, [dummy] * 4)
assert args == (strength, [0, 1, 3, 4, 6, 7], [op[0] for op in term], ["Id"] * (len(term) - 1))
args = mc.multi_coupling_term_handle_JW(strength, term, sites)
print(args)
assert args == (strength, [0, 1, 3, 4, 6,
7], ["Y_0 JW", "X_1 JW", "Y_3", "X_0", "Y_2 JW",
"Y_3"], ["JW", "JW", "Id", "Id", "JW"])
term = [
("Y_3", 7),
("X_1", 1),
("Y_0", 0),
("X_0", 4),
("Y_2", 6),
("Y_3", 3),
]
term, sign = order_combine_term(term, sites)
args = mc.multi_coupling_term_handle_JW(strength * sign, term, sites)
print(args)
assert args == (strength, [0, 1, 3, 4, 6,
7], ["Y_0 JW", "X_1 JW", "Y_3", "X_0", "Y_2 JW",
"Y_3"], ["JW", "JW", "Id", "Id", "JW"])