def testGetImage(self):
"""PlotWidget.getImage and PlotWidget.getActiveImage tests"""
plot = PlotWidget(backend='none')
# No image
image = plot.getImage()
self.assertIsNone(image)
plot.addImage(((0, 1), (2, 3)), legend='image 0')
plot.addImage(((0, 1), (2, 3)), legend='image 1')
# Active image
active = plot.getActiveImage()
self.assertEqual(active.getLegend(), 'image 0')
image = plot.getImage()
self.assertEqual(image.getLegend(), 'image 0')
# No active image
plot.addImage(((0, 1), (2, 3)), legend='image 2')
plot.setActiveImage(None)
active = plot.getActiveImage()
self.assertIsNone(active)
image = plot.getImage()
self.assertEqual(image.getLegend(), 'image 2')
# Active image
plot.setActiveImage('image 1')
active = plot.getActiveImage()
self.assertEqual(active.getLegend(), 'image 1')
image = plot.getImage()
self.assertEqual(image.getLegend(), 'image 1')
def testGetImage(self):
"""PlotWidget.getImage and PlotWidget.getActiveImage tests"""
plot = PlotWidget(backend='none')
# No image
image = plot.getImage()
self.assertIsNone(image)
plot.addImage(((0, 1), (2, 3)), legend='image 0')
plot.addImage(((0, 1), (2, 3)), legend='image 1')
# Active image
active = plot.getActiveImage()
self.assertEqual(active.getLegend(), 'image 0')
image = plot.getImage()
self.assertEqual(image.getLegend(), 'image 0')
# No active image
plot.addImage(((0, 1), (2, 3)), legend='image 2')
plot.setActiveImage(None)
active = plot.getActiveImage()
self.assertIsNone(active)
image = plot.getImage()
self.assertEqual(image.getLegend(), 'image 2')
# Active image
plot.setActiveImage('image 1')
active = plot.getActiveImage()
self.assertEqual(active.getLegend(), 'image 1')
image = plot.getImage()
self.assertEqual(image.getLegend(), 'image 1')
def testGetImageOldApi(self):
"""PlotWidget.getImage and PlotWidget.getActiveImage old API tests"""
plot = PlotWidget(backend='none')
# No image
image = plot.getImage()
self.assertIsNone(image)
image = numpy.arange(10).astype(numpy.float32)
image.shape = 5, 2
plot.addImage(image, legend='image 0', info=["Hi!"])
# Active image
data, legend, info, something, params = plot.getActiveImage()
self.assertEqual(legend, 'image 0')
self.assertEqual(info, ["Hi!"])
self.assertTrue(numpy.allclose(data, image), "image 0 data not correct")
def testGetImageOldApi(self):
"""PlotWidget.getImage and PlotWidget.getActiveImage old API tests"""
plot = PlotWidget(backend='none')
# No image
image = plot.getImage()
self.assertIsNone(image)
image = numpy.arange(10).astype(numpy.float32)
image.shape = 5, 2
plot.addImage(image, legend='image 0', info=["Hi!"])
# Active image
data, legend, info, something, params = plot.getActiveImage()
self.assertEqual(legend, 'image 0')
self.assertEqual(info, ["Hi!"])
self.assertTrue(numpy.allclose(data, image), "image 0 data not correct")
