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)