def test_has_antenna(self):
network = create_network([0,2,3,7,8,9])
assert has_antenna(0, network)
assert has_antenna(3, network)
assert has_antenna(9, network)
assert not has_antenna(4, network)
assert not has_antenna(10, network)
def get_random_network(box_size, random, Generator):
antenna_generator = Generator(box_size, random)
antennas = []
antenna, network_prob = antenna_generator.get_random_antenna()
while antenna != -1:
antennas.append(antenna)
antenna, p = antenna_generator.get_random_antenna()
network_prob *= p
return create_network(antennas), network_prob
def test_can_antenna_be_turned_on(self):
# □-□-□-□-■
# □-□-□-□-□
# □-□-■-□-□
# □-□-□-□-■
# □-□-□-□-□
network = create_network([4, 12, 20])
assert can_antenna_be_turned_on(0, network, 5)
assert can_antenna_be_turned_on(1, network, 5)
assert can_antenna_be_turned_on(5, network, 5)
assert can_antenna_be_turned_on(19, network, 5)
assert not can_antenna_be_turned_on(3, network, 5)
assert not can_antenna_be_turned_on(7, network, 5)
assert not can_antenna_be_turned_on(11, network, 5)
assert not can_antenna_be_turned_on(13, network, 5)
assert not can_antenna_be_turned_on(17, network, 5)
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
def test_create_network(self):
assert create_network([]) == 0
assert create_network([0,2,3,7,8,9]) == 909