def test_plotly_pane_single_trace(document, comm):
trace = go.Scatter(x=[0, 1], y=[2, 3], uid='Test')
pane = Pane({'data': [trace], 'layout': {'width': 350}})
# Create pane
model = pane.get_root(document, comm=comm)
assert isinstance(model, PlotlyPlot)
assert pane._models[model.ref['id']][0] is model
assert len(model.data) == 1
assert model.data[0]['type'] == 'scatter'
assert model.data[0]['x'] == [0, 1]
assert model.data[0]['y'] == [2, 3]
assert model.layout == {'width': 350}
assert len(model.data_sources) == 1
assert model.data_sources[0].data == {}
# Replace Pane.object
new_trace = go.Bar(x=[2, 3], y=[4, 5])
pane.object = {'data': new_trace, 'layout': {'width': 350}}
assert len(model.data) == 1
assert model.data[0]['type'] == 'bar'
assert model.data[0]['x'] == [2, 3]
assert model.data[0]['y'] == [4, 5]
assert model.layout == {'width': 350}
assert len(model.data_sources) == 1
assert model.data_sources[0].data == {}
assert pane._models[model.ref['id']][0] is model
# Cleanup
pane._cleanup(model)
assert pane._models == {}
def test_vega_pane(document, comm):
pane = Pane(vega_example)
# Create pane
model = pane.get_root(document, comm=comm)
assert isinstance(model, VegaPlot)
expected = dict(vega_example, data={})
assert dict(model.data, **blank_schema) == dict(expected, **blank_schema)
cds_data = model.data_sources['data'].data
assert np.array_equal(cds_data['x'], np.array(['A', 'B', 'C', 'D', 'E']))
assert np.array_equal(cds_data['y'], np.array([5, 3, 6, 7, 2]))
point_example = dict(vega_example, mark='point')
point_example['data']['values'][0]['x'] = 'C'
pane.object = point_example
point_example['data'].pop('values')
assert model.data == point_example
cds_data = model.data_sources['data'].data
assert np.array_equal(cds_data['x'], np.array(['C', 'B', 'C', 'D', 'E']))
assert np.array_equal(cds_data['y'], np.array([5, 3, 6, 7, 2]))
pane._cleanup(model)
assert pane._models == {}
def test_altair_pane(document, comm):
pane = Pane(altair_example())
# Create pane
model = pane.get_root(document, comm=comm)
assert isinstance(model, VegaPlot)
expected = dict(vega_example, data={})
if altair_version >= '4.0.0':
expected['config'] = vega4_config
assert dict(model.data, **blank_schema) == dict(expected, **blank_schema)
cds_data = model.data_sources['data'].data
assert np.array_equal(cds_data['x'], np.array(['A', 'B', 'C', 'D', 'E']))
assert np.array_equal(cds_data['y'], np.array([5, 3, 6, 7, 2]))
chart = altair_example()
chart.mark = 'point'
chart.data.values[0]['x'] = 'C'
pane.object = chart
point_example = dict(vega_example, mark='point')
if altair_version >= '4.0.0':
point_example['config'] = vega4_config
assert dict(model.data, **blank_schema) == dict(point_example,
**blank_schema)
cds_data = model.data_sources['data'].data
assert np.array_equal(cds_data['x'], np.array(['C', 'B', 'C', 'D', 'E']))
assert np.array_equal(cds_data['y'], np.array([5, 3, 6, 7, 2]))
pane._cleanup(model)
assert pane._models == {}
def test_param_method_pane_mpl(document, comm):
test = View()
pane = Pane(test.mpl_view)
inner_pane = pane._pane
assert isinstance(inner_pane, Matplotlib)
# Create pane
row = pane.get_root(document, comm=comm)
assert isinstance(row, BkRow)
assert len(row.children) == 1
model = row.children[0]
assert pane._models[row.ref['id']][0] is row
text = model.text
# Update pane
test.a = 5
new_model = row.children[0]
assert inner_pane is pane._pane
assert new_model is model
assert new_model.text != text
assert pane._models[row.ref['id']][0] is row
# Cleanup pane
pane._cleanup(row)
assert pane._models == {}
assert inner_pane._models == {}
def test_param_method_pane_changing_type(document, comm):
test = View()
pane = Pane(test.mixed_view)
inner_pane = pane._pane
assert isinstance(inner_pane, Matplotlib)
# Create pane
row = pane.get_root(document, comm=comm)
assert isinstance(row, BkRow)
assert len(row.children) == 1
model = row.children[0]
text = model.text
assert text.startswith('<img src=')
# Update pane
test.a = 5
new_model = row.children[0]
new_pane = pane._pane
assert isinstance(new_pane, Bokeh)
assert isinstance(new_model, Div)
assert new_model.text != text
# Cleanup pane
new_pane._cleanup(row)
assert new_pane._models == {}
def test_param_method_pane(document, comm):
test = View()
pane = Pane(test.view)
inner_pane = pane._pane
assert isinstance(inner_pane, Bokeh)
# Create pane
row = pane.get_root(document, comm=comm)
assert isinstance(row, BkRow)
assert len(row.children) == 1
model = row.children[0]
assert pane._models[row.ref['id']][0] is row
assert isinstance(model, Div)
assert model.text == '0'
# Update pane
test.a = 5
new_model = row.children[0]
assert inner_pane is pane._pane
assert new_model.text == '5'
assert pane._models[row.ref['id']][0] is row
# Cleanup pane
pane._cleanup(row)
assert pane._models == {}
assert inner_pane._models == {}
def test_param_method_pane_subobject(document, comm):
subobject = View(name='Nested', a=42)
test = View(b=subobject)
pane = Pane(test.subobject_view)
inner_pane = pane._pane
assert isinstance(inner_pane, Bokeh)
# Create pane
row = pane.get_root(document, comm=comm)
assert isinstance(row, BkRow)
assert len(row.children) == 1
model = row.children[0]
assert isinstance(model, Div)
assert model.text == '42'
# Ensure that subobject is being watched
watchers = pane._callbacks
assert any(w.inst is subobject for w in watchers)
assert pane._models[row.ref['id']][0] is row
# Ensure that switching the subobject triggers update in watchers
new_subobject = View(name='Nested', a=42)
test.b = new_subobject
assert pane._models[row.ref['id']][0] is row
watchers = pane._callbacks
assert not any(w.inst is subobject for w in watchers)
assert any(w.inst is new_subobject for w in watchers)
# Cleanup pane
pane._cleanup(row)
assert pane._models == {}
assert inner_pane._models == {}
def test_expand_param_subobject_tabs(document, comm):
class Test(param.Parameterized):
abc = param.Parameter()
test = Test(abc=Test(name='Nested'), name='A')
test_pane = Pane(test, expand_layout=Tabs)
model = test_pane.get_root(document, comm=comm)
toggle = model.tabs[0].child.children[0].children[1]
assert isinstance(toggle, Toggle)
# Expand subpanel
test_pane._widgets['abc'][1].value = True
assert len(model.tabs) == 2
_, subpanel = test_pane.layout.objects
subtabs = model.tabs[1].child
assert model.tabs[1].title == 'Abc'
assert isinstance(subtabs, BkTabs)
assert len(subtabs.tabs) == 1
assert subtabs.tabs[0].title == 'Nested'
box = subtabs.tabs[0].child
assert isinstance(box, BkColumn)
assert len(box.children) == 1
widget = box.children[0]
assert isinstance(widget, BkTextInput)
# Collapse subpanel
test_pane._widgets['abc'][1].value = False
assert len(model.tabs) == 1
def test_param_function_pane(document, comm):
test = View()
@param.depends(test.param.a)
def view(a):
return Div(text='%d' % a)
pane = Pane(view)
inner_pane = pane._pane
assert isinstance(inner_pane, Bokeh)
# Create pane
row = pane.get_root(document, comm=comm)
assert isinstance(row, BkRow)
assert len(row.children) == 1
model = row.children[0]
assert pane._models[row.ref['id']][0] is row
assert isinstance(model, Div)
assert model.text == '0'
# Update pane
test.a = 5
new_model = row.children[0]
assert inner_pane is pane._pane
assert new_model.text == '5'
assert pane._models[row.ref['id']][0] is row
# Cleanup pane
pane._cleanup(row)
assert pane._models == {}
assert inner_pane._models == {}
def test_list_selector_param(document, comm):
class Test(param.Parameterized):
a = param.ListSelector(default=['b', 1], objects=[1, 'b', 'c'])
test = Test()
test_pane = Pane(test)
model = test_pane.get_root(document, comm=comm)
slider = model.children[1]
assert isinstance(slider, MultiSelect)
assert slider.options == ['1', 'b', 'c']
assert slider.value == ['b', '1']
assert slider.disabled == False
# Check changing param value updates widget
test.a = ['c', 1]
assert slider.value == ['c', '1']
# Check changing param attribute updates widget
a_param = test.param['a']
a_param.objects = ['c', 'd', 1]
assert slider.options == ['c', 'd', '1']
a_param.constant = True
assert slider.disabled == True
# Ensure cleanup works
test_pane._cleanup(model)
a_param.constant = False
a_param.objects = [1, 'c', 'd']
test.a = ['d']
assert slider.value == ['c', '1']
assert slider.options == ['c', 'd', '1']
assert slider.disabled == True
def test_expand_param_subobject_expand(document, comm):
class Test(param.Parameterized):
a = param.Parameter()
test = Test(a=Test(name='Nested'))
test_pane = Pane(test, expand=True, expand_button=True)
model = test_pane.get_root(document, comm=comm)
toggle = model.children[1].children[1]
assert isinstance(toggle, Toggle)
# Expand subpane
assert len(model.children) == 3
_, _, subpanel = test_pane.layout.objects
col = model.children[2]
assert isinstance(col, BkColumn)
assert len(col.children) == 2
div, widget = col.children
assert div.text == '<b>Nested</b>'
assert isinstance(widget, BkTextInput)
# Collapse subpanel
test_pane._widgets['a'][1].value = False
assert len(model.children) == 2
assert subpanel._models == {}
def test_object_selector_param(document, comm):
class Test(param.Parameterized):
a = param.ObjectSelector(default='b', objects=[1, 'b', 'c'])
test = Test()
test_pane = Pane(test)
model = test_pane.get_root(document, comm=comm)
select = model.children[1]
assert isinstance(select, Select)
assert select.options == [('1', '1'), ('b', 'b'), ('c', 'c')]
assert select.value == 'b'
assert select.disabled == False
# Check changing param value updates widget
test.a = 1
assert select.value == '1'
# Check changing param attribute updates widget
a_param = test.param['a']
a_param.objects = ['c', 'd', 1]
assert select.options == [('c', 'c'), ('d', 'd'), ('1', '1')]
a_param.constant = True
assert select.disabled == True
# Ensure cleanup works
test_pane._cleanup(model)
a_param.constant = False
a_param.objects = [1, 'c', 'd']
test.a = 'd'
assert select.value == '1'
assert select.options == [('c', 'c'), ('d', 'd'), ('1', '1')]
assert select.disabled == True
def test_range_param(document, comm):
class Test(param.Parameterized):
a = param.Range(default=(0.1, 0.5), bounds=(0, 1.1))
test = Test()
test_pane = Pane(test)
model = test_pane.get_root(document, comm=comm)
widget = model.children[1]
assert isinstance(widget, RangeSlider)
assert widget.start == 0
assert widget.end == 1.1
assert widget.value == (0.1, 0.5)
# Check changing param value updates widget
test.a = (0.2, 0.4)
assert widget.value == (0.2, 0.4)
# Check changing param attribute updates widget
a_param = test.param['a']
a_param.bounds = (0.1, 0.6)
assert widget.start == 0.1
assert widget.end == 0.6
a_param.constant = True
assert widget.disabled == True
# Ensure cleanup works
test_pane._cleanup(model)
a_param.constant = False
a_param.bounds = (-1, 1)
test.a = (0.05, 0.2)
assert widget.value == (0.2, 0.4)
assert widget.start == 0.1
assert widget.end == 0.6
assert widget.disabled == True
def test_boolean_param(document, comm):
class Test(param.Parameterized):
a = param.Boolean(default=False)
test = Test()
test_pane = Pane(test)
model = test_pane.get_root(document, comm=comm)
checkbox = model.children[1]
assert isinstance(checkbox, CheckboxGroup)
assert checkbox.labels == ['A']
assert checkbox.active == []
assert checkbox.disabled == False
# Check changing param value updates widget
test.a = True
assert checkbox.active == [0]
# Check changing param attribute updates widget
a_param = test.param['a']
a_param.constant = True
assert checkbox.disabled == True
# Ensure cleanup works
test_pane._cleanup(model)
a_param.constant = False
test.a = False
assert checkbox.active == [0]
assert checkbox.disabled == True
def test_holoviews_pane_bokeh_renderer(document, comm):
curve = hv.Curve([1, 2, 3])
pane = Pane(curve)
# Create pane
row = pane.get_root(document, comm=comm)
assert isinstance(row, BkRow)
assert len(row.children) == 1
model = row.children[0]
assert isinstance(model, Figure)
assert pane._models[row.ref['id']][0] is model
renderers = [r for r in model.renderers if isinstance(r, GlyphRenderer)]
assert len(renderers) == 1
assert isinstance(renderers[0].glyph, Line)
# Replace Pane.object
scatter = hv.Scatter([1, 2, 3])
pane.object = scatter
model = row.children[0]
assert isinstance(model, Figure)
renderers = [r for r in model.renderers if isinstance(r, GlyphRenderer)]
assert len(renderers) == 1
assert isinstance(renderers[0].glyph, Scatter)
assert pane._models[row.ref['id']][0] is model
# Cleanup
pane._cleanup(row)
assert pane._models == {}
def test_holoviews_link_within_pane(document, comm):
from bokeh.models.tools import RangeTool
from holoviews.plotting.links import RangeToolLink
c1 = hv.Curve([])
c2 = hv.Curve([])
RangeToolLink(c1, c2)
pane = Pane(Pane(hv.Layout([c1, c2]), backend='bokeh'))
column = pane.get_root(document, comm=comm)
assert len(column.children) == 1
subcolumn = column.children[0]
assert isinstance(subcolumn, BkColumn)
assert len(subcolumn.children) == 2
toolbar, subsubcolumn = subcolumn.children
assert isinstance(subsubcolumn, GridBox)
assert len(subsubcolumn.children) == 2
(p1, _, _), (p2, _, _) = subsubcolumn.children
assert isinstance(p1, Figure)
assert isinstance(p2, Figure)
range_tool = subsubcolumn.select_one({'type': RangeTool})
assert isinstance(range_tool, RangeTool)
assert range_tool.x_range == p2.x_range
def test_altair_pane(document, comm):
pane = Pane(altair_example())
# Create pane
model = pane.get_root(document, comm=comm)
assert isinstance(model, VegaPlot)
expected = dict(vega_example, data={})
assert model.data == expected
cds_data = model.data_sources['data'].data
assert np.array_equal(cds_data['x'], np.array(['A', 'B', 'C', 'D', 'E']))
assert np.array_equal(cds_data['y'], np.array([5, 3, 6, 7, 2]))
chart = altair_example()
chart.mark = 'point'
chart.data.values[0]['x'] = 'C'
pane.object = chart
point_example = dict(vega_example, mark='point')
assert model.data == point_example
cds_data = model.data_sources['data'].data
assert np.array_equal(cds_data['x'], np.array(['C', 'B', 'C', 'D', 'E']))
assert np.array_equal(cds_data['y'], np.array([5, 3, 6, 7, 2]))
pane._cleanup(model)
assert pane._models == {}
def test_plotly_pane_datetime_array_transform(document, comm):
index = np.array([dt.datetime(2019, 1, i) for i in range(1, 11)])
data = np.random.randn(10)
traces = [go.Scatter(x=index, y=data)]
fig = go.Figure(traces)
pane = Pane(fig)
model = pane.get_root(document, comm)
assert model.data_sources[0].data['x'][0].dtype.kind in 'SU'
def test_plotly_pane_datetime64_list_transform(document, comm):
index = [np.datetime64(dt.datetime(2019, 1, i)) for i in range(1, 11)]
data = np.random.randn(10)
traces = [go.Scatter(x=index, y=data)]
fig = go.Figure(traces)
pane = Pane(fig)
model = pane.get_root(document, comm)
assert all(isinstance(v, str) for v in model.data[0]['x'])
def test_holoviews_property_override(document, comm):
c1 = hv.Curve([])
pane = Pane(c1, backend='bokeh', background='red',
css_classes=['test_class'])
model = pane.get_root(document, comm=comm)
assert model.background == 'red'
assert model.css_classes == ['test_class']
def test_vega_geometry_data(document, comm):
pane = Pane(gdf_example)
# Create pane
model = pane.get_root(document, comm=comm)
assert isinstance(model, VegaPlot)
# Ensure geometries are not packed into CDS
assert model.data_sources == {}
def test_markdown_pane_dedent(document, comm):
pane = Pane(" ABC")
# Create pane
model = pane.get_root(document, comm=comm)
assert pane._models[model.ref['id']][0] is model
assert model.text.endswith("<p>ABC</p>")
pane.dedent = False
assert model.text.startswith('<div class="codehilite')
def test_vtk_pane_from_url(document, comm):
url = r'https://raw.githubusercontent.com/Kitware/vtk-js/master/Data/StanfordDragon.vtkjs'
pane = Pane(url)
# Create pane
model = pane.get_root(document, comm=comm)
assert isinstance(model, VTKPlot)
assert pane._models[model.ref['id']][0] is model
assert isinstance(model.data, string_types)
def test_explicit_params(document, comm):
class Test(param.Parameterized):
a = param.Boolean(default=False)
b = param.Integer(default=1)
test = Test()
test_pane = Pane(test, parameters=['a'])
model = test_pane.get_root(document, comm=comm)
assert len(model.children) == 2
assert isinstance(model.children[1], CheckboxGroup)
def test_action_param(document, comm):
class Test(param.Parameterized):
a = param.Action(lambda x: x.b.append(1))
b = param.List(default=[])
test = Test()
test_pane = Pane(test)
model = test_pane.get_root(document, comm=comm)
slider = model.children[1]
assert isinstance(slider, Button)
def test_vega_pane_inline(document, comm):
pane = Pane(vega_inline_example)
# Create pane
model = pane.get_root(document, comm=comm)
assert isinstance(model, VegaPlot)
assert dict(model.data, **blank_schema) == dict(vega_inline_example, **blank_schema)
assert model.data_sources == {}
pane._cleanup(model)
assert pane._models == {}
def test_markdown_pane_extensions(document, comm):
pane = Pane("""
```python
None
```
""")
# Create pane
model = pane.get_root(document, comm=comm)
assert pane._models[model.ref['id']][0] is model
assert model.text.startswith('<div class="codehilite')
pane.extensions = ["extra", "smarty"]
assert model.text.startswith('<pre><code class="python')
def test_single_param(document, comm):
class Test(param.Parameterized):
a = param.Parameter(default=0)
test = Test()
test_pane = Pane(test.param.a)
model = test_pane.get_root(document, comm=comm)
assert isinstance(model, BkColumn)
assert len(model.children) == 1
widget = model.children[0]
assert isinstance(widget, TextInput)
assert widget.value == '0'
def test_get_root(document, comm):
class Test(param.Parameterized):
pass
test = Test()
test_pane = Pane(test)
model = test_pane.get_root(document, comm=comm)
assert isinstance(model, BkColumn)
assert len(model.children) == 1
div = model.children[0]
assert isinstance(div, Div)
assert div.text == '<b>' + test.name[:-5] + '</b>'
def test_get_root_tabs(document, comm):
class Test(param.Parameterized):
pass
test = Test()
test_pane = Pane(test, expand_layout=Tabs)
model = test_pane.get_root(document, comm=comm)
assert isinstance(model, BkTabs)
assert len(model.tabs) == 1
box = model.tabs[0].child
assert isinstance(box, BkColumn)
assert len(box.children) == 0
