def test_lamp_turn_on(self): b = Button() b.set_condition(True) self.assertEqual("on", b.lamp.state())