def test_set_mapping():
d = PUBO({('a', 'b'): 1, ('a', ): 2})
d.set_mapping({'a': 0, 'b': 2})
assert d.to_pubo() == {(0, 2): 1, (0, ): 2}
d = PUBO({('a', 'b'): 1, ('a', ): 2})
d.set_reverse_mapping({0: 'a', 2: 'b'})
assert d.to_pubo() == {(0, 2): 1, (0, ): 2}
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}