source: sasview/src/sas/qtgui/UnitTesting/PlotHelperTest.py @ 161713c

import unittest

# Tested module
import PlotHelper

class PlotHelperTest(unittest.TestCase):
    '''Test the Plot helper functions'''
    def setUp(self):
        '''Empty'''
        pass

    def tearDown(self):
        '''Empty'''
        pass

    def testDefaults(self):
        """ default method variables values """
        self.assertIsInstance(PlotHelper._plots, dict)
        self.assertEqual(PlotHelper._plots, {})
        self.assertEqual(PlotHelper._plot_id, 0)

    def testFunctions(self):
        """ Adding a plot """
        plot = "I am a plot. Really."
        PlotHelper.addPlot(plot)
        plot_id = PlotHelper.idOfPlot(plot)

        # We can't guarantee unique values for plot_id, as
        # the tests are executed in random order, so just
        # make sure we get consecutive values for 2 plots
        plot2 = "I am also a plot."
        PlotHelper.addPlot(plot2)
        plot_id_2 = PlotHelper.idOfPlot(plot2)
        self.assertEqual(plot_id_2 - plot_id, 1)

        # Other properties
        self.assertEqual(PlotHelper.currentPlots(), [plot_id, plot_id_2])
        self.assertEqual(PlotHelper.plotById(plot_id), plot)
        self.assertEqual(PlotHelper.plotById(plot_id_2), plot2)

        # Delete a graph
        PlotHelper.deletePlot(plot_id)
        self.assertEqual(PlotHelper.currentPlots(), [plot_id_2])

        # Add another graph to see the counter
        plot3 = "Just another plot. Move along."
        PlotHelper.addPlot(plot3)
        self.assertEqual(PlotHelper.idOfPlot(plot3), plot_id_2 + 1)

if __name__ == "__main__":
    unittest.main()