示例#1
0
def test_graycode():
    g = GrayCode(2)
    got = []
    for i in g.generate_gray():
        if i.startswith('0'):
            g.skip()
        got.append(i)
    assert got == '00 11 10'.split()
    a = GrayCode(6)
    assert a.current == '0'*6
    assert a.rank == 0
    assert len(list(a.generate_gray())) == 64
    codes = ['011001', '011011', '011010',
    '011110', '011111', '011101', '011100', '010100', '010101', '010111',
    '010110', '010010', '010011', '010001', '010000', '110000', '110001',
    '110011', '110010', '110110', '110111', '110101', '110100', '111100',
    '111101', '111111', '111110', '111010', '111011', '111001', '111000',
    '101000', '101001', '101011', '101010', '101110', '101111', '101101',
    '101100', '100100', '100101', '100111', '100110', '100010', '100011',
    '100001', '100000']
    assert list(a.generate_gray(start='011001')) == codes
    assert list(
        a.generate_gray(rank=GrayCode(6, start='011001').rank)) == codes
    assert a.next().current == '000001'
    assert a.next(2).current == '000011'
    assert a.next(-1).current == '100000'

    a = GrayCode(5, start='10010')
    assert a.rank == 28
    a = GrayCode(6, start='101000')
    assert a.rank == 48

    assert GrayCode(6, rank=4).current == '000110'
    assert GrayCode(6, rank=4).rank == 4
    assert [GrayCode(4, start=s).rank for s in
            GrayCode(4).generate_gray()] == [0, 1, 2, 3, 4, 5, 6, 7, 8,
                                             9, 10, 11, 12, 13, 14, 15]
    a = GrayCode(15, rank=15)
    assert a.current == '000000000001000'

    assert bin_to_gray('111') == '100'

    a = random_bitstring(5)
    assert type(a) is str
    assert len(a) == 5
    assert all(i in ['0', '1'] for i in a)

    assert get_subset_from_bitstring(
        ['a', 'b', 'c', 'd'], '0011') == ['c', 'd']
    assert get_subset_from_bitstring('abcd', '1001') == ['a', 'd']
    assert list(graycode_subsets(['a', 'b', 'c'])) == \
        [[], ['c'], ['b', 'c'], ['b'], ['a', 'b'], ['a', 'b', 'c'],
         ['a', 'c'], ['a']]

    raises(ValueError, lambda: GrayCode(0))
    raises(ValueError, lambda: GrayCode(2.2))
    raises(ValueError, lambda: GrayCode(2, start=[1, 1, 0]))
    raises(ValueError, lambda: GrayCode(2, rank=2.5))
    raises(ValueError, lambda: get_subset_from_bitstring(['c', 'a', 'c'], '1100'))
    raises(ValueError, lambda: list(GrayCode(3).generate_gray(start="1111")))
示例#2
0
def test_graycode():
    g = GrayCode(2)
    got = []
    for i in g.generate_gray():
        if i.startswith('0'):
            g.skip()
        got.append(i)
    assert got == '00 11 10'.split()
    a = GrayCode(6)
    assert a.current == '0'*6
    assert a.rank == 0
    assert len(list(a.generate_gray())) == 64
    codes = ['011001', '011011', '011010',
    '011110', '011111', '011101', '011100', '010100', '010101', '010111',
    '010110', '010010', '010011', '010001', '010000', '110000', '110001',
    '110011', '110010', '110110', '110111', '110101', '110100', '111100',
    '111101', '111111', '111110', '111010', '111011', '111001', '111000',
    '101000', '101001', '101011', '101010', '101110', '101111', '101101',
    '101100', '100100', '100101', '100111', '100110', '100010', '100011',
    '100001', '100000']
    assert list(a.generate_gray(start='011001')) == codes
    assert list(
        a.generate_gray(rank=GrayCode(6, start='011001').rank)) == codes
    assert a.next().current == '000001'
    assert a.next(2).current == '000011'
    assert a.next(-1).current == '100000'

    a = GrayCode(5, start='10010')
    assert a.rank == 28
    a = GrayCode(6, start='101000')
    assert a.rank == 48

    assert GrayCode(6, rank=4).current == '000110'
    assert GrayCode(6, rank=4).rank == 4
    assert [GrayCode(4, start=s).rank for s in
            GrayCode(4).generate_gray()] == [0, 1, 2, 3, 4, 5, 6, 7, 8,
                                             9, 10, 11, 12, 13, 14, 15]
    a = GrayCode(15, rank=15)
    assert a.current == '000000000001000'

    assert bin_to_gray('111') == '100'

    a = random_bitstring(5)
    assert type(a) is str
    assert len(a) == 5
    assert all(i in ['0', '1'] for i in a)

    assert get_subset_from_bitstring(
        ['a', 'b', 'c', 'd'], '0011') == ['c', 'd']
    assert get_subset_from_bitstring('abcd', '1001') == ['a', 'd']
    assert list(graycode_subsets(['a', 'b', 'c'])) == \
        [[], ['c'], ['b', 'c'], ['b'], ['a', 'b'], ['a', 'b', 'c'],
    ['a', 'c'], ['a']]
