def test_spread_padding_logx(self):
spread = Spread([(1, 1, 0.5), (2, 2, 0.5),
(3, 3, 0.5)]).options(padding=0.1, logx=True)
plot = bokeh_renderer.get_plot(spread)
x_range, y_range = plot.handles['x_range'], plot.handles['y_range']
self.assertEqual(x_range.start, 0.89595845984076228)
self.assertEqual(x_range.end, 3.3483695221017129)
self.assertEqual(y_range.start, 0.19999999999999996)
self.assertEqual(y_range.end, 3.8)
def test_spread_padding_soft_range(self):
spread = Spread([(1, 1, 0.5), (2, 2, 0.5), (3, 3, 0.5)
]).redim.soft_range(y=(0, 3.5)).options(padding=0.1)
plot = bokeh_renderer.get_plot(spread)
x_range, y_range = plot.handles['x_range'], plot.handles['y_range']
self.assertEqual(x_range.start, 0.8)
self.assertEqual(x_range.end, 3.2)
self.assertEqual(y_range.start, 0)
self.assertEqual(y_range.end, 3.8)
def test_spread_padding_nonsquare(self):
spread = Spread([(1, 1, 0.5), (2, 2, 0.5),
(3, 3, 0.5)]).options(padding=0.1, width=600)
plot = bokeh_renderer.get_plot(spread)
x_range, y_range = plot.handles['x_range'], plot.handles['y_range']
self.assertEqual(x_range.start, 0.9)
self.assertEqual(x_range.end, 3.1)
self.assertEqual(y_range.start, 0.19999999999999996)
self.assertEqual(y_range.end, 3.8)
def test_spread_padding_logy(self):
spread = Spread([(1, 1, 0.5), (2, 2, 0.5),
(3, 3, 0.5)]).options(padding=0.1, logy=True)
plot = bokeh_renderer.get_plot(spread)
x_range, y_range = plot.handles['x_range'], plot.handles['y_range']
self.assertEqual(x_range.start, 0.8)
self.assertEqual(x_range.end, 3.2)
self.assertEqual(y_range.start, 0.41158562699652224)
self.assertEqual(y_range.end, 4.2518491541367327)
def test_spread_with_nans(self):
spread = Spread([(0, 0, 0, 1), (1, 0, 0, 2), (2, 0, 0, 3), (3, np.nan, np.nan, np.nan),
(4, 0, 0, 5), (5, 0, 0, 6), (6, 0, 0, 7)], vdims=['y', 'neg', 'pos'])
plot = bokeh_renderer.get_plot(spread)
cds = plot.handles['cds']
self.assertEqual(cds.data['x'], np.array([0., 1., 2., 2., 1., 0., np.nan,
4., 5., 6., 6., 5., 4.]))
self.assertEqual(cds.data['y'], np.array([0., 0., 0., 3., 2., 1., np.nan,
0., 0., 0., 7., 6., 5.]))
def test_spread_fill_between_ys(self):
spread = Spread([(0, 1, 0.5), (1, 2, 1), (2, 3, 2.25)],
vdims=['y', 'y2'])
state = self._get_plot_state(spread)
self.assertEqual(state['data'][0]['y'], np.array([0.5, 1, 0.75]))
self.assertEqual(state['data'][0]['mode'], 'lines')
self.assertEqual(state['data'][0].get('fill', None), None)
self.assertEqual(state['data'][1]['y'], np.array([1.5, 3, 5.25]))
self.assertEqual(state['data'][1]['mode'], 'lines')
self.assertEqual(state['data'][1]['fill'], 'tonexty')
self.assertEqual(state['layout']['yaxis']['range'], [0.5, 5.25])
def test_spread_fill_between_xs(self):
spread = Spread([(0, 1, 0.5), (1, 2, 1), (2, 3, 2.25)],
vdims=['y', 'y2']).options(invert_axes=True)
state = self._get_plot_state(spread)
self.assertEqual(state['data'][0]['x'], np.array([0.5, 1, 0.75]))
self.assertEqual(state['data'][0]['mode'], 'lines')
self.assertEqual(state['data'][0]['fill'], None)
self.assertEqual(state['data'][1]['x'], np.array([1.5, 3, 5.25]))
self.assertEqual(state['data'][1]['mode'], 'lines')
self.assertEqual(state['data'][1]['fill'], 'tonextx')
self.assertEqual(state['layout']['xaxis']['range'], [0.5, 5.25])
self.assertEqual(state['layout']['yaxis']['range'], [0, 2])
def test_spread_empty(self):
spread = Spread([])
plot = bokeh_renderer.get_plot(spread)
cds = plot.handles['cds']
self.assertEqual(cds.data['x'], [])
self.assertEqual(cds.data['y'], [])
def test_visible(self):
element = Spread([(0, 1, 0.5), (1, 2, 1), (2, 3, 2.25)],
vdims=['y', 'y2']).options(visible=False)
state = self._get_plot_state(element)
self.assertEqual(state['data'][0]['visible'], False)
