"""Test suite for lesson 2."""importunittestfromunittest.mockimportMagicMockfromlessons.lesson_2.mainimport(LED)
[docs]classLesson2TestCase(unittest.TestCase):""" Testing led functionality by mocking Pin methods """
[docs]deftest_led_on(self):""" Mock the led.value() function and return 1 always :return: """LED.value=MagicMock()LED.on()LED.value.return_value=1self.assertEqual(LED.value(),1)
[docs]deftest_led_off(self):""" Mock the led.value() function and return 0 always :return: """LED.value=MagicMock()LED.off()LED.value.return_value=0self.assertEqual(LED.value(),0)