def setUp(self):
self.dmap_element = DynamicMap(lambda: Image([]))
self.dmap_overlay = DynamicMap(lambda: Overlay([Curve([]), Points([])]))
self.dmap_ndoverlay = DynamicMap(lambda: NdOverlay({0: Curve([]), 1: Curve([])}))
self.element = Scatter([])
self.el1, self.el2 = Path([]), HLine(0)
self.overlay = Overlay([self.el1, self.el2])
self.ndoverlay = NdOverlay({0: VectorField([]), 1: VectorField([])})
def test_vectorfield_line_width_op_update(self):
vectorfield = HoloMap({
0:
VectorField([(0, 0, 0, 1, 1), (0, 1, 0, 1, 4), (0, 2, 0, 1, 8)],
vdims=['A', 'M', 'line_width']),
1:
VectorField([(0, 0, 0, 1, 3), (0, 1, 0, 1, 2), (0, 2, 0, 1, 5)],
vdims=['A', 'M', 'line_width'])
}).options(linewidth='line_width')
plot = mpl_renderer.get_plot(vectorfield)
artist = plot.handles['artist']
self.assertEqual(artist.get_linewidths(), [1, 4, 8])
plot.update((1, ))
self.assertEqual(artist.get_linewidths(), [3, 2, 5])
def test_vectorfield_alpha_op(self):
vectorfield = VectorField([(0, 0, 0, 1, 0), (0, 1, 0, 1, 0.2),
(0, 2, 0, 1, 0.7)],
vdims=['A', 'M',
'alpha']).options(alpha='alpha')
with self.assertRaises(Exception):
mpl_renderer.get_plot(vectorfield)
def test_vectorfield_linear_color_op_update(self):
vectorfield = HoloMap({
0:
VectorField([(0, 0, 0, 1, 0), (0, 1, 0, 1, 1), (0, 2, 0, 1, 2)],
vdims=['A', 'M', 'color']),
1:
VectorField([(0, 0, 0, 1, 3.2), (0, 1, 0, 1, 2), (0, 2, 0, 1, 4)],
vdims=['A', 'M', 'color'])
}).options(color='color', framewise=True)
plot = mpl_renderer.get_plot(vectorfield)
artist = plot.handles['artist']
self.assertEqual(np.asarray(artist.get_array()), np.array([0, 1, 2]))
self.assertEqual(artist.get_clim(), (0, 2))
plot.update((1, ))
self.assertEqual(np.asarray(artist.get_array()), np.array([3.2, 2, 4]))
self.assertEqual(artist.get_clim(), (2, 4))
def test_vectorfield_line_width_op(self):
vectorfield = VectorField([(0, 0, 0, 1, 1), (0, 1, 0, 1, 4),
(0, 2, 0, 1, 8)],
vdims=['A', 'M', 'line_width'
]).options(linewidth='line_width')
plot = mpl_renderer.get_plot(vectorfield)
artist = plot.handles['artist']
self.assertEqual(artist.get_linewidths(), [1, 4, 8])
def test_vectorfield_categorical_color_op(self):
vectorfield = VectorField([(0, 0, 0, 1, 'A'), (0, 1, 0, 1, 'B'),
(0, 2, 0, 1, 'C')],
vdims=['A', 'M',
'color']).options(color='color')
plot = mpl_renderer.get_plot(vectorfield)
artist = plot.handles['artist']
self.assertEqual(np.asarray(artist.get_array()), np.array([0, 1, 2]))
self.assertEqual(artist.get_clim(), (0, 2))
def test_vectorfield_color_op_update(self):
vectorfield = HoloMap({
0:
VectorField([(0, 0, 0, 1, '#000000'), (0, 1, 0, 1, '#FF0000'),
(0, 2, 0, 1, '#00FF00')],
vdims=['A', 'M', 'color']),
1:
VectorField([(0, 0, 0, 1, '#0000FF'), (0, 1, 0, 1, '#00FF00'),
(0, 2, 0, 1, '#FF0000')],
vdims=['A', 'M', 'color'])
}).options(color='color')
plot = mpl_renderer.get_plot(vectorfield)
artist = plot.handles['artist']
self.assertEqual(artist.get_facecolors(),
np.array([[0, 0, 0, 1], [1, 0, 0, 1], [0, 1, 0, 1]]))
plot.update((1, ))
self.assertEqual(artist.get_facecolors(),
np.array([[0, 0, 1, 1], [0, 1, 0, 1], [1, 0, 0, 1]]))
def test_vectorfield_color_index_color_clash(self):
vectorfield = VectorField([(0, 0, 0), (0, 1, 1), (0, 2, 2)],
vdims='color').options(line_color='color',
color_index='color')
with ParamLogStream() as log:
bokeh_renderer.get_plot(vectorfield)
log_msg = log.stream.read()
warning = ("Cannot declare style mapping for 'line_color' option "
"and declare a color_index; ignoring the color_index.\n")
self.assertEqual(log_msg, warning)
def test_vectorfield_line_width_op(self):
vectorfield = VectorField([(0, 0, 0, 1, 1), (0, 1, 0, 1, 4),
(0, 2, 0, 1, 8)],
vdims=['A', 'M', 'line_width'
]).options(line_width='line_width')
plot = bokeh_renderer.get_plot(vectorfield)
cds = plot.handles['cds']
glyph = plot.handles['glyph']
self.assertEqual(cds.data['line_width'],
np.array([1, 4, 8, 1, 4, 8, 1, 4, 8]))
self.assertEqual(glyph.line_width, {'field': 'line_width'})
def test_vectorfield_alpha_op(self):
vectorfield = VectorField([(0, 0, 0, 1, 0), (0, 1, 0, 1, 0.2),
(0, 2, 0, 1, 0.7)],
vdims=['A', 'M',
'alpha']).options(alpha='alpha')
plot = bokeh_renderer.get_plot(vectorfield)
cds = plot.handles['cds']
glyph = plot.handles['glyph']
self.assertEqual(cds.data['alpha'],
np.array([0, 0.2, 0.7, 0, 0.2, 0.7, 0, 0.2, 0.7]))
self.assertEqual(glyph.line_alpha, {'field': 'alpha'})
def test_vectorfield_color_index_color_clash(self):
vectorfield = VectorField([(0, 0, 0, 1, 0), (0, 1, 0, 1, 1),
(0, 2, 0, 1, 2)],
vdims=['A', 'M',
'color']).options(color='color',
color_index='A')
with ParamLogStream() as log:
plot = mpl_renderer.get_plot(vectorfield)
log_msg = log.stream.read()
warning = ("%s: Cannot declare style mapping for 'color' option "
"and declare a color_index; ignoring the color_index.\n" %
plot.name)
self.assertEqual(log_msg, warning)
def test_vectorfield_color_op(self):
vectorfield = VectorField([(0, 0, 0, 1, '#000'), (0, 1, 0, 1, '#F00'),
(0, 2, 0, 1, '#0F0')],
vdims=['A', 'M',
'color']).options(color='color')
plot = bokeh_renderer.get_plot(vectorfield)
cds = plot.handles['cds']
glyph = plot.handles['glyph']
self.assertEqual(
cds.data['color'],
np.array([
'#000', '#F00', '#0F0', '#000', '#F00', '#0F0', '#000', '#F00',
'#0F0'
]))
self.assertEqual(glyph.line_color, {'field': 'color'})
def test_vectorfield_categorical_color_op(self):
vectorfield = VectorField([(0, 0, 0, 1, 'A'), (0, 1, 0, 1, 'B'),
(0, 2, 0, 1, 'C')],
vdims=['A', 'M',
'color']).options(color='color')
plot = bokeh_renderer.get_plot(vectorfield)
cds = plot.handles['cds']
glyph = plot.handles['glyph']
cmapper = plot.handles['color_color_mapper']
self.assertTrue(cmapper, CategoricalColorMapper)
self.assertEqual(cmapper.factors, ['A', 'B', 'C'])
self.assertEqual(
cds.data['color'],
np.array(['A', 'B', 'C', 'A', 'B', 'C', 'A', 'B', 'C']))
self.assertEqual(glyph.line_color, {
'field': 'color',
'transform': cmapper
})
def test_vectorfield_linear_color_op(self):
vectorfield = VectorField([(0, 0, 0, 1, 0), (0, 1, 0, 1, 1),
(0, 2, 0, 1, 2)],
vdims=['A', 'M',
'color']).options(color='color')
plot = bokeh_renderer.get_plot(vectorfield)
cds = plot.handles['cds']
glyph = plot.handles['glyph']
cmapper = plot.handles['color_color_mapper']
self.assertTrue(cmapper, LinearColorMapper)
self.assertEqual(cmapper.low, 0)
self.assertEqual(cmapper.high, 2)
self.assertEqual(cds.data['color'],
np.array([0, 1, 2, 0, 1, 2, 0, 1, 2]))
self.assertEqual(glyph.line_color, {
'field': 'color',
'transform': cmapper
})
