def test_xymodelchooser_config(qtbot):
w1 = tpg.TaurusPlot()
qtbot.addWidget(w1)
models1 = [
"eval:1*rand(22)",
("eval:linspace(-10,20,10)", "eval:2*rand(10)"),
]
w1.setModel(models1)
# test createConfig
cfg = w1.createConfig()
xymccfg1 = get_sub_config(cfg, "XYmodelchooser")
modelscfg1 = get_sub_config(xymccfg1, "CurveInfo")
assert modelscfg1[0] == (
None,
"eval://localhost/@DefaultEvaluator/1*rand(22)",
"1*rand(22)",
)
assert modelscfg1[1] == (
"eval://localhost/@DefaultEvaluator/linspace(-10,20,10)",
"eval://localhost/@DefaultEvaluator/2*rand(10)",
"2*rand(10)",
)
# test applyConfig
w2 = tpg.TaurusPlot()
qtbot.addWidget(w2)
assert len(w2) == 0
assert len(w2._model_chooser_tool.getModelNames()) == 0
# add a model to w2
w2.setModel("eval:9*rand(11)")
assert len(w2) == 1
assert len(w2._model_chooser_tool.getModelNames()) == 1
# apply config (the previously added tauruscurve of w2 should be removed)
w2.applyConfig(cfg)
assert len(w2) == 2
assert len(w2._model_chooser_tool.getModelNames()) == 2
assert w2._model_chooser_tool._getCurveInfo() == modelscfg1
def test_multiple_setModel(qtbot):
"""
Check that repeated calls to setModel do not duplicate the items
in the plot
"""
w = tpg.TaurusPlot()
qtbot.addWidget(w)
for i in range(5):
w.setModel(["eval:rand(2)"])
assert w.getPlotItem().listDataItems() == [
w[0]
], "Found duplicates after {} calls to setModel".format(i + 1)
def test_y2legend(qtbot):
"""
Check that legend items are not duplicated or lost on changes of Y axis
"""
w = tpg.TaurusPlot()
qtbot.addWidget(w)
vb1 = w.getViewBox()
vb2 = w._y2
plot_item = w.getPlotItem()
legend = plot_item.legend
assert w[:] == []
assert vb1.addedItems == []
assert vb2.addedItems == []
assert legend.items == []
# Add a curve to Y1
w.setModel([(None, "eval:1*rand(11)", "foo")])
c = w[0]
assert w[:] == [c]
assert vb1.addedItems == [c]
assert vb2.addedItems == []
assert len(legend.items) == 1
assert legend.items[0][1].text == "foo"
# move c to Y2
tpg.set_y_axis_for_curve(y2=True,
dataItem=c,
plotItem=plot_item,
y2Axis=w._y2)
assert w[:] == [c]
assert vb1.addedItems == []
assert vb2.addedItems == [c]
assert len(legend.items) == 1
assert legend.items[0][1].text == "foo"
# move c back to Y1
tpg.set_y_axis_for_curve(y2=False,
dataItem=c,
plotItem=plot_item,
y2Axis=w._y2)
assert w[:] == [c]
assert vb1.addedItems == [c]
assert vb2.addedItems == []
assert len(legend.items) == 1
assert legend.items[0][1].text == "foo"
def test_duplicate_curve_titles(qtbot):
"""
Check that curves with same title are changed to have unique titles
"""
w = tpg.TaurusPlot()
qtbot.addWidget(w)
w.setModel(
[
"eval:@foo/a=0+rand(3);a",
"eval:@bar/a=1+rand(3);a",
(None, "eval:2+rand(3)", "a"),
"eval:3+rand(3)",
]
)
assert len(w[:]) == 4
mod_items = w._cprop_tool.getModifiableItems()
assert len(mod_items) == 4, "{}".format(mod_items)
assert list(sorted(mod_items.keys())) == [
"3+rand(3)",
"a",
"a (2)",
"a (3)",
]
def test_plot_model_setting_with_y2(qtbot):
w = tpg.TaurusPlot()
qtbot.addWidget(w)
vb1 = w.getViewBox()
vb2 = w._y2
assert w[:] == []
assert w._model_chooser_tool.getModelNames() == []
assert vb1.addedItems == []
assert vb2.addedItems == []
models1 = [
"eval:1*rand(22)",
("eval:linspace(-10,20,10)", "eval:2*rand(10)"),
]
w.setModel(models1)
c0 = w[0]
c1 = w[1]
# move the first curve to Y2
vb1.removeItem(c0)
vb2.addItem(c0)
assert vb1.addedItems == [c1]
assert vb2.addedItems == [c0]
assert w[:] == [c0, c1]
assert w._model_chooser_tool.getModelNames() == [
c0.getFullModelNames(),
c1.getFullModelNames(),
]
# add a regular data item (non-taurus) to y2
c2 = pg.PlotDataItem(name="pg item", pen="b", fillLevel=0, brush="c")
c2.setData(numpy.linspace(0, 20, 10))
vb2.addItem(c2)
assert vb1.addedItems == [c1]
assert vb2.addedItems == [c0, c2]
assert len(w) == 3
assert w[:] == [c0, c1, c2]
assert w._model_chooser_tool.getModelNames() == [
c0.getFullModelNames(),
c1.getFullModelNames(),
]
# add a taurus data item to y2
c3 = tpg.TaurusPlotDataItem(name="taurus item", pen="r", symbol="o")
c3.setModel('eval:Quantity(rand(16),"m")')
vb2.addItem(c3)
assert vb1.addedItems == [c1]
assert vb2.addedItems == [c0, c2, c3]
assert w[:] == [c0, c1, c2, c3]
assert w._model_chooser_tool.getModelNames() == [
c0.getFullModelNames(),
c1.getFullModelNames(),
c3.getFullModelNames(),
]
# Add existing (c0) model again
w.addModels([models1[0]])
assert vb1.addedItems == [c1]
assert vb2.addedItems == [c2, c0, c3]
assert w[:] == [c2, c0, c1, c3] # there is reordering, non taurus first
assert w._model_chooser_tool.getModelNames() == [
c0.getFullModelNames(),
c1.getFullModelNames(),
c3.getFullModelNames(),
]
# set (not adding!) 1 model which is already on y2
# only non-taurus curve and the set curve remain. Both stay in y2
w.setModel([models1[0]])
assert vb1.addedItems == []
assert vb2.addedItems == [c2, c0]
assert w[:] == [c2, c0] # c0 **is** still the same object!
assert w._model_chooser_tool.getModelNames() == [c0.getFullModelNames()]
# set empty model (not adding!, only the non-taurus curve remains)
w.setModel([])
assert vb1.addedItems == []
assert vb2.addedItems == [c2]
assert w[:] == [c2]
assert w._model_chooser_tool.getModelNames() == []
# remove non-taurus curve
c2.getViewBox().removeItem(c2)
w.getPlotItem().removeItem(c2)
assert vb1.addedItems == []
assert vb2.addedItems == []
assert w[:] == []
assert w._model_chooser_tool.getModelNames() == []
def test_curveproperties_configfile(qtbot, tmp_path):
w1 = tpg.TaurusPlot()
qtbot.addWidget(w1)
w1.setBackground("y")
# add a regular data item (non-taurus) to y1
c0 = pg.PlotDataItem(name="pg item1", pen="m", fillLevel=0, brush="c")
c0.setData(numpy.linspace(0, 20, 10) / 20.0)
w1.addItem(c0)
# add a regular data item (non-taurus) to y2
c1 = pg.PlotDataItem(name="pg item2", pen="y", symbol="d", symbolBrush="r")
c1.setData(1 - numpy.linspace(0, 20, 10) / 20.0)
w1._y2.addItem(c1)
# add a taurus data item to y1
c2 = tpg.TaurusPlotDataItem(name="taurus item1", pen="r", symbol="o")
c2.setModel('eval:Quantity(rand(16),"m")')
w1.addItem(c2)
# add a taurus data item to y2
c3 = tpg.TaurusPlotDataItem(name="taurus item2", pen="y", symbol="s")
c3.setModel('eval:Quantity(rand(20),"km")')
w1._y2.addItem(c3)
# Add 2 more items using setModel
models1 = [
"eval:1*rand(22)",
("eval:linspace(-10,20,10)", "eval:2*rand(10)"),
]
w1.addModels(models1)
c4 = w1[4]
c5 = w1[5]
c4.setPen("g")
c5.setPen(None)
c5.setSymbol("t")
c5.setSymbolSize(7)
c5.setSymbolBrush("r")
# move c5 to y2
w1._y2.addItem(c5)
assert len(w1) == 6
assert w1[:] == [c0, c1, c2, c3, c4, c5]
w1_mod_items = w1._cprop_tool.getModifiableItems()
for c in [c0, c1, c2, c3, c4, c5]:
assert c0 in w1_mod_items.values()
assert w1.getViewBox().addedItems == [c0, c2, c4]
assert w1._y2.addedItems == [c1, c3, c5]
# test saveConfigFile
f = tmp_path / "plot.pck"
with open(str(f), "wb") as ofile:
w1.saveConfigFile(ofile=ofile)
assert f.exists()
assert len(list(tmp_path.iterdir())) == 1
# test loadConfigFile
w2 = tpg.TaurusPlot()
qtbot.addWidget(w2)
with open(str(f), "rb") as ifile:
w2.loadConfigFile(ifile=ifile)
w1_props = w1._cprop_tool._getCurveAppearanceProperties()
w2_props = w2._cprop_tool._getCurveAppearanceProperties()
assert len(w1_props) == 6
assert len(w2_props) == 4
for k, p_aft in w2_props.items():
assert k in w1_props
p_ini = w1_props[k]
conflicts = p_ini.conflictsWith(p_aft, strict=True)
msg = "Mismatch in saved/restored curve properties for '{}':".format(k)
msg += "\n\t Saved: {}".format(p_ini)
msg += "\n\t Restored: {}".format(p_aft)
assert conflicts == [], msg
def test_curveproperties_config(qtbot):
w1 = tpg.TaurusPlot()
qtbot.addWidget(w1)
# add a regular data item (non-taurus) to y1
c0 = pg.PlotDataItem(name="pg item1", pen="m", fillLevel=0, brush="c")
c0.setData(numpy.linspace(0, 20, 10) / 20.0)
w1.addItem(c0)
# add a regular data item (non-taurus) to y2
c1 = pg.PlotDataItem(name="pg item2", pen="y", symbol="d", symbolBrush="r")
c1.setData(1 - numpy.linspace(0, 20, 10) / 20.0)
w1._y2.addItem(c1)
# add a taurus data item to y1
c2 = tpg.TaurusPlotDataItem(name="taurus item1", pen="r", symbol="o")
c2.setModel('eval:Quantity(rand(16),"m")')
w1.addItem(c2)
# add a taurus data item to y2
c3 = tpg.TaurusPlotDataItem(name="taurus item2", pen="y", symbol="s")
c3.setModel('eval:Quantity(rand(20),"km")')
w1._y2.addItem(c3)
# Add 2 more items using setModel
models1 = [
"eval:1*rand(22)",
("eval:linspace(-10,20,10)", "eval:2*rand(10)"),
]
w1.addModels(models1)
c4 = w1[4]
c5 = w1[5]
c4.setPen("g")
c5.setPen(None)
c5.setSymbol("t")
c5.setSymbolSize(7)
c5.setSymbolBrush("r")
# move c5 to y2
w1._y2.addItem(c5)
assert len(w1) == 6
assert w1[:] == [c0, c1, c2, c3, c4, c5]
w1_mod_items = w1._cprop_tool.getModifiableItems()
for c in [c0, c1, c2, c3, c4, c5]:
assert c0 in w1_mod_items.values()
assert w1.getViewBox().addedItems == [c0, c2, c4]
assert w1._y2.addedItems == [c1, c3, c5]
# test createConfig
cfg = w1.createConfig()
propcfg = get_sub_config(cfg, "CurvePropertiesTool")
_ = get_sub_config(propcfg, "CurveProperties")
# # Debugging
# from pprint import pprint
# # pprint(cfg)
# print("-" * 80)
# pprint(curvescfg)
# test applyConfig
w2 = tpg.TaurusPlot()
qtbot.addWidget(w2)
# assert len(w2._model_chooser_tool.getModelNames()) == 0
assert len(w2._cprop_tool._getCurveAppearanceProperties()) == 0
w2.applyConfig(cfg)
# assert len(w2._model_chooser_tool.getModelNames()) == 2
# assert w2._model_chooser_tool.getModelNames() == modelscfg1
w1_props = w1._cprop_tool._getCurveAppearanceProperties()
w2_props = w2._cprop_tool._getCurveAppearanceProperties()
assert len(w1_props) == 6
assert len(w2_props) == 4
for k, p_aft in w2_props.items():
assert k in w1_props
p_ini = w1_props[k]
conflicts = p_ini.conflictsWith(p_aft, strict=True)
msg = "Mismatch in saved/restored curve properties for '{}':".format(k)
msg += "\n\t Saved: {}".format(p_ini)
msg += "\n\t Restored: {}".format(p_aft)
assert conflicts == [], msg
# test applyConfig
w3 = tpg.TaurusPlot()
qtbot.addWidget(w3)
# Manually add regular data items matching the names used in w1
# but do not match the properties, which should be updated by applyConfig
c0_w3 = pg.PlotDataItem(
name="pg item1", y=numpy.linspace(0, 20, 10) / 20.0
)
w3.addItem(c0_w3)
c1_w3 = pg.PlotDataItem(name="pg item2", y=numpy.zeros(15))
w3._y2.addItem(c1_w3)
assert len(w3._cprop_tool._getCurveAppearanceProperties()) == 2
w3.applyConfig(cfg)
w3_props = w3._cprop_tool._getCurveAppearanceProperties()
assert len(w3_props) == 6
for k, p_aft in w3_props.items():
assert k in w1_props
p_ini = w1_props[k]
conflicts = p_ini.conflictsWith(p_aft, strict=True)
msg = "Mismatch in saved/restored curve properties for '{}':".format(k)
msg += "\n\t Saved: {}".format(p_ini)
msg += "\n\t Restored: {}".format(p_aft)
assert conflicts == [], msg
def test_curveproperties(qtbot):
w = tpg.TaurusPlot()
qtbot.addWidget(w)
# add a regular data item (non-taurus) to y1
c0 = pg.PlotDataItem(name="pg item1", pen="b", fillLevel=0, brush="c")
c0.setData(numpy.linspace(0, 20, 10))
w.addItem(c0)
# add a regular data item (non-taurus) to y2
c1 = pg.PlotDataItem(name="pg item2", pen="y", symbol="d", symbolBrush="r")
c1.setData(20 - numpy.linspace(0, 20, 10))
w._y2.addItem(c1)
# add a taurus data item to y1
c2 = tpg.TaurusPlotDataItem(name="taurus item1", pen="r", symbol="o")
c2.setModel('eval:Quantity(rand(16),"m")')
w.addItem(c2)
# add a taurus data item to y2
c3 = tpg.TaurusPlotDataItem(name="taurus item2", pen="y", symbol="s")
c3.setModel('eval:Quantity(rand(20),"km")')
w._y2.addItem(c3)
# Add 2 more items using setModel
models1 = [
"eval:1*rand(22)",
("eval:linspace(-10,20,10)", "eval:2*rand(10)"),
]
w.addModels(models1)
c4 = w[4]
c5 = w[5]
c4.setPen("g")
c5.setPen(None)
c5.setSymbol("t")
c5.setSymbolSize(7)
# move c5 to y2
w._y2.addItem(c5)
assert len(w) == 6
assert w[:] == [c0, c1, c2, c3, c4, c5]
assert len(w._cprop_tool.getModifiableItems()) == 6
for c in [c0, c1, c2, c3, c4, c5]:
assert c.name() in w._cprop_tool.getModifiableItems().keys()
assert c in w._cprop_tool.getModifiableItems().values()
assert w.getViewBox().addedItems == [c0, c2, c4]
assert w._y2.addedItems == [c1, c3, c5]
prop = w._cprop_tool._getCurveAppearanceProperties()
# check lColor
assert pg.mkColor(prop[c0.name()].lColor) == pg.mkColor("b")
assert pg.mkColor(prop[c1.name()].lColor) == pg.mkColor("y")
assert pg.mkColor(prop[c2.name()].lColor) == pg.mkColor("r")
assert pg.mkColor(prop[c3.name()].lColor) == pg.mkColor("y")
assert pg.mkColor(prop[c4.name()].lColor) == pg.mkColor("g")
# assert pg.mkColor(prop[c5.name()].lColor) == pg.mkColor('b')
# check lStyle
for c in c0, c1, c2, c3, c4:
assert prop[c.name()].lStyle == Qt.Qt.SolidLine
assert prop[c5.name()].lStyle == Qt.Qt.NoPen
# check y2
for c in c0, c2, c4:
assert prop[c.name()].y2 is False
for c in c1, c3, c5:
assert prop[c.name()].y2 is True
def test_y2_config(qtbot):
# create a plot with 3 curves
w1 = tpg.TaurusPlot()
qtbot.addWidget(w1)
models1 = [
"eval:1*rand(11)",
"eval:2*rand(22)",
("eval:linspace(-10,20,10)", "eval:2*rand(10)"),
]
w1.setModel(models1)
w1_vb1 = w1.getViewBox()
w1_vb2 = w1._y2
# check that the 3 curves are on Y1
assert [c.getViewBox() for c in w1] == [w1_vb1] * 3
assert len(w1_vb1.addedItems) == 3
assert len(w1_vb2.addedItems) == 0
# move the second curve to Y2
c = w1[1]
w1_vb1.removeItem(c)
w1_vb2.addItem(c)
# check that the move was ok and there are no duplicates
assert [c.getViewBox() for c in w1] == [w1_vb1, w1_vb2, w1_vb1]
assert w1_vb2._getCurvesNames() == [w1[1].getFullModelNames()]
assert len(w1_vb1.addedItems) == 2
assert len(w1_vb2.addedItems) == 1
# test createConfig
plot_cfg = w1.createConfig()
y2_cfg = get_sub_config(plot_cfg, "Y2Axis")
curvescfg1 = get_sub_config(y2_cfg, "Y2Curves")
assert curvescfg1 == [w1[1].getFullModelNames()]
# # Debugging
# from pprint import pprint
# pprint(plot_cfg)
# pprint(curvescfg1)
# create a second, empty plot
w2 = tpg.TaurusPlot()
qtbot.addWidget(w2)
assert len(w2) == 0
# check applyConfig on the new plot
w2.applyConfig(plot_cfg)
assert len(w2) == 3
w2_vb1 = w2.getViewBox()
w2_vb2 = w2._y2
w1_all_names = [c.getFullModelNames() for c in w1]
w2_all_names = [c.getFullModelNames() for c in w2]
assert w1_all_names == w2_all_names
assert len(w2_all_names) == 3
assert len(w2_vb1.addedItems) == 2
assert len(w2_vb2.addedItems) == 1
assert w2_vb2._getCurvesNames() == w1_vb2._getCurvesNames()
assert w2_vb2._getCurvesNames() == [w1[1].getFullModelNames()]
def test_plot_model_setting(qtbot):
w = tpg.TaurusPlot()
qtbot.addWidget(w)
assert len(w) == 0
assert len(w._model_chooser_tool.getModelNames()) == 0
models1 = [
"eval:1*rand(22)",
("eval:linspace(-10,20,10)", "eval:2*rand(10)"),
]
w.setModel(models1)
c0 = w[0]
c1 = w[1]
assert len(w) == 2
assert w._model_chooser_tool.getModelNames() == [
c0.getFullModelNames(),
c1.getFullModelNames(),
]
# add a regular data item (non-taurus)
c2 = pg.PlotDataItem(name="pg item", pen="b", fillLevel=0, brush="c")
c2.setData(numpy.linspace(0, 20, 10))
w.addItem(c2)
assert w[:] == [c0, c1, c2]
assert w._model_chooser_tool.getModelNames() == [
c0.getFullModelNames(),
c1.getFullModelNames(),
]
# add a taurus data item
c3 = tpg.TaurusPlotDataItem(name="taurus item", pen="r", symbol="o")
c3.setModel('eval:Quantity(rand(16),"m")')
w.addItem(c3)
assert w[:] == [c0, c1, c2, c3]
assert w._model_chooser_tool.getModelNames() == [
c0.getFullModelNames(),
c1.getFullModelNames(),
c3.getFullModelNames(),
]
# Add existing (c0) model again (it is ignored)
w.addModels([models1[0]])
assert w[:] == [c2, c0, c1, c3] # there is reordering, non taurus first
assert w._model_chooser_tool.getModelNames() == [
c0.getFullModelNames(),
c1.getFullModelNames(),
c3.getFullModelNames(),
]
# set 1 model (not adding!, the non-taurus curve is kept)
w.setModel([models1[1]])
assert w[:] == [c2, c1] # c1 **is** still the same object!
assert w._model_chooser_tool.getModelNames() == [c1.getFullModelNames()]
# set empty model (not adding!, the non taurus curve is kept)
w.setModel([])
assert w[:] == [c2]
assert w._model_chooser_tool.getModelNames() == []
# remove non-taurus curve
w.removeItem(c2)
assert w[:] == []
assert w._model_chooser_tool.getModelNames() == []
