def testGetCurveOldApi(self):
"""old API PlotWidget.getCurve and Plot.getActiveCurve tests"""
plot = PlotWidget(backend='none')
# No curve
curve = plot.getCurve()
self.assertIsNone(curve) # No curve
plot.setActiveCurveHandling(True)
x = numpy.arange(10.).astype(numpy.float32)
y = x * x
plot.addCurve(x=x, y=y, legend='curve 0', info=["whatever"])
plot.addCurve(x=x, y=2 * x, legend='curve 1', info="anything")
plot.setActiveCurve('curve 0')
# Active curve (4 elements)
xOut, yOut, legend, info = plot.getActiveCurve()[:4]
self.assertEqual(legend, 'curve 0')
self.assertTrue(numpy.allclose(xOut, x), 'curve 0 wrong x data')
self.assertTrue(numpy.allclose(yOut, y), 'curve 0 wrong y data')
# Active curve (5 elements)
xOut, yOut, legend, info, params = plot.getCurve("curve 1")
self.assertEqual(legend, 'curve 1')
self.assertEqual(info, 'anything')
self.assertTrue(numpy.allclose(xOut, x), 'curve 1 wrong x data')
self.assertTrue(numpy.allclose(yOut, 2 * x), 'curve 1 wrong y data')
def testGetCurveOldApi(self):
"""old API PlotWidget.getCurve and Plot.getActiveCurve tests"""
plot = PlotWidget(backend='none')
# No curve
curve = plot.getCurve()
self.assertIsNone(curve) # No curve
plot.setActiveCurveHandling(True)
x = numpy.arange(10.).astype(numpy.float32)
y = x * x
plot.addCurve(x=x, y=y, legend='curve 0', info=["whatever"])
plot.addCurve(x=x, y=2*x, legend='curve 1', info="anything")
plot.setActiveCurve('curve 0')
# Active curve (4 elements)
xOut, yOut, legend, info = plot.getActiveCurve()[:4]
self.assertEqual(legend, 'curve 0')
self.assertTrue(numpy.allclose(xOut, x), 'curve 0 wrong x data')
self.assertTrue(numpy.allclose(yOut, y), 'curve 0 wrong y data')
# Active curve (5 elements)
xOut, yOut, legend, info, params = plot.getCurve("curve 1")
self.assertEqual(legend, 'curve 1')
self.assertEqual(info, 'anything')
self.assertTrue(numpy.allclose(xOut, x), 'curve 1 wrong x data')
self.assertTrue(numpy.allclose(yOut, 2 * x), 'curve 1 wrong y data')
def testGetCurve(self):
"""PlotWidget.getCurve and Plot.getActiveCurve tests"""
plot = PlotWidget(backend='none')
# No curve
curve = plot.getCurve()
self.assertIsNone(curve) # No curve
plot.setActiveCurveHandling(True)
plot.addCurve(x=(0, 1), y=(0, 1), legend='curve 0')
plot.addCurve(x=(0, 1), y=(0, 1), legend='curve 1')
plot.addCurve(x=(0, 1), y=(0, 1), legend='curve 2')
plot.setActiveCurve('curve 0')
# Active curve
active = plot.getActiveCurve()
self.assertEqual(active.getLegend(), 'curve 0')
curve = plot.getCurve()
self.assertEqual(curve.getLegend(), 'curve 0')
# No active curve and curves
plot.setActiveCurveHandling(False)
active = plot.getActiveCurve()
self.assertIsNone(active) # No active curve
curve = plot.getCurve()
self.assertEqual(curve.getLegend(), 'curve 2') # Last added curve
# Last curve hidden
plot.hideCurve('curve 2', True)
curve = plot.getCurve()
self.assertEqual(curve.getLegend(), 'curve 1') # Last added curve
# All curves hidden
plot.hideCurve('curve 1', True)
plot.hideCurve('curve 0', True)
curve = plot.getCurve()
self.assertIsNone(curve)
def testGetCurve(self):
"""PlotWidget.getCurve and Plot.getActiveCurve tests"""
plot = PlotWidget(backend='none')
# No curve
curve = plot.getCurve()
self.assertIsNone(curve) # No curve
plot.setActiveCurveHandling(True)
plot.addCurve(x=(0, 1), y=(0, 1), legend='curve 0')
plot.addCurve(x=(0, 1), y=(0, 1), legend='curve 1')
plot.addCurve(x=(0, 1), y=(0, 1), legend='curve 2')
plot.setActiveCurve('curve 0')
# Active curve
active = plot.getActiveCurve()
self.assertEqual(active.getLegend(), 'curve 0')
curve = plot.getCurve()
self.assertEqual(curve.getLegend(), 'curve 0')
# No active curve and curves
plot.setActiveCurveHandling(False)
active = plot.getActiveCurve()
self.assertIsNone(active) # No active curve
curve = plot.getCurve()
self.assertEqual(curve.getLegend(), 'curve 2') # Last added curve
# Last curve hidden
plot.hideCurve('curve 2', True)
curve = plot.getCurve()
self.assertEqual(curve.getLegend(), 'curve 1') # Last added curve
# All curves hidden
plot.hideCurve('curve 1', True)
plot.hideCurve('curve 0', True)
curve = plot.getCurve()
self.assertIsNone(curve)
