def testGetAverage(self):
f = filter.MovingAverageFilter(5, default=0.0)
self.assertEqual(0, f.getAverage())
inVals = [5, 5, 5, 5, 5]
avgVals = [1, 2, 3, 4, 5]
for i, a in zip(inVals, avgVals):
f.append(i)
self.assertAlmostEqual(a, f.getAverage(), 5)
f = filter.MovingAverageFilter(5)
self.assertEqual(0, f.getAverage())
inVals = [5, 5, 5, 5, 5]
avgVals = [5, 5, 5, 5, 5]
for i, a in zip(inVals, avgVals):
f.append(i)
self.assertAlmostEqual(a, f.getAverage(), 5)
f = filter.MovingAverageFilter(5)
self.assertEqual(0, f.getAverage())
inVals = [1, 2, 3, 4, 5]
avgVals = [1, 1.5, 2, 2.5, 3]
for i, a in zip(inVals, avgVals):
f.append(i)
self.assertAlmostEqual(a, f.getAverage(), 5)
def testInit(self):
# Test that setting a default works properly
f = filter.MovingAverageFilter(5, default=0.0)
self.assertEqual([0, 0, 0, 0, 0], f.get())
# Test that setting no default value works properly
f = filter.MovingAverageFilter(5)
self.assertEqual([], f.get())
def enter(self):
# Read in filter size
filterSize = self._config.get('filterSize', 1)
# Create filters
self._xFilter = filter.MovingAverageFilter(filterSize)
self._yFilter = filter.MovingAverageFilter(filterSize)
self._rangeFilter = filter.MovingAverageFilter(filterSize)
self._alignFilter = filter.MovingAverageFilter(filterSize)
# Zero values
self._filterdX = self._xFilter.getAverage()
self._filterdY = self._yFilter.getAverage()
self._filterdRange = self._rangeFilter.getAverage()
self._filterdAlign = self._alignFilter.getAverage()
def testSingleLength(self):
f = filter.MovingAverageFilter(1)
inVals = [3, 1, 8, 10, 2]
avgVals = inVals
for i, a in zip(inVals, avgVals):
f.append(i)
self.assertAlmostEqual(a, f.getAverage(), 5)
def enter(self):
# Read in filter size
filterSize = self._config.get('filterSize', 1)
self._haveLeftPipe = False
self._haveRightPipe = False
# Create filters
self._xFilter = filter.MovingAverageFilter(filterSize)
self._yFilter = filter.MovingAverageFilter(filterSize)
self._rangeFilter = filter.MovingAverageFilter(filterSize)
self._alignFilter = filter.MovingAverageFilter(filterSize)
# Zero values
self._filterdX = self._xFilter.getAverage()
self._filterdY = self._yFilter.getAverage()
self._filterdRange = self._rangeFilter.getAverage()
self._filterdAlign = self._alignFilter.getAverage()
self._filterdAngle = None
