def testDataRangeHiddenCurve(self): """curves with a hidden curve""" plot = PlotWidget(backend='none') plot.addCurve((0, 1), (0, 1), legend='shown') plot.addCurve((0, 1, 2), (5, 5, 5), legend='hidden') range1 = plot.getDataRange() self.assertEqual(range1.x, (0, 2)) self.assertEqual(range1.y, (0, 5)) plot.hideCurve('hidden') range2 = plot.getDataRange() self.assertEqual(range2.x, (0, 1)) self.assertEqual(range2.y, (0, 1))
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)