def test_recursive_traverse_large(self): from bokeh.objects import recursively_traverse_plot_object from bokeh.session.session import Session context, objects = large_plot(500) objects1 = recursively_traverse_plot_object(context) objects2 = set(Session._collect_objs(context)) self.assertEqual(objects1, objects) self.assertEqual(objects1, objects)
def test_r_traverse(self): from bokeh.objects import PlotObject, recursively_traverse_plot_object pobject1 = PlotObject() pobject2 = PlotObject() pobject3 = PlotObject() pobject4 = PlotObject() pobject1.pobject2 = pobject2 pobject1.pobject3 = pobject3 pobject3.pobject4 = pobject4 pobject1.properties_with_refs = Mock(return_value=['pobject2', 'pobject3']) pobject3.properties_with_refs = Mock(return_value=['pobject4']) resultset = recursively_traverse_plot_object(pobject1) expectedset = set([pobject1, pobject2, pobject3, pobject4]) self.assertEqual(resultset, expectedset)
def test_r_traverse(self): from bokeh.objects import PlotObject, recursively_traverse_plot_object pobject1 = PlotObject() pobject2 = PlotObject() pobject3 = PlotObject() pobject4 = PlotObject() pobject1.pobject2 = pobject2 pobject1.pobject3 = pobject3 pobject3.pobject4 = pobject4 pobject1.properties_with_refs = Mock( return_value=['pobject2', 'pobject3']) pobject3.properties_with_refs = Mock(return_value=['pobject4']) resultset = recursively_traverse_plot_object(pobject1) expectedset = set([pobject1, pobject2, pobject3, pobject4]) self.assertEqual(resultset, expectedset)