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