def setUp(self): self.collisionDetector = CollisionDetector(1000, 600, 90, [Obstacle((400, 400))]) self.graph = MagicMock() self.bottomPathGenerator = BottomPathGenerator(90, 600, self.graph, self.collisionDetector)
def __init__(self, obstaclesList, mapSizeX, mapSizeY): self.MAP_SIZE_X = mapSizeX self.MAP_SIZE_Y = mapSizeY self.graph = Graph(obstaclesList, self.SAFE_MARGIN) self.collisionDetector = CollisionDetector(self.MAP_SIZE_X, self.MAP_SIZE_Y, self.SAFE_MARGIN, obstaclesList) self.endNodeGenerator = EndNodeGenerator(self.MAP_SIZE_X, self.MAP_SIZE_Y, self.SAFE_MARGIN, obstaclesList, self.collisionDetector, self.graph) self.bottomPathGenerator = BottomPathGenerator(self.SAFE_MARGIN, mapSizeY, self.graph, self.collisionDetector) self.topPathGenerator = TopPathGenerator(self.SAFE_MARGIN, mapSizeY, self.graph, self.collisionDetector) self.obstaclesList = obstaclesList self.obstaclesList.sort(key=lambda obstacle: obstacle.positionX)