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)
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)
def tearDown(self): # Restore saved preferred colormaps setPreferredColormaps(self._colormaps)