def testRemovePointOnlyOnePointShouldBeOnlyPoint(self): frontier = GreedyFrontier() frontier.set_goal(Point(1, 2)) frontier.add_point(Point(1, 1)) self.assertEqual([Point(1, 1)], frontier._points) self.assertEqual(1, len(frontier._points)) point = frontier.remove_point() self.assertEqual(Point(1, 1), point) self.assertEqual([], frontier._points) self.assertEqual(0, len(frontier._points))
def testRemovePointMultiplePointsShouldBeSecondPoint(self): frontier = GreedyFrontier() frontier.set_goal(Point(2, 3)) frontier.add_point(Point(1, 1)) frontier.add_point(Point(3, 3)) frontier.add_point(Point(5, 5)) self.assertEqual([Point(1, 1), Point(3, 3), Point(5, 5)], frontier._points) self.assertEqual(3, len(frontier._points)) point = frontier.remove_point() self.assertEqual(Point(3, 3), point) self.assertEqual([Point(1, 1), Point(5, 5)], frontier._points) self.assertEqual(2, len(frontier._points))