pyqtgraph/tests/graphicsItems/PlotItem/test_PlotItem.py

88 lines
2.7 KiB
Python
Raw Normal View History

# -*- 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()
_ = 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