pyqtgraph/tests/graphicsItems/test_PlotCurveItem.py

33 lines
1.2 KiB
Python
Raw Normal View History

2016-02-12 11:03:52 +00:00
import numpy as np
import pyqtgraph as pg
from tests.image_testing import assertImageApproved
2016-02-12 11:03:52 +00:00
def test_PlotCurveItem():
p = pg.GraphicsLayoutWidget()
2016-02-12 11:03:52 +00:00
p.resize(200, 150)
p.ci.setContentsMargins(4, 4, 4, 4) # default margins vary by platform
p.show()
v = p.addViewBox()
2016-02-12 11:03:52 +00:00
data = np.array([1,4,2,3,np.inf,5,7,6,-np.inf,8,10,9,np.nan,-1,-2,0])
c = pg.PlotCurveItem(data)
v.addItem(c)
v.autoRange()
2016-02-12 11:03:52 +00:00
2016-02-14 21:29:20 +00:00
# Check auto-range works. Some platform differences may be expected..
checkRange = np.array([[-1.1457564053237301, 16.145756405323731], [-3.076811473165955, 11.076811473165955]])
assert np.allclose(v.viewRange(), checkRange)
2016-02-14 21:29:20 +00:00
2016-02-12 11:03:52 +00:00
assertImageApproved(p, 'plotcurveitem/connectall', "Plot curve with all points connected.")
c.setData(data, connect='pairs')
assertImageApproved(p, 'plotcurveitem/connectpairs', "Plot curve with pairs connected.")
c.setData(data, connect='finite')
assertImageApproved(p, 'plotcurveitem/connectfinite', "Plot curve with finite points connected.")
c.setData(data, connect=np.array([1,1,1,0,1,1,0,0,1,0,0,0,1,1,0,0]))
assertImageApproved(p, 'plotcurveitem/connectarray', "Plot curve with connection array.")
p.close()