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)
