class SelectorTest(unittest.TestCase):
def setUp(self):
self.empty_bs=BrewSelector(empty=True)
self.brew_9=Brew('sequential', ['#F7FCFD', '#E5F5F9', '#CCECE6','#99D8C9', '#66C2A4', '#41AE76', '#238B45', '#006D2C', '#00441B'], 'BuGn', colorblind_safe=True)
self.bs=BrewSelector()
def test_add_brew(self):
self.empty_bs.add_brew(self.brew_9)
self.assertEqual(self.empty_bs.brews, {'sequential':{9:{'BuGn':self.brew_9}}, 'diverging': defaultdict(dict), 'qualitative': defaultdict(dict)})
def test_get_brew(self):
self.assertEqual(self.bs.get_brew('sequential', 9, 'BuGn'), self.brew_9)
def test_list_schemes_sequential(self):
self.assertEqual(sorted(self.bs.list_schemes('sequential', 9)), sorted(('BuGn', 'BuPu', 'GnBu','OrRd', 'PuBu', 'PuBuGn', 'PuRd', 'RdPu', 'YlGn', 'YlGnBu', 'YlOrBr', 'YlOrRd', 'Blues', 'Greens', 'Greys', 'Oranges', 'Purples', 'Reds')))
def test_list_schemes_qualitative(self):
self.assertEqual(sorted(self.bs.list_schemes('qualitative', 5)), sorted(('Accent', 'Dark2', 'Paired', 'Pastel1', 'Pastel2', 'Set1', 'Set2', 'Set3')))
def test_list_schemes_diverging(self):
self.assertEqual(sorted(self.bs.list_schemes('diverging', 3)), sorted(('BrBG', 'PiYG', 'PRGn', 'PuOr', 'RdBu', 'RdGy', 'RdYlBu', 'RdYlGn', 'Spectral')))
def test_get_colorblind_safe(self):
self.assertEqual(sorted(self.bs.list_schemes('qualitative', 3, colorblind_safe=True)), sorted(('Dark2', 'Paired', 'Set2')))
def test_get_photocopy_safe(self):
schemes=self.bs.list_schemes('sequential', 4, photocopy_safe=True)
self.assertEqual(schemes, ('OrRd',))
def test_get_print_safe(self):
schemes=self.bs.list_schemes('sequential', 5, print_safe=True)
self.assertEqual(sorted(schemes), sorted(('GnBu','PuRd', 'YlGnBu')))
def setUp(self):
self.empty_bs=BrewSelector(empty=True)
self.brew_9=Brew('sequential', ['#F7FCFD', '#E5F5F9', '#CCECE6','#99D8C9', '#66C2A4', '#41AE76', '#238B45', '#006D2C', '#00441B'], 'BuGn', colorblind_safe=True)
self.bs=BrewSelector()
