Ejemplo n.º 1
0
def test_loop():
    assert_equal(
        loop(1),
        {0: [0]}
    )
    assert_equal(
        loop(2),
        {0: [1], 1: [0]}
    )
    assert_equal(
        loop(3),
        {0: [1], 1: [2], 2: [0]}
    )
Ejemplo n.º 2
0
def test_simulate_simple():
    result = list(simulate(
        loop(1),
        [1],
        3,
    ))
    assert_equal(
        result,
        [
            [1],
            [1],
            [1],
        ]
    )

    result = list(simulate(
        {0: [0, 0]},
        [1],
        3,
    ))
    assert_equal(
        result,
        [
            [1],
            [1],
            [1],
        ]
    )

    result = list(simulate(
        loop(2),
        [1, 0],
        4,
    ))
    assert_equal(
        result,
        [
            [1, 0],
            [0, 1],
            [1, 0],
            [0, 1],
        ]
    )

    result = list(simulate(
        loop(3),
        [1, 0, 0],
        3,
    ))
    assert_equal(
        result,
        [
            [1, 0, 0],
            [0, 1, 0],
            [0, 0, 1],
        ]
    )

    result = list(simulate(
        loop(3),
        [1, 1, 0],
        3,
    ))
    assert_equal(
        result,
        [
            [1, 1, 0],
            [0, 1, 1],
            [1, 0, 1],
        ]
    )

    graph = {
        0: [1],
        1: [2],
        2: [0, 1],
    }
    result = list(simulate(
        graph,
        [1, 0, 0],
        8,
    ))
    assert_equal(
        result,
        [
            [1, 0, 0],
            [0, 1, 0],
            [0, 0, 1],
            [1, 1, 0],
            [0, 1, 1],
            [1, 1, 1],
            [1, 1, 1],
            [1, 1, 1],
        ]
    )