def test_pubo_degree_reduction_pairs():
pubo = PUBO({
('x0', 'x1'): -1,
('x1', ): 1,
('x1', 'x2'): -1,
('x2', ): 1,
('x3', 'x2'): -1,
('x3', ): 1,
('x4', 'x3'): -1,
('x4', ): 1,
('x4', 'x5'): -1,
('x5', ): 1,
('x5', 'x6'): -1,
('x6', ): 1,
('x7', 'x6'): -1,
('x7', ): 1,
('x8', 'x7'): -1,
('x8', ): 1,
('x9', 'x8'): -1,
('x9', ): 1
})**2
pairs = {('x0', 'x1'), ('x1', 'x2'), ('x2', 'x3'), ('x3', 'x4'),
('x4', 'x5'), ('x5', 'x6'), ('x6', 'x7'), ('x7', 'x8'),
('x8', 'x9')}
qubo1 = pubo.to_qubo()
qubo2 = pubo.to_qubo(pairs=pairs)
assert qubo1.num_binary_variables - pubo.num_binary_variables > 9
assert qubo2.num_binary_variables - pubo.num_binary_variables == 9
quso1 = pubo.to_quso()
quso2 = pubo.to_quso(pairs=pairs)
assert quso1.num_binary_variables - pubo.num_binary_variables > 9
assert quso2.num_binary_variables - pubo.num_binary_variables == 9
def test_properties():
temp = PUBO({('0', '0'): 1, ('0', 1): 2})
assert not temp.offset
d = PUBO()
d[(0, )] += 1
d[(1, )] += 2
assert d == d.to_qubo() == {(0, ): 1, (1, ): 2}
assert d.mapping == d.reverse_mapping == {0: 0, 1: 1}
d.set_mapping({1: 0, 0: 1})
assert d.to_qubo() == {(1, ): 1, (0, ): 2}
assert d.mapping == d.reverse_mapping == {0: 1, 1: 0}
def test_properties():
temp = PUBO({('0', '0'): 1, ('0', 1): 2})
assert not temp.offset
d = PUBO()
d[(0,)] += 1
d[(1,)] += 2
assert d == d.to_qubo() == {(0,): 1, (1,): 2}
assert d.mapping == d.reverse_mapping == {0: 0, 1: 1}
d.set_mapping({1: 0, 0: 1})
assert d.to_qubo() == {(1,): 1, (0,): 2}
assert d.mapping == d.reverse_mapping == {0: 1, 1: 0}
# an old bug
d = PUBO()
d.set_mapping({0: 0})
d[(0,)] += 1
assert d.num_binary_variables == 1
assert d.variables == {0}
def test_pubo_degree_reduction_lam():
pubo = PUBO({
('x0', 'x1'): -1, ('x1',): 1, ('x1', 'x2'): -1, ('x2',): 1,
('x3', 'x2'): -1, ('x3',): 1, ('x4', 'x3'): -1, ('x4',): 1,
}) ** 2
# just make sure it runs
pubo.to_qubo(lam=4)
pubo.to_qubo(lam=lambda v: v)
pubo.to_qubo(lam=Symbol('lam'))
pubo.to_qubo(lam=lambda v: v * Symbol('lam'))
pubo.to_quso(lam=4)
pubo.to_quso(lam=lambda v: v)
pubo.to_quso(lam=Symbol('lam'))
pubo.to_quso(lam=lambda v: v * Symbol('lam'))
