ASCEE/pyqtgraph
5
0
Fork 0
Code Issues Pull Requests Projects Releases Wiki Activity

add test for export to qimage

Browse Source
This commit is contained in:
KIU Shueng Chuan 2021-06-10 03:16:22 +08:00
parent 765f9648cd
commit b3e8f332fb
1 changed files with 15 additions and 0 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

15
tests/exporters/test_image.py
View File

@ -1,6 +1,9 @@
# -*- coding: utf-8 -*- # -*- coding: utf-8 -*-
import numpy as np
import pyqtgraph as pg import pyqtgraph as pg
from pyqtgraph.Qt import QtGui
from pyqtgraph.exporters import ImageExporter from pyqtgraph.exporters import ImageExporter
import pyqtgraph.functions as fn
app = pg.mkQApp() app = pg.mkQApp()
@ -11,3 +14,15 @@ def test_ImageExporter_filename_dialog():
p = pg.plot() p = pg.plot()
exp = ImageExporter(p.getPlotItem()) exp = ImageExporter(p.getPlotItem())
exp.export() exp.export()
def test_ImageExporter_toBytes():
p = pg.plot()
p.hideAxis('bottom')
p.hideAxis('left')
exp = ImageExporter(p.getPlotItem())
qimg = exp.export(toBytes=True)
qimg = qimg.convertToFormat(QtGui.QImage.Format.Format_RGBA8888)
data = fn.qimage_to_ndarray(qimg)
black = (0, 0, 0, 255)
assert np.all(data == black), "Exported image should be entirely black."