def test_three_close(self): rt = [PointXyz(0, 0, 0), PointXyz(10, 0, 0), PointXyz(20, 0, 0)] routeutils.remove_close_points(rt, 100) self.assertEqual(2, len(rt)) self.assertEqual(PointXyz(0, 0, 0), rt[0]) self.assertEqual(PointXyz(20, 0, 0), rt[1])
def test_three_close_end(self): rt = [PointXyz(0, 0, 0), PointXyz(90, 0, 0), PointXyz(100, 0, 0)] routeutils.remove_close_points(rt, 20) self.assertEqual(3, len(rt)) self.assertEqual(PointXyz(0, 0, 0), rt[0]) self.assertEqual(PointXyz(90, 0, 0), rt[1]) self.assertEqual(PointXyz(100, 0, 0), rt[2])
def test_two_points_between_first_kept(self): rt = [ PointXyz(0, 0, 0), PointXyz(20, 0, 0), PointXyz(22, 0, 0), PointXyz(30, 0, 0) ] routeutils.remove_close_points(rt, 25) self.assertEqual(3, len(rt)) self.assertEqual(PointXyz(0, 0, 0), rt[0]) self.assertEqual(PointXyz(20, 0, 0), rt[1]) self.assertEqual(PointXyz(30, 0, 0), rt[2])
def test_many_mid(self): rt = [ PointXyz(0, 0, 0), PointXyz(30, 0, 0), PointXyz(40, 0, 0), PointXyz(50, 0, 0), PointXyz(100, 0, 0) ] routeutils.remove_close_points(rt, 25) self.assertEqual(4, len(rt)) self.assertEqual(PointXyz(0, 0, 0), rt[0]) self.assertEqual(PointXyz(30, 0, 0), rt[1]) self.assertEqual(PointXyz(50, 0, 0), rt[2]) self.assertEqual(PointXyz(100, 0, 0), rt[3])
def test_empty(self): rt = [] routeutils.remove_close_points(rt, 10) self.assertEqual(0, len(rt))
def test_one(self): rt = [PointXyz(0, 0, 0)] routeutils.remove_close_points(rt, 10) self.assertEqual(1, len(rt)) self.assertEqual(PointXyz(0, 0, 0), rt[0])