def test_chart_tools_linear(self, mock_warn): base_args = dict(title="title", xlabel="xlabel", ylabel="ylabel", legend="top_left", xscale="linear", yscale="linear", xgrid=True, ygrid=True, width=800, height=600, filename=False, server=False, notebook=False) expected = [ [ PanTool, WheelZoomTool, BoxZoomTool, PreviewSaveTool, ResizeTool, ResetTool, HelpTool ], [], [ResizeTool, PanTool, BoxZoomTool, ResetTool, LassoSelectTool], ] scenarios = zip( [True, False, "resize,pan,box_zoom,reset,lasso_select"], expected) self.check_tools_scenario(base_args, scenarios) self.check_tools_scenario(base_args, scenarios, categorical=True) msg_repeat = "LassoSelectTool are being repeated" expected_tools = [ ResizeTool, PanTool, BoxZoomTool, ResetTool, LassoSelectTool, LassoSelectTool ] mock_warn.reset_mock() # Finally check repeated tools base_args[ 'tools'] = "resize,pan,box_zoom,reset,lasso_select,lasso_select" chart = LegacyChart(**base_args) chart.x_range = FactorRange() chart.tools = [] chart.create_tools(chart._options.tools) self.compare_tools(chart.tools, expected_tools) mock_warn.assert_any_call(msg_repeat)
def setUp(self): self.source = ColumnDataSource() self.xdr = Range1d() self.ydr = Range1d() self.glyph = GlyphRenderer() self._groups = [self.glyph] * 3 self.chart = LegacyChart(title="title", xlabel="xlabel", ylabel="ylabel", legend="top_left", xscale="linear", yscale="linear", width=800, height=600, tools=True, filename=False, server=False, notebook=False, responsive=True, xgrid=True, ygrid=False)
def check_tools_scenario(self, base_args, scenarios, categorical=False): for tools, expected_tools in scenarios: base_args['tools'] = tools chart = LegacyChart(**base_args) self.compare_tools(chart.tools, expected_tools)