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_on_antenna(self): self.assertEqual(turn_on_antenna(5, 111), 111) self.assertEqual(turn_on_antenna(4, 111), 111+2**4)