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

infiniteline API testing

Browse Source
This commit is contained in:
Luke Campagnola 2016-02-05 00:12:21 -08:00
parent c1de24e825
commit ad8e169160
1 changed files with 50 additions and 2 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

52
pyqtgraph/graphicsItems/tests/test_InfiniteLine.py
View File

@ -1,10 +1,49 @@
import pyqtgraph as pg
from pyqtgraph.Qt import QtTest, QtGui, QtCore
from pyqtgraph.Qt import QtGui, QtCore, QtTest
from pyqtgraph.tests import mouseDrag, mouseMove
pg.mkQApp()
qWait = QtTest.QTest.qWait
def test_InfiniteLine():
plt = pg.plot()
plt.setXRange(-10, 10)
plt.setYRange(-10, 10)
vline = plt.addLine(x=1)
plt.resize(600, 600)
QtGui.QApplication.processEvents()
QtTest.QTest.qWaitForWindowShown(plt)
QtTest.QTest.qWait(100)
assert vline.angle == 90
br = vline.mapToView(QtGui.QPolygonF(vline.boundingRect()))
print(vline.boundingRect())
print(list(QtGui.QPolygonF(vline.boundingRect())))
print(list(br))
assert br.containsPoint(pg.Point(1, 5), QtCore.Qt.OddEvenFill)
assert not br.containsPoint(pg.Point(5, 0), QtCore.Qt.OddEvenFill)
hline = plt.addLine(y=0)
assert hline.angle == 0
assert hline.boundingRect().contains(pg.Point(5, 0))
assert not hline.boundingRect().contains(pg.Point(0, 5))
vline.setValue(2)
assert vline.value() == 2
vline.setPos(pg.Point(4, -5))
assert vline.value() == 4
oline = pg.InfiniteLine(angle=30)
plt.addItem(oline)
oline.setPos(pg.Point(1, -1))
assert oline.angle == 30
assert oline.pos() == pg.Point(1, -1)
assert oline.value() == [1, -1]
br = oline.mapToScene(oline.boundingRect())
pos = oline.mapToScene(pg.Point(2, 0))
assert br.containsPoint(pos, QtCore.Qt.OddEvenFill)
px = oline.pixelVectors(pg.Point(1, 0))[0]
assert br.containsPoint(pos + 4 * px, QtCore.Qt.OddEvenFill)
assert not br.containsPoint(pos + 7 * px, QtCore.Qt.OddEvenFill)
def test_mouseInteraction():
plt = pg.plot()
@ -12,6 +51,7 @@ def test_mouseInteraction():
vline = plt.addLine(x=0, movable=True)
plt.addItem(vline)
hline = plt.addLine(y=0, movable=True)
hline2 = plt.addLine(y=-1, movable=False)
plt.setXRange(-10, 10)
plt.setYRange(-10, 10)
@ -42,6 +82,14 @@ def test_mouseInteraction():
px = hline.pixelLength(pg.Point(1, 0), ortho=True)
assert abs(hline.value() - plt.plotItem.vb.mapSceneToView(pos2).y()) <= px
# test non-interactive line
pos = plt.plotItem.vb.mapViewToScene(pg.Point(5,-1)).toPoint()
pos2 = pos - QtCore.QPoint(50, 50)
mouseMove(plt, pos)
assert hline2.mouseHovering == False
mouseDrag(plt, pos, pos2, QtCore.Qt.LeftButton)
assert hline2.value() == -1
if __name__ == '__main__':
test_mouseInteraction()