def test_fitness(self):
terrain_map = [
[0.1, 0.1, 0.1, 0.1, 0.1],
[0.1, 1, 1, 0.1, 0.1],
[1, 1, 1, 0.1, 1],
[1, 1, 0.1, 0.1, 1],
[1, 1, 0.1, 0.1, 0.1]
]
start = Point(3, 4)
finish = Point(0, 1)
model = Model(terrain_map, n_turns=3, start=start, finish=finish,
weight_segment=1, weight_turn=0)
turns = [Point(*x) for x in [(3, 2), (3, 0), (1, 0)]]
fitness = model.get_fitness(turns)
# self.assertAlmostEqual(fitness, 6 + math.sqrt(2))
self.assertAlmostEqual(fitness, 4.4) #TODO wyliczyc co tu powinno
