def testAddGetScatter(self): plot = Plot(backend='none') # No curve scatter = plot._getItem(kind="scatter") self.assertIsNone(scatter) # No curve plot.addScatter(x=(0, 1), y=(0, 1), value=(0, 1), legend='scatter 0') plot.addScatter(x=(0, 1), y=(0, 1), value=(0, 1), legend='scatter 1') plot.addScatter(x=(0, 1), y=(0, 1), value=(0, 1), legend='scatter 2') plot._setActiveItem('scatter', 'scatter 0') # Active curve active = plot._getActiveItem(kind='scatter') self.assertEqual(active.getLegend(), 'scatter 0') scatter1 = plot._getItem(kind='scatter', legend='scatter 1') self.assertEqual(scatter1.getLegend(), 'scatter 1')
def testGetAllScatters(self): """Plot.getAllImages test""" plot = Plot(backend='none') scatters = plot._getItems(kind='scatter') self.assertEqual(len(scatters), 0) plot.addScatter(x=(0, 1), y=(0, 1), value=(0, 1), legend='scatter 0') plot.addScatter(x=(0, 1), y=(0, 1), value=(0, 1), legend='scatter 1') plot.addScatter(x=(0, 1), y=(0, 1), value=(0, 1), legend='scatter 2') scatters = plot._getItems(kind='scatter') self.assertEqual(len(scatters), 3) self.assertEqual(scatters[0].getLegend(), 'scatter 0') self.assertEqual(scatters[2].getLegend(), 'scatter 2') scatters = plot._getItems(kind='scatter', just_legend=True) self.assertEqual(len(scatters), 3) self.assertEqual(list(scatters), ['scatter 0', 'scatter 1', 'scatter 2'])