some tests for my hanoi implementation

hanoi_tests.py

from itertools import chain

from my_hanoi import (
    pad_disk,
    format_row,
    buffer_peg,
    pad_left_reverse,
    solve_tower,
)


def test_solve():
    steps = tuple(solve_tower(2))
    assert steps == (
        ((2, 1), (), ()),
        ((2,), (), (1,)),
        ((), (2,), (1,)),
        ((), (2, 1), ()),
    )


def test_pad_disk():
    assert tuple(pad_disk(1, num_disks=4)) == (1, 1, 1, 0, 0, 1, 1, 1)


def test_row():
    row = [2, 0, 1]
    num_disks = 4
    assert tuple(format_row(row, num_disks)) == tuple(chain(
        buffer_peg(),
        (1, 1, 0, 0, 0, 0, 1, 1,),
        buffer_peg(),
        (1, 1, 1, 1, 1, 1, 1, 1,),
        buffer_peg(),
        (1, 1, 1, 0, 0, 1, 1, 1,),
        buffer_peg(),
    ))


def test_left_pad():
    assert tuple(tuple(pad_left_reverse([2, 1], 5, 0))) == (0, 0, 0, 1, 2)