示例#3
0
def test_graycode():
    a = GrayCode(6)
    assert a.current == '0' * 6
    assert a.rank == 0
    assert len(list(a.generate_gray())) == 64
    codes = [
        '011001', '011011', '011010', '011110', '011111', '011101', '011100',
        '010100', '010101', '010111', '010110', '010010', '010011', '010001',
        '010000', '110000', '110001', '110011', '110010', '110110', '110111',
        '110101', '110100', '111100', '111101', '111111', '111110', '111010',
        '111011', '111001', '111000', '101000', '101001', '101011', '101010',
        '101110', '101111', '101101', '101100', '100100', '100101', '100111',
        '100110', '100010', '100011', '100001', '100000'
    ]
    assert list(a.generate_gray(start='011001')) == codes
    assert list(
        a.generate_gray(rank=GrayCode(6, start='011001').rank)) == codes
    assert a.next().current == '000001'
    assert a.next(2).current == '000011'
    assert a.next(-1).current == '100000'

    a = GrayCode(5, start='10010')
    assert a.rank == 28
    a = GrayCode(6, start='101000')
    assert a.rank == 48

    assert GrayCode(6, rank=4).current == '000110'
    assert GrayCode(6, rank=4).rank == 4
    assert [GrayCode(4, start=s).rank for s in \
            GrayCode(4).generate_gray()] == [0, 1, 2, 3, 4, 5, 6, 7, 8, \
                                             9, 10, 11, 12, 13, 14, 15]
    a = GrayCode(15, rank=15)
    assert a.current == '000000000001000'

    assert bin_to_gray('111') == '100'

    a = random_bitstring(5)
    assert type(a) is str
    assert len(a) == 5
    assert all(i in ['0', '1'] for i in a)

    assert get_subset_from_bitstring(['a', 'b', 'c', 'd'],
                                     '0011') == ['c', 'd']
    assert get_subset_from_bitstring('abcd', '1001') == ['a', 'd']
    assert list(graycode_subsets(['a','b','c'])) == \
    [[], ['c'], ['b', 'c'], ['b'], ['a', 'b'], ['a', 'b', 'c'], \
    ['a', 'c'], ['a']]
示例#4
0
def test_graycode():
    a = GrayCode(6)
    assert a.current == '0'*6
    assert a.rank == 0
    assert len(list(a.generate_gray())) == 64
    codes = ['011001', '011011', '011010',
    '011110', '011111', '011101', '011100', '010100', '010101', '010111',
    '010110', '010010', '010011', '010001', '010000', '110000', '110001',
    '110011', '110010', '110110', '110111', '110101', '110100', '111100',
    '111101', '111111', '111110', '111010', '111011', '111001', '111000',
    '101000', '101001', '101011', '101010', '101110', '101111', '101101',
    '101100', '100100', '100101', '100111', '100110', '100010', '100011',
    '100001', '100000']
    assert list(a.generate_gray(start='011001')) == codes
    assert list(a.generate_gray(rank=GrayCode(6, start='011001').rank)) == codes
    assert a.next().current == '000001'
    assert a.next(2).current == '000011'
    assert a.next(-1).current == '100000'

    a = GrayCode(5, start='10010')
    assert a.rank == 28
    a = GrayCode(6, start='101000')
    assert a.rank == 48

    assert GrayCode(6, rank=4).current == '000110'
    assert GrayCode(6, rank=4).rank == 4
    assert [GrayCode(4, start=s).rank for s in \
            GrayCode(4).generate_gray()] == [0, 1, 2, 3, 4, 5, 6, 7, 8, \
                                             9, 10, 11, 12, 13, 14, 15]
    a = GrayCode(15, rank=15)
    assert a.current == '000000000001000'

    assert bin_to_gray('111') == '100'
    random.seed(0)
    assert [random_bitstring(5) for i in range(3)] == ['11001', '01001', '11011']
    assert get_subset_from_bitstring(['a','b','c','d'], '0011') == ['c', 'd']
    assert get_subset_from_bitstring('abcd','1001') == ['a', 'd']
    assert list(graycode_subsets(['a','b','c'])) == \
    [[], ['c'], ['b', 'c'], ['b'], ['a', 'b'], ['a', 'b', 'c'], \
    ['a', 'c'], ['a']]
