Python PUBO.set_mapping Examples

Example #1

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}

Example #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}

Example #3

File: test_pubo.py Project: jtiosue/qubovert

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}