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."