def test_set_mapping(): d = QUSO({('a', 'b'): 1, ('a',): 2}) d.set_mapping({'a': 0, 'b': 2}) assert d.to_quso() == {(0, 2): 1, (0,): 2} d = QUSO({('a', 'b'): 1, ('a',): 2}) d.set_reverse_mapping({0: 'a', 2: 'b'}) assert d.to_quso() == {(0, 2): 1, (0,): 2}
def test_properties(): temp = QUSO({('0', '0'): 1, ('0', 1): 2}) assert temp.offset == 1 d = QUSO() d[(0,)] += 1 d[(1,)] += 2 assert d == d.to_quso() == {(0,): 1, (1,): 2} assert d.mapping == d.reverse_mapping == {0: 0, 1: 1} d.set_mapping({1: 0, 0: 1}) assert d.to_quso() == {(1,): 1, (0,): 2} assert d.mapping == d.reverse_mapping == {0: 1, 1: 0}
def test_properties(): temp = QUSO({('0', '0'): 1, ('0', 1): 2}) assert temp.offset == 1 d = QUSO() d[(0,)] += 1 d[(1,)] += 2 assert d == d.to_quso() == {(0,): 1, (1,): 2} assert d.mapping == d.reverse_mapping == {0: 0, 1: 1} d.set_mapping({1: 0, 0: 1}) assert d.to_quso() == {(1,): 1, (0,): 2} assert d.mapping == d.reverse_mapping == {0: 1, 1: 0} # an old bug d = QUSO() d.set_mapping({0: 0}) d[(0,)] += 1 assert d.num_binary_variables == 1 assert d.variables == {0}