class SignalWindow(object):
def __init__(self, collectedQueue, windowSize, fields):
self.collectedQueue = collectedQueue
self.windowSize = windowSize
self.fields = fields
self.index = 0
self._createDto()
def _createDto(self):
self.dto = WindowDto(self.windowSize, self.fields)
def _resetWindow(self):
self.index = 0
self._createDto()
def addData(self, data):
'''
expects data like this
{
"X": {
"value": 1,
"quality": 2
},
"F3": {
"value": 3,
"quality": 4
}, ...
}
:param dict data:
'''
#TODO potential bottleneck
self.dto.addData(data)
self.index += 1
if self.isFull():
data = self._doWindowFunction(self.dto)
self.collectedQueue.put(data)
self._resetWindow()
def isFull(self):
return self.index >= self.windowSize
def _doWindowFunction(self, data):
pass
def __repr__(self): # pragma: no cover
return "%s: { windowSize = %d, numValue = %d }" % (self.__class__.__name__, self.windowSize, self.index)
class TestWindowDto(BaseTest):
def setUp(self):
self.winSize = 4
self.header = ["X", "AF3"]
self.dto = WindowDto(self.winSize, self.header)
def _fillDto(self, start=0, count=2):
for i in range(start, count):
self.dto.addData({
"X": {
"value": i,
"quality": i * 2
},
"AF3": {
"value": i * 3,
"quality": i * 4
}
})
def test_init(self):
self._fillDto(count=self.winSize)
def test_len(self):
self.assertEqual(len(self.dto), len(self.header))
def test_shape(self):
self.assertEqual(self.dto.shape(), (len(self.dto), len(self.header)))
def test_contains(self):
self.assertTrue("X" in self.dto)
self.assertTrue("AF3" in self.dto)
self.assertFalse("x" in self.dto)
self.assertFalse("Y" in self.dto)
def test_getitem(self):
self._fillDto(count=self.winSize)
self.assertEqual(self.dto["X"]["value"], range(self.winSize))
def test_iter(self):
for key in self.dto:
self.assertTrue(key in self.header)
def test_eq(self):
self.assertNotEqual(self.dto, None)
self.assertNotEqual(self.dto, 1)
cp = self.dto.copy()
cp.header = ["AF3", "X"]
self.assertNotEqual(self.dto, cp)
cp = self.dto.copy()
self.assertEqual(cp, self.dto)
self._fillDto(count=self.winSize)
self.assertNotEqual(cp, self.dto)
def test_cp(self):
cp = self.dto.copy()
self.assertEqual(cp, self.dto)
self.assertNotEquals(id(cp), id(self.dto))
def test_getField(self):
self._fillDto(count=self.winSize)
assert_array_equal(self.dto.getValue("X"), range(self.winSize))
assert_array_equal(self.dto.getQuality("X"),
np.array(range(self.winSize)) * 2)
testVal = np.array([17]) * self.winSize
self.dto.addNewField("X", "test", testVal)
assert_array_equal(self.dto.getField("X", "test"), testVal)