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()