def test_smiley_shown_if_button_b_pressed(self):
script.update(0)
self.assertEqual(0, mbit.display.show.call_count)
mbit.button_b.was_pressed.return_value = True
timer = script.update(0)
mbit.display.show.assert_called_once_with(mbit.Image.SAD)
self.assertEqual(script.TICKS_TO_DISPLAY, timer)
def test_smiley_shown_if_button_a_pressed(self):
script.update(0)
self.assertEqual(0, mbit.display.show.call_count)
mbit.button_a.was_pressed.return_value = True
timer = script.update(0)
mbit.display.show.assert_called_once_with(mbit.Image.HAPPY)
self.assertEqual(script.SHOW_SMILEY_TICKS, timer)
def main():
args = sys.argv
if len(args) == 2:
if (args[1] == 'gtcl'):
script.gtcl()
elif (args[1] == 'go'):
script.go()
elif (args[1] == 'setup'):
script.setup()
elif (args[1] == 'update'):
script.update()
elif (args[1] == 'confirm'):
script.confirm()
else:
print(config.COMMANDLINE_MESSAGE_TEMPLATE)
else:
print(config.COMMANDLINE_MESSAGE_TEMPLATE)
def test_ticks_count_down(self):
timer = script.SHOW_SMILEY_TICKS
for i in range(1, script.SHOW_SMILEY_TICKS):
timer = script.update(timer)
self.assertEqual(0, mbit.display.show.call_count)
self.assertEqual(0, mbit.display.clear.call_count)
self.assertEqual(script.SHOW_SMILEY_TICKS - i, timer)
# One more tick and display is cleared.
timer = script.update(timer)
self.assertEqual(0, timer)
self.assertEqual(1, mbit.display.clear.call_count)
# Some more ticks, but nothing happens.
timer = script.update(timer)
timer = script.update(timer)
timer = script.update(timer)
self.assertEqual(0, timer)
self.assertEqual(0, mbit.display.show.call_count)
def test_no_button_press_no_image_displayed(self):
for _ in range(0, 10):
script.update(0)
self.assertEqual(0, mbit.display.show.call_count)