def test_defaults():
    c1 = Chart()
    defaults.plot_height = 1000
    defaults.plot_width = 1000
    defaults.tools = False
    c2 = Chart()
    c3 = Chart()

    assert c1.plot_height == 600
    assert c2.plot_height == c3.plot_height == 1000

    assert c1.plot_width == 600
    assert c2.plot_width == c3.plot_width == 1000

    assert c1.tools
    assert c2.tools == c3.tools == []
 def setUp(self):
     self.source = ColumnDataSource()
     self.xdr = Range1d()
     self.ydr = Range1d()
     self.glyph = GlyphRenderer()
     self._groups = [self.glyph] * 3
     self.chart = Chart(title="title",
                        xlabel="xlabel",
                        ylabel="ylabel",
                        legend="top_left",
                        xscale="linear",
                        yscale="linear",
                        width=800,
                        height=600,
                        tools=True,
                        responsive=True,
                        xgrid=True,
                        ygrid=False)
    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,
        )
        expected = [
            [
                PanTool, WheelZoomTool, BoxZoomTool, SaveTool, ResetTool,
                HelpTool
            ],
            [],
            [PanTool, BoxZoomTool, ResetTool, LassoSelectTool],
        ]
        scenarios = zip([True, False, "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 = [
            PanTool, BoxZoomTool, ResetTool, LassoSelectTool, LassoSelectTool
        ]
        mock_warn.reset_mock()

        # Finally check removing tools
        base_args['tools'] = "pan,box_zoom,reset,lasso_select,lasso_select"

        chart = Chart(**base_args)
        chart.x_range = FactorRange()

        self.compare_tools(chart.tools, expected_tools)
        mock_warn.assert_any_call(msg_repeat)
class TestChart(unittest.TestCase):
    def setUp(self):
        self.source = ColumnDataSource()
        self.xdr = Range1d()
        self.ydr = Range1d()
        self.glyph = GlyphRenderer()
        self._groups = [self.glyph] * 3
        self.chart = Chart(title="title",
                           xlabel="xlabel",
                           ylabel="ylabel",
                           legend="top_left",
                           xscale="linear",
                           yscale="linear",
                           width=800,
                           height=600,
                           tools=True,
                           responsive=True,
                           xgrid=True,
                           ygrid=False)

    def test_title(self):
        self.chart.title.text = "new_title"
        self.assertEqual(self.chart.title.text, "new_title")

    def test_sizing_mode(self):
        self.assertEqual(self.chart.sizing_mode, 'scale_width')

    def check_chart_elements(self, expected_tools):
        self.assertIsInstance(self.chart.left[0], LinearAxis)
        self.assertIsInstance(self.chart.renderers[0], LinearAxis)
        self.assertIsInstance(self.chart.below[0], LinearAxis)
        self.assertIsInstance(self.chart.renderers[1], LinearAxis)
        self.assertIsInstance(self.chart.renderers[2], Grid)
        self.assertIsInstance(self.chart.renderers[3], Grid)
        for i, type_ in enumerate(expected_tools):
            self.assertIsInstance(self.chart.tools[i], type_)

    def test_ranges(self):
        """Test ranges are not created buy the chart."""
        self.assertEqual(self.chart.x_range, None)
        self.assertEqual(self.chart.y_range, None)

    def test_axis_requires_range(self):

        # the axis creation depends on ranges
        with pytest.raises(ValueError):
            self.chart.make_axis("x", "left", "datetime", "foo")

    def test_make_axis(self):

        self.chart.add_ranges('x', Range1d())

        axis = self.chart.make_axis("x", "left", "datetime", "foo")
        self.assertEqual(axis.axis_label, "foo")

        axis = self.chart.make_axis("x", "left", "categorical", "bar")
        self.assertEqual(axis.axis_label, "bar")
        self.assertEqual(axis.major_label_orientation, np.pi / 4)

        axis = self.chart.make_axis("x", "left", "linear", "foobar")
        self.assertEqual(axis.axis_label, "foobar")

    def test_make_grid(self):
        self.chart.add_ranges('x', Range1d())
        axis = self.chart.make_axis("x", "left", "datetime", "foo")
        grid = self.chart.make_grid(0, axis.ticker)
        self.assertEqual(grid.dimension, 0)
        self.assertIsInstance(grid.ticker, Ticker)

    def check_tools_scenario(self, base_args, scenarios, categorical=False):
        for tools, expected_tools in scenarios:
            base_args['tools'] = tools
            chart = Chart(**base_args)
            self.compare_tools(chart.tools, expected_tools)

    def compare_tools(self, tools, expected_tools):
        self.assertEqual(len(tools), len(expected_tools))
        for i, _type in enumerate(expected_tools):
            self.assertIsInstance(tools[i], _type)

    @patch('bokeh.plotting.helpers.warnings.warn')
    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,
        )
        expected = [
            [
                PanTool, WheelZoomTool, BoxZoomTool, SaveTool, ResetTool,
                HelpTool
            ],
            [],
            [PanTool, BoxZoomTool, ResetTool, LassoSelectTool],
        ]
        scenarios = zip([True, False, "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 = [
            PanTool, BoxZoomTool, ResetTool, LassoSelectTool, LassoSelectTool
        ]
        mock_warn.reset_mock()

        # Finally check removing tools
        base_args['tools'] = "pan,box_zoom,reset,lasso_select,lasso_select"

        chart = Chart(**base_args)
        chart.x_range = FactorRange()

        self.compare_tools(chart.tools, expected_tools)
        mock_warn.assert_any_call(msg_repeat)
def test_title_kwarg_no_warning(recwarn):
    Chart(title="title")
    assert len(recwarn) == 0
def test_chart_id():
    chart = Chart(id='1234', title="title")
    assert chart._id == '1234'
 def check_tools_scenario(self, base_args, scenarios, categorical=False):
     for tools, expected_tools in scenarios:
         base_args['tools'] = tools
         chart = Chart(**base_args)
         self.compare_tools(chart.tools, expected_tools)