def test_segments(self):
terrain_map = [
[0, 0, 0, 0, 0],
[0, 1, 1, 0, 0],
[1, 1, 1, 0, 1],
[1, 1, 0, 0, 1],
[1, 1, 0, 0, 0]
]
start = (3, 4)
finish = (0, 1)
model = Model(terrain_map, n_turns=3, start=start, finish=finish,
weight_segment=1, weight_turn=1)
turns = [(3, 2), (3, 0), (1, 0)]
# turns = [Point(*x) for x in turns]
expected = [((3, 4), (3, 2)),
((3, 2), (3, 0)),
((3, 0), (1, 0)),
((1, 0), (0, 1))]
# expected = [(Point(*x), Point(*y)) for x, y in expected]
self.assertSequenceEqual(model._segments(turns), expected)
