Mercurial > hg > nsaunier > traffic-intelligence
diff python/extrapolation.py @ 255:13ec22bec5d4
corrected typos and bugs and added a test
| author | Nicolas Saunier <nicolas.saunier@polymtl.ca> |
|---|---|
| date | Mon, 23 Jul 2012 23:07:19 -0400 |
| parents | 59f547aebaac |
| children | dc1faa7287bd |
line wrap: on
line diff
--- a/python/extrapolation.py Mon Jul 23 12:35:45 2012 -0400 +++ b/python/extrapolation.py Mon Jul 23 23:07:19 2012 -0400 @@ -26,7 +26,8 @@ def predictPosition(self, nTimeSteps): if nTimeSteps > 0 and not nTimeSteps in self.predictedPositions.keys(): - self.predictedPositions[nTimeSteps], self.predictedspeedOrientations[nTimeSteps] = predictPosition(self.predictedPositions[nTimeSteps-1], self.predictedspeedOrientations[nTimeSteps-1], self.control, maxSpeed) + self.predictPosition(nTimeSteps-1) + self.predictedPositions[nTimeSteps], self.predictedSpeedOrientations[nTimeSteps] = moving.predictPosition(self.predictedPositions[nTimeSteps-1], self.predictedSpeedOrientations[nTimeSteps-1], self.control, self.maxSpeed) return self.predictedPositions[nTimeSteps] class ExtrapolatedTrajectoryAdaptation(ExtrapolatedTrajectoryConstant): @@ -204,3 +205,12 @@ return POC +if __name__ == "__main__": + import doctest + import unittest + suite = doctest.DocFileSuite('tests/extrapolation.txt') + #suite = doctest.DocTestSuite() + unittest.TextTestRunner().run(suite) + #doctest.testmod() + #doctest.testfile("example.txt") +