示例#5
0
def test_graycode():
    g = GrayCode(2)
    got = []
    for i in g.generate_gray():
        if i.startswith("0"):
            g.skip()
        got.append(i)
    assert got == "00 11 10".split()
    a = GrayCode(6)
    assert a.current == "0" * 6
    assert a.rank == 0
    assert len(list(a.generate_gray())) == 64
    codes = [
        "011001",
        "011011",
        "011010",
        "011110",
        "011111",
        "011101",
        "011100",
        "010100",
        "010101",
        "010111",
        "010110",
        "010010",
        "010011",
        "010001",
        "010000",
        "110000",
        "110001",
        "110011",
        "110010",
        "110110",
        "110111",
        "110101",
        "110100",
        "111100",
        "111101",
        "111111",
        "111110",
        "111010",
        "111011",
        "111001",
        "111000",
        "101000",
        "101001",
        "101011",
        "101010",
        "101110",
        "101111",
        "101101",
        "101100",
        "100100",
        "100101",
        "100111",
        "100110",
        "100010",
        "100011",
        "100001",
        "100000",
    ]
    assert list(a.generate_gray(start="011001")) == codes
    assert list(
        a.generate_gray(rank=GrayCode(6, start="011001").rank)) == codes
    assert a.next().current == "000001"
    assert a.next(2).current == "000011"
    assert a.next(-1).current == "100000"

    a = GrayCode(5, start="10010")
    assert a.rank == 28
    a = GrayCode(6, start="101000")
    assert a.rank == 48

    assert GrayCode(6, rank=4).current == "000110"
    assert GrayCode(6, rank=4).rank == 4
    assert [GrayCode(4, start=s).rank
            for s in GrayCode(4).generate_gray()] == [
                0,
                1,
                2,
                3,
                4,
                5,
                6,
                7,
                8,
                9,
                10,
                11,
                12,
                13,
                14,
                15,
            ]
    a = GrayCode(15, rank=15)
    assert a.current == "000000000001000"

    assert bin_to_gray("111") == "100"

    a = random_bitstring(5)
    assert type(a) is str
    assert len(a) == 5
    assert all(i in ["0", "1"] for i in a)

    assert get_subset_from_bitstring(["a", "b", "c", "d"],
                                     "0011") == ["c", "d"]
    assert get_subset_from_bitstring("abcd", "1001") == ["a", "d"]
    assert list(graycode_subsets(["a", "b", "c"])) == [
        [],
        ["c"],
        ["b", "c"],
        ["b"],
        ["a", "b"],
        ["a", "b", "c"],
        ["a", "c"],
        ["a"],
    ]

    raises(ValueError, lambda: GrayCode(0))
    raises(ValueError, lambda: GrayCode(2.2))
    raises(ValueError, lambda: GrayCode(2, start=[1, 1, 0]))
    raises(ValueError, lambda: GrayCode(2, rank=2.5))
    raises(ValueError,
           lambda: get_subset_from_bitstring(["c", "a", "c"], "1100"))
    raises(ValueError, lambda: list(GrayCode(3).generate_gray(start="1111")))
示例#6
0
def part1():
    all_commands = []
    p = []
    vm = intcode.run(inp, p)

    def add_command(cmd):
        p.extend([ord(c) for c in cmd])
        if cmd[-1] != '\n':
            p.append(ord('\n'))

    for cmd in path:
        add_command(cmd)

    def take(item):
        add_command(f'take {item}')

    def drop(item):
        add_command(f'drop {item}')

    for item in items:
        drop(item)

    def run_until_command(show=False):
        lastn = []
        bad = False
        outp = []
        for c in vm:
            cr = chr(c)
            outp.append(cr)

            if len(lastn) == n:
                lastn.pop(0)
            lastn.append(chr(c))

            s = ''.join(lastn)
            if s == ' heavier' or s == ' lighter':
                bad = True
            if s == 'Command?' and len(p) == 0:
                break
        if not bad and show:
            print(''.join(outp), end='')
        return bad

    run_until_command()

    cur_set = set()

    for subset in graycode_subsets(items):
        if not subset:
            continue
        #print(subset)
        diff = cur_set.symmetric_difference(subset)
        for item in diff:
            if item in cur_set:
                drop(item)
            else:
                take(item)
        cur_set = set(subset)
        run_until_command()
        add_command('west')
        bad = run_until_command(show=True)
        if not bad:
            break