def test_set_get_all():
from itertools import chain
a = App()
w = Waffle(a)
w.set_all("red")
# turn 2d list into 1d list
pixels = chain.from_iterable(zip(*w.get_all()))
count = 0
for pixel in pixels:
assert pixel == "red"
count += 1
assert count == w.width * w.height
a.destroy()
def test_reset():
from itertools import chain
a = App()
w = Waffle(a)
w.set_pixel(0, 1, "red")
w.reset()
pixels = chain.from_iterable(zip(*w.get_all()))
count = 0
for pixel in pixels:
assert pixel == w.color
count += 1
assert count == w.width * w.height
a.destroy()
