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'])
