def test_Test4Over3(self):
acq = btk.btkAcquisition()
acq.Init(0, 25, 3, 1)
data = numpy.empty([25, 1])
data.fill(5.0)
acq.GetAnalog(0).SetValues(data)
data.fill(4.0)
acq.GetAnalog(1).SetValues(data)
data.fill(3.0)
acq.GetAnalog(2).SetValues(data)
acq2 = btk.btkAcquisition()
acq2.Init(0, 25, 4, 1)
data.fill(1.0)
acq2.GetAnalog(0).SetValues(data)
data.fill(2.0)
acq2.GetAnalog(1).SetValues(data)
data.fill(3.0)
acq2.GetAnalog(2).SetValues(data)
data.fill(4.0)
acq2.GetAnalog(3).SetValues(data)
remover = btk.btkAnalogOffsetRemover()
remover.SetRawInput(acq)
remover.SetOffsetInput(acq2)
remover.Update()
output = remover.GetOutput()
self.assertEqual(output.GetAnalogNumber(), 3)
self.assertEqual(output.GetAnalog(0).GetValues().sum() / 25.0, 4.0)
self.assertEqual(output.GetAnalog(1).GetValues().sum() / 25.0, 2.0)
self.assertEqual(output.GetAnalog(2).GetValues().sum() / 25.0, 0.0)
def test_Test4Over3(self): acq = btk.btkAcquisition() acq.Init(0,25,3,1) data = numpy.empty([25,1]); data.fill(5.0) acq.GetAnalog(0).SetValues(data) data.fill(4.0) acq.GetAnalog(1).SetValues(data) data.fill(3.0) acq.GetAnalog(2).SetValues(data) acq2 = btk.btkAcquisition() acq2.Init(0,25,4,1) data.fill(1.0) acq2.GetAnalog(0).SetValues(data) data.fill(2.0) acq2.GetAnalog(1).SetValues(data) data.fill(3.0) acq2.GetAnalog(2).SetValues(data) data.fill(4.0) acq2.GetAnalog(3).SetValues(data) remover = btk.btkAnalogOffsetRemover() remover.SetRawInput(acq) remover.SetOffsetInput(acq2) remover.Update() output = remover.GetOutput() self.assertEqual(output.GetAnalogNumber(), 3) self.assertEqual(output.GetAnalog(0).GetValues().sum() / 25.0, 4.0) self.assertEqual(output.GetAnalog(1).GetValues().sum() / 25.0, 2.0) self.assertEqual(output.GetAnalog(2).GetValues().sum() / 25.0, 0.0)
def test_Test0(self):
acq = btk.btkAcquisition()
acq.Init(0, 25, 1, 1)
data = numpy.empty([25, 1])
data.fill(5.0)
acq.GetAnalog(0).SetValues(data)
remover = btk.btkAnalogOffsetRemover()
remover.SetRawInput(acq)
remover.SetOffsetInput(acq)
remover.Update()
output = remover.GetOutput()
self.assertEqual(output.GetAnalogNumber(), 1)
self.assertEqual(output.GetAnalog(0).GetValues().sum(), 0.0)
def test_Test0(self): acq = btk.btkAcquisition() acq.Init(0,25,1,1) data = numpy.empty([25,1]); data.fill(5.0) acq.GetAnalog(0).SetValues(data) remover = btk.btkAnalogOffsetRemover() remover.SetRawInput(acq) remover.SetOffsetInput(acq) remover.Update() output = remover.GetOutput() self.assertEqual(output.GetAnalogNumber(), 1) self.assertEqual(output.GetAnalog(0).GetValues().sum(), 0.0)
def test_TestNoCommonLabel(self):
acq = btk.btkAcquisition()
acq.Init(0, 25, 3, 1)
acq.GetAnalog(0).SetLabel("FOO")
acq.GetAnalog(1).SetLabel("BAR")
acq.GetAnalog(2).SetLabel("FOOBAR")
acq2 = btk.btkAcquisition()
acq2.Init(0, 25, 3, 1)
data = numpy.empty([25, 1])
data.fill(1.0)
acq2.GetAnalog(0).SetValues(data)
data.fill(2.0)
acq2.GetAnalog(1).SetValues(data)
data.fill(3.0)
acq2.GetAnalog(2).SetValues(data)
remover = btk.btkAnalogOffsetRemover()
remover.SetRawInput(acq)
remover.SetOffsetInput(acq2)
remover.Update()
output = remover.GetOutput()
self.assertEqual(output.GetAnalogNumber(), 3)
self.assertEqual(output.GetAnalog(0).GetValues().sum(), 0.0)
self.assertEqual(output.GetAnalog(1).GetValues().sum(), 0.0)
self.assertEqual(output.GetAnalog(2).GetValues().sum(), 0.0)
def test_TestNoCommonLabel(self):
acq = btk.btkAcquisition()
acq.Init(0,25,3,1)
acq.GetAnalog(0).SetLabel("FOO")
acq.GetAnalog(1).SetLabel("BAR")
acq.GetAnalog(2).SetLabel("FOOBAR")
acq2 = btk.btkAcquisition()
acq2.Init(0,25,3,1)
data = numpy.empty([25,1]);
data.fill(1.0)
acq2.GetAnalog(0).SetValues(data)
data.fill(2.0)
acq2.GetAnalog(1).SetValues(data)
data.fill(3.0)
acq2.GetAnalog(2).SetValues(data)
remover = btk.btkAnalogOffsetRemover()
remover.SetRawInput(acq)
remover.SetOffsetInput(acq2)
remover.Update()
output = remover.GetOutput()
self.assertEqual(output.GetAnalogNumber(), 3)
self.assertEqual(output.GetAnalog(0).GetValues().sum(), 0.0)
self.assertEqual(output.GetAnalog(1).GetValues().sum(), 0.0)
self.assertEqual(output.GetAnalog(2).GetValues().sum(), 0.0)
