def setUp(self): self.device = PseudoSwitch()
class TestPseudoSwitch(unittest.TestCase): def setUp(self): self.device = PseudoSwitch() def test_device_type(self): self.assertEqual(self.device.getDeviceType(), "Switch") def test_power_commands(self): """Silly tests exercising PseudoSwitch logic""" self.assertEqual( self.device.getPower(), 'Off') self.device.setPowerOff() self.assertEqual( self.device.getPower(), 'Off') self.assertEqual( self.device.isPowerOn(), False) self.assertEqual( self.device.isPowerOff(), True) self.device.setPowerOn() self.assertEqual( self.device.getPower(), 'On') self.assertEqual( self.device.isPowerOn(), True) self.assertEqual( self.device.isPowerOff(), False)