def use_data_source_with_color_mapper_legends_auto(d, map_width):
#data = _make_random_data(d.keys())
p = figure(width=map_width, height=int(round(4 / 5 * map_width)))
color_mapper = LinearColorMapper(palette=Blues8)
color_mapper.nan_color = 'gray'
xs = []
ys = []
data = []
for key in d.keys():
if key == '04015':
data.append(nan)
else:
data.append(random.randint(1, 8))
xs.append(d[key]['x'])
ys.append(d[key]['y'])
legends = make_legends(data, Blues8)
source = ColumnDataSource(data=dict(x=xs, y=ys, data=data, legend=legends))
p.patches('x',
'y',
source=source,
fill_color={
'field': 'data',
'transform': color_mapper
},
fill_alpha=0.7,
line_color='white',
legend='legend')
p.legend.location = "bottom_right"
return p
def test_3_colors_dataset_9_returns_right_list(self):
data = [1, 2, 3, 4, 5, 6, 7, 8,9]
palette = ['blue', 'green', 'red']
l = map_legends.make_legends(data, palette)
needed = [ '1: 3.7',
'1: 3.7',
'1: 3.7',
'3.7: 6.3',
'3.7: 6.3',
'3.7: 6.3',
'6.3: 9.0',
'6.3: 9.0',
'6.3: 9.0']
self.assertEqual(needed, l)
def test_nan(self):
data = [1, 2, 3, 4, 5, 6, 7, nan, 8,9]
palette = ['blue', 'green', 'red']
l = map_legends.make_legends(data, palette)
needed = [ '1.0: 3.7',
'1.0: 3.7',
'1.0: 3.7',
'3.7: 6.3',
'3.7: 6.3',
'3.7: 6.3',
'6.3: 9.0',
nan,
'6.3: 9.0',
'6.3: 9.0']
self.assertEqual(needed, l)
def test_2_colors_dataset_5_returns_right_list(self):
data = [1, 2, 3, 4, 1]
palette = ['blue', 'green']
l = map_legends.make_legends(data, palette)
needed = ['1: 2.5', '1: 2.5', '2.5: 4.0', '2.5: 4.0', '1: 2.5']
self.assertEqual(needed, l)