14 lines
354 B
Python
14 lines
354 B
Python
from src.codewars.kata_the_lamp import Lamp
|
|
|
|
|
|
def test_lamp():
|
|
my_lamp = Lamp("Blue")
|
|
|
|
assert my_lamp.color == "Blue"
|
|
assert not my_lamp.on
|
|
assert my_lamp.state() == "The lamp is off."
|
|
my_lamp.toggle_switch()
|
|
assert my_lamp.state() == "The lamp is on."
|
|
my_lamp.toggle_switch()
|
|
assert my_lamp.state() == "The lamp is off."
|