88 lines
2.7 KiB
Python
88 lines
2.7 KiB
Python
# -*- coding: utf-8 -*-
|
|
import numpy as np
|
|
import pytest
|
|
|
|
import pyqtgraph as pg
|
|
|
|
app = pg.mkQApp()
|
|
|
|
|
|
@pytest.mark.parametrize('orientation', ['left', 'right', 'top', 'bottom'])
|
|
def test_PlotItem_shared_axis_items(orientation):
|
|
"""Adding an AxisItem to multiple plots raises RuntimeError"""
|
|
ax1 = pg.AxisItem(orientation)
|
|
ax2 = pg.AxisItem(orientation)
|
|
|
|
layout = pg.GraphicsLayoutWidget()
|
|
|
|
pi1 = layout.addPlot(axisItems={orientation: ax1})
|
|
|
|
pi2 = layout.addPlot()
|
|
# left or bottom replaces, right or top adds new
|
|
pi2.setAxisItems({orientation: ax2})
|
|
|
|
with pytest.raises(RuntimeError):
|
|
pi2.setAxisItems({orientation: ax1})
|
|
|
|
|
|
def test_PlotItem_maxTraces():
|
|
item = pg.PlotItem()
|
|
|
|
curve1 = pg.PlotDataItem(np.random.normal(size=10))
|
|
item.addItem(curve1)
|
|
assert curve1.isVisible(), "curve1 should be visible"
|
|
|
|
item.ctrl.maxTracesCheck.setChecked(True)
|
|
item.ctrl.maxTracesSpin.setValue(0)
|
|
assert not curve1.isVisible(), "curve1 should not be visible"
|
|
|
|
item.ctrl.maxTracesCheck.setChecked(False)
|
|
assert curve1.isVisible(), "curve1 should be visible"
|
|
|
|
curve2 = pg.PlotDataItem(np.random.normal(size=10))
|
|
item.addItem(curve2)
|
|
assert curve2.isVisible(), "curve2 should be visible"
|
|
|
|
item.ctrl.maxTracesCheck.setChecked(True)
|
|
item.ctrl.maxTracesSpin.setValue(1)
|
|
assert curve2.isVisible(), "curve2 should be visible"
|
|
assert not curve1.isVisible(), "curve1 should not be visible"
|
|
assert curve1 in item.curves, "curve1 should be in the item's curves"
|
|
|
|
item.ctrl.forgetTracesCheck.setChecked(True)
|
|
assert curve2 in item.curves, "curve2 should be in the item's curves"
|
|
assert curve1 not in item.curves, "curve1 should not be in the item's curves"
|
|
|
|
|
|
def test_plotitem_menu_initialize():
|
|
"""Test the menu initialization of the plotitem"""
|
|
item = pg.PlotItem()
|
|
assert item.menuEnabled() is True
|
|
viewbox = item.vb
|
|
assert viewbox is not None
|
|
assert viewbox.menu is not None
|
|
assert viewbox.menuEnabled() is True
|
|
|
|
item = pg.PlotItem(enableMenu=False)
|
|
assert item.menuEnabled() is False
|
|
viewbox = item.vb
|
|
assert viewbox is not None
|
|
assert viewbox.menu is None
|
|
assert viewbox.menuEnabled() is False
|
|
|
|
viewbox = pg.ViewBox()
|
|
item = pg.PlotItem(viewBox=viewbox, enableMenu=False)
|
|
assert item.menuEnabled() is False
|
|
viewbox = item.vb
|
|
assert viewbox is not None
|
|
assert viewbox.menu is not None
|
|
assert viewbox.menuEnabled() is True
|
|
|
|
viewbox = pg.ViewBox(enableMenu=False)
|
|
item = pg.PlotItem(viewBox=viewbox)
|
|
assert item.menuEnabled() is True
|
|
viewbox = item.vb
|
|
assert viewbox is not None
|
|
assert viewbox.menu is None
|
|
assert viewbox.menuEnabled() is False
|