def testClear(self):
buf = py_metrics.NumpyDeque(maxlen=4, dtype=np.float64)
buf.add(2)
buf.add(3)
buf.clear()
buf.add(5)
self.assertEqual(5, buf.mean())
def testFullLength(self):
buf = py_metrics.NumpyDeque(maxlen=4, dtype=np.float64)
buf.add(2)
buf.add(3)
buf.add(5)
buf.add(6)
self.assertEqual(4, buf.mean())
def testSimple(self):
buf = py_metrics.NumpyDeque(maxlen=10, dtype=np.float64)
buf.add(2)
buf.add(3)
buf.add(5)
buf.add(6)
self.assertEqual(4, buf.mean())
def testUnboundedClear(self):
buf = py_metrics.NumpyDeque(maxlen=np.inf, dtype=np.float64)
for i in range(101):
buf.add(i)
buf.clear()
buf.add(4)
buf.add(6)
self.assertEqual(5, buf.mean())
def testPastMaxLen(self):
buf = py_metrics.NumpyDeque(maxlen=4, dtype=np.float64)
buf.add(2)
buf.add(3)
buf.add(5)
buf.add(6)
buf.add(8)
buf.add(9)
self.assertEqual(7, buf.mean())
def __init__(self, name='PolicyUsageFrequency', buffer_size=10):
super(PolicyUsageFrequency, self).__init__(name)
self._buffer = py_metrics.NumpyDeque(maxlen=buffer_size,
dtype=np.float64)
self.reset()
