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)
