def test_command_rotate_column():
with patch.object(Display, 'rotate_column') as mock_method:
display = Display(rows=3, columns=7)
display.command('rotate column x=19 by 2')
mock_method.assert_called_once_with(19, 2)
def test_command_rect():
with patch.object(Display, 'rect') as mock_method:
display = Display(rows=3, columns=7)
display.command('rect 1x2')
mock_method.assert_called_once_with(1, 2)
def test_command_rotate_row():
with patch.object(Display, 'rotate_row') as mock_method:
display = Display(rows=3, columns=7)
display.command('rotate row y=1 by 20')
mock_method.assert_called_once_with(1, 20)
from day08.display import Display
with open('input.txt') as f:
lines = f.readlines()
display = Display(columns=50, rows=6)
for line in lines:
display.command(line)
print(display.lit_pixels())
print(str(display))