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)