def test_sankey_label_index(self):
sankey = Sankey(([
(0, 2, 5), (0, 3, 7), (0, 4, 6),
(1, 2, 2), (1, 3, 9), (1, 4, 4)],
Dataset(enumerate('ABXYZ'), 'index', 'label'))
).options(label_index='label', tools=['hover'])
plot = list(bokeh_renderer.get_plot(sankey).subplots.values())[0]
scatter_source = plot.handles['scatter_1_source']
text_source = plot.handles['text_1_source']
patch_source = plot.handles['patches_1_source']
scatter_index = np.arange(5)
self.assertEqual(scatter_source.data['index'], scatter_index)
text_data = {'x': np.array([18.75, 18.75, 1003.75, 1003.75, 1003.75]),
'y': np.array([145.454545, 395.454545, 48.787879, 229.090909, 430.30303]),
'text': ['A - 18', 'B - 15', 'X - 7', 'Y - 16', 'Z - 10']}
for k in text_data:
self.assertEqual(text_source.data[k], text_data[k])
patch_data = {'start_values': ['A', 'A', 'A', 'B', 'B', 'B'],
'end_values': ['X', 'Y', 'Z', 'X', 'Y', 'Z'],
'Value': np.array([5, 7, 6, 2, 9, 4])}
for k in patch_data:
self.assertEqual(patch_source.data[k], patch_data[k])
renderers = plot.state.renderers
quad_renderer = plot.handles['quad_1_glyph_renderer']
text_renderer = plot.handles['text_1_glyph_renderer']
graph_renderer = plot.handles['glyph_renderer']
self.assertTrue(renderers.index(graph_renderer)<renderers.index(quad_renderer))
self.assertTrue(renderers.index(quad_renderer)<renderers.index(text_renderer))
def test_sankey_label_index(self):
sankey = Sankey(
([(0, 2, 5), (0, 3, 7), (0, 4, 6), (1, 2, 2), (1, 3, 9),
(1, 4, 4)], Dataset(enumerate('ABXYZ'), 'index',
'label'))).options(label_index='label',
tools=['hover'])
plot = list(bokeh_renderer.get_plot(sankey).subplots.values())[0]
scatter_source = plot.handles['scatter_1_source']
text_source = plot.handles['text_1_source']
patch_source = plot.handles['patches_1_source']
scatter_index = np.arange(5)
self.assertEqual(scatter_source.data['index'], scatter_index)
text_data = {
'x':
np.array([18.75, 18.75, 1003.75, 1003.75, 1003.75]),
'y':
np.array([
140.90909091, 390.90909091, 50.90909091, 228.18181818,
427.27272727
]),
'text': ['A - 18', 'B - 15', 'X - 7', 'Y - 16', 'Z - 10']
}
for k in text_data:
self.assertEqual(text_source.data[k], text_data[k])
patch_data = {
'start_values': ['A', 'A', 'A', 'B', 'B', 'B'],
'end_values': ['X', 'Y', 'Z', 'X', 'Y', 'Z'],
'Value': np.array([5, 7, 6, 2, 9, 4])
}
for k in patch_data:
self.assertEqual(patch_source.data[k], patch_data[k])
def test_sankey_simple(self):
sankey = Sankey([('A', 'X', 5), ('A', 'Y', 7), ('A', 'Z', 6),
('B', 'X', 2), ('B', 'Y', 9), ('B', 'Z', 4)])
plot = list(bokeh_renderer.get_plot(sankey).subplots.values())[0]
scatter_source = plot.handles['scatter_1_source']
quad_source = plot.handles['quad_1_source']
text_source = plot.handles['text_1_source']
patch_source = plot.handles['patches_1_source']
scatter_index = np.array(['A', 'B', 'X', 'Y', 'Z'])
self.assertEqual(scatter_source.data['index'], scatter_index)
text_data = {
'x':
np.array([18.75, 18.75, 1003.75, 1003.75, 1003.75]),
'y':
np.array([
140.90909091, 390.90909091, 50.90909091, 228.18181818,
427.27272727
]),
'text': ['A - 18', 'B - 15', 'X - 7', 'Y - 16', 'Z - 10']
}
for k in text_data:
self.assertEqual(text_source.data[k], text_data[k])
quad_data = {
'index':
scatter_index,
'x0': [0, 0, 985.0, 985.0, 985.0],
'x1': [15, 15, 1000.0, 1000.0, 1000.0],
'y0': [
10.000000000000355, 281.8181818181821, -5.6843418860808015e-14,
111.8181818181817, 354.54545454545433
],
'y1': [
271.8181818181821, 500.0, 101.8181818181817,
344.54545454545433, 500.0
]
}
for k in quad_data:
self.assertEqual(quad_source.data[k], quad_data[k])
self.assertEqual(patch_source.data['Value'],
np.array([5, 7, 6, 2, 9, 4]))
renderers = plot.state.renderers
quad_renderer = plot.handles['quad_1_glyph_renderer']
text_renderer = plot.handles['text_1_glyph_renderer']
graph_renderer = plot.handles['glyph_renderer']
self.assertTrue(
renderers.index(graph_renderer) < renderers.index(quad_renderer))
self.assertTrue(
renderers.index(quad_renderer) < renderers.index(text_renderer))
def test_sankey_simple(self):
sankey = Sankey([('A', 'X', 5), ('A', 'Y', 7), ('A', 'Z', 6),
('B', 'X', 2), ('B', 'Y', 9), ('B', 'Z', 4)])
plot = bokeh_renderer.get_plot(sankey)
scatter_source = plot.handles['scatter_1_source']
quad_source = plot.handles['quad_1_source']
text_source = plot.handles['text_1_source']
patch_source = plot.handles['patches_1_source']
scatter_index = np.array(['A', 'B', 'X', 'Y', 'Z'])
self.assertEqual(scatter_source.data['index'], scatter_index)
text_data = {
'x':
np.array([18.75, 18.75, 1003.75, 1003.75, 1003.75]),
'y':
np.array(
[145.454545, 395.454545, 48.787879, 229.090909, 430.30303]),
'text': ['A - 18', 'B - 15', 'X - 7', 'Y - 16', 'Z - 10']
}
for k in text_data:
self.assertEqual(text_source.data[k], text_data[k])
quad_data = {
'index':
scatter_index,
'x0': [0, 0, 985.0, 985.0, 985.0],
'x1': [15, 15, 1000.0, 1000.0, 1000.0],
'y0': [
19.999999999999588, 290.90909090909054, 2.842170943040401e-14,
117.57575757575768, 360.60606060606057
],
'y1': [
270.90909090909054, 500.0, 97.57575757575768,
340.60606060606057, 500.0
]
}
for k in quad_data:
self.assertEqual(quad_source.data[k], quad_data[k])
self.assertEqual(patch_source.data['Value'],
np.array([5, 7, 6, 2, 9, 4]))
renderers = plot.state.renderers
quad_renderer = plot.handles['quad_1_glyph_renderer']
text_renderer = plot.handles['text_1_glyph_renderer']
graph_renderer = plot.handles['glyph_renderer']
self.assertTrue(
renderers.index(graph_renderer) < renderers.index(quad_renderer))
self.assertTrue(
renderers.index(quad_renderer) < renderers.index(text_renderer))
def test_sankey_label_index(self):
sankey = Sankey(
([(0, 2, 5), (0, 3, 7), (0, 4, 6), (1, 2, 2), (1, 3, 9),
(1, 4, 4)], Dataset(enumerate('ABXYZ'), 'index',
'label'))).options(label_index='label')
plot = list(mpl_renderer.get_plot(sankey).subplots.values())[0]
labels = plot.handles['labels']
text_data = {
'x':
np.array([18.75, 18.75, 1003.75, 1003.75, 1003.75]),
'y':
np.array(
[145.454545, 395.454545, 48.787879, 229.090909, 430.30303]),
'text': ['A - 18', 'B - 15', 'X - 7', 'Y - 16', 'Z - 10']
}
for i, text in enumerate(labels):
self.assertEqual(text.xy[0], text_data['x'][i])
self.assertEqual(text.xy[1], text_data['y'][i])
self.assertEqual(text.get_text(), text_data['text'][i])
def test_sankey_simple(self):
sankey = Sankey([('A', 'X', 5), ('A', 'Y', 7), ('A', 'Z', 6),
('B', 'X', 2), ('B', 'Y', 9), ('B', 'Z', 4)])
plot = list(mpl_renderer.get_plot(sankey).subplots.values())[0]
rects = plot.handles['rects']
labels = plot.handles['labels']
text_data = {
'x':
np.array([18.75, 18.75, 1003.75, 1003.75, 1003.75]),
'y':
np.array([
140.90909091, 390.90909091, 50.90909091, 228.18181818,
427.27272727
]),
'text': ['A - 18', 'B - 15', 'X - 7', 'Y - 16', 'Z - 10']
}
for i, text in enumerate(labels):
self.assertEqual(text.xy[0], text_data['x'][i])
self.assertEqual(text.xy[1], text_data['y'][i])
self.assertEqual(text.get_text(), text_data['text'][i])
quad_data = {
'x0': [0, 0, 985.0, 985.0, 985.0],
'x1': [15, 15, 1000.0, 1000.0, 1000.0],
'y0': [
10.000000000000355, 281.8181818181821, -5.6843418860808015e-14,
111.8181818181817, 354.54545454545433
],
'y1': [
271.8181818181821, 500.0, 101.8181818181817,
344.54545454545433, 500.0
]
}
for i, rect in enumerate(rects.get_paths()):
x0, x1, y0, y1 = [
quad_data[c][i] for c in ('x0', 'x1', 'y0', 'y1')
]
arr = np.array([[x0, y0], [x1, y0], [x1, y1], [x0, y1], [x0, y0]])
self.assertEqual(rect.vertices, arr)
def test_sankey_simple(self):
sankey = Sankey([('A', 'X', 5), ('A', 'Y', 7), ('A', 'Z', 6),
('B', 'X', 2), ('B', 'Y', 9), ('B', 'Z', 4)])
plot = list(mpl_renderer.get_plot(sankey).subplots.values())[0]
rects = plot.handles['rects']
labels = plot.handles['labels']
text_data = {
'x':
np.array([18.75, 18.75, 1003.75, 1003.75, 1003.75]),
'y':
np.array(
[145.454545, 395.454545, 48.787879, 229.090909, 430.30303]),
'text': ['A - 18', 'B - 15', 'X - 7', 'Y - 16', 'Z - 10']
}
for i, text in enumerate(labels):
self.assertEqual(text.xy[0], text_data['x'][i])
self.assertEqual(text.xy[1], text_data['y'][i])
self.assertEqual(text.get_text(), text_data['text'][i])
quad_data = {
'x0': [0, 0, 985.0, 985.0, 985.0],
'x1': [15, 15, 1000.0, 1000.0, 1000.0],
'y0': [
19.999999999999588, 290.90909090909054, 2.842170943040401e-14,
117.57575757575768, 360.60606060606057
],
'y1': [
270.90909090909054, 500.0, 97.57575757575768,
340.60606060606057, 500.0
]
}
for i, rect in enumerate(rects.get_paths()):
x0, x1, y0, y1 = [
quad_data[c][i] for c in ('x0', 'x1', 'y0', 'y1')
]
arr = np.array([[x0, y0], [x1, y0], [x1, y1], [x0, y1], [x0, y0]])
self.assertEqual(rect.vertices, arr)
def test_sankey_simple(self):
sankey = Sankey([('A', 'X', 5), ('A', 'Y', 7), ('A', 'Z', 6),
('B', 'X', 2), ('B', 'Y', 9), ('B', 'Z', 4)])
plot = mpl_renderer.get_plot(sankey)
rects = plot.handles['rects']
labels = plot.handles['labels']
text_data = {
'x':
np.array([18.75, 18.75, 1003.75, 1003.75, 1003.75]),
'y':
np.array(
[125.454545, 375.454545, 48.787879, 229.090909, 430.30303]),
'text': ['A - 18', 'B - 15', 'X - 7', 'Y - 16', 'Z - 10']
}
for i, text in enumerate(labels):
self.assertEqual(text.xy[0], text_data['x'][i])
self.assertEqual(text.xy[1], text_data['y'][i])
self.assertEqual(text.get_text(), text_data['text'][i])
quad_data = {
'x0': [0, 0, 985.0, 985.0, 985.0],
'x1': [15, 15, 1000.0, 1000.0, 1000.0],
'y0': [
0.0, 270.9090909090908, -7.105427357601002e-15,
117.57575757575756, 360.6060606060606
],
'y1': [250.909091, 480.0, 97.575758, 340.606061, 500.0]
}
for i, rect in enumerate(rects.get_paths()):
x0, x1, y0, y1 = [
quad_data[c][i] for c in ('x0', 'x1', 'y0', 'y1')
]
arr = np.array([[x0, y0], [x1, y0], [x1, y1], [x0, y1], [x0, y0]])
self.assertEqual(rect.vertices, arr)