745b04baa5
* Update PlotItem.py make update decimate to not unhide curves when items added/removed, while preserving the Max Traces well behaviour * Update PlotItem.py fix typo * Update PlotItem.py fix: typo with self as argument * give better name for the function which handles MaxTraces checkstate change rename it to _handle_max_traces_toggle * add doc string to updateDecimation * add test for PlotItem for external curve visibility control check if hidden curve would stay hidden when adding or removing other items. * remove additional empty line between methods
100 lines
3.0 KiB
Python
100 lines
3.0 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()
|
|
|
|
_ = 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_preserve_external_visibility_control():
|
|
item = pg.PlotItem()
|
|
curve1 = pg.PlotDataItem(np.random.normal(size=10))
|
|
curve2 = pg.PlotDataItem(np.random.normal(size=10))
|
|
item.addItem(curve1)
|
|
curve1.hide()
|
|
item.addItem(curve2)
|
|
assert not curve1.isVisible()
|
|
item.removeItem(curve2)
|
|
assert not curve1.isVisible()
|
|
|
|
|
|
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
|