def test(self):
        colormaps = 'viridis', 'magma'

        colors.setPreferredColormaps(colormaps)
        self.assertEqual(colors.preferredColormaps(), colormaps)

        with self.assertRaises(ValueError):
            colors.setPreferredColormaps(())

        with self.assertRaises(ValueError):
            colors.setPreferredColormaps(('This is not a colormap',))

        colormaps = 'red', 'green'
        colors.setPreferredColormaps(('This is not a colormap',) + colormaps)
        self.assertEqual(colors.preferredColormaps(), colormaps)
Esempio n. 2
0
    def test(self):
        colormaps = 'viridis', 'magma'

        setPreferredColormaps(colormaps)
        self.assertEqual(preferredColormaps(), colormaps)

        with self.assertRaises(ValueError):
            setPreferredColormaps(())

        with self.assertRaises(ValueError):
            setPreferredColormaps(('This is not a colormap',))

        colormaps = 'red', 'green'
        setPreferredColormaps(('This is not a colormap',) + colormaps)
        self.assertEqual(preferredColormaps(), colormaps)
 def tearDown(self):
     # Restore saved preferred colormaps
     colors.setPreferredColormaps(self._colormaps)
Esempio n. 4
0
 def tearDown(self):
     # Restore saved preferred colormaps
     setPreferredColormaps(self._colormaps)