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 })