def test_given_one_cell_panel_then_it_can_be_ON_light():
panel = Panel(1)
start_coordinate = Coordinate(0, 0)
end_coordinate = Coordinate(0, 0)
panel.turn_on(start_coordinate, end_coordinate)
assert panel.coordinate(x=0, y=0).status == Light.ON
def test_given_four_cell_panel_and_power_on_all_then_four_lights_on():
panel = Panel(4)
start_coordinate = Coordinate(0, 0)
end_coordinate = Coordinate(1, 1)
panel.turn_on(start_coordinate, end_coordinate)
assert panel.lights_on() == 4
def test_given_four_cell_panel_then_power_on_and_then_off():
panel = Panel(4)
panel.turn_on(Coordinate(0, 0), Coordinate(1, 0))
panel.turn_off(Coordinate(0, 0), Coordinate(1, 1))
assert panel.coordinate(0, 0).status == Light.OFF
assert panel.coordinate(0, 1).status == Light.OFF
assert panel.coordinate(1, 0).status == Light.OFF
assert panel.coordinate(1, 1).status == Light.OFF
def test_given_four_cell_panel_then_power_on_first_row():
panel = Panel(4)
start_coordinate = Coordinate(0, 0)
end_coordinate = Coordinate(1, 0)
panel.turn_on(start_coordinate, end_coordinate)
assert panel.coordinate(1, 1).status == Light.OFF
assert panel.coordinate(0, 1).status == Light.OFF
assert panel.coordinate(1, 0).status == Light.ON
assert panel.coordinate(0, 0).status == Light.ON
def test_given_a_coordinate_it_has_x_and_y():
x = 0
y = 1
coord = Coordinate(x, y)
assert coord.x == x
assert coord.y == y