def create_new_network(network, box_size, random):
number_of_square = box_size * box_size
for antenna in range(number_of_square):
if can_antenna_be_turned_on(antenna, network, box_size) and _flip_coin(random):
network = turn_on_antenna(antenna, network)
else:
network = turn_off_antenna(antenna, network)
return network
def test_turn_off_antenna(self):
network = create_network([0,2,3,7,8,9])
assert turn_off_antenna(2, network) == 905
assert turn_off_antenna(7, network) == 781
assert turn_off_antenna(10, network) == 909
assert turn_off_antenna(4, network) == 